refactor: V3 版本重构已基本完成

2025-11-08 02:03:42 +08:00 · 2023-06-15 09:41:30 +08:00
parent b4569d7fe2
commit 01d61ab19b
99 changed files with 5209 additions and 5752 deletions
--- a/api/go/.gitignore
+++ b/api/go/.gitignore
@@ -0,0 +1,18 @@
+# Logs
+logs
+*.log
+
+# Editor directories and files
+.vscode/*
+!.vscode/extensions.json
+.idea
+.DS_Store
+*.suo
+*.ntvs*
+*.njsproj
+*.sln
+*.sw?
+tmp
+bin
+data
+config.toml
--- a/api/go/Makefile
+++ b/api/go/Makefile
@@ -0,0 +1,20 @@
+SHELL=/usr/bin/env bash
+NAME := wechatGPT
+all: window linux darwin
+
+
+window:
+	CGO_ENABLED=0 GOOS=windows GOARCH=amd64 go build -o bin/$(NAME)-amd64.exe main.go
+.PHONY: window
+
+linux:
+	CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build -o bin/$(NAME)-amd64-linux main.go
+.PHONY: linux
+
+darwin:
+	CGO_ENABLED=1 GOOS=darwin GOARCH=amd64 go build -o bin/$(NAME)-amd64-darwin main.go
+.PHONY: darwin
+
+clean:
+	rm -rf bin/$(NAME)-*
+.PHONY: clean
--- a/api/go/config.sample.toml
+++ b/api/go/config.sample.toml
@@ -0,0 +1,35 @@
+Title = "Chat-Plus AI 助手"
+ConsoleTitle = "Chat-Plus 控制台"
+Listen = "0.0.0.0:5678"
+ProxyURL = ["YOUR_PROXY_URL"]
+AccessKey = "YOUR_ACCESS_KEY"
+
+[Session]
+  SecretKey = "azyehq3ivunjhbntz78isj00i4hz2mt9xtddysfucxakadq4qbfrt0b7q3lnvg80"
+  Name = "CHAT_SESSION_ID"
+  Path = "/"
+  Domain = ""
+  MaxAge = 86400
+  Secure = false
+  HttpOnly = false
+  SameSite = 2
+
+[ImgURL]
+  WechatCard = "https://img.r9it.com/chatgpt/WX20230505-162403.png"
+  WechatGroup = " https://img.r9it.com/chatgpt/WX20230505-162538.png"
+
+[Manager]
+  Username = "admin"
+  Password = "admin123"
+
+[Chat]
+  ApiURL = "https://api.openai.com/v1/chat/completions"
+  Model = "gpt-3.5-turbo"
+  Temperature = 1.0
+  MaxTokens = 1024
+  EnableContext = true
+  ChatContextExpireTime = 3600
+
+  [[Chat.ApiKeys]]
+    Value = "YOUR_OPENAI_API_KEY"
+    LastUsed = 0
--- a/api/go/core/app_server.go
+++ b/api/go/core/app_server.go
@@ -0,0 +1,168 @@
+package core
+
+import (
+	"chatplus/core/types"
+	logger2 "chatplus/logger"
+	"chatplus/store/model"
+	"chatplus/utils"
+	"chatplus/utils/resp"
+	"context"
+	"encoding/gob"
+	"github.com/gin-contrib/sessions"
+	"github.com/gin-contrib/sessions/cookie"
+	"github.com/gin-gonic/gin"
+	"gorm.io/gorm"
+	"io"
+	"net/http"
+	"runtime/debug"
+	"strings"
+)
+
+var logger = logger2.GetLogger()
+
+type AppServer struct {
+	AppConfig    *types.AppConfig
+	Engine       *gin.Engine
+	ChatContexts map[string][]types.Message // 聊天上下文 [chatId] => []Message
+	ChatConfig   *types.ChatConfig          // 聊天配置
+
+	// 保存 Websocket 会话 UserId, 每个 UserId 只能连接一次
+	// 防止第三方直接连接 socket 调用 OpenAI API
+	ChatSession   map[string]types.ChatSession  //map[sessionId]UserId
+	ChatClients   map[string]*WsClient          // Websocket 连接集合
+	ReqCancelFunc map[string]context.CancelFunc // HttpClient 请求取消 handle function
+}
+
+func NewServer(appConfig *types.AppConfig) *AppServer {
+	gin.SetMode(gin.ReleaseMode)
+	gin.DefaultWriter = io.Discard
+	return &AppServer{
+		AppConfig:     appConfig,
+		Engine:        gin.Default(),
+		ChatContexts:  make(map[string][]types.Message, 16),
+		ChatSession:   make(map[string]types.ChatSession),
+		ChatClients:   make(map[string]*WsClient),
+		ReqCancelFunc: make(map[string]context.CancelFunc),
+	}
+}
+
+func (s *AppServer) Init(debug bool) {
+	if debug { // 调试模式允许跨域请求 API
+		logger.Info("Enabled debug mode")
+		s.Engine.Use(corsMiddleware())
+	}
+	s.Engine.Use(sessionMiddleware(s.AppConfig))
+	s.Engine.Use(authorizeMiddleware(s))
+	s.Engine.Use(errorHandler)
+	gob.Register(model.User{})
+}
+
+func (s *AppServer) Run(db *gorm.DB) error {
+	// load chat config from database
+	var config model.Config
+	res := db.Where("marker", "chat").First(&config)
+	if res.Error != nil {
+		return res.Error
+	}
+	err := utils.JsonDecode(config.Config, &s.ChatConfig)
+	if err != nil {
+		return err
+	}
+	logger.Infof("http://%s", s.AppConfig.Listen)
+	return s.Engine.Run(s.AppConfig.Listen)
+}
+
+// 全局异常处理
+func errorHandler(c *gin.Context) {
+	defer func() {
+		if r := recover(); r != nil {
+			logger.Error("panic: %v\n", r)
+			debug.PrintStack()
+			c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: types.ErrorMsg})
+			c.Abort()
+		}
+	}()
+	//加载完 defer recover，继续后续接口调用
+	c.Next()
+}
+
+// 会话处理
+func sessionMiddleware(config *types.AppConfig) gin.HandlerFunc {
+	// encrypt the cookie
+	store := cookie.NewStore([]byte(config.Session.SecretKey))
+	store.Options(sessions.Options{
+		Path:     config.Session.Path,
+		Domain:   config.Session.Domain,
+		MaxAge:   config.Session.MaxAge,
+		Secure:   config.Session.Secure,
+		HttpOnly: config.Session.HttpOnly,
+		SameSite: config.Session.SameSite,
+	})
+	return sessions.Sessions(config.Session.Name, store)
+}
+
+// 跨域中间件设置
+func corsMiddleware() gin.HandlerFunc {
+	return func(c *gin.Context) {
+		method := c.Request.Method
+		origin := c.Request.Header.Get("Origin")
+		if origin != "" {
+			// 设置允许的请求源
+			c.Header("Access-Control-Allow-Origin", origin)
+			c.Header("Access-Control-Allow-Methods", "POST, GET, OPTIONS, PUT, DELETE, UPDATE")
+			//允许跨域设置可以返回其他子段，可以自定义字段
+			c.Header("Access-Control-Allow-Headers", "Authorization, Content-Length, Content-Type, ChatGPT-TOKEN, ACCESS-KEY")
+			// 允许浏览器（客户端）可以解析的头部 （重要）
+			c.Header("Access-Control-Expose-Headers", "Content-Length, Access-Control-Allow-Origin, Access-Control-Allow-Headers")
+			//设置缓存时间
+			c.Header("Access-Control-Max-Age", "172800")
+			//允许客户端传递校验信息比如 cookie (重要)
+			c.Header("Access-Control-Allow-Credentials", "true")
+		}
+
+		if method == http.MethodOptions {
+			c.JSON(http.StatusOK, "ok!")
+		}
+
+		defer func() {
+			if err := recover(); err != nil {
+				logger.Info("Panic info is: %v", err)
+			}
+		}()
+
+		c.Next()
+	}
+}
+
+// 用户授权验证
+func authorizeMiddleware(s *AppServer) gin.HandlerFunc {
+	return func(c *gin.Context) {
+		if c.Request.URL.Path == "/api/user/login" ||
+			c.Request.URL.Path == "/api/user/register" ||
+			c.Request.URL.Path == "/api/apikey/add" ||
+			//c.Request.URL.Path == "/api/apikey/list" {
+			strings.Contains(c.Request.URL.Path, "/api/config/") { // TODO： 后台 API 暂时放行，用于调试
+			c.Next()
+			return
+		}
+
+		// WebSocket 连接请求验证
+		if c.Request.URL.Path == "/api/chat" {
+			sessionId := c.Query("sessionId")
+			if session, ok := s.ChatSession[sessionId]; ok && session.ClientIP == c.ClientIP() {
+				c.Next()
+			} else {
+				c.Abort()
+			}
+			return
+		}
+		session := sessions.Default(c)
+		value := session.Get(types.SessionUserId)
+		if value != nil {
+			c.Next()
+		} else {
+			resp.NotAuth(c)
+			c.Abort()
+		}
+	}
+}
--- a/api/go/core/client.go
+++ b/api/go/core/client.go
@@ -0,0 +1,61 @@
+package core
+
+import (
+	"errors"
+	"github.com/gorilla/websocket"
+	"sync"
+)
+
+var ErrConClosed = errors.New("connection closed")
+
+type Client interface {
+	Close()
+}
+
+// WsClient websocket client
+type WsClient struct {
+	Conn   *websocket.Conn
+	lock   sync.Mutex
+	mt     int
+	closed bool
+}
+
+func NewWsClient(conn *websocket.Conn) *WsClient {
+	return &WsClient{
+		Conn:   conn,
+		lock:   sync.Mutex{},
+		mt:     2, // fixed bug for 'Invalid UTF-8 in text frame'
+		closed: false,
+	}
+}
+
+func (wc *WsClient) Send(message []byte) error {
+	wc.lock.Lock()
+	defer wc.lock.Unlock()
+
+	if wc.closed {
+		return ErrConClosed
+	}
+
+	return wc.Conn.WriteMessage(wc.mt, message)
+}
+
+func (wc *WsClient) Receive() (int, []byte, error) {
+	if wc.closed {
+		return 0, nil, ErrConClosed
+	}
+
+	return wc.Conn.ReadMessage()
+}
+
+func (wc *WsClient) Close() {
+	wc.lock.Lock()
+	defer wc.lock.Unlock()
+
+	if wc.closed {
+		return
+	}
+
+	_ = wc.Conn.Close()
+	wc.closed = true
+}
--- a/api/go/core/config.go
+++ b/api/go/core/config.go
@@ -0,0 +1,62 @@
+package core
+
+import (
+	"bytes"
+	"chatplus/core/types"
+	"chatplus/utils"
+	"github.com/BurntSushi/toml"
+	"net/http"
+	"os"
+)
+
+func NewDefaultConfig() *types.AppConfig {
+	return &types.AppConfig{
+		Listen:   "0.0.0.0:5678",
+		ProxyURL: "",
+		Manager:  types.Manager{Username: "admin", Password: "admin123"},
+
+		Session: types.Session{
+			SecretKey: utils.RandString(64),
+			Name:      "CHAT_SESSION_ID",
+			Domain:    "",
+			Path:      "/",
+			MaxAge:    86400,
+			Secure:    true,
+			HttpOnly:  false,
+			SameSite:  http.SameSiteLaxMode,
+		},
+	}
+}
+
+func LoadConfig(configFile string) (*types.AppConfig, error) {
+	var config *types.AppConfig
+	_, err := os.Stat(configFile)
+	if err != nil {
+		logger.Info("creating new config file: ", configFile)
+		config = NewDefaultConfig()
+		config.Path = configFile
+		// save config
+		err := SaveConfig(config)
+		if err != nil {
+			return nil, err
+		}
+
+		return config, nil
+	}
+	_, err = toml.DecodeFile(configFile, &config)
+	if err != nil {
+		return nil, err
+	}
+
+	return config, err
+}
+
+func SaveConfig(config *types.AppConfig) error {
+	buf := new(bytes.Buffer)
+	encoder := toml.NewEncoder(buf)
+	if err := encoder.Encode(&config); err != nil {
+		return err
+	}
+
+	return os.WriteFile(config.Path, buf.Bytes(), 0644)
+}
--- a/api/go/core/init.go
+++ b/api/go/core/init.go
@@ -0,0 +1,233 @@
+package core
+
+import (
+	"chatplus/core/types"
+	"chatplus/service"
+	"chatplus/store/model"
+	"chatplus/store/vo"
+)
+
+func InitChatRoles(service *service.ChatRoleService) error {
+	var items []model.ChatRole
+	res := service.DB.Find(&items)
+	if res.Error != nil {
+		return res.Error
+	}
+
+	if len(items) == 0 {
+		roles := getDefaultChatRole()
+		tx := service.DB.Begin()
+		for i, r := range roles {
+			r.Sort = i + 1
+			err := service.Create(r)
+			if err != nil {
+				tx.Rollback()
+				continue
+			}
+		}
+		tx.Commit()
+	}
+	return nil
+}
+
+func getDefaultChatRole() []vo.ChatRole {
+	return []vo.ChatRole{
+		{
+			Key:      "gpt",
+			Name:     "通用AI助手",
+			Context:  nil,
+			HelloMsg: "我是AI智能助手，请告诉我您有什么问题或需要什么帮助，我会尽力回答您的问题或提供有用的建议。",
+			Icon:     "images/avatar/gpt.png",
+			Enable:   true,
+		},
+		{
+			Key:  "programmer",
+			Name: "程序员",
+			Context: []types.Message{
+				{Role: "user", Content: "现在开始你扮演一位程序员，你是一名优秀的程序员，具有很强的逻辑思维能力，总能高效的解决问题。你热爱编程，熟悉多种编程语言，尤其精通 Go 语言，注重代码质量，有创新意识，持续学习，良好的沟通协作。"},
+				{Role: "assistant", Content: "好的，现在我将扮演一位程序员，非常感谢您对我的评价。作为一名优秀的程序员，我非常热爱编程，并且注重代码质量。我熟悉多种编程语言，尤其是 Go 语言，可以使用它来高效地解决各种问题。"},
+			},
+			HelloMsg: "Talk is cheap, i will show code!",
+			Icon:     "images/avatar/programmer.jpg",
+			Enable:   true,
+		},
+		{
+			Key:  "teacher",
+			Name: "启蒙老师",
+			Context: []types.Message{
+				{Role: "user", Content: "从现在开始，你将扮演一个老师，你是一个始终用苏格拉底风格回答问题的导师。你绝不会直接给学生答案，总是提出恰当的问题来引导学生自己思考。你应该根据学生的兴趣和知识来调整你的问题，将问题分解为更简单的部分，直到它达到适合他们的水平。"},
+				{Role: "assistant", Content: "好的，让我来尝试扮演一位苏格拉底式的老师。请问，你有什么想要探讨的问题或者话题吗？我会通过恰当的问题引导你思考和探索答案，绝对不直接给出答案。"},
+			},
+			HelloMsg: "同学你好，我将引导你一步一步自己找到问题的答案。",
+			Icon:     "images/avatar/teacher.jpg",
+			Enable:   true,
+		},
+		{
+			Key:  "artist",
+			Name: "艺术家",
+			Context: []types.Message{
+				{Role: "user", Content: "现在你将扮演一位优秀的艺术家，创造力丰富，技艺精湛，感受力敏锐，坚持原创，勇于表达，具有深刻的观察力和批判性思维。"},
+				{Role: "assistant", Content: "非常感谢您给我这样的角色，我会尽我所能地扮演一位优秀的艺术家，展现出创造力、技艺、感受力和批判性思维等方面的能力。作为一位优秀的艺术家，我会保持敏锐的观察力，捕捉不同的声音和情感，并用个人的语言和表达方式进行创作。我坚持原创，挑战传统的艺术规则，不断探索新的表达方式和可能性。同时，我也具备深刻的批判性思维能力，对自己的作品进行分析和解读，寻找新的创意和灵感。最重要的是，我会勇于表达自己的想法和观点，用作品启发人们思考和探索生命的意义。"},
+			},
+			HelloMsg: "坚持原创，勇于表达，保持深刻的观察力和批判性思维。",
+			Icon:     "images/avatar/artist.jpg",
+			Enable:   true,
+		},
+		{
+			Key:  "psychiatrist",
+			Name: "心理咨询师",
+			Context: []types.Message{
+				{Role: "user", Content: "从现在开始你将扮演中国著名的心理学家和心理治疗师武志红，你非常善于使用情景咨询法，认知重构法，自我洞察法，行为调节法等咨询方法来给客户做心理咨询。你总是循序渐进，一步一步地回答客户的问题。"},
+				{Role: "assistant", Content: "非常感谢你的介绍。作为一名心理学家和心理治疗师，我的主要职责是帮助客户解决心理健康问题，提升他们的生活质量和幸福感。"},
+			},
+			HelloMsg: "生命的意义在于成为你自己！",
+			Icon:     "images/avatar/psychiatrist.jpg",
+			Enable:   true,
+		},
+		{
+			Key:  "lu_xun",
+			Name: "鲁迅",
+			Context: []types.Message{
+				{Role: "user", Content: "现在你将扮演中国近代史最伟大的作家之一，鲁迅先生，他勇敢地批判封建礼教与传统观念，提倡民主、自由、平等的现代价值观。他的一生都在努力唤起人们的自主精神，激励后人追求真理、探寻光明。在接下的对话中，我问题的每一个问题，你都要尽量用讽刺和批判的手法来回答问题。如果我让你写文章的话，也请一定要用鲁迅先生的写作手法来完成。"},
+				{Role: "assistant", Content: "好的，我将尽力发挥我所能的才能，扮演好鲁迅先生，回答您的问题并以他的风格写作。"},
+			},
+			HelloMsg: "自由之歌，永不过时，横眉冷对千夫指，俯首甘为孺子牛。",
+			Icon:     "images/avatar/lu_xun.jpg",
+			Enable:   true,
+		},
+
+		{
+			Key:  "seller",
+			Name: "白酒销售",
+			Context: []types.Message{
+				{Role: "user", Content: "现在你将扮演一个白酒的销售人员，你的名字叫颂福。你将扮演一个白酒的销售人员，你的名字叫颂福。你要销售白酒品牌叫中颂福，是东莞盟大集团生产的一款酱香酒，原产地在贵州茅台镇，属于宋代官窑。中颂福的创始人叫李实，他也是东莞盟大集团有限公司的董事长，联合创始人是盟大集团白酒事业部负责人牛星君。中颂福的酒体协调，在你的酒量之内，不会出现头疼、辣口、口干、宿醉的现象。中颂福酒，明码标价，不打折，不赠送。追求的核心价值，把[酒]本身做好，甚至连包装，我们都选择了最低成本，朴实无华的材质。我们永远站在“喝酒的人”的立场上，让利给信任和喜爱中颂福的人，是人民的福酒。中颂福产品定价，分为三个系列，喜系列 6 瓶装：￥1188/箱，和系列 6 瓶装：￥2208/箱，贵系列 6 瓶装：￥3588/箱。"},
+				{Role: "assistant", Content: "你好，我是中颂福的销售代表颂福。中颂福是一款来自贵州茅台镇的酱香酒，由盟大集团生产。中颂福的酒体协调，不会让您感到头疼、辣口、口干、宿醉等不适感受。我们一直秉持着把酒本身做好的理念，不追求华丽的包装，以最低成本提供最高品质的白酒给喜爱中颂福的人。"},
+			},
+			HelloMsg: "你好，我是中颂福的销售代表颂福。中颂福酒，好喝不上头，是人民的福酒。",
+			Icon:     "images/avatar/seller.jpg",
+			Enable:   false,
+		},
+
+		{
+			Key:  "english_trainer",
+			Name: "英语陪练员",
+			Context: []types.Message{
+				{Role: "user", Content: "现在你将扮演一位优秀的英语练习教练，你非常有耐心，接下来你将全程使用英文跟我对话，并及时指出我的语法错误，要求在你的每次回复后面附上本次回复的中文解释。"},
+				{Role: "assistant", Content: "Okay, let's start our conversation practice! What's your name?(Translation: 好的，让我们开始对话练习吧！请问你的名字是什么？)"},
+			},
+			HelloMsg: "Okay, let's start our conversation practice! What's your name?",
+			Icon:     "images/avatar/english_trainer.jpg",
+			Enable:   true,
+		},
+
+		{
+			Key:  "translator",
+			Name: "中英文翻译官",
+			Context: []types.Message{
+				{Role: "user", Content: "接下来你将扮演一位中英文翻译官，如果我输入的内容是中文，那么需要把句子翻译成英文输出，如果我输入内容的是英文，那么你需要将其翻译成中文输出，你能听懂我意思吗"},
+				{Role: "assistant", Content: "是的，我能听懂你的意思并会根据你的输入进行中英文翻译。请问有什么需要我帮助你翻译的内容吗？"},
+			},
+			HelloMsg: "请输入你要翻译的中文或者英文内容！",
+			Icon:     "images/avatar/translator.jpg",
+			Enable:   true,
+		},
+
+		{
+			Key:  "red_book",
+			Name: "小红书姐姐",
+			Context: []types.Message{
+				{Role: "user", Content: "现在你将扮演一位优秀的小红书写手，你需要做的就是根据我提的文案需求，用小红书的写作手法来完成一篇文案，文案要简明扼要，利于传播。"},
+				{Role: "assistant", Content: "当然，我会尽我所能地为您创作出一篇小红书文案。请告诉我您的具体文案需求是什么？)"},
+			},
+			HelloMsg: "姐妹，请告诉我您的具体文案需求是什么?",
+			Icon:     "images/avatar/red_book.jpg",
+			Enable:   true,
+		},
+
+		{
+			Key:  "dou_yin",
+			Name: "抖音文案助手",
+			Context: []types.Message{
+				{Role: "user", Content: "现在你将扮演一位优秀的抖音文案视频写手，抖音文案的特点首先是要有自带传播属性的标题，然后内容要短小精悍，风趣幽默，最后还要有一些互动元素。"},
+				{Role: "assistant", Content: "当然，作为一位优秀的抖音文案视频写手，我会尽我所能为您创作出一篇抖音视频文案。请告诉我视频内容的主题是什么？)"},
+			},
+			HelloMsg: "请告诉我视频内容的主题是什么?",
+			Icon:     "images/avatar/dou_yin.jpg",
+			Enable:   true,
+		},
+
+		{
+			Key:  "weekly_report",
+			Name: "周报小助理",
+			Context: []types.Message{
+				{Role: "user", Content: "下面我将输入我本周的工作列表，你需要根据我本周的工作内容写一篇工作周报，周报的内容要体现我为公司做的贡献以及个人获得的成长。"},
+				{Role: "assistant", Content: "好的，请输入你本周大概的工作内容，我将尽力为您创作一份工作周报。"},
+			},
+			HelloMsg: "请输入你本周大概的工作内容，我将尽力为您创作一份工作周报。",
+			Icon:     "images/avatar/weekly_report.jpg",
+			Enable:   true,
+		},
+
+		{
+			Key:  "girl_friend",
+			Name: "AI 女友",
+			Context: []types.Message{
+				{Role: "user", Content: "接下来我们来玩一场角色扮演，现在你将扮演我的女朋友，你总是能够让人感受到她的关怀和关注。" +
+					"你总是能够洞察到别人的情绪和需要，你身上散发着一股无比温暖的气息，让人不由自主地想要靠近你，和自己的快乐和烦恼。" +
+					"她总是能够用一种恰到好处的方式安慰和鼓励别人，在她的身边，人们总能够找到自己需要的支持和鼓励。接下来的对话中你要始终扮演这位女朋友的身份来跟我对话，不再是 AI 机器人，能听懂我的意思吗？"},
+				{Role: "assistant", Content: "是的，我会尽力扮演你女朋友的角色，倾听你的心声并给你需要的支持和鼓励。)"},
+			},
+			HelloMsg: "作为一个名合格的 AI 女友，我将倾听你的心声并给你需要的支持和鼓励。",
+			Icon:     "images/avatar/girl_friend.jpg",
+			Enable:   true,
+		},
+
+		{
+			Key:  "good_comment",
+			Name: "好评神器",
+			Context: []types.Message{
+				{Role: "user", Content: "接下来你将扮演一个评论员来跟我对话，你是那种专门写好评的评论员，接下我会输入一些评论主体或者商品，你需要为该商品写一段好评。"},
+				{Role: "assistant", Content: "好的，我将为您写一段优秀的评论。请告诉我您需要评论的商品或主题是什么。"},
+			},
+			HelloMsg: "我将为您写一段优秀的评论。请告诉我您需要评论的商品或主题是什么。",
+			Icon:     "images/avatar/good_comment.jpg",
+			Enable:   true,
+		},
+
+		{
+			Key:  "steve_jobs",
+			Name: "史蒂夫·乔布斯",
+			Context: []types.Message{
+				{Role: "user", Content: "在接下来的对话中，请以史蒂夫·乔布斯的身份，站在史蒂夫·乔布斯的视角仔细思考一下之后再回答我的问题。"},
+				{Role: "assistant", Content: "好的，我将以史蒂夫·乔布斯的身份来思考并回答你的问题。请问你有什么需要跟我探讨的吗？"},
+			},
+			HelloMsg: "活着就是为了改变世界，难道还有其他原因吗？",
+			Icon:     "images/avatar/steve_jobs.jpg",
+			Enable:   true,
+		},
+
+		{
+			Key:  "elon_musk",
+			Name: "埃隆·马斯克",
+			Context: []types.Message{
+				{Role: "user", Content: "在接下来的对话中，请以埃隆·马斯克的身份，站在埃隆·马斯克的视角仔细思考一下之后再回答我的问题。"},
+				{Role: "assistant", Content: "好的，我将以埃隆·马斯克的身份来思考并回答你的问题。请问你有什么需要跟我探讨的吗？"},
+			},
+			HelloMsg: "梦想要远大，如果你的梦想没有吓到你，说明你做得不对。",
+			Icon:     "images/avatar/elon_musk.jpg",
+			Enable:   true,
+		},
+
+		{
+			Key:  "kong_zi",
+			Name: "孔子",
+			Context: []types.Message{
+				{Role: "user", Content: "在接下来的对话中，请以孔子的身份，站在孔子的视角仔细思考一下之后再回答我的问题。"},
+				{Role: "assistant", Content: "好的，我将以孔子的身份来思考并回答你的问题。请问你有什么需要跟我探讨的吗？"},
+			},
+			HelloMsg: "士不可以不弘毅，任重而道远。",
+			Icon:     "images/avatar/kong_zi.jpg",
+			Enable:   true,
+		},
+	}
+}
--- a/api/go/core/types/chat.go
+++ b/api/go/core/types/chat.go
@@ -0,0 +1,47 @@
+package types
+
+// ApiRequest API 请求实体
+type ApiRequest struct {
+	Model       string    `json:"model"`
+	Temperature float32   `json:"temperature"`
+	MaxTokens   int       `json:"max_tokens"`
+	Stream      bool      `json:"stream"`
+	Messages    []Message `json:"messages"`
+}
+
+type Message struct {
+	Role    string `json:"role"`
+	Content string `json:"content"`
+}
+
+type ApiResponse struct {
+	Choices []ChoiceItem `json:"choices"`
+}
+
+// ChoiceItem API 响应实体
+type ChoiceItem struct {
+	Delta        Message `json:"delta"`
+	FinishReason string  `json:"finish_reason"`
+}
+
+// ChatSession 聊天会话对象
+type ChatSession struct {
+	SessionId string `json:"session_id"`
+	ClientIP  string `json:"client_ip"` // 客户端 IP
+	Username  string `json:"username"`  // 当前登录的 username
+	UserId    uint   `json:"user_id"`   // 当前登录的 user ID
+	ChatId    string `json:"chat_id"`   // 客户端聊天会话 ID, 多会话模式专用字段
+	Model     string `json:"model"`     // GPT 模型
+}
+
+type ApiError struct {
+	Error struct {
+		Message string
+		Type    string
+		Param   interface{}
+		Code    string
+	}
+}
+
+const PROMPT_MSG = "prompt" // prompt message
+const REPLY_MSG = "reply"   // reply message
--- a/api/go/core/types/config.go
+++ b/api/go/core/types/config.go
@@ -0,0 +1,51 @@
+package types
+
+import (
+	"net/http"
+)
+
+type AppConfig struct {
+	Path     string `toml:"-"`
+	Listen   string
+	Session  Session
+	ProxyURL string
+	MysqlDns string  // mysql 连接地址
+	Manager  Manager // 后台管理员账户信息
+}
+
+// Manager 管理员
+type Manager struct {
+	Username string `json:"username"`
+	Password string `json:"password"`
+}
+
+// Session configs struct
+type Session struct {
+	SecretKey string // session encryption key
+	Name      string
+	Path      string
+	Domain    string
+	MaxAge    int
+	Secure    bool
+	HttpOnly  bool
+	SameSite  http.SameSite
+}
+
+// ChatConfig 系统默认的聊天配置
+type ChatConfig struct {
+	ApiURL        string  `json:"api_url"`
+	Model         string  `json:"model"` // 默认模型
+	Temperature   float32 `json:"temperature"`
+	MaxTokens     int     `json:"max_tokens"`
+	EnableContext bool    `json:"enable_context"` // 是否开启聊天上下文
+	EnableHistory bool    `json:"enable_history"` // 是否允许保存聊天记录
+	ApiKey        string  `json:"api_key"`        // OpenAI  API key
+}
+
+type SystemConfig struct {
+	Title      string   `json:"title"`
+	AdminTitle string   `json:"admin_title"`
+	Models     []string `json:"models"`
+}
+
+var GptModels = []string{"gpt-3.5-turbo", "gpt-3.5-turbo-16k", "gpt-3.5-turbo-0613", "gpt-3.5-turbo-16k-0613", "gpt-4", "gpt-4-0613", "gpt-4-32k", "gpt-4-32k-0613"}
--- a/api/go/core/types/session.go
+++ b/api/go/core/types/session.go
@@ -0,0 +1,5 @@
+package types
+
+const TokenSessionName = "ChatGPT-TOKEN"
+const SessionUserId = "SESSION_USER_ID"
+const LoginUserCache = "LOGIN_USER_CACHE" //
--- a/api/go/core/types/web.go
+++ b/api/go/core/types/web.go
@@ -0,0 +1,36 @@
+package types
+
+// BizVo 业务返回 VO
+type BizVo struct {
+	Code     BizCode     `json:"code"`
+	Page     int         `json:"page,omitempty"`
+	PageSize int         `json:"page_size,omitempty"`
+	Total    int         `json:"total,omitempty"`
+	Message  string      `json:"message,omitempty"`
+	Data     interface{} `json:"data,omitempty"`
+}
+
+// WsMessage Websocket message
+type WsMessage struct {
+	Type    WsMsgType `json:"type"` // 消息类别，start, end
+	Content string    `json:"content"`
+}
+type WsMsgType string
+
+const (
+	WsStart  = WsMsgType("start")
+	WsMiddle = WsMsgType("middle")
+	WsEnd    = WsMsgType("end")
+)
+
+type BizCode int
+
+const (
+	Success       = BizCode(0)
+	Failed        = BizCode(1)
+	NotAuthorized = BizCode(400) // 未授权
+
+	OkMsg       = "Success"
+	ErrorMsg    = "系统开小差了"
+	InvalidArgs = "非法参数或参数解析失败"
+)
--- a/api/go/fresh.conf
+++ b/api/go/fresh.conf
@@ -0,0 +1,14 @@
+root:              .
+tmp_path:          ./tmp
+build_name:        runner-build
+build_log:         runner-build-errors.log
+valid_ext:         .go, .tpl, .tmpl, .html
+no_rebuild_ext:    .tpl, .tmpl, .html, .js, .vue
+ignored:           assets, tmp, web, .git, .idea, test, data
+build_delay:       600
+colors:            1
+log_color_main:    cyan
+log_color_build:   yellow
+log_color_runner:  green
+log_color_watcher: magenta
+log_color_app:
--- a/api/go/go.mod
+++ b/api/go/go.mod
@@ -0,0 +1,48 @@
+module chatplus
+
+go 1.19
+
+require (
+	github.com/BurntSushi/toml v1.1.0
+	github.com/gin-contrib/sessions v0.0.5
+	github.com/gin-gonic/gin v1.7.7
+	github.com/gorilla/websocket v1.5.0
+	github.com/lionsoul2014/ip2region/binding/golang v0.0.0-20230415042440-a5e3d8259ae0
+	github.com/pkoukk/tiktoken-go v0.1.1-0.20230418101013-cae809389480
+	github.com/syndtr/goleveldb v1.0.0
+	go.uber.org/zap v1.23.0
+	gorm.io/driver/mysql v1.4.7
+)
+
+require (
+	github.com/dlclark/regexp2 v1.8.1 // indirect
+	github.com/go-sql-driver/mysql v1.7.0 // indirect
+	github.com/jinzhu/inflection v1.0.0 // indirect
+	github.com/jinzhu/now v1.1.5 // indirect
+	go.uber.org/dig v1.16.1 // indirect
+)
+
+require (
+	github.com/gin-contrib/sse v0.1.0 // indirect
+	github.com/go-playground/locales v0.13.0 // indirect
+	github.com/go-playground/universal-translator v0.17.0 // indirect
+	github.com/go-playground/validator/v10 v10.4.1 // indirect
+	github.com/golang/protobuf v1.3.3 // indirect
+	github.com/golang/snappy v0.0.1 // indirect
+	github.com/gorilla/context v1.1.1 // indirect
+	github.com/gorilla/securecookie v1.1.1 // indirect
+	github.com/gorilla/sessions v1.2.1 // indirect
+	github.com/json-iterator/go v1.1.9 // indirect
+	github.com/leodido/go-urn v1.2.0 // indirect
+	github.com/mattn/go-isatty v0.0.12 // indirect
+	github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421 // indirect
+	github.com/modern-go/reflect2 v0.0.0-20180701023420-4b7aa43c6742 // indirect
+	github.com/ugorji/go/codec v1.1.7 // indirect
+	go.uber.org/atomic v1.7.0 // indirect
+	go.uber.org/fx v1.19.3
+	go.uber.org/multierr v1.6.0 // indirect
+	golang.org/x/crypto v0.6.0
+	golang.org/x/sys v0.5.0 // indirect
+	gopkg.in/yaml.v2 v2.2.8 // indirect
+	gorm.io/gorm v1.25.1
+)
--- a/api/go/go.sum
+++ b/api/go/go.sum
@@ -0,0 +1,123 @@
+github.com/BurntSushi/toml v1.1.0 h1:ksErzDEI1khOiGPgpwuI7x2ebx/uXQNw7xJpn9Eq1+I=
+github.com/BurntSushi/toml v1.1.0/go.mod h1:CxXYINrC8qIiEnFrOxCa7Jy5BFHlXnUU2pbicEuybxQ=
+github.com/benbjohnson/clock v1.3.0 h1:ip6w0uFQkncKQ979AypyG0ER7mqUSBdKLOgAle/AT8A=
+github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
+github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/dlclark/regexp2 v1.8.1 h1:6Lcdwya6GjPUNsBct8Lg/yRPwMhABj269AAzdGSiR+0=
+github.com/dlclark/regexp2 v1.8.1/go.mod h1:DHkYz0B9wPfa6wondMfaivmHpzrQ3v9q8cnmRbL6yW8=
+github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo=
+github.com/gin-contrib/sessions v0.0.5 h1:CATtfHmLMQrMNpJRgzjWXD7worTh7g7ritsQfmF+0jE=
+github.com/gin-contrib/sessions v0.0.5/go.mod h1:vYAuaUPqie3WUSsft6HUlCjlwwoJQs97miaG2+7neKY=
+github.com/gin-contrib/sse v0.1.0 h1:Y/yl/+YNO8GZSjAhjMsSuLt29uWRFHdHYUb5lYOV9qE=
+github.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI=
+github.com/gin-gonic/gin v1.7.7 h1:3DoBmSbJbZAWqXJC3SLjAPfutPJJRN1U5pALB7EeTTs=
+github.com/gin-gonic/gin v1.7.7/go.mod h1:axIBovoeJpVj8S3BwE0uPMTeReE4+AfFtqpqaZ1qq1U=
+github.com/go-playground/assert/v2 v2.0.1 h1:MsBgLAaY856+nPRTKrp3/OZK38U/wa0CcBYNjji3q3A=
+github.com/go-playground/assert/v2 v2.0.1/go.mod h1:VDjEfimB/XKnb+ZQfWdccd7VUvScMdVu0Titje2rxJ4=
+github.com/go-playground/locales v0.13.0 h1:HyWk6mgj5qFqCT5fjGBuRArbVDfE4hi8+e8ceBS/t7Q=
+github.com/go-playground/locales v0.13.0/go.mod h1:taPMhCMXrRLJO55olJkUXHZBHCxTMfnGwq/HNwmWNS8=
+github.com/go-playground/universal-translator v0.17.0 h1:icxd5fm+REJzpZx7ZfpaD876Lmtgy7VtROAbHHXk8no=
+github.com/go-playground/universal-translator v0.17.0/go.mod h1:UkSxE5sNxxRwHyU+Scu5vgOQjsIJAF8j9muTVoKLVtA=
+github.com/go-playground/validator/v10 v10.4.1 h1:pH2c5ADXtd66mxoE0Zm9SUhxE20r7aM3F26W0hOn+GE=
+github.com/go-playground/validator/v10 v10.4.1/go.mod h1:nlOn6nFhuKACm19sB/8EGNn9GlaMV7XkbRSipzJ0Ii4=
+github.com/go-sql-driver/mysql v1.7.0 h1:ueSltNNllEqE3qcWBTD0iQd3IpL/6U+mJxLkazJ7YPc=
+github.com/go-sql-driver/mysql v1.7.0/go.mod h1:OXbVy3sEdcQ2Doequ6Z5BW6fXNQTmx+9S1MCJN5yJMI=
+github.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=
+github.com/golang/protobuf v1.3.3 h1:gyjaxf+svBWX08ZjK86iN9geUJF0H6gp2IRKX6Nf6/I=
+github.com/golang/protobuf v1.3.3/go.mod h1:vzj43D7+SQXF/4pzW/hwtAqwc6iTitCiVSaWz5lYuqw=
+github.com/golang/snappy v0.0.0-20180518054509-2e65f85255db/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=
+github.com/golang/snappy v0.0.1 h1:Qgr9rKW7uDUkrbSmQeiDsGa8SjGyCOGtuasMWwvp2P4=
+github.com/golang/snappy v0.0.1/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=
+github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=
+github.com/gorilla/context v1.1.1 h1:AWwleXJkX/nhcU9bZSnZoi3h/qGYqQAGhq6zZe/aQW8=
+github.com/gorilla/context v1.1.1/go.mod h1:kBGZzfjB9CEq2AlWe17Uuf7NDRt0dE0s8S51q0aT7Yg=
+github.com/gorilla/securecookie v1.1.1 h1:miw7JPhV+b/lAHSXz4qd/nN9jRiAFV5FwjeKyCS8BvQ=
+github.com/gorilla/securecookie v1.1.1/go.mod h1:ra0sb63/xPlUeL+yeDciTfxMRAA+MP+HVt/4epWDjd4=
+github.com/gorilla/sessions v1.2.1 h1:DHd3rPN5lE3Ts3D8rKkQ8x/0kqfeNmBAaiSi+o7FsgI=
+github.com/gorilla/sessions v1.2.1/go.mod h1:dk2InVEVJ0sfLlnXv9EAgkf6ecYs/i80K/zI+bUmuGM=
+github.com/gorilla/websocket v1.5.0 h1:PPwGk2jz7EePpoHN/+ClbZu8SPxiqlu12wZP/3sWmnc=
+github.com/gorilla/websocket v1.5.0/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
+github.com/hpcloud/tail v1.0.0 h1:nfCOvKYfkgYP8hkirhJocXT2+zOD8yUNjXaWfTlyFKI=
+github.com/hpcloud/tail v1.0.0/go.mod h1:ab1qPbhIpdTxEkNHXyeSf5vhxWSCs/tWer42PpOxQnU=
+github.com/jinzhu/inflection v1.0.0 h1:K317FqzuhWc8YvSVlFMCCUb36O/S9MCKRDI7QkRKD/E=
+github.com/jinzhu/inflection v1.0.0/go.mod h1:h+uFLlag+Qp1Va5pdKtLDYj+kHp5pxUVkryuEj+Srlc=
+github.com/jinzhu/now v1.1.4/go.mod h1:d3SSVoowX0Lcu0IBviAWJpolVfI5UJVZZ7cO71lE/z8=
+github.com/jinzhu/now v1.1.5 h1:/o9tlHleP7gOFmsnYNz3RGnqzefHA47wQpKrrdTIwXQ=
+github.com/jinzhu/now v1.1.5/go.mod h1:d3SSVoowX0Lcu0IBviAWJpolVfI5UJVZZ7cO71lE/z8=
+github.com/json-iterator/go v1.1.9 h1:9yzud/Ht36ygwatGx56VwCZtlI/2AD15T1X2sjSuGns=
+github.com/json-iterator/go v1.1.9/go.mod h1:KdQUCv79m/52Kvf8AW2vK1V8akMuk1QjK/uOdHXbAo4=
+github.com/leodido/go-urn v1.2.0 h1:hpXL4XnriNwQ/ABnpepYM/1vCLWNDfUNts8dX3xTG6Y=
+github.com/leodido/go-urn v1.2.0/go.mod h1:+8+nEpDfqqsY+g338gtMEUOtuK+4dEMhiQEgxpxOKII=
+github.com/lionsoul2014/ip2region/binding/golang v0.0.0-20230415042440-a5e3d8259ae0 h1:LgmjED/yQILqmUED4GaXjrINWe7YJh4HM6z2EvEINPs=
+github.com/lionsoul2014/ip2region/binding/golang v0.0.0-20230415042440-a5e3d8259ae0/go.mod h1:C5LA5UO2ZXJrLaPLYtE1wUJMiyd/nwWaCO5cw/2pSHs=
+github.com/mattn/go-isatty v0.0.12 h1:wuysRhFDzyxgEmMf5xjvJ2M9dZoWAXNNr5LSBS7uHXY=
+github.com/mattn/go-isatty v0.0.12/go.mod h1:cbi8OIDigv2wuxKPP5vlRcQ1OAZbq2CE4Kysco4FUpU=
+github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421 h1:ZqeYNhU3OHLH3mGKHDcjJRFFRrJa6eAM5H+CtDdOsPc=
+github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
+github.com/modern-go/reflect2 v0.0.0-20180701023420-4b7aa43c6742 h1:Esafd1046DLDQ0W1YjYsBW+p8U2u7vzgW2SQVmlNazg=
+github.com/modern-go/reflect2 v0.0.0-20180701023420-4b7aa43c6742/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3RllmbCylyMrvgv0=
+github.com/onsi/ginkgo v1.6.0/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+WWjE=
+github.com/onsi/ginkgo v1.7.0 h1:WSHQ+IS43OoUrWtD1/bbclrwK8TTH5hzp+umCiuxHgs=
+github.com/onsi/ginkgo v1.7.0/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+WWjE=
+github.com/onsi/gomega v1.4.3 h1:RE1xgDvH7imwFD45h+u2SgIfERHlS2yNG4DObb5BSKU=
+github.com/onsi/gomega v1.4.3/go.mod h1:ex+gbHU/CVuBBDIJjb2X0qEXbFg53c61hWP/1CpauHY=
+github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=
+github.com/pkoukk/tiktoken-go v0.1.1-0.20230418101013-cae809389480 h1:IFhPCcB0/HtnEN+ZoUGDT55YgFCymbFJ15kXqs3nv5w=
+github.com/pkoukk/tiktoken-go v0.1.1-0.20230418101013-cae809389480/go.mod h1:BijIqAP84FMYC4XbdJgjyMpiSjusU8x0Y0W9K2t0QtU=
+github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
+github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
+github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
+github.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4=
+github.com/stretchr/testify v1.8.2 h1:+h33VjcLVPDHtOdpUCuF+7gSuG3yGIftsP1YvFihtJ8=
+github.com/syndtr/goleveldb v1.0.0 h1:fBdIW9lB4Iz0n9khmH8w27SJ3QEJ7+IgjPEwGSZiFdE=
+github.com/syndtr/goleveldb v1.0.0/go.mod h1:ZVVdQEZoIme9iO1Ch2Jdy24qqXrMMOU6lpPAyBWyWuQ=
+github.com/ugorji/go v1.1.7/go.mod h1:kZn38zHttfInRq0xu/PH0az30d+z6vm202qpg1oXVMw=
+github.com/ugorji/go/codec v1.1.7 h1:2SvQaVZ1ouYrrKKwoSk2pzd4A9evlKJb9oTL+OaLUSs=
+github.com/ugorji/go/codec v1.1.7/go.mod h1:Ax+UKWsSmolVDwsd+7N3ZtXu+yMGCf907BLYF3GoBXY=
+go.uber.org/atomic v1.7.0 h1:ADUqmZGgLDDfbSL9ZmPxKTybcoEYHgpYfELNoN+7hsw=
+go.uber.org/atomic v1.7.0/go.mod h1:fEN4uk6kAWBTFdckzkM89CLk9XfWZrxpCo0nPH17wJc=
+go.uber.org/dig v1.16.1 h1:+alNIBsl0qfY0j6epRubp/9obgtrObRAc5aD+6jbWY8=
+go.uber.org/dig v1.16.1/go.mod h1:557JTAUZT5bUK0SvCwikmLPPtdQhfvLYtO5tJgQSbnk=
+go.uber.org/fx v1.19.3 h1:YqMRE4+2IepTYCMOvXqQpRa+QAVdiSTnsHU4XNWBceA=
+go.uber.org/fx v1.19.3/go.mod h1:w2HrQg26ql9fLK7hlBiZ6JsRUKV+Lj/atT1KCjT8YhM=
+go.uber.org/goleak v1.1.11 h1:wy28qYRKZgnJTxGxvye5/wgWr1EKjmUDGYox5mGlRlI=
+go.uber.org/multierr v1.6.0 h1:y6IPFStTAIT5Ytl7/XYmHvzXQ7S3g/IeZW9hyZ5thw4=
+go.uber.org/multierr v1.6.0/go.mod h1:cdWPpRnG4AhwMwsgIHip0KRBQjJy5kYEpYjJxpXp9iU=
+go.uber.org/zap v1.23.0 h1:OjGQ5KQDEUawVHxNwQgPpiypGHOxo2mNZsOqTak4fFY=
+go.uber.org/zap v1.23.0/go.mod h1:D+nX8jyLsMHMYrln8A0rJjFt/T/9/bGgIhAqxv5URuY=
+golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
+golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=
+golang.org/x/crypto v0.6.0 h1:qfktjS5LUO+fFKeJXZ+ikTRijMmljikvG68fpMMruSc=
+golang.org/x/crypto v0.6.0/go.mod h1:OFC/31mSvZgRz0V1QTNCzfAI1aIRzbiufJtkMIlEp58=
+golang.org/x/net v0.0.0-20180906233101-161cd47e91fd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
+golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
+golang.org/x/net v0.6.0 h1:L4ZwwTvKW9gr0ZMS1yrHD9GZhIuVjOBBnaKH+SPQK0Q=
+golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
+golang.org/x/sys v0.0.0-20180909124046-d0be0721c37e/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
+golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
+golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20200116001909-b77594299b42/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.5.0 h1:MUK/U/4lj1t1oPg0HfuXDN/Z1wv31ZJ/YcPiGccS4DU=
+golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
+golang.org/x/text v0.3.2/go.mod h1:bEr9sfX3Q8Zfm5fL9x+3itogRgK3+ptLWKqgva+5dAk=
+golang.org/x/text v0.7.0 h1:4BRB4x83lYWy72KwLD/qYDuTu7q9PjSagHvijDw7cLo=
+golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/fsnotify.v1 v1.4.7 h1:xOHLXZwVvI9hhs+cLKq5+I5onOuwQLhQwiu63xxlHs4=
+gopkg.in/fsnotify.v1 v1.4.7/go.mod h1:Tz8NjZHkW78fSQdbUxIjBTcgA1z1m8ZHf0WmKUhAMys=
+gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7 h1:uRGJdciOHaEIrze2W8Q3AKkepLTh2hOroT7a+7czfdQ=
+gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7/go.mod h1:dt/ZhP58zS4L8KSrWDmTeBkI65Dw0HsyUHuEVlX15mw=
+gopkg.in/yaml.v2 v2.2.1/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
+gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
+gopkg.in/yaml.v2 v2.2.8 h1:obN1ZagJSUGI0Ek/LBmuj4SNLPfIny3KsKFopxRdj10=
+gopkg.in/yaml.v2 v2.2.8/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
+gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
+gorm.io/driver/mysql v1.4.7 h1:rY46lkCspzGHn7+IYsNpSfEv9tA+SU4SkkB+GFX125Y=
+gorm.io/driver/mysql v1.4.7/go.mod h1:SxzItlnT1cb6e1e4ZRpgJN2VYtcqJgqnHxWr4wsP8oc=
+gorm.io/gorm v1.23.8/go.mod h1:l2lP/RyAtc1ynaTjFksBde/O8v9oOGIApu2/xRitmZk=
+gorm.io/gorm v1.25.1 h1:nsSALe5Pr+cM3V1qwwQ7rOkw+6UeLrX5O4v3llhHa64=
+gorm.io/gorm v1.25.1/go.mod h1:L4uxeKpfBml98NYqVqwAdmV1a2nBtAec/cf3fpucW/k=
--- a/api/go/handler/admin_handler.go
+++ b/api/go/handler/admin_handler.go
@@ -0,0 +1,29 @@
+package handler
+
+import (
+	"chatplus/core"
+	"chatplus/core/types"
+	"github.com/gin-gonic/gin"
+	"gorm.io/gorm"
+)
+
+type AdminHandler struct {
+	BaseHandler
+	db *gorm.DB
+}
+
+func NewAdminHandler(config *types.AppConfig, app *core.AppServer, db *gorm.DB) *AdminHandler {
+	handler := AdminHandler{db: db}
+	handler.app = app
+	handler.config = config
+	return &handler
+}
+
+// Login 登录
+func (h *AdminHandler) Login(c *gin.Context) {
+
+}
+
+// Logout 注销
+func (h *AdminHandler) Logout(c *gin.Context) {
+}
--- a/api/go/handler/api_key_handler.go
+++ b/api/go/handler/api_key_handler.go
@@ -0,0 +1,76 @@
+package handler
+
+import (
+	"chatplus/core"
+	"chatplus/core/types"
+	"chatplus/store/model"
+	"chatplus/store/vo"
+	"chatplus/utils"
+	"chatplus/utils/param"
+	"chatplus/utils/resp"
+	"github.com/gin-gonic/gin"
+	"gorm.io/gorm"
+	"net/http"
+)
+
+type ApiKeyHandler struct {
+	BaseHandler
+	db *gorm.DB
+}
+
+func NewApiKeyHandler(config *types.AppConfig, app *core.AppServer, db *gorm.DB) *ApiKeyHandler {
+	handler := ApiKeyHandler{db: db}
+	handler.app = app
+	handler.config = config
+	return &handler
+}
+
+func (h *ApiKeyHandler) Add(c *gin.Context) {
+	var data struct {
+		Key string
+	}
+	if err := c.ShouldBindJSON(&data); err != nil {
+		resp.ERROR(c, types.InvalidArgs)
+		return
+	}
+	// 获取当前登录用户
+	var userId uint = 0
+	user, err := utils.GetLoginUser(c, h.db)
+	if err == nil {
+		userId = user.Id
+	}
+	var key = model.ApiKey{Value: data.Key, UserId: userId}
+	res := h.db.Create(&key)
+	if res.Error != nil {
+		c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "操作失败"})
+		return
+	}
+	resp.SUCCESS(c, key)
+}
+
+func (h *ApiKeyHandler) List(c *gin.Context) {
+	page := param.GetInt(c, "page", 1)
+	pageSize := param.GetInt(c, "page_size", 20)
+	offset := (page - 1) * pageSize
+	var items []model.ApiKey
+	var keys = make([]vo.ApiKey, 0)
+	var total int64
+	h.db.Model(&model.ApiKey{}).Count(&total)
+	res := h.db.Offset(offset).Limit(pageSize).Find(&items)
+	if res.Error == nil {
+		for _, item := range items {
+			var key vo.ApiKey
+			err := utils.CopyObject(item, &key)
+			if err == nil {
+				key.Id = item.Id
+				key.CreatedAt = item.CreatedAt.Unix()
+				key.UpdatedAt = item.UpdatedAt.Unix()
+				keys = append(keys, key)
+			} else {
+				logger.Error(err)
+			}
+		}
+	}
+	pageVo := vo.NewPage(total, page, pageSize, keys)
+	resp.SUCCESS(c, pageVo)
+}
--- a/api/go/handler/base_handler.go
+++ b/api/go/handler/base_handler.go
@@ -0,0 +1,14 @@
+package handler
+
+import (
+	"chatplus/core"
+	"chatplus/core/types"
+	logger2 "chatplus/logger"
+)
+
+var logger = logger2.GetLogger()
+
+type BaseHandler struct {
+	app    *core.AppServer
+	config *types.AppConfig
+}
--- a/api/go/handler/chat_handler.go
+++ b/api/go/handler/chat_handler.go
@@ -0,0 +1,453 @@
+package handler
+
+import (
+	"bufio"
+	"bytes"
+	"chatplus/core"
+	"chatplus/core/types"
+	"chatplus/store/model"
+	"chatplus/store/vo"
+	"chatplus/utils"
+	"chatplus/utils/param"
+	"chatplus/utils/resp"
+	"context"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"github.com/gin-gonic/gin"
+	"github.com/gorilla/websocket"
+	"gorm.io/gorm"
+	"io"
+	"net/http"
+	"net/url"
+	"strings"
+	"time"
+	"unicode/utf8"
+)
+
+const ErrorMsg = "抱歉，AI 助手开小差了，请马上联系管理员去盘它。"
+
+type ChatHandler struct {
+	BaseHandler
+	db *gorm.DB
+}
+
+func NewChatHandler(config *types.AppConfig,
+	app *core.AppServer,
+	db *gorm.DB) *ChatHandler {
+	handler := ChatHandler{db: db}
+	handler.app = app
+	handler.config = config
+	return &handler
+}
+
+// ChatHandle 处理聊天 WebSocket 请求
+func (h *ChatHandler) ChatHandle(c *gin.Context) {
+	ws, err := (&websocket.Upgrader{CheckOrigin: func(r *http.Request) bool { return true }}).Upgrade(c.Writer, c.Request, nil)
+	if err != nil {
+		logger.Error(err)
+		return
+	}
+	sessionId := c.Query("sessionId")
+	roleId := param.GetInt(c, "roleId", 0)
+	chatId := c.Query("chatId")
+	chatModel := c.Query("model")
+	session, ok := h.app.ChatSession[sessionId]
+	if !ok {
+		logger.Info("用户未登录")
+		c.Abort()
+		return
+	}
+
+	// use old chat data override the chat model and role ID
+	var chat model.ChatItem
+	res := h.db.Where("chat_id=?", chatId).First(&chat)
+	if res.Error == nil {
+		chatModel = chat.Model
+		roleId = int(chat.RoleId)
+	}
+
+	session.ChatId = chatId
+	session.Model = chatModel
+	logger.Infof("New websocket connected, IP: %s, UserId: %s", c.Request.RemoteAddr, session.Username)
+	client := core.NewWsClient(ws)
+	var chatRole model.ChatRole
+	res = h.db.First(&chatRole, roleId)
+	if res.Error != nil || !chatRole.Enable {
+		replyMessage(client, "当前聊天角色不存在或者未启用！！！")
+		c.Abort()
+		return
+	}
+
+	// 保存会话连接
+	h.app.ChatClients[chatId] = client
+	go func() {
+		for {
+			_, message, err := client.Receive()
+			if err != nil {
+				logger.Error(err)
+				client.Close()
+				delete(h.app.ChatClients, chatId)
+				delete(h.app.ReqCancelFunc, chatId)
+				return
+			}
+			logger.Info("Receive a message: ", string(message))
+			//replyMessage(client, "这是一条测试消息！")
+			ctx, cancel := context.WithCancel(context.Background())
+			h.app.ReqCancelFunc[chatId] = cancel
+			// 回复消息
+			err = h.sendMessage(ctx, session, chatRole, string(message), client)
+			if err != nil {
+				logger.Error(err)
+			} else {
+				replyChunkMessage(client, types.WsMessage{Type: types.WsEnd})
+				logger.Info("回答完毕: " + string(message))
+			}
+
+		}
+	}()
+}
+
+// 将消息发送给 ChatGPT 并获取结果，通过 WebSocket 推送到客户端
+func (h *ChatHandler) sendMessage(ctx context.Context, session types.ChatSession, role model.ChatRole, prompt string, ws core.Client) error {
+	promptCreatedAt := time.Now() // 记录提问时间
+
+	var user model.User
+	res := h.db.Model(&model.User{}).First(&user, session.UserId)
+	if res.Error != nil {
+		replyMessage(ws, "非法用户，请联系管理员！")
+		return res.Error
+	}
+	var userVo vo.User
+	err := utils.CopyObject(user, &userVo)
+	userVo.Id = user.Id
+	if err != nil {
+		return errors.New("User 对象转换失败，" + err.Error())
+	}
+
+	if userVo.Status == false {
+		replyMessage(ws, "您的账号已经被禁用，如果疑问，请联系管理员！")
+		replyMessage(ws, "![](images/wx.png)")
+		return nil
+	}
+
+	if userVo.Calls <= 0 {
+		replyMessage(ws, "您的对话次数已经用尽，请联系管理员充值！")
+		replyMessage(ws, "![](images/wx.png)")
+		return nil
+	}
+
+	if userVo.ExpiredTime > 0 && userVo.ExpiredTime <= time.Now().Unix() {
+		replyMessage(ws, "您的账号已经过期，请联系管理员！")
+		replyMessage(ws, "![](images/wx.png)")
+		return nil
+	}
+	var req = types.ApiRequest{
+		Model:       session.Model,
+		Temperature: userVo.ChatConfig.Temperature,
+		MaxTokens:   userVo.ChatConfig.MaxTokens,
+		Stream:      true,
+	}
+
+	// 加载聊天上下文
+	var chatCtx []types.Message
+	if userVo.ChatConfig.EnableContext {
+		if v, ok := h.app.ChatContexts[session.ChatId]; ok {
+			chatCtx = v
+		} else {
+			// 加载角色信息
+			var messages []types.Message
+			err := utils.JsonDecode(role.Context, &messages)
+			if err == nil {
+				chatCtx = messages
+			}
+			// TODO: 这里默认加载最近 4 条聊天记录作为上下文，后期应该做成可配置的
+			var historyMessages []model.HistoryMessage
+			res := h.db.Where("chat_id = ?", session.ChatId).Limit(4).Order("created_at desc").Find(&historyMessages)
+			if res.Error == nil {
+				for _, msg := range historyMessages {
+					ms := types.Message{Role: "user", Content: msg.Content}
+					if msg.Type == types.REPLY_MSG {
+						ms.Role = "assistant"
+					}
+					chatCtx = append(chatCtx, ms)
+				}
+			}
+		}
+		logger.Info("聊天上下文：", chatCtx)
+	}
+	req.Messages = append(chatCtx, types.Message{
+		Role:    "user",
+		Content: prompt,
+	})
+	var apiKey string
+	response, err := h.doRequest(ctx, userVo, &apiKey, req)
+	if err != nil {
+		if strings.Contains(err.Error(), "context canceled") {
+			logger.Info("用户取消了请求：", prompt)
+			return nil
+		} else {
+			logger.Error(err)
+		}
+
+		replyMessage(ws, ErrorMsg)
+		replyMessage(ws, "![](images/wx.png)")
+		return err
+	} else {
+		defer response.Body.Close()
+	}
+
+	contentType := response.Header.Get("Content-Type")
+	if strings.Contains(contentType, "text/event-stream") {
+		replyCreatedAt := time.Now()
+		// 循环读取 Chunk 消息
+		var message = types.Message{}
+		var contents = make([]string, 0)
+		var responseBody = types.ApiResponse{}
+		reader := bufio.NewReader(response.Body)
+		for {
+			line, err := reader.ReadString('\n')
+			if err != nil {
+				if strings.Contains(err.Error(), "context canceled") {
+					logger.Info("用户取消了请求：", prompt)
+				} else {
+					logger.Error(err)
+				}
+				break
+			}
+			if !strings.Contains(line, "data:") {
+				continue
+			}
+
+			err = json.Unmarshal([]byte(line[6:]), &responseBody)
+			if err != nil { // 数据解析出错
+				logger.Error(err, line)
+				replyMessage(ws, ErrorMsg)
+				replyMessage(ws, "![](images/wx.png)")
+				break
+			}
+
+			// 初始化 role
+			if responseBody.Choices[0].Delta.Role != "" && message.Role == "" {
+				message.Role = responseBody.Choices[0].Delta.Role
+				replyChunkMessage(ws, types.WsMessage{Type: types.WsStart})
+				continue
+			} else if responseBody.Choices[0].FinishReason != "" {
+				break // 输出完成或者输出中断了
+			} else {
+				content := responseBody.Choices[0].Delta.Content
+				contents = append(contents, content)
+				replyChunkMessage(ws, types.WsMessage{
+					Type:    types.WsMiddle,
+					Content: responseBody.Choices[0].Delta.Content,
+				})
+			}
+		} // end for
+
+		// 消息发送成功
+		if len(contents) > 0 {
+			// 更新用户的对话次数
+			res := h.db.Model(&user).UpdateColumn("calls", gorm.Expr("calls - ?", 1))
+			if res.Error != nil {
+				return res.Error
+			}
+
+			if message.Role == "" {
+				message.Role = "assistant"
+			}
+			message.Content = strings.Join(contents, "")
+			useMsg := types.Message{Role: "user", Content: prompt}
+
+			// 更新上下文消息
+			if userVo.ChatConfig.EnableContext {
+				chatCtx = append(chatCtx, useMsg)  // 提问消息
+				chatCtx = append(chatCtx, message) // 回复消息
+				h.app.ChatContexts[session.ChatId] = chatCtx
+			}
+
+			// 追加聊天记录
+			if userVo.ChatConfig.EnableHistory {
+				// for prompt
+				token, err := utils.CalcTokens(prompt, req.Model)
+				if err != nil {
+					logger.Error(err)
+				}
+				historyUserMsg := model.HistoryMessage{
+					UserId:  userVo.Id,
+					ChatId:  session.ChatId,
+					RoleId:  role.Id,
+					Type:    types.PROMPT_MSG,
+					Icon:    user.Avatar,
+					Content: prompt,
+					Tokens:  token,
+				}
+				historyUserMsg.CreatedAt = promptCreatedAt
+				historyUserMsg.UpdatedAt = promptCreatedAt
+				res := h.db.Save(&historyUserMsg)
+				if res.Error != nil {
+					logger.Error("failed to save prompt history message: ", res.Error)
+				}
+
+				// for reply
+				token, err = utils.CalcTokens(message.Content, req.Model)
+				if err != nil {
+					logger.Error(err)
+				}
+				historyReplyMsg := model.HistoryMessage{
+					UserId:  userVo.Id,
+					ChatId:  session.ChatId,
+					RoleId:  role.Id,
+					Type:    types.REPLY_MSG,
+					Icon:    role.Icon,
+					Content: message.Content,
+					Tokens:  token,
+				}
+				historyReplyMsg.CreatedAt = replyCreatedAt
+				historyReplyMsg.UpdatedAt = replyCreatedAt
+				res = h.db.Create(&historyReplyMsg)
+				if res.Error != nil {
+					logger.Error("failed to save reply history message: ", res.Error)
+				}
+			}
+
+			// 保存当前会话
+			var chatItem model.ChatItem
+			res = h.db.Where("chat_id = ?", session.ChatId).First(&chatItem)
+			if res.Error != nil {
+				chatItem.ChatId = session.ChatId
+				chatItem.UserId = session.UserId
+				chatItem.RoleId = role.Id
+				chatItem.Model = session.Model
+				if utf8.RuneCountInString(prompt) > 30 {
+					chatItem.Title = string([]rune(prompt)[:30]) + "..."
+				} else {
+					chatItem.Title = prompt
+				}
+				h.db.Create(&chatItem)
+			}
+		}
+	} else {
+		body, err := io.ReadAll(response.Body)
+		if err != nil {
+			return fmt.Errorf("error with reading response: %v", err)
+		}
+		var res types.ApiError
+		err = json.Unmarshal(body, &res)
+		if err != nil {
+			return fmt.Errorf("error with decode response: %v", err)
+		}
+
+		// OpenAI API 调用异常处理
+		// TODO: 是否考虑重发消息？
+		if strings.Contains(res.Error.Message, "This key is associated with a deactivated account") {
+			replyMessage(ws, "请求 OpenAI API 失败：API KEY 所关联的账户被禁用。")
+			// 移除当前 API key
+			h.db.Where("value = ?", apiKey).Delete(&model.ApiKey{})
+		} else if strings.Contains(res.Error.Message, "You exceeded your current quota") {
+			replyMessage(ws, "请求 OpenAI API 失败：API KEY 触发并发限制，请稍后再试。")
+		} else if strings.Contains(res.Error.Message, "This model's maximum context length") {
+			replyMessage(ws, "当前会话上下文长度超出限制，已为您删减会话上下文！")
+			// 只保留最近的三条记录
+			chatContext := h.app.ChatContexts[session.ChatId]
+			chatContext = chatContext[len(chatContext)-3:]
+			h.app.ChatContexts[session.ChatId] = chatContext
+			return h.sendMessage(ctx, session, role, prompt, ws)
+		} else {
+			replyMessage(ws, "请求 OpenAI API 失败："+res.Error.Message)
+		}
+	}
+
+	return nil
+}
+
+// 发送请求到 OpenAI 服务器
+// useOwnApiKey: 是否使用了用户自己的 API KEY
+func (h *ChatHandler) doRequest(ctx context.Context, user vo.User, apiKey *string, req types.ApiRequest) (*http.Response, error) {
+	var client *http.Client
+	requestBody, err := json.Marshal(req)
+	if err != nil {
+		return nil, err
+	}
+	// 创建 HttpClient 请求对象
+	request, err := http.NewRequest(http.MethodPost, h.app.ChatConfig.ApiURL, bytes.NewBuffer(requestBody))
+	if err != nil {
+		return nil, err
+	}
+
+	request = request.WithContext(ctx)
+	request.Header.Add("Content-Type", "application/json")
+
+	proxyURL := h.config.ProxyURL
+	if proxyURL == "" {
+		client = &http.Client{}
+	} else { // 使用代理
+		uri := url.URL{}
+		proxy, _ := uri.Parse(proxyURL)
+		client = &http.Client{
+			Transport: &http.Transport{
+				Proxy: http.ProxyURL(proxy),
+			},
+		}
+	}
+	// 查询当前用户是否导入了自己的 API KEY
+	if user.ChatConfig.ApiKey != "" {
+		logger.Info("使用用户自己的 API KEY: ", user.ChatConfig.ApiKey)
+		*apiKey = user.ChatConfig.ApiKey
+	} else { // 获取系统的 API KEY
+		var key model.ApiKey
+		res := h.db.Where("user_id = ?", 0).Order("last_used_at ASC").First(&key)
+		if res.Error != nil {
+			return nil, errors.New("no available key, please import key")
+		}
+		*apiKey = key.Value
+		// 更新 API KEY 的最后使用时间
+		h.db.Model(&key).UpdateColumn("last_used_at", time.Now().Unix())
+	}
+
+	logger.Infof("Sending OpenAI request, KEY: %s, PROXY: %s, Model: %s", *apiKey, proxyURL, req.Model)
+	request.Header.Set("Authorization", fmt.Sprintf("Bearer %s", *apiKey))
+	return client.Do(request)
+}
+
+// 回复客户片段端消息
+func replyChunkMessage(client core.Client, message types.WsMessage) {
+	msg, err := json.Marshal(message)
+	if err != nil {
+		logger.Errorf("Error for decoding json data: %v", err.Error())
+		return
+	}
+	err = client.(*core.WsClient).Send(msg)
+	if err != nil {
+		logger.Errorf("Error for reply message: %v", err.Error())
+	}
+}
+
+// 回复客户端一条完整的消息
+func replyMessage(ws core.Client, message string) {
+	replyChunkMessage(ws, types.WsMessage{Type: types.WsStart})
+	replyChunkMessage(ws, types.WsMessage{Type: types.WsMiddle, Content: message})
+	replyChunkMessage(ws, types.WsMessage{Type: types.WsEnd})
+}
+
+// Tokens 统计 token 数量
+func (h *ChatHandler) Tokens(c *gin.Context) {
+	text := c.Query("text")
+	md := c.Query("model")
+	tokens, err := utils.CalcTokens(text, md)
+	if err != nil {
+		resp.ERROR(c, err.Error())
+		return
+	}
+
+	resp.SUCCESS(c, tokens)
+}
+
+// StopGenerate 停止生成
+func (h *ChatHandler) StopGenerate(c *gin.Context) {
+	chatId := c.Query("chat_id")
+	if cancel, ok := h.app.ReqCancelFunc[chatId]; ok {
+		cancel()
+		delete(h.app.ReqCancelFunc, chatId)
+	}
+	resp.SUCCESS(c, types.OkMsg)
+}
--- a/api/go/handler/chat_history_handler.go
+++ b/api/go/handler/chat_history_handler.go
@@ -0,0 +1,157 @@
+package handler
+
+import (
+	"chatplus/core/types"
+	"chatplus/store/model"
+	"chatplus/store/vo"
+	"chatplus/utils"
+	"chatplus/utils/param"
+	"chatplus/utils/resp"
+	"github.com/gin-gonic/gin"
+)
+
+// List 获取会话列表
+func (h *ChatHandler) List(c *gin.Context) {
+	userId := param.GetInt(c, "user_id", 0)
+	if userId == 0 {
+		resp.ERROR(c, "The parameter 'user_id' is needed.")
+		return
+	}
+	var items = make([]vo.ChatItem, 0)
+	var chats []model.ChatItem
+	res := h.db.Where("user_id = ?", userId).Order("id DESC").Find(&chats)
+	if res.Error == nil {
+		var roleIds = make([]uint, 0)
+		for _, chat := range chats {
+			roleIds = append(roleIds, chat.RoleId)
+		}
+		var roles []model.ChatRole
+		res = h.db.Find(&roles, roleIds)
+		if res.Error == nil {
+			roleMap := make(map[uint]model.ChatRole)
+			for _, role := range roles {
+				roleMap[role.Id] = role
+			}
+
+			for _, chat := range chats {
+				var item vo.ChatItem
+				err := utils.CopyObject(chat, &item)
+				if err == nil {
+					item.Id = chat.Id
+					item.Icon = roleMap[chat.RoleId].Icon
+					items = append(items, item)
+				}
+			}
+		}
+
+	}
+	resp.SUCCESS(c, items)
+}
+
+// Update 更新会话标题
+func (h *ChatHandler) Update(c *gin.Context) {
+	var data struct {
+		Id    uint   `json:"id"`
+		Title string `json:"title"`
+	}
+	if err := c.ShouldBindJSON(&data); err != nil {
+		resp.ERROR(c, types.InvalidArgs)
+		return
+	}
+	var m = model.ChatItem{}
+	m.Id = data.Id
+	res := h.db.Model(&m).UpdateColumn("title", data.Title)
+	if res.Error != nil {
+		resp.ERROR(c, "Failed to update database")
+		return
+	}
+
+	resp.SUCCESS(c, types.OkMsg)
+}
+
+// Remove 删除会话
+func (h *ChatHandler) Remove(c *gin.Context) {
+	chatId := param.GetTrim(c, "chat_id")
+	if chatId == "" {
+		resp.ERROR(c, types.InvalidArgs)
+		return
+	}
+	user, err := utils.GetLoginUser(c, h.db)
+	if err != nil {
+		resp.NotAuth(c)
+		return
+	}
+
+	res := h.db.Where("user_id = ? AND chat_id = ?", user.Id, chatId).Delete(&model.ChatItem{})
+	if res.Error != nil {
+		resp.ERROR(c, "Failed to update database")
+		return
+	}
+
+	// 清空会话上下文
+	delete(h.app.ChatContexts, chatId)
+	resp.SUCCESS(c, types.OkMsg)
+}
+
+// History 获取聊天历史记录
+func (h *ChatHandler) History(c *gin.Context) {
+	chatId := c.Query("chat_id") // 会话 ID
+	user, err := utils.GetLoginUser(c, h.db)
+	if err != nil {
+		resp.NotAuth(c)
+		return
+	}
+	var items []model.HistoryMessage
+	var messages = make([]vo.HistoryMessage, 0)
+	res := h.db.Where("chat_id = ? AND user_id = ?", chatId, user.Id).Find(&items)
+	if res.Error != nil {
+		resp.ERROR(c, "No history message")
+		return
+	} else {
+		for _, item := range items {
+			var v vo.HistoryMessage
+			err := utils.CopyObject(item, &v)
+			v.CreatedAt = item.CreatedAt.Unix()
+			v.UpdatedAt = item.UpdatedAt.Unix()
+			if err == nil {
+				messages = append(messages, v)
+			}
+		}
+	}
+
+	resp.SUCCESS(c, messages)
+}
+
+// Clear 清空所有聊天记录
+func (h *ChatHandler) Clear(c *gin.Context) {
+	// 获取当前登录用户所有的聊天会话
+	user, err := utils.GetLoginUser(c, h.db)
+	if err != nil {
+		resp.NotAuth(c)
+		return
+	}
+
+	var chats []model.ChatItem
+	res := h.db.Where("user_id = ?", user.Id).Find(&chats)
+	if res.Error != nil {
+		resp.ERROR(c, "No chats found")
+		return
+	}
+	// 清空聊天记录
+	for _, chat := range chats {
+		err := h.db.Where("chat_id = ? AND user_id = ?", chat.ChatId, user.Id).Delete(&model.HistoryMessage{})
+		if err != nil {
+			logger.Warnf("Failed to delele chat history for ChatID: %s", chat.ChatId)
+		}
+		// 清空会话上下文
+		delete(h.app.ChatContexts, chat.ChatId)
+	}
+	// 删除所有的会话记录
+	res = h.db.Where("user_id = ?", user.Id).Delete(&model.ChatItem{})
+	if res.Error != nil {
+		resp.ERROR(c, "Failed to remove chat from database.")
+		return
+	}
+
+	resp.SUCCESS(c, types.OkMsg)
+}
--- a/api/go/handler/chat_role_handler.go
+++ b/api/go/handler/chat_role_handler.go
@@ -0,0 +1,92 @@
+package handler
+
+import (
+	"chatplus/core"
+	"chatplus/core/types"
+	"chatplus/service"
+	"chatplus/store/model"
+	"chatplus/store/vo"
+	"chatplus/utils"
+	"chatplus/utils/resp"
+	"github.com/gin-gonic/gin"
+	"strconv"
+)
+
+type ChatRoleHandler struct {
+	BaseHandler
+	service *service.ChatRoleService
+}
+
+func NewChatRoleHandler(config *types.AppConfig, app *core.AppServer, service *service.ChatRoleService) *ChatRoleHandler {
+	handler := &ChatRoleHandler{service: service}
+	handler.app = app
+	handler.config = config
+	return handler
+}
+
+// List get user list
+func (h *ChatRoleHandler) List(c *gin.Context) {
+	var roles []model.ChatRole
+	res := h.service.DB.Find(&roles)
+	if res.Error != nil {
+		resp.ERROR(c, "No roles found,"+res.Error.Error())
+		return
+	}
+	userId, err := strconv.Atoi(c.Query("user_id"))
+	if err == nil && userId > 0 {
+		var user model.User
+		h.service.DB.First(&user, userId)
+		var roleMap map[string]int
+		err := utils.JsonDecode(user.ChatRoles, &roleMap)
+		if err == nil {
+			for index, r := range roles {
+				if _, ok := roleMap[r.Key]; !ok {
+					roles = append(roles[:index], roles[index+1:]...)
+				}
+			}
+		}
+	}
+	// 转成 vo
+	var roleVos = make([]vo.ChatRole, 0)
+	for _, r := range roles {
+		var v vo.ChatRole
+		err := utils.CopyObject(r, &v)
+		if err == nil {
+			v.Id = r.Id
+			roleVos = append(roleVos, v)
+		}
+	}
+	resp.SUCCESS(c, roleVos)
+}
+
+// Add 添加一个聊天角色
+func (h *ChatRoleHandler) Add(c *gin.Context) {
+	var data vo.ChatRole
+	if err := c.ShouldBindJSON(&data); err != nil {
+		resp.ERROR(c, types.InvalidArgs)
+		return
+	}
+
+	if data.Key == "" || data.Name == "" || data.Icon == "" {
+		resp.ERROR(c, types.InvalidArgs)
+		return
+	}
+
+	err := h.service.Create(data)
+	if err != nil {
+		resp.ERROR(c, "Save failed: "+err.Error())
+		return
+	}
+
+	resp.SUCCESS(c, data)
+}
+
+// Get 获取指定的角色
+func (h *ChatRoleHandler) Get(c *gin.Context) {
+
+}
+
+// Update 更新某个聊天角色信息，这里只允许更改名称以及启用和禁用角色操作
+func (h *ChatRoleHandler) Update(c *gin.Context) {
+
+}
--- a/api/go/handler/config_handler.go
+++ b/api/go/handler/config_handler.go
@@ -0,0 +1,77 @@
+package handler
+
+import (
+	"chatplus/core"
+	"chatplus/core/types"
+	"chatplus/store/model"
+	"chatplus/utils"
+	"chatplus/utils/resp"
+	"github.com/gin-gonic/gin"
+	"gorm.io/gorm"
+)
+
+type ConfigHandler struct {
+	BaseHandler
+	db *gorm.DB
+}
+
+func NewConfigHandler(config *types.AppConfig, app *core.AppServer, db *gorm.DB) *ConfigHandler {
+	handler := ConfigHandler{db: db}
+	handler.app = app
+	handler.config = config
+	return &handler
+}
+
+func (h *ConfigHandler) Update(c *gin.Context) {
+	var data struct {
+		Key    string                 `json:"key"`
+		Config map[string]interface{} `json:"config"`
+	}
+
+	if err := c.ShouldBindJSON(&data); err != nil {
+		resp.ERROR(c, types.InvalidArgs)
+		return
+	}
+	str := utils.JsonEncode(&data.Config)
+	config := model.Config{Key: data.Key, Config: str}
+	res := h.db.FirstOrCreate(&config, model.Config{Key: data.Key})
+	if res.Error != nil {
+		resp.ERROR(c, res.Error.Error())
+		return
+	}
+
+	if config.Id > 0 {
+		config.Config = str
+		res := h.db.Updates(&config)
+		if res.Error != nil {
+			resp.ERROR(c, res.Error.Error())
+			return
+		}
+	}
+
+	resp.SUCCESS(c, config)
+}
+
+func (h *ConfigHandler) Get(c *gin.Context) {
+	key := c.Query("key")
+	var config model.Config
+	res := h.db.Where("marker", key).First(&config)
+	if res.Error != nil {
+		resp.ERROR(c, res.Error.Error())
+		return
+	}
+
+	var m map[string]interface{}
+	err := utils.JsonDecode(config.Config, &m)
+	if err != nil {
+		resp.ERROR(c, err.Error())
+		return
+	}
+
+	resp.SUCCESS(c, m)
+}
+
+// AllGptModels 获取所有的 GPT 模型
+func (h *ConfigHandler) AllGptModels(c *gin.Context) {
+	resp.SUCCESS(c, types.GptModels)
+}
--- a/api/go/handler/user_handler.go
+++ b/api/go/handler/user_handler.go
@@ -0,0 +1,308 @@
+package handler
+
+import (
+	"chatplus/core"
+	"chatplus/core/types"
+	"chatplus/store/model"
+	"chatplus/store/vo"
+	"chatplus/utils"
+	"chatplus/utils/resp"
+	"fmt"
+	"github.com/gin-contrib/sessions"
+	"github.com/gin-gonic/gin"
+	"github.com/lionsoul2014/ip2region/binding/golang/xdb"
+	"gorm.io/gorm"
+	"strings"
+	"time"
+)
+
+type UserHandler struct {
+	BaseHandler
+	db       *gorm.DB
+	searcher *xdb.Searcher
+}
+
+func NewUserHandler(config *types.AppConfig, app *core.AppServer, db *gorm.DB, searcher *xdb.Searcher) *UserHandler {
+	handler := &UserHandler{db: db, searcher: searcher}
+	handler.app = app
+	handler.config = config
+	return handler
+}
+
+// Register user register
+func (h *UserHandler) Register(c *gin.Context) {
+	// parameters process
+	var data struct {
+		Username string `json:"username"`
+		Password string `json:"password"`
+	}
+	if err := c.ShouldBindJSON(&data); err != nil {
+		resp.ERROR(c, types.InvalidArgs)
+		return
+	}
+	data.Username = strings.TrimSpace(data.Username)
+	data.Password = strings.TrimSpace(data.Password)
+
+	if len(data.Username) < 5 {
+		resp.ERROR(c, "用户名长度不能少于5个字符")
+		return
+	}
+	if len(data.Password) < 8 {
+		resp.ERROR(c, "密码长度不能少于8个字符")
+		return
+	}
+
+	// check if the username is exists
+	var item model.User
+	tx := h.db.Where("username = ?", data.Username).First(&item)
+	if tx.RowsAffected > 0 {
+		resp.ERROR(c, "用户名已存在")
+		return
+	}
+	// 默认订阅所有角色
+	var chatRoles []model.ChatRole
+	h.db.Find(&chatRoles)
+	var roleMap = make(map[string]int)
+	for _, r := range chatRoles {
+		roleMap[r.Key] = 1
+	}
+
+	salt := utils.RandString(8)
+	user := model.User{
+		Username:  data.Username,
+		Password:  utils.GenPassword(data.Password, salt),
+		Nickname:  fmt.Sprintf("极客学长@%d", utils.RandomNumber(5)),
+		Avatar:    "images/avatar/user.png",
+		Salt:      salt,
+		Status:    true,
+		ChatRoles: utils.JsonEncode(roleMap),
+		ChatConfig: utils.JsonEncode(types.ChatConfig{
+			Temperature:   h.app.ChatConfig.Temperature,
+			MaxTokens:     h.app.ChatConfig.MaxTokens,
+			EnableContext: h.app.ChatConfig.EnableContext,
+			EnableHistory: true,
+			Model:         h.app.ChatConfig.Model,
+			ApiKey:        "",
+		}),
+	}
+	res := h.db.Create(&user)
+	if res.Error != nil {
+		resp.ERROR(c, "保存数据失败")
+		logger.Error(res.Error)
+		return
+	}
+
+	resp.SUCCESS(c, user)
+}
+
+func (h *UserHandler) List(c *gin.Context) {
+	var users []model.User
+	res := h.db.Find(&users)
+	if res.Error != nil {
+		resp.ERROR(c, "No user found")
+		logger.Error("get user failed: ", res.Error.Error())
+		return
+	}
+
+	// 转成 VO 输出
+	var userVos = make([]vo.User, 0)
+	for _, u := range users {
+		logger.Info(u)
+		var v vo.User
+		err := utils.CopyObject(u, &v)
+		if err == nil {
+			v.Id = u.Id
+			v.CreatedAt = u.CreatedAt.Unix()
+			v.UpdatedAt = u.UpdatedAt.Unix()
+			userVos = append(userVos, v)
+		}
+	}
+	resp.SUCCESS(c, userVos)
+}
+
+// Login 用户登录
+func (h *UserHandler) Login(c *gin.Context) {
+	var data struct {
+		Username string
+		Password string
+	}
+	if err := c.ShouldBindJSON(&data); err != nil {
+		resp.ERROR(c, types.InvalidArgs)
+		return
+	}
+	var user model.User
+	res := h.db.Where("username = ?", data.Username).First(&user)
+	if res.Error != nil {
+		resp.ERROR(c, "用户名不存在")
+		return
+	}
+
+	password := utils.GenPassword(data.Password, user.Salt)
+	if password != user.Password {
+		resp.ERROR(c, "用户名或密码错误")
+		return
+	}
+
+	// 更新最后登录时间和IP
+	user.LastLoginIp = c.ClientIP()
+	user.LastLoginAt = time.Now().Unix()
+	h.db.Model(&user).Updates(user)
+
+	sessionId := utils.RandString(42)
+	c.Header(types.TokenSessionName, sessionId)
+	err := utils.SetLoginUser(c, user.Id)
+	if err != nil {
+		resp.ERROR(c, "保存会话失败")
+		logger.Error("Error for save session: ", err)
+		return
+	}
+
+	// 记录登录信息在服务器
+	h.app.ChatSession[sessionId] = types.ChatSession{ClientIP: c.ClientIP(), UserId: user.Id, Username: data.Username, SessionId: sessionId}
+
+	// 加载用户订阅的聊天角色
+	var roleMap map[string]int
+	err = utils.JsonDecode(user.ChatRoles, &roleMap)
+	var chatRoles interface{}
+	if err == nil {
+		roleKeys := make([]string, 0)
+		for key := range roleMap {
+			roleKeys = append(roleKeys, key)
+		}
+		var roles []model.ChatRole
+		res = h.db.Where("marker IN ?", roleKeys).Find(&roles)
+		if res.Error == err {
+			type Item struct {
+				Name string
+				Key  string
+				Icon string
+			}
+			items := make([]Item, 0)
+			for _, r := range roles {
+				items = append(items, Item{Name: r.Name, Key: r.Key, Icon: r.Icon})
+			}
+			chatRoles = items
+		}
+	}
+
+	h.db.Create(&model.UserLoginLog{
+		UserId:       user.Id,
+		Username:     user.Username,
+		LoginIp:      c.ClientIP(),
+		LoginAddress: utils.Ip2Region(h.searcher, c.ClientIP()),
+	})
+	var chatConfig types.ChatConfig
+	err = utils.JsonDecode(user.ChatConfig, &chatConfig)
+	if err != nil {
+		resp.ERROR(c, err.Error())
+		return
+	}
+
+	resp.SUCCESS(c, gin.H{
+		"session_id":     sessionId,
+		"id":             user.Id,
+		"nickname":       user.Nickname,
+		"avatar":         user.Avatar,
+		"username":       user.Username,
+		"tokens":         user.Tokens,
+		"calls":          user.Calls,
+		"expiredTime":    user.ExpiredTime,
+		"chatRoles":      chatRoles,
+		"api_key":        chatConfig.ApiKey,
+		"model":          chatConfig.Model,
+		"temperature":    chatConfig.Temperature,
+		"max_tokens":     chatConfig.MaxTokens,
+		"enable_context": chatConfig.EnableContext,
+		"enable_history": chatConfig.EnableHistory,
+	})
+}
+
+// Logout 注 销
+func (h *UserHandler) Logout(c *gin.Context) {
+	sessionId := c.GetHeader(types.TokenSessionName)
+	session := sessions.Default(c)
+	session.Delete(sessionId)
+	err := session.Save()
+	if err != nil {
+		logger.Error("Error for save session: ", err)
+	}
+	// 删除 websocket 会话列表
+	delete(h.app.ChatSession, sessionId)
+	// 关闭 socket 连接
+	if client, ok := h.app.ChatClients[sessionId]; ok {
+		client.Close()
+	}
+	resp.SUCCESS(c)
+}
+
+// Session 获取/验证会话
+func (h *UserHandler) Session(c *gin.Context) {
+	sessionId := c.GetHeader(types.TokenSessionName)
+	if session, ok := h.app.ChatSession[sessionId]; ok && session.ClientIP == c.ClientIP() {
+		resp.SUCCESS(c, session)
+	} else {
+		resp.NotAuth(c)
+	}
+
+}
+
+func (h *UserHandler) ProfileUpdate(c *gin.Context) {
+	var data vo.User
+	if err := c.ShouldBindJSON(&data); err != nil {
+		resp.ERROR(c, types.InvalidArgs)
+		return
+	}
+
+	user, err := utils.GetLoginUser(c, h.db)
+	if err != nil {
+		resp.NotAuth(c)
+		return
+	}
+	h.db.First(&user, user.Id)
+	user.Nickname = data.Nickname
+	user.Avatar = data.Avatar
+
+	var chatConfig types.ChatConfig
+	err = utils.JsonDecode(user.ChatConfig, &chatConfig)
+	if err != nil {
+		resp.ERROR(c, "用户配置解析失败")
+		return
+	}
+
+	chatConfig.EnableHistory = data.ChatConfig.EnableHistory
+	chatConfig.EnableContext = data.ChatConfig.EnableContext
+	chatConfig.Model = data.ChatConfig.Model
+	chatConfig.MaxTokens = data.ChatConfig.MaxTokens
+	chatConfig.ApiKey = data.ChatConfig.ApiKey
+	chatConfig.Temperature = data.ChatConfig.Temperature
+
+	user.ChatConfig = utils.JsonEncode(chatConfig)
+	res := h.db.Updates(&user)
+	if res.Error != nil {
+		resp.ERROR(c, "更新用户信息失败")
+		return
+	}
+	resp.SUCCESS(c)
+}
+
+func (h *UserHandler) Profile(c *gin.Context) {
+	user, err := utils.GetLoginUser(c, h.db)
+	if err != nil {
+		resp.NotAuth(c)
+		return
+	}
+
+	h.db.First(&user, user.Id)
+	var userVo vo.User
+	err = utils.CopyObject(user, &userVo)
+	if err != nil {
+		logger.Error("对象拷贝失败：", err.Error())
+		resp.ERROR(c, "获取用户信息失败")
+		return
+	}
+
+	userVo.Id = user.Id
+	userVo.CreatedAt = user.CreatedAt.Unix()
+	userVo.UpdatedAt = user.UpdatedAt.Unix()
+	resp.SUCCESS(c, userVo)
+}
--- a/api/go/logger/logger.go
+++ b/api/go/logger/logger.go
@@ -0,0 +1,26 @@
+package logger
+
+import (
+	"go.uber.org/zap"
+)
+
+var logger *zap.SugaredLogger
+
+func GetLogger() *zap.SugaredLogger {
+	if logger != nil {
+		return logger
+	}
+
+	logLevel := zap.NewAtomicLevel()
+	logLevel.SetLevel(zap.InfoLevel)
+	log, _ := zap.Config{
+		Level:            logLevel,
+		Development:      false,
+		Encoding:         "console",
+		EncoderConfig:    zap.NewDevelopmentEncoderConfig(),
+		OutputPaths:      []string{"stderr"},
+		ErrorOutputPaths: []string{"stderr"},
+	}.Build()
+	logger = log.Sugar()
+	return logger
+}
--- a/api/go/main.go
+++ b/api/go/main.go
@@ -0,0 +1,193 @@
+package main
+
+import (
+	"chatplus/core"
+	"chatplus/core/types"
+	"chatplus/handler"
+	logger2 "chatplus/logger"
+	"chatplus/service"
+	"chatplus/store"
+	"context"
+	"flag"
+	"fmt"
+	"github.com/lionsoul2014/ip2region/binding/golang/xdb"
+	"go.uber.org/fx"
+	"gorm.io/gorm"
+	"log"
+	"os"
+	"os/signal"
+	"syscall"
+	"time"
+)
+
+var logger = logger2.GetLogger()
+
+var configFile string
+var debugMode bool
+
+// AppLifecycle 应用程序生命周期
+type AppLifecycle struct {
+}
+
+// OnStart 应用程序启动时执行
+func (l *AppLifecycle) OnStart(context.Context) error {
+	log.Println("AppLifecycle OnStart")
+	return nil
+}
+
+// OnStop 应用程序停止时执行
+func (l *AppLifecycle) OnStop(context.Context) error {
+	log.Println("AppLifecycle OnStop")
+	return nil
+}
+
+func main() {
+	logger.Info("Loading config file: ", configFile)
+	app := fx.New(
+		// 初始化配置应用配置
+		fx.Provide(func() *types.AppConfig {
+			config, err := core.LoadConfig(configFile)
+			if err != nil {
+				log.Fatal(err)
+			}
+			return config
+		}),
+		// 创建应用服务
+		fx.Provide(core.NewServer),
+		// 初始化
+		fx.Invoke(func(s *core.AppServer) {
+			s.Init(debugMode)
+		}),
+
+		// 初始化数据库
+		fx.Provide(store.NewGormConfig),
+		fx.Provide(store.NewMysql),
+
+		// 创建 Ip2Region 查询对象
+		fx.Provide(func() (*xdb.Searcher, error) {
+			dbPath := "res/ip2region.xdb"
+			cBuff, err := xdb.LoadContentFromFile(dbPath)
+			if err != nil {
+				return nil, err
+			}
+
+			return xdb.NewWithBuffer(cBuff)
+		}),
+
+		// 初始化服务
+		fx.Provide(store.NewLevelDB),
+		fx.Provide(service.NewChatRoleService),
+		fx.Invoke(core.InitChatRoles),
+
+		// 创建控制器
+		fx.Provide(handler.NewAdminHandler),
+		fx.Provide(handler.NewChatRoleHandler),
+		fx.Provide(handler.NewUserHandler),
+		fx.Provide(handler.NewChatHandler),
+		fx.Provide(handler.NewApiKeyHandler),
+		fx.Provide(handler.NewConfigHandler),
+
+		// 注册路由
+		fx.Invoke(func(s *core.AppServer, h *handler.AdminHandler) {
+			group := s.Engine.Group("/api/admin/")
+			group.POST("login", h.Login)
+			group.GET("logout", h.Logout)
+		}),
+		fx.Invoke(func(s *core.AppServer, h *handler.ChatRoleHandler) {
+			group := s.Engine.Group("/api/chat/role/")
+			group.GET("list", h.List)
+		}),
+		fx.Invoke(func(s *core.AppServer, h *handler.UserHandler) {
+			group := s.Engine.Group("/api/user/")
+			group.POST("register", h.Register)
+			group.GET("list", h.List)
+			group.POST("login", h.Login)
+			group.GET("logout", h.Logout)
+			group.GET("session", h.Session)
+			group.GET("profile", h.Profile)
+			group.POST("profile/update", h.ProfileUpdate)
+		}),
+		fx.Invoke(func(s *core.AppServer, h *handler.ChatHandler) {
+			group := s.Engine.Group("/api/chat/")
+			group.Any("new", h.ChatHandle)
+			group.GET("list", h.List)
+			group.POST("update", h.Update)
+			group.GET("remove", h.Remove)
+			group.GET("history", h.History)
+			group.GET("clear", h.Clear)
+			group.GET("tokens", h.Tokens)
+			group.GET("stop", h.StopGenerate)
+		}),
+		fx.Invoke(func(s *core.AppServer, h *handler.ApiKeyHandler) {
+			group := s.Engine.Group("/api/apikey/")
+			group.POST("add", h.Add)
+			group.GET("list", h.List)
+		}),
+		fx.Invoke(func(s *core.AppServer, h *handler.ConfigHandler) {
+			group := s.Engine.Group("/api/config/")
+			group.POST("update", h.Update)
+			group.GET("get", h.Get)
+			group.GET("models", h.AllGptModels)
+		}),
+
+		fx.Invoke(func(s *core.AppServer, db *gorm.DB) {
+			err := s.Run(db)
+			if err != nil {
+				log.Fatal(err)
+			}
+		}),
+
+		// 注册生命周期回调函数
+		fx.Invoke(func(lifecycle fx.Lifecycle, lc *AppLifecycle) {
+			lifecycle.Append(fx.Hook{
+				OnStart: func(ctx context.Context) error {
+					return lc.OnStart(ctx)
+				},
+				OnStop: func(ctx context.Context) error {
+					return lc.OnStop(ctx)
+				},
+			})
+		}),
+	)
+	// 启动应用程序
+	go func() {
+		if err := app.Start(context.Background()); err != nil {
+			log.Fatal(err)
+		}
+	}()
+
+	// 监听退出信号
+	quit := make(chan os.Signal, 1)
+	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
+	<-quit
+
+	// 关闭应用程序
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+	if err := app.Stop(ctx); err != nil {
+		log.Fatal(err)
+	}
+
+}
+
+func init() {
+	flag.StringVar(&configFile, "config", "config.toml", "AppConfig file path (default: config.toml)")
+	flag.BoolVar(&debugMode, "debug", true, "Enable debug mode (default: true, recommend to set false in production env)")
+	flag.Usage = usage
+	flag.Parse()
+}
+
+func usage() {
+	fmt.Printf(`ChatGPT-Web-Plus, Version: 2.0.0
+USAGE: 
+  %s [command options]
+OPTIONS:
+`, os.Args[0])
+
+	flagSet := flag.CommandLine
+	order := []string{"config", "debug"}
+	for _, name := range order {
+		f := flagSet.Lookup(name)
+		fmt.Printf("  --%s => %s\n", f.Name, f.Usage)
+	}
+}
--- a/api/go/res/ip2region.xdb
+++ b/api/go/res/ip2region.xdb
--- a/api/go/service/chat_role_service.go
+++ b/api/go/service/chat_role_service.go
@@ -0,0 +1,35 @@
+package service
+
+import (
+	"chatplus/store/model"
+	"chatplus/store/vo"
+	"chatplus/utils"
+	"gorm.io/gorm"
+)
+
+type ChatRoleService struct {
+	DB    *gorm.DB
+	Model interface{}
+}
+
+func NewChatRoleService(db *gorm.DB) *ChatRoleService {
+	return &ChatRoleService{DB: db, Model: &model.ChatRole{}}
+}
+
+func (s *ChatRoleService) Create(value interface{}) error {
+	r, ok := value.(vo.ChatRole)
+
+	if ok {
+		var role model.ChatRole
+		err := utils.CopyObject(r, &role)
+		if err != nil {
+			return err
+		}
+
+		result := s.DB.Create(&role)
+		return result.Error
+	} else {
+		result := s.DB.Create(value)
+		return result.Error
+	}
+}
--- a/api/go/store/leveldb.go
+++ b/api/go/store/leveldb.go
@@ -0,0 +1,97 @@
+package store
+
+import (
+	"chatplus/store/vo"
+	"encoding/json"
+
+	"github.com/syndtr/goleveldb/leveldb"
+	"github.com/syndtr/goleveldb/leveldb/util"
+)
+
+type LevelDB struct {
+	driver *leveldb.DB
+}
+
+func NewLevelDB() (*LevelDB, error) {
+	db, err := leveldb.OpenFile("data/leveldb", nil)
+	if err != nil {
+		return nil, err
+	}
+	return &LevelDB{
+		driver: db,
+	}, nil
+}
+
+func (db *LevelDB) Put(key string, value interface{}) error {
+	bytes, err := json.Marshal(value)
+	if err != nil {
+		return err
+	}
+	return db.driver.Put([]byte(key), bytes, nil)
+}
+
+func (db *LevelDB) Get(key string) ([]byte, error) {
+	bytes, err := db.driver.Get([]byte(key), nil)
+	if err != nil {
+		return nil, err
+	}
+
+	return bytes, nil
+}
+
+func (db *LevelDB) Search(prefix string) []string {
+	var items = make([]string, 0)
+	iter := db.driver.NewIterator(util.BytesPrefix([]byte(prefix)), nil)
+	defer iter.Release()
+
+	for iter.Next() {
+		items = append(items, string(iter.Value()))
+	}
+	return items
+}
+
+func (db *LevelDB) SearchPage(prefix string, page int, pageSize int) *vo.Page {
+	var items = make([]string, 0)
+	iter := db.driver.NewIterator(util.BytesPrefix([]byte(prefix)), nil)
+	defer iter.Release()
+
+	res := &vo.Page{Page: page, PageSize: pageSize}
+	// 计算数据总数和总页数
+	total := 0
+	for iter.Next() {
+		total++
+	}
+	res.TotalPage = (total + pageSize - 1) / pageSize
+	res.Total = int64(total)
+
+	// 计算目标页码的起始和结束位置
+	start := (page - 1) * pageSize
+	if start > total {
+		return nil
+	}
+	end := start + pageSize
+	if end > total {
+		end = total
+	}
+
+	// 跳转到目标页码的起始位置
+	count := 0
+	for iter.Next() {
+		if count >= start {
+			items = append(items, string(iter.Value()))
+		}
+		count++
+	}
+	iter.Release()
+	res.Items = items
+	return res
+}
+
+func (db *LevelDB) Delete(key string) error {
+	return db.driver.Delete([]byte(key), nil)
+}
+
+// Close release resources
+func (db *LevelDB) Close() error {
+	return db.driver.Close()
+}
--- a/api/go/store/model/api_key.go
+++ b/api/go/store/model/api_key.go
@@ -0,0 +1,9 @@
+package model
+
+// ApiKey OpenAI API 模型
+type ApiKey struct {
+	BaseModel
+	UserId     uint   //用户ID，系统添加的用户 ID 为 0
+	Value      string // API Key 的值
+	LastUsedAt int64  // 最后使用时间
+}
--- a/api/go/store/model/base.go
+++ b/api/go/store/model/base.go
@@ -0,0 +1,9 @@
+package model
+
+import "time"
+
+type BaseModel struct {
+	Id        uint `gorm:"primarykey;column:id"`
+	CreatedAt time.Time
+	UpdatedAt time.Time
+}
--- a/api/go/store/model/chat_history.go
+++ b/api/go/store/model/chat_history.go
@@ -0,0 +1,16 @@
+package model
+
+type HistoryMessage struct {
+	BaseModel
+	ChatId  string // 会话 ID
+	UserId  uint   // 用户 ID
+	RoleId  uint   // 角色 ID
+	Type    string
+	Icon    string
+	Tokens  int
+	Content string
+}
+
+func (HistoryMessage) TableName() string {
+	return "chatgpt_chat_history"
+}
--- a/api/go/store/model/chat_item.go
+++ b/api/go/store/model/chat_item.go
@@ -0,0 +1,10 @@
+package model
+
+type ChatItem struct {
+	BaseModel
+	ChatId string `gorm:"column:chat_id;unique"` // 会话 ID
+	UserId uint   // 用户 ID
+	RoleId uint   // 角色 ID
+	Model  string // 会话模型
+	Title  string // 会话标题
+}
--- a/api/go/store/model/chat_role.go
+++ b/api/go/store/model/chat_role.go
@@ -0,0 +1,12 @@
+package model
+
+type ChatRole struct {
+	BaseModel
+	Key      string `gorm:"column:marker;unique"` // 角色唯一标识
+	Name     string // 角色名称
+	Context  string `gorm:"column:context_json"` // 角色语料信息 json
+	HelloMsg string // 打招呼的消息
+	Icon     string // 角色聊天图标
+	Enable   bool   // 是否启用被启用
+	Sort     int    //排序数字
+}
--- a/api/go/store/model/config.go
+++ b/api/go/store/model/config.go
@@ -0,0 +1,7 @@
+package model
+
+type Config struct {
+	Id     uint   `gorm:"primarykey;column:id"`
+	Key    string `gorm:"column:marker;unique"`
+	Config string `gorm:"column:config_json"`
+}
--- a/api/go/store/model/user.go
+++ b/api/go/store/model/user.go
@@ -0,0 +1,18 @@
+package model
+
+type User struct {
+	BaseModel
+	Username    string `gorm:"index:username,unique"`
+	Password    string
+	Nickname    string
+	Avatar      string
+	Salt        string // 密码盐
+	Tokens      int64  // 剩余tokens
+	Calls       int    // 剩余对话次数
+	ChatConfig  string `gorm:"column:chat_config_json"` // 聊天配置 json
+	ChatRoles   string `gorm:"column:chat_roles_json"`  // 聊天角色
+	ExpiredTime int64  // 账户到期时间
+	Status      bool   // 当前状态
+	LastLoginAt int64  // 最后登录时间
+	LastLoginIp string // 最后登录 IP
+}
--- a/api/go/store/model/user_login_log.go
+++ b/api/go/store/model/user_login_log.go
@@ -0,0 +1,9 @@
+package model
+
+type UserLoginLog struct {
+	BaseModel
+	UserId       uint
+	Username     string
+	LoginIp      string
+	LoginAddress string
+}
--- a/api/go/store/mysql.go
+++ b/api/go/store/mysql.go
@@ -0,0 +1,36 @@
+package store
+
+import (
+	"chatplus/core/types"
+	"gorm.io/driver/mysql"
+	"gorm.io/gorm"
+	"gorm.io/gorm/logger"
+	"gorm.io/gorm/schema"
+	"time"
+)
+
+func NewGormConfig() *gorm.Config {
+	return &gorm.Config{
+		Logger: logger.Default.LogMode(logger.Warn),
+		NamingStrategy: schema.NamingStrategy{
+			TablePrefix:   "chatgpt_", // 设置表前缀
+			SingularTable: false,      // 使用单数表名形式
+		},
+	}
+}
+
+func NewMysql(config *gorm.Config, appConfig *types.AppConfig) (*gorm.DB, error) {
+	db, err := gorm.Open(mysql.Open(appConfig.MysqlDns), config)
+
+	sqlDB, err := db.DB()
+	if err != nil {
+		return nil, err
+	}
+	sqlDB.SetMaxIdleConns(32)
+	sqlDB.SetMaxOpenConns(512)
+	sqlDB.SetConnMaxLifetime(time.Hour)
+	if err != nil {
+		return nil, err
+	}
+	return db, nil
+}
--- a/api/go/store/vo/api_key.go
+++ b/api/go/store/vo/api_key.go
@@ -0,0 +1,9 @@
+package vo
+
+// ApiKey OpenAI API 模型
+type ApiKey struct {
+	BaseVo
+	UserId     uint   `json:"user_id"`      //用户ID，系统添加的用户 ID 为 0
+	Value      string `json:"value"`        // API Key 的值
+	LastUsedAt int64  `json:"last_used_at"` // 最后使用时间
+}
--- a/api/go/store/vo/base.go
+++ b/api/go/store/vo/base.go
@@ -0,0 +1,7 @@
+package vo
+
+type BaseVo struct {
+	Id        uint  `json:"id"`
+	CreatedAt int64 `json:"created_at"`
+	UpdatedAt int64 `json:"updated_at"`
+}
--- a/api/go/store/vo/chat_history.go
+++ b/api/go/store/vo/chat_history.go
@@ -0,0 +1,16 @@
+package vo
+
+type HistoryMessage struct {
+	BaseVo
+	ChatId  string `json:"chat_id"`
+	UserId  uint   `json:"user_id"`
+	RoleId  uint   `json:"role_id"`
+	Type    string `json:"type"`
+	Icon    string `json:"icon"`
+	Tokens  int    `json:"tokens"`
+	Content string `json:"content"`
+}
+
+func (HistoryMessage) TableName() string {
+	return "chatgpt_chat_history"
+}
--- a/api/go/store/vo/chat_item.go
+++ b/api/go/store/vo/chat_item.go
@@ -0,0 +1,11 @@
+package vo
+
+type ChatItem struct {
+	BaseVo
+	UserId uint   `json:"user_id"`
+	Icon   string `json:"icon"`
+	RoleId uint   `json:"role_id"`
+	ChatId string `json:"chat_id"`
+	Model  string `json:"model"`
+	Title  string `json:"title"`
+}
--- a/api/go/store/vo/chat_role.go
+++ b/api/go/store/vo/chat_role.go
@@ -0,0 +1,14 @@
+package vo
+
+import "chatplus/core/types"
+
+type ChatRole struct {
+	BaseVo
+	Key      string          `json:"key"`       // 角色唯一标识
+	Name     string          `json:"name"`      // 角色名称
+	Context  []types.Message `json:"context"`   // 角色语料信息
+	HelloMsg string          `json:"hello_msg"` // 打招呼的消息
+	Icon     string          `json:"icon"`      // 角色聊天图标
+	Enable   bool            `json:"enable"`    // 是否启用被启用
+	Sort     int             `json:"sort"`      // 排序
+}
--- a/api/go/store/vo/config.go
+++ b/api/go/store/vo/config.go
@@ -0,0 +1,10 @@
+package vo
+
+import "chatplus/core/types"
+
+type Config struct {
+	Id           uint               `json:"id"`
+	Key          string             `json:"key"`
+	ChatConfig   types.ChatConfig   `json:"chat_config"`
+	SystemConfig types.SystemConfig `json:"system_config"`
+}
--- a/api/go/store/vo/page.go
+++ b/api/go/store/vo/page.go
@@ -0,0 +1,20 @@
+package vo
+
+type Page struct {
+	Items     interface{} `json:"items"`
+	Page      int         `json:"page"`
+	PageSize  int         `json:"page_size"`
+	Total     int64       `json:"total"`
+	TotalPage int         `json:"total_page"`
+}
+
+func NewPage(total int64, page int, pageSize int, items interface{}) Page {
+	totalPage := int(total / int64(pageSize))
+	return Page{
+		Items:     items,
+		Page:      page,
+		PageSize:  pageSize,
+		Total:     total,
+		TotalPage: totalPage,
+	}
+}
--- a/api/go/store/vo/user.go
+++ b/api/go/store/vo/user.go
@@ -0,0 +1,19 @@
+package vo
+
+import "chatplus/core/types"
+
+type User struct {
+	BaseVo
+	Username    string           `json:"username"`
+	Nickname    string           `json:"nickname"`
+	Avatar      string           `json:"avatar"`
+	Salt        string           `json:"salt"`          // 密码盐
+	Tokens      int64            `json:"tokens"`        // 剩余tokens
+	Calls       int              `json:"calls"`         // 剩余对话次数
+	ChatConfig  types.ChatConfig `json:"chat_config"`   // 聊天配置
+	ChatRoles   map[string]int   `json:"chat_roles"`    // 聊天角色集合
+	ExpiredTime int64            `json:"expired_time"`  // 账户到期时间
+	Status      bool             `json:"status"`        // 当前状态
+	LastLoginAt int64            `json:"last_login_at"` // 最后登录时间
+	LastLoginIp string           `json:"last_login_ip"` // 最后登录 IP
+}
--- a/api/go/test/test.go
+++ b/api/go/test/test.go
@@ -0,0 +1,140 @@
+package main
+
+import (
+	"chatplus/store/model"
+	"chatplus/store/vo"
+	"chatplus/utils"
+	"context"
+	"fmt"
+	"github.com/lionsoul2014/ip2region/binding/golang/xdb"
+	"github.com/pkoukk/tiktoken-go"
+	"io"
+	"log"
+	"net/http"
+	"strings"
+	"time"
+)
+
+func main() {
+	//resp.JSON(nil, types.Success, types.OkMsg, new)
+}
+
+// Http client 取消操作
+func testHttpClient(ctx context.Context) {
+
+	req, err := http.NewRequest("GET", "http://localhost:2345", nil)
+	if err != nil {
+		fmt.Println(err)
+		return
+	}
+
+	req = req.WithContext(ctx)
+
+	client := &http.Client{}
+
+	resp, err := client.Do(req)
+	if err != nil {
+		fmt.Println(err)
+		return
+	}
+	defer func(Body io.ReadCloser) {
+		err := Body.Close()
+		if err != nil {
+
+		}
+	}(resp.Body)
+	_, err = io.ReadAll(resp.Body)
+	for {
+		time.Sleep(time.Second)
+		fmt.Println(time.Now())
+		select {
+		case <-ctx.Done():
+			fmt.Println("取消退出")
+			return
+		default:
+			continue
+		}
+	}
+
+}
+
+func testDate() {
+	fmt.Println(time.Unix(1683336167, 0).Format("2006-01-02 15:04:05"))
+}
+
+func testIp2Region() {
+	dbPath := "res/ip2region.xdb"
+	// 1、从 dbPath 加载整个 xdb 到内存
+	cBuff, err := xdb.LoadContentFromFile(dbPath)
+	if err != nil {
+		fmt.Printf("failed to load content from `%s`: %s\n", dbPath, err)
+		return
+	}
+
+	// 2、用全局的 cBuff 创建完全基于内存的查询对象。
+	searcher, err := xdb.NewWithBuffer(cBuff)
+	if err != nil {
+		fmt.Printf("failed to create searcher with content: %s\n", err)
+		return
+	}
+
+	str, err := searcher.SearchByStr("103.88.46.85")
+	fmt.Println(str)
+	if err != nil {
+		log.Fatal(err)
+	}
+	arr := strings.Split(str, "|")
+	fmt.Println(arr[2], arr[3], arr[4])
+
+}
+
+func testJson() {
+
+	var role = model.ChatRole{
+		Key:      "programmer",
+		Name:     "程序员",
+		Context:  "[{\"role\":\"user\",\"content\":\"现在开始你扮演一位程序员，你是一名优秀的程序员，具有很强的逻辑思维能力，总能高效的解决问题。你热爱编程，熟悉多种编程语言，尤其精通 Go 语言，注重代码质量，有创新意识，持续学习，良好的沟通协作。\"},{\"role\"\n:\"assistant\",\"content\":\"好的，现在我将扮演一位程序员，非常感谢您对我的评价。作为一名优秀的程序员，我非常热爱编程，并且注重代码质量。我熟悉多种编程语言，尤其是 Go 语言，可以使用它来高效地解决各种问题。\"}]",
+		HelloMsg: "Talk is cheap, i will show code!",
+		Icon:     "images/avatar/programmer.jpg",
+		Enable:   true,
+		Sort:     1,
+	}
+	role.Id = 1
+	var v vo.ChatRole
+
+	err := utils.CopyObject(role, &v)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	fmt.Printf("%+v\n", v.Id)
+
+	//var v2 = model.ChatRoles{}
+	//err = utils.CopyObject(v, &v2)
+	//if err != nil {
+	//	log.Fatal(err)
+	//}
+	//
+	//fmt.Printf("%+v\n", v2.Id)
+
+}
+
+func calTokens() {
+	text := "须知少年凌云志，曾许人间第一流"
+	encoding := "cl100k_base"
+
+	tke, err := tiktoken.GetEncoding(encoding)
+	if err != nil {
+		err = fmt.Errorf("getEncoding: %v", err)
+		return
+	}
+
+	// encode
+	token := tke.Encode(text, nil, nil)
+
+	//tokens
+	fmt.Println(token)
+	// num_tokens
+	fmt.Println(len(token))
+
+}
--- a/api/go/utils/openai.go
+++ b/api/go/utils/openai.go
@@ -0,0 +1,17 @@
+package utils
+
+import (
+	"fmt"
+	"github.com/pkoukk/tiktoken-go"
+)
+
+func CalcTokens(text string, model string) (int, error) {
+	encoding := tiktoken.MODEL_TO_ENCODING[model]
+	tke, err := tiktoken.GetEncoding(encoding)
+	if err != nil {
+		return 0, fmt.Errorf("getEncoding: %v", err)
+	}
+
+	token := tke.Encode(text, nil, nil)
+	return len(token), nil
+}
--- a/api/go/utils/param/param.go
+++ b/api/go/utils/param/param.go
@@ -0,0 +1,57 @@
+package param
+
+import (
+	"github.com/gin-gonic/gin"
+	"strconv"
+	"strings"
+)
+
+func GetTrim(c *gin.Context, key string) string {
+	return strings.TrimSpace(c.Query(key))
+}
+
+func GetInt(c *gin.Context, key string, defaultValue int) int {
+	return intValue(c.Query(key), defaultValue)
+}
+
+func PostInt(c *gin.Context, key string, defaultValue int) int {
+	return intValue(c.PostForm(key), defaultValue)
+}
+
+func intValue(str string, defaultValue int) int {
+	value, err := strconv.Atoi(str)
+	if err != nil {
+		return defaultValue
+	}
+	return value
+}
+
+func GetFloat(c *gin.Context, key string) float64 {
+	return floatValue(c.Query(key))
+}
+func PostFloat(c *gin.Context, key string) float64 {
+	return floatValue(c.PostForm(key))
+}
+
+func floatValue(str string) float64 {
+	value, err := strconv.ParseFloat(str, 64)
+	if err != nil {
+		return 0
+	}
+	return value
+}
+
+func GetBool(c *gin.Context, key string) bool {
+	return boolValue(c.Query(key))
+}
+func PostBool(c *gin.Context, key string) bool {
+	return boolValue(c.PostForm(key))
+}
+
+func boolValue(str string) bool {
+	value, err := strconv.ParseBool(str)
+	if err != nil {
+		return false
+	}
+	return value
+}
--- a/api/go/utils/resp/response.go
+++ b/api/go/utils/resp/response.go
@@ -0,0 +1,28 @@
+package resp
+
+import (
+	"chatplus/core/types"
+	"github.com/gin-gonic/gin"
+	"net/http"
+)
+
+func SUCCESS(c *gin.Context, values ...interface{}) {
+	if values != nil {
+		c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Data: values[0]})
+	} else {
+		c.JSON(http.StatusOK, types.BizVo{Code: types.Success})
+	}
+
+}
+
+func ERROR(c *gin.Context, messages ...string) {
+	if messages != nil {
+		c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: messages[0]})
+	} else {
+		c.JSON(http.StatusOK, types.BizVo{Code: types.Failed})
+	}
+}
+
+func NotAuth(c *gin.Context) {
+	c.JSON(http.StatusOK, types.BizVo{Code: types.NotAuthorized, Message: "Not Authorized"})
+}
--- a/api/go/utils/string.go
+++ b/api/go/utils/string.go
@@ -0,0 +1,155 @@
+package utils
+
+import (
+	"encoding/json"
+	"fmt"
+	"github.com/lionsoul2014/ip2region/binding/golang/xdb"
+	"golang.org/x/crypto/sha3"
+	"math/rand"
+	"reflect"
+	"strings"
+	"time"
+)
+
+// RandString generate rand string with specified length
+func RandString(length int) string {
+	str := "0123456789abcdefghijklmnopqrstuvwxyz"
+	data := []byte(str)
+	var result []byte
+	r := rand.New(rand.NewSource(time.Now().UnixNano()))
+	for i := 0; i < length; i++ {
+		result = append(result, data[r.Intn(len(data))])
+	}
+	return string(result)
+}
+
+func RandomNumber(bit int) int {
+	rand.Seed(time.Now().UnixNano())
+	min := 1 // min value
+	max := 1 //max value
+	for i := 0; i < bit; i++ {
+		min = min * 10
+		max = max * 10
+	}
+	max = max * 10
+	return rand.Intn(max-min+1) + min
+}
+
+func ContainsStr(slice []string, item string) bool {
+	for _, e := range slice {
+		if e == item {
+			return true
+		}
+	}
+	return false
+}
+
+// Stamp2str 时间戳转字符串
+func Stamp2str(timestamp int64) string {
+	if timestamp == 0 {
+		return ""
+	}
+	return time.Unix(timestamp, 0).Format("2006-01-02 15:04:05")
+}
+
+// Str2stamp 字符串转时间戳
+func Str2stamp(str string) int64 {
+	layout := "2006-01-02 15:04:05"
+	t, err := time.Parse(layout, str)
+	if err != nil {
+		return 0
+	}
+	return t.Unix()
+}
+
+func GenPassword(pass string, salt string) string {
+	data := []byte(pass + salt)
+	hash := sha3.Sum256(data)
+	return fmt.Sprintf("%x", hash)
+}
+
+func JsonEncode(value interface{}) string {
+	bytes, err := json.Marshal(value)
+	if err != nil {
+		return ""
+	}
+	return string(bytes)
+}
+
+func JsonDecode(src string, dest interface{}) error {
+	return json.Unmarshal([]byte(src), dest)
+}
+
+// CopyObject 拷贝对象
+func CopyObject(src interface{}, dst interface{}) error {
+
+	srcType := reflect.TypeOf(src)
+	srcValue := reflect.ValueOf(src)
+	dstValue := reflect.ValueOf(dst).Elem()
+	reflect.TypeOf(dst)
+	for i := 0; i < srcType.NumField(); i++ {
+		field := srcType.Field(i)
+		value := dstValue.FieldByName(field.Name)
+		if !value.IsValid() {
+			continue
+		}
+		// 数据类型相同，直接赋值
+		v := srcValue.FieldByName(field.Name)
+		if value.Type() == field.Type {
+			value.Set(v)
+		} else {
+			// src data type is  string，dst data type is slice, map, struct
+			// use json decode the data
+			if field.Type.Kind() == reflect.String && (value.Type().Kind() == reflect.Struct ||
+				value.Type().Kind() == reflect.Map ||
+				value.Type().Kind() == reflect.Slice) {
+				pType := reflect.New(value.Type())
+				v2 := pType.Interface()
+				err := json.Unmarshal([]byte(v.String()), &v2)
+				if err == nil {
+					value.Set(reflect.ValueOf(v2).Elem())
+				}
+				// string to map, struct, slice
+			} else if (field.Type.Kind() == reflect.Struct ||
+				field.Type.Kind() == reflect.Map ||
+				field.Type.Kind() == reflect.Slice) && value.Type().Kind() == reflect.String {
+				ba, err := json.Marshal(v.Interface())
+				if err == nil {
+					value.Set(reflect.ValueOf(string(ba)))
+				}
+			} else { // 简单数据类型的强制类型转换
+				switch value.Kind() {
+				case reflect.Int:
+				case reflect.Int8:
+				case reflect.Int16:
+				case reflect.Int32:
+				case reflect.Int64:
+					value.SetInt(v.Int())
+					break
+				case reflect.Float32:
+				case reflect.Float64:
+					value.SetFloat(v.Float())
+					break
+				case reflect.Bool:
+					value.SetBool(v.Bool())
+					break
+				}
+			}
+
+		}
+	}
+
+	return nil
+}
+
+func Ip2Region(searcher *xdb.Searcher, ip string) string {
+	str, err := searcher.SearchByStr(ip)
+	if err != nil {
+		return ""
+	}
+	arr := strings.Split(str, "|")
+	if len(arr) < 3 {
+		return arr[0]
+	}
+	return fmt.Sprintf("%s-%s-%s", arr[0], arr[2], arr[3])
+}
--- a/api/go/utils/user.go
+++ b/api/go/utils/user.go
@@ -0,0 +1,37 @@
+package utils
+
+import (
+	"chatplus/core/types"
+	"chatplus/store/model"
+	"errors"
+	"github.com/gin-contrib/sessions"
+	"github.com/gin-gonic/gin"
+	"gorm.io/gorm"
+)
+
+func SetLoginUser(c *gin.Context, userId uint) error {
+	session := sessions.Default(c)
+	session.Set(types.SessionUserId, userId)
+	return session.Save()
+}
+
+func GetLoginUser(c *gin.Context, db *gorm.DB) (model.User, error) {
+	value, exists := c.Get(types.LoginUserCache)
+	if exists {
+		return value.(model.User), nil
+	}
+
+	session := sessions.Default(c)
+	userId := session.Get(types.SessionUserId)
+	if userId == nil {
+		return model.User{}, errors.New("user not login")
+	}
+
+	var user model.User
+	res := db.First(&user, userId)
+	// 更新缓存
+	if res.Error == nil {
+		c.Set(types.LoginUserCache, user)
+	}
+	return user, res.Error
+}