Repository navigation
Expand file tree
/
Copy pathquery.go
More file actions
109 lines (95 loc) · 3.38 KB
/
Copy pathquery.go
File metadata and controls
109 lines (95 loc) · 3.38 KB
1
2
3
<
8000
div data-line-number="4" class="react-line-number react-code-text" style="padding-right:16px">45
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
package sql
import (
"database/sql"
"github.com/acronis/perfkit/db"
)
// sqlResult wraps sql.Result to implement the db.Result interface
type sqlResult struct {
result sql.Result // Underlying SQL result
}
// LastInsertId returns the ID generated by an INSERT operation
// Implements db.Result interface
func (r *sqlResult) LastInsertId() (int64, error) {
return r.result.LastInsertId()
}
// RowsAffected returns the number of rows affected by the operation
// Implements db.Result interface
func (r *sqlResult) RowsAffected() (int64, error) {
return r.result.RowsAffected()
}
// Exec executes a query that doesn't return rows
// Implements querier.execContext through sqlGateway
// Parameters:
// - format: SQL query string with placeholders
// - args: Values for the placeholders
//
// Returns:
// - db.Result: Wrapper around sql.Result
// - error: Any error that occurred during execution
func (g *sqlGateway) Exec(format string, args ...interface{}) (db.Result, error) {
var sqlRes, err = g.rw.execContext(g.ctx, format, args...)
return &sqlResult{result: sqlRes}, err
}
// QueryRow executes a query that returns a single row
// Implements querier.queryRowContext through sqlGateway
// Supports EXPLAIN mode using dialect interface
// Parameters:
// - format: SQL query string with placeholders
// - args: Values for the placeholders
//
// Returns:
// - db.Row: Interface for scanning row values
func (g *sqlGateway) QueryRow(format string, args ...interface{}) db.Row {
if g.explain {
var err error
// Use dialect interface to get database-specific EXPLAIN syntax
if format, err = addExplainPrefix(g.dialect.name(), format); err != nil {
return &db.EmptyRow{}
}
var rows *sql.Rows
// Execute EXPLAIN query using querier interface
if rows, err = g.rw.queryContext(g.ctx, format, args...); err != nil {
return &db.EmptyRow{}
}
// Log EXPLAIN results if logger is configured
if g.explainLogger != nil {
if err = logExplainResults(g.explainLogger, g.dialect.name(), rows, format, args...); err != nil {
return &db.EmptyRow{}
}
}
}
// Execute actual query using querier interface
var row = g.rw.queryRowContext(g.ctx, format, args...)
return &wrappedRow{row: row, logTime: g.LogTime, readRowsLogger: g.readRowsLogger}
}
// Query executes a query that returns multiple rows
// Implements querier.queryContext through sqlGateway
// Supports EXPLAIN mode using dialect interface
// Parameters:
// - format: SQL query string with placeholders
// - args: Values for the placeholders
//
// Returns:
// - db.Rows: Interface for iterating over result rows
// - error: Any error that occurred during execution
func (g *sqlGateway) Query(format string, args ...interface{}) (db.Rows, err
4484
or) {
if g.explain {
var err error
// Use dialect interface to get database-specific EXPLAIN syntax
if format, err = addExplainPrefix(g.dialect.name(), format); err != nil {
return &db.EmptyRows{}, err
}
}
// Execute query using querier interface
var rows, err = g.rw.queryContext(g.ctx, format, args...)
if err != nil {
return rows, err
}
// Log EXPLAIN results if in explain mode and logger is configured
if g.explain && g.explainLogger != nil {
if err = logExplainResults(g.explainLogger, g.dialect.name(), rows, format, args...); err != nil {
return &db.EmptyRows{}, err
}
}
return &wrappedRows{rows: rows, logTime: g.LogTime, readRowsLogger: g.readRowsLogger}, err
}