package mainimport ("fmt""github.com/gorilla/websocket""log""net/http"
)// 我们去定义一个 Upgrader
// 这需要一个 Read 和 Write 的缓冲大小
var upgrader = websocket.Upgrader{ReadBufferSize:  1024,WriteBufferSize: 1024,// 我们需要检查连接的来源// 这将允许我们根据我们的 React 发出请求// 现在,我们将不检查就允许任何连接CheckOrigin: func(r *http.Request) bool { return true },
}// 此处为收到消息之后的处理逻辑
func reader(conn *websocket.Conn) {for {// 读取信息messageType, p, err := conn.ReadMessage()if err != nil {log.Println(err)return}str := fmt.Sprintf("%s %s", p, p)fmt.Println(str)if err := conn.WriteMessage(messageType, []byte(str)); err != nil {log.Println(err)return}}
}// 定义我们的 WebSocket 端点
func serveWs(w http.ResponseWriter, r *http.Request) {fmt.Println(r.Host)// 将此连接升级到 WebSocket// 连接ws, err := upgrader.Upgrade(w, r, nil)if err != nil {log.Println(err)}// 持续监听新消息// 通过我们的 WebSocket 连接reader(ws)
}func setupRoutes() {// 将我们的 `/ws` 端点映射到 `serveWs` 函数http.HandleFunc("/ws", serveWs)
}func main() {setupRoutes()http.ListenAndServe(":8080", nil)
}