From f1776d4b3efc28e6a17ee48ef7ce229f6326b4ce Mon Sep 17 00:00:00 2001
From: Ildar Nurislamov <absorbb@gmail.com>
Date: Wed, 17 Jul 2024 20:12:25 +0400
Subject: [PATCH] tests workaround

---
 bulkerlib/implementations/sql/clickhouse.go |  2 +-
 bulkerlib/implementations/sql/tx_wrapper.go | 36 +++++++--------------
 2 files changed, 12 insertions(+), 26 deletions(-)

diff --git a/bulkerlib/implementations/sql/clickhouse.go b/bulkerlib/implementations/sql/clickhouse.go
index 573873f..5875e79 100644
--- a/bulkerlib/implementations/sql/clickhouse.go
+++ b/bulkerlib/implementations/sql/clickhouse.go
@@ -318,7 +318,7 @@ func (ch *ClickHouse) OpenTx(ctx context.Context) (*TxSQLAdapter, error) {
 		if err != nil {
 			return nil, fmt.Errorf("failed to open connection: %v", err)
 		}
-		db = NewConWithDB(sessionDb, c, sessionId)
+		db = NewConWithDB(sessionDb, c)
 		time.Sleep(1 * time.Second)
 	} else {
 		var err error
diff --git a/bulkerlib/implementations/sql/tx_wrapper.go b/bulkerlib/implementations/sql/tx_wrapper.go
index c189fa2..eed5f67 100644
--- a/bulkerlib/implementations/sql/tx_wrapper.go
+++ b/bulkerlib/implementations/sql/tx_wrapper.go
@@ -179,42 +179,28 @@ func (t *TxWrapper) Rollback() error {
 }
 
 type ConWithDB struct {
-	db        *sql.DB
-	con       *sql.Conn
-	sessionId string
+	db  *sql.DB
+	con *sql.Conn
 }
 
-func NewConWithDB(db *sql.DB, con *sql.Conn, sessionId string) *ConWithDB {
-	fmt.Println("OPEN", sessionId)
-	return &ConWithDB{db: db, con: con, sessionId: sessionId}
+func NewConWithDB(db *sql.DB, con *sql.Conn) *ConWithDB {
+	return &ConWithDB{db: db, con: con}
 }
 
-func (c *ConWithDB) ExecContext(ctx context.Context, query string, args ...any) (res sql.Result, err error) {
-	fmt.Println("START", c.sessionId, query)
-	res, err = c.con.ExecContext(ctx, query, args...)
-	fmt.Println("END", c.sessionId, query)
-	return
+func (c *ConWithDB) ExecContext(ctx context.Context, query string, args ...any) (sql.Result, error) {
+	return c.con.ExecContext(ctx, query, args...)
 }
 
-func (c *ConWithDB) QueryContext(ctx context.Context, query string, args ...any) (res *sql.Rows, err error) {
-	fmt.Println("START", c.sessionId, query)
-	res, err = c.con.QueryContext(ctx, query, args...)
-	fmt.Println("END", c.sessionId, query)
-	return
+func (c *ConWithDB) QueryContext(ctx context.Context, query string, args ...any) (*sql.Rows, error) {
+	return c.con.QueryContext(ctx, query, args...)
 }
 
 func (c *ConWithDB) QueryRowContext(ctx context.Context, query string, args ...any) *sql.Row {
-	fmt.Println("START", c.sessionId, query)
-	row := c.con.QueryRowContext(ctx, query, args...)
-	fmt.Println("END", c.sessionId, query)
-	return row
+	return c.con.QueryRowContext(ctx, query, args...)
 }
 
-func (c *ConWithDB) PrepareContext(ctx context.Context, query string) (res *sql.Stmt, err error) {
-	fmt.Println("START", c.sessionId, query)
-	res, err = c.con.PrepareContext(ctx, query)
-	fmt.Println("END", c.sessionId, query)
-	return
+func (c *ConWithDB) PrepareContext(ctx context.Context, query string) (*sql.Stmt, error) {
+	return c.con.PrepareContext(ctx, query)
 }
 
 func (c *ConWithDB) Close() error {