From bed85a4f7fa033a82c222c36d5e2eab8bff5f90d Mon Sep 17 00:00:00 2001 From: philhuan Date: Wed, 27 Mar 2024 10:24:24 +0800 Subject: [PATCH] feat: prepare_stmt tx support ping --- prepare_stmt.go | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/prepare_stmt.go b/prepare_stmt.go index 1db682b9..4d533885 100644 --- a/prepare_stmt.go +++ b/prepare_stmt.go @@ -182,6 +182,14 @@ func (db *PreparedStmtDB) QueryRowContext(ctx context.Context, query string, arg return &sql.Row{} } +func (db *PreparedStmtDB) Ping() error { + conn, err := db.GetDBConn() + if err != nil { + return err + } + return conn.Ping() +} + type PreparedStmtTX struct { Tx PreparedStmtDB *PreparedStmtDB @@ -243,7 +251,7 @@ func (tx *PreparedStmtTX) QueryRowContext(ctx context.Context, query string, arg return &sql.Row{} } -func (tx *PreparedStmtDB) Ping() error { +func (tx *PreparedStmtTX) Ping() error { conn, err := tx.GetDBConn() if err != nil { return err