支持lru缓存

prepare_stmt.go

@@ -24,25 +24,26 @@ type PreparedStmtDB struct {
 	ConnPool
 }
 
+func newPrepareStmtCache(prepareStmtLruConfig *PrepareStmtLruConfig) *StmtStore {
+	var stmts StmtStore
+	if prepareStmtLruConfig != nil && prepareStmtLruConfig.Open {
+		if prepareStmtLruConfig.Size <= 0 {
+			panic("LRU prepareStmtLruConfig.Size must > 0")
+		}
+		lru := &LruStmtStore{}
+		lru.NewLru(prepareStmtLruConfig.Size, prepareStmtLruConfig.TTL)
+		stmts = lru
+	} else {
+		defaultStmtStore := &DefaultStmtStore{}
+		stmts = defaultStmtStore.init()
+	}
+	return &stmts
+}
 func NewPreparedStmtDB(connPool ConnPool, prepareStmtLruConfig *PrepareStmtLruConfig) *PreparedStmtDB {
 	return &PreparedStmtDB{
 		ConnPool: connPool,
-		Stmts: func() StmtStore {
+		Stmts:    *newPrepareStmtCache(prepareStmtLruConfig),
-			var stmts StmtStore
+		Mux:      &sync.RWMutex{},
-			if prepareStmtLruConfig != nil && prepareStmtLruConfig.Open {
-				if prepareStmtLruConfig.Size <= 0 {
-					panic("LRU prepareStmtLruConfig.Size must > 0")
-				}
-				lru := &LruStmtStore{}
-				lru.NewLru(prepareStmtLruConfig.Size, prepareStmtLruConfig.TTL)
-				stmts = lru
-			} else {
-				defaultStmtStore := &DefaultStmtStore{}
-				stmts = defaultStmtStore.init()
-			}
-			return stmts
-		}(),
-		Mux: &sync.RWMutex{},
 	}
 }