return a concrete type from Wrap() to enable non-Oblast DB operations

1 files changed, 3 insertions, 3 deletions

diff --git a/select.go b/select.go
index 17195d0..35c0671 100644
--- a/select.go
+++ b/select.go
@@ -105,7 +105,7 @@ func (s Store[R]) SelectWhere(ctx context.Context, db Handle, partialQuery strin
 }
 
 func startSelectQuery(ctx context.Context, db Handle, plan plan, query string, args ...any) (handle.Rows, [][]int, error) {
-	rows, err := db.Query(ctx, query, args)
+	rows, err := db.OblastQuery(ctx, query, args)
 	if err != nil {
 		return nil, nil, fmt.Errorf("during Query(): %w", err)
 	}
@@ -136,7 +136,7 @@ func startSelectWhereQuery(ctx context.Context, db Handle, plan plan, partialQue
 		return nil, nil, errors.New("cannot execute SelectWhere() because query could not be autogenerated")
 	}
 	query := plan.Select.Query + partialQuery
-	rows, err = db.Query(ctx, query, args)
+	rows, err = db.OblastQuery(ctx, query, args)
 	if err != nil {
 		err = fmt.Errorf("during Query(): %w", err)
 	}
@@ -240,7 +240,7 @@ func selectOne(ctx context.Context, db Handle, plan plan, v reflect.Value, query
 	for idx, index := range plan.Select.ScanIndexes {
 		slots[idx] = v.FieldByIndex(index).Addr().Interface()
 	}
-	stmt, err := db.Prepare(ctx, query, false)
+	stmt, err := db.OblastPrepare(ctx, query, false)
 	if err != nil {
 		return err
 	}