diff --git a/pkg/query_transaction/sql/querier.go b/pkg/query_transaction/sql/querier.go index a795e86..a4f1163 100644 --- a/pkg/query_transaction/sql/querier.go +++ b/pkg/query_transaction/sql/querier.go @@ -43,7 +43,7 @@ func (q *querier) Begin() *querier { q.txRequested.Store(true) return q } -func (q *querier) Continue(ctx context.Context, conn conn) (*querier, error) { +func (q *querier) Continue(ctx context.Context, conn *SqlDB) (*querier, error) { var iErr error q.initOnce.Do(func() { if q.txRequested.Load() { @@ -54,7 +54,7 @@ func (q *querier) Continue(ctx context.Context, conn conn) (*querier, error) { } q.conn = &SqlTx{tx} } else { - q.conn = conn.(*SqlDB) + q.conn = conn } }) return q, iErr