tokenizer

ai-chat-service/pkg/db/mysql/mysql.go

@@ -0,0 +1,28 @@
+package mysql
+
+import (
+	"ai-chat-service/pkg/config"
+	"database/sql"
+	_ "github.com/go-sql-driver/mysql"
+	"time"
+)
+
+var db *sql.DB
+
+func InitMysql(cnf *config.Config) {
+	var err error
+	if cnf.Mysql.DSN == "" {
+		panic("数据库连接字符串不能为空")
+	}
+	db, err = sql.Open("mysql", cnf.Mysql.DSN)
+	if err != nil {
+		panic(err)
+	}
+	db.SetMaxOpenConns(cnf.Mysql.MaxOpenConn)
+	db.SetMaxIdleConns(cnf.Mysql.MaxIdleConn)
+	db.SetConnMaxLifetime(time.Second * time.Duration(cnf.Mysql.MaxLifeTime))
+}
+
+func GetDB() *sql.DB {
+	return db
+}