Skip to content

Commit a356b59

Browse files
authored
add defer for sql logQuery (#347)
Fix time calculation for sql queries.
1 parent 9ce5abb commit a356b59

File tree

1 file changed

+5
-5
lines changed
  • pkg/gofr/datasource/sql

1 file changed

+5
-5
lines changed

pkg/gofr/datasource/sql/db.go

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -54,27 +54,27 @@ func (d *DB) Query(query string, args ...interface{}) (*sql.Rows, error) {
5454
}
5555

5656
func (d *DB) QueryRow(query string, args ...interface{}) *sql.Row {
57-
d.logQuery(time.Now(), "QueryRow", query, args...)
57+
defer d.logQuery(time.Now(), "QueryRow", query, args...)
5858
return d.DB.QueryRow(query, args...)
5959
}
6060

6161
func (d *DB) QueryRowContext(ctx context.Context, query string, args ...interface{}) *sql.Row {
62-
d.logQuery(time.Now(), "QueryRowContext", query, args...)
62+
defer d.logQuery(time.Now(), "QueryRowContext", query, args...)
6363
return d.DB.QueryRowContext(ctx, query, args...)
6464
}
6565

6666
func (d *DB) Exec(query string, args ...interface{}) (sql.Result, error) {
67-
d.logQuery(time.Now(), "Exec", query, args...)
67+
defer d.logQuery(time.Now(), "Exec", query, args...)
6868
return d.DB.Exec(query, args...)
6969
}
7070

7171
func (d *DB) ExecContext(ctx context.Context, query string, args ...interface{}) (sql.Result, error) {
72-
d.logQuery(time.Now(), "ExecContext", query, args...)
72+
defer d.logQuery(time.Now(), "ExecContext", query, args...)
7373
return d.DB.ExecContext(ctx, query, args...)
7474
}
7575

7676
func (d *DB) Prepare(query string) (*sql.Stmt, error) {
77-
d.logQuery(time.Now(), "Prepare", query)
77+
defer d.logQuery(time.Now(), "Prepare", query)
7878
return d.DB.Prepare(query)
7979
}
8080

0 commit comments

Comments
 (0)