完成前后端框架搭建，完成聊天页面布局

server/client.go

@@ -0,0 +1,65 @@
+package server
+
+import (
+	"errors"
+	"github.com/gorilla/websocket"
+	"sync"
+)
+
+var ErrConClosed = errors.New("connection closed")
+
+type Client interface {
+	Close()
+}
+
+// WsClient websocket client
+type WsClient struct {
+	NodeId    string
+	SessionId string
+	Conn      *websocket.Conn
+	lock      sync.Mutex
+	mt        int
+	closed    bool
+}
+
+func NewWsClient(nodeId string, sessionId string, conn *websocket.Conn) *WsClient {
+	return &WsClient{
+		NodeId:    nodeId,
+		SessionId: sessionId,
+		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
+}