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)
}
使用go将http升级为websocket的代码
最新推荐文章于 2024-05-17 10:04:07 发布