redis缓存替换+pgvector向量替换

2026-04-04 22:39:16 +08:00
parent e993eb6c5c
commit 9d7c416737
124 changed files with 5460 additions and 141 deletions
--- a/ai-chat-backend/cmd/main.go
+++ b/ai-chat-backend/cmd/main.go
@@ -0,0 +1,128 @@
+package main
+
+import (
+	"ai-chat-backend/pkg/config"
+	"ai-chat-backend/pkg/log"
+	"context"
+	"flag"
+	"fmt"
+	"net/http"
+	"os"
+	"os/signal"
+	"strings"
+	"time"
+
+	"ai-chat-backend/pkg/controllers"
+	"ai-chat-backend/pkg/middlewares"
+	"github.com/gin-gonic/gin"
+)
+
+type ChatGPTWebServer struct {
+	config *config.Config
+	log    log.ILogger
+}
+
+func (r *ChatGPTWebServer) Run(ctx context.Context) error {
+	go r.httpServer(ctx)
+	return nil
+}
+
+func (r *ChatGPTWebServer) httpServer(ctx context.Context) {
+	chatService, err := controllers.NewChatService(r.config, r.log)
+	if err != nil {
+		r.log.Fatal(err)
+	}
+
+	addr := fmt.Sprintf("%s:%d", r.config.Http.IP, r.config.Http.Port)
+	r.log.InfoF("ChatGPT Web Server on: %s", addr)
+	server := &http.Server{
+		Addr: addr,
+	}
+	gin.SetMode(gin.ReleaseMode)
+	entry := gin.Default()
+	entry.Use(middlewares.Cors())
+	chat := entry.Group("/api")
+	if strings.TrimSpace(r.config.BasicAuthUser) != "" {
+		accounts := gin.Accounts{}
+		users := strings.Split(r.config.BasicAuthUser, ",")
+		passwords := strings.Split(r.config.BasicAuthPassword, ",")
+		if len(users) != len(passwords) {
+			r.log.Fatal("basic auth setting error")
+		}
+		for i := 0; i < len(users); i++ {
+			accounts[strings.TrimSpace(users[i])] = strings.TrimSpace(passwords[i])
+		}
+		chat.POST("/chat-process", gin.BasicAuth(accounts), middlewares.RateLimitMiddleware(1, 2), chatService.ChatProcess)
+	} else {
+		chat.POST("/chat-process", middlewares.RateLimitMiddleware(1, 2), chatService.ChatProcess)
+	}
+	chat.POST("/config", func(ctx *gin.Context) {
+		ctx.JSON(200, gin.H{
+			"status": "Success",
+			"data": map[string]string{
+				"apiModel":   "ChatGPTAPI",
+				"socksProxy": "",
+			},
+		})
+	})
+	chat.POST("/session", func(ctx *gin.Context) {
+		ctx.JSON(200, gin.H{
+			"status":  "Success",
+			"message": "",
+			"data": gin.H{
+				"auth": false,
+			},
+		})
+	})
+	chat.GET("/health", func(c *gin.Context) {})
+	fs := http.FileServer(http.Dir(r.config.FrontendPath))
+	entry.NoRoute(func(ctx *gin.Context) {
+		fs.ServeHTTP(ctx.Writer, ctx.Request)
+	})
+	entry.GET("/", func(ctx *gin.Context) {
+		http.ServeFile(ctx.Writer, ctx.Request, r.config.FrontendPath+"/index.html")
+	})
+	server.Handler = entry
+	go func(ctx context.Context) {
+		<-ctx.Done()
+		ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+		defer cancel()
+		if err := server.Shutdown(ctx); err != nil {
+			r.log.InfoF("Server shutdown with error %v", err)
+		}
+	}(ctx)
+	if err := server.ListenAndServe(); err != http.ErrServerClosed {
+		r.log.FatalF("Server listen and serve error %v", err)
+	}
+}
+
+var (
+	configFile = flag.String("config", "dev.config.yaml", "")
+)
+
+func main() {
+	flag.Parse()
+	ctx, stop := signal.NotifyContext(context.Background(), os.Kill, os.Interrupt)
+	defer stop()
+
+	config.InitConfig(*configFile)
+	cnf := config.GetConfig()
+	fmt.Printf("%+v\n", cnf)
+
+	//初始化日志
+	log.SetLevel(cnf.Log.Level)
+	log.SetOutput(log.GetRotateWriter(cnf.Log.LogPath))
+	log.SetPrintCaller(true)
+
+	logger := log.NewLogger()
+	logger.SetLevel(cnf.Log.Level)
+	logger.SetOutput(log.GetRotateWriter(cnf.Log.LogPath))
+	logger.SetPrintCaller(true)
+
+	webServer := &ChatGPTWebServer{
+		config: cnf,
+		log:    logger,
+	}
+	webServer.Run(ctx)
+	<-ctx.Done()
+}