129 lines
3.1 KiB
Go
129 lines
3.1 KiB
Go
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()
|
|
}
|