Golang Websocket框架：实时通信的新选择，2024年最新不看绝对血亏

package main
 
import (
    "fmt"
    "log"
    "net/http"
 
    "github.com/gorilla/websocket"
)
 
var upgrader = websocket.Upgrader{
    CheckOrigin: func(r *http.Request) bool {
        return true // 允许跨域请求
    },
}
 
// 处理Websocket连接
func handleConnections(w http.ResponseWriter, r *http.Request) {
    // 尝试升级HTTP连接到Websocket
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println(err)
        return
    }
    defer conn.Close()
 
    // 创建一个goroutine来处理发送和接收消息
    go func() {
        for {
            messageType, p, err := conn.ReadMessage()
            if err != nil {
                log.Println(err)
                return
            }
            log.Printf("接收到消息: %s\n", p)
 
            err = conn.WriteMessage(messageType, p)
            if err != nil {
                log.Println(err)
                return
            }
        }
    }()
}
 
func main() {
    http.HandleFunc("/ws", handleConnections)
    fmt.Println("服务器启动中...")
    log.Fatal(http.ListenAndServe(":8080", nil))
}

这段代码使用了Gorilla Websocket库来创建一个简单的Websocket服务器。它定义了一个upgrader用于处理HTTP连接升级到Websocket连接，允许跨域请求。handleConnections函数处理单个Websocket连接，包括消息的接收和发送。代码简洁，注重原理，适合学习和教学。