prioritize db with context over non-context db

Masterminds · Dec 6, 2019 · 83644f2 · 83644f2
1 parent 5d1d880
commit 83644f2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/squirrel.go b/squirrel.go
@@ -84,10 +84,10 @@ func (r *stdsqlRunner) QueryRow(query string, args ...interface{}) RowScanner {
 
 func setRunWith(b interface{}, runner BaseRunner) interface{} {
 	switch r := runner.(type) {
+	case StdSqlCtx:
+		runner = WrapStdSqlCtx(r)
 	case StdSql:
 		runner = WrapStdSql(r)
-	case StdSqlCtx:
-		runner = &stdsqlCtxRunner{r}
 	}
 	return builder.Set(b, "RunWith", runner)
 }