完成聊天 websocket API 对接

server/chat_handler.go

@@ -1,11 +1,34 @@
 package server
 
 import (
-	"fmt"
 	"github.com/gin-gonic/gin"
+	"github.com/gorilla/websocket"
 	"net/http"
 )
 
 func (s *Server) Chat(c *gin.Context) {
-	c.JSON(http.StatusOK, gin.H{"code": 0, "message": fmt.Sprintf("HELLO, ChatGPT !!!")})
+	ws, err := (&websocket.Upgrader{CheckOrigin: func(r *http.Request) bool { return true }}).Upgrade(c.Writer, c.Request, nil)
+	if err != nil {
+		logger.Fatal(err)
+		return
+	}
+	logger.Infof("New websocket connected, IP: %s", c.Request.RemoteAddr)
+	client := NewWsClient(ws)
+	go func() {
+		for {
+			_, message, err := client.Receive()
+			if err != nil {
+				logger.Error(err)
+				client.Close()
+				return
+			}
+
+			// TODO: 接受消息，调用 ChatGPT 返回消息
+			logger.Info(string(message))
+			err = client.Send(message)
+			if err != nil {
+				logger.Error(err)
+			}
+		}
+	}()
 }