lian/mchat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

service 修改 Redis 存储 KV

Browse Source
This commit is contained in:
1iaan
2026-04-10 11:12:10 +00:00
parent c888ca8844
commit bc82e3e708
25 changed files with 322 additions and 3666 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
ai-chat-service/chat-server/main.go
View File

@@ -5,13 +5,13 @@ import (
metrics_app "ai-chat-service/chat-server/metrics-app"
metrics_bus "ai-chat-service/chat-server/metrics-bus"
"ai-chat-service/chat-server/server"
vector_data "ai-chat-service/chat-server/vector-data"
"ai-chat-service/interceptor"
"ai-chat-service/pkg/config"
"ai-chat-service/pkg/db/mysql"
"ai-chat-service/pkg/db/redis"
"ai-chat-service/pkg/log"
"ai-chat-service/proto"
"ai-chat-service/services/embedding"
"ai-chat-service/services/faiss"
"flag"
"fmt"
"net/http"
@@ -52,23 +52,22 @@ func main() {
logger.SetOutput(log.GetRotateWriter(cnf.Log.LogPath))
logger.SetPrintCaller(true)
// 初始化Mysql
mysql.InitMysql(cnf)
// 初始化redis
redis.InitRedisPool(cnf)
recordsData := data.NewChatRecordsData(mysql.GetDB())
vectorRecordsData, err := vector_data.NewChatRecordsData(cnf)
recordsData := data.NewChatRecordsData()
embedder, err := embedding.NewEmbedder(cnf)
if err != nil {
log.Fatal(err)
}
faissClient := faiss.NewClient(cnf)
lis, err := net.Listen("tcp", fmt.Sprintf("%s:%d", cnf.Server.IP, cnf.Server.Port))
if err != nil {
log.Fatal(err)
}
s := grpc.NewServer(grpc.UnaryInterceptor(interceptor.UnaryAuthInterceptor), grpc.StreamInterceptor(metrics_app.NewStreamMiddleware(registry).WrapHandler()))
service := server.NewChatService(recordsData, vectorRecordsData, cnf, logger, busMetrics)
service := server.NewChatService(recordsData, embedder, faissClient, cnf, logger, busMetrics)
proto.RegisterChatServer(s, service)
healthCheckSrv := health.NewServer()