使用go将http升级为websocket的代码

package main

import (
	"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)
}