format code and add session level prepare stmt config

This commit is contained in:
Jinzhu 2025-04-25 16:15:00 +08:00
parent f4e7282442
commit 29a3f7e7c4
2 changed files with 5 additions and 3 deletions

View File

@ -110,6 +110,8 @@ type DB struct {
type Session struct { type Session struct {
DryRun bool DryRun bool
PrepareStmt bool PrepareStmt bool
PrepareStmtMaxSize int
PrepareStmtTTL time.Duration
NewDB bool NewDB bool
Initialized bool Initialized bool
SkipHooks bool SkipHooks bool
@ -273,7 +275,7 @@ func (db *DB) Session(config *Session) *DB {
if v, ok := db.cacheStore.Load(preparedStmtDBKey); ok { if v, ok := db.cacheStore.Load(preparedStmtDBKey); ok {
preparedStmt = v.(*PreparedStmtDB) preparedStmt = v.(*PreparedStmtDB)
} else { } else {
preparedStmt = NewPreparedStmtDB(db.ConnPool, db.Config.PrepareStmtMaxSize, db.Config.PrepareStmtTTL) preparedStmt = NewPreparedStmtDB(db.ConnPool, config.PrepareStmtMaxSize, config.PrepareStmtTTL)
db.cacheStore.Store(preparedStmtDBKey, preparedStmt) db.cacheStore.Store(preparedStmtDBKey, preparedStmt)
} }