feat: prepare_stmt support ping (#6924)

* feat: prepare_stmt support ping * feat: prepare_stmt tx support ping
go-gorm · Mar 28, 2024 · 1b48aa0 · 1b48aa0
1 parent 26195e6
commit 1b48aa0
Showing 1 changed file with 16 additions and 0 deletions.
diff --git 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
@@ -242,3 +250,11 @@ func (tx *PreparedStmtTX) QueryRowContext(ctx context.Context, query string, arg
 	}
 	return &sql.Row{}
 }
+
+func (tx *PreparedStmtTX) Ping() error {
+	conn, err := tx.GetDBConn()
+	if err != nil {
+		return err
+	}
+	return conn.Ping()
+}