瀏覽代碼

wip:并发限制提交

wangkaiyue 2 年之前
父節點
當前提交
6417f6175d
共有 1 個文件被更改,包括 2 次插入4 次删除
  1. 2 4
      internal/model/ws.go

+ 2 - 4
internal/model/ws.go

@@ -119,6 +119,8 @@ func (m *WsChat) Handle(ws *ghttp.WebSocket, msg []byte) {
 		for {
 			line, _, err := buf.ReadLine()
 			if err == io.EOF {
+				//放回链接池
+				ChatGptPool.Add()
 				replyId := ChatHistory.Save(m.Ctx, &ChatRecord{
 					Content:    lastData.Response,
 					Type:       2,
@@ -140,10 +142,6 @@ func (m *WsChat) Handle(ws *ghttp.WebSocket, msg []byte) {
 	}
 }
 
-func saveAndReturnMap() {
-
-}
-
 func parseEventStream(line []byte) (event string, date *BufRes) {
 	// 如果行以 "event:" 开头,表示这是一个事件的标识符
 	if len(line) > 6 && string(line[:6]) == "event:" {