feat(PreparedStmtDB): support reset (#5782)

* feat(PreparedStmtDB): support reset * fix: close all stmt * test: fix test * fix: delete one by one
2022-10-19 14:46:59 +08:00
parent 3f20a543fa
commit 5dd2bb4827
2 changed files with 39 additions and 1 deletions
@@ -44,6 +44,18 @@ func (db *PreparedStmtDB) Close() {
 	}
 }

+func (db *PreparedStmtDB) Reset() {
+	db.Mux.Lock()
+	defer db.Mux.Unlock()
+	for query, stmt := range db.Stmts {
+		delete(db.Stmts, query)
+		go stmt.Close()
+	}
+
+	db.PreparedSQL = make([]string, 0, 100)
+	db.Stmts = map[string](*Stmt){}
+}
+
 func (db *PreparedStmtDB) prepare(ctx context.Context, conn ConnPool, isTransaction bool, query string) (Stmt, error) {
 	db.Mux.RLock()
 	if stmt, ok := db.Stmts[query]; ok && (!stmt.Transaction || isTransaction) {