service 修改 Redis 存储 KV

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

@@ -1,28 +0,0 @@
-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
-}

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

@@ -1,29 +0,0 @@
-package vector
-
-import (
-	"ai-chat-service/pkg/config"
-	"ai-chat-service/pkg/log"
-	"github.com/tencent/vectordatabase-sdk-go/tcvectordb"
-	"time"
-)
-
-var vdb *tcvectordb.Client
-
-func InitDB(config *config.Config) {
-	var defaultOption = &tcvectordb.ClientOption{
-		Timeout:            time.Second * time.Duration(config.VectorDB.Timeout),
-		MaxIdldConnPerHost: config.VectorDB.MaxIdleConnPerHost,
-		IdleConnTimeout:    time.Second * time.Duration(config.VectorDB.IdleConnTimeout),
-		ReadConsistency:    tcvectordb.ReadConsistency(config.VectorDB.ReadConsistency),
-	}
-	var err error
-	vdb, err = tcvectordb.NewClient(config.VectorDB.Url, config.VectorDB.Username, config.VectorDB.Pwd, defaultOption)
-	if err != nil {
-		log.Error(err)
-		return
-	}
-}
-
-func GetVdb() *tcvectordb.Client {
-	return vdb
-}