fix: fix ut case

lqs · Feb 18, 2024 · 5b9c119 · 5b9c119
1 parent 89a04c2
commit 5b9c119
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/database.go b/database.go
@@ -196,7 +196,8 @@ func (d database) Query(sqlString string) (Cursor, error) {
 func (d database) QueryContext(ctx context.Context, sqlString string) (Cursor, error) {
 	isRetry := false
 	for {
-		rows, err := d.queryContextOnce(ctx, sqlString, isRetry)
+		sqlStringWithCallerInfo := getCallerInfo(d, isRetry) + sqlString
+		rows, err := d.queryContextOnce(ctx, sqlStringWithCallerInfo, isRetry)
 		if err != nil {
 			isRetry = d.retryPolicy != nil && d.retryPolicy(err)
 			if isRetry {