Go语言的WebSocket与实时通信

package main
 
import (
    "fmt"
    "log"
    "net/http"
 
    "github.com/gorilla/websocket"
)
 
var upgrader = websocket.Upgrader{
    CheckOrigin: func(r *http.Request) bool {
        return true // 允许跨域请求，注意生产环境应当设置为安全的域名
    },
}
 
func echo(w http.ResponseWriter, r *http.Request) {
    c, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println(err)
        return
    }
    defer c.Close()
 
    for {
        mt, message, err := c.ReadMessage()
        if err != nil {
            log.Println(err)
            break
        }
 
        err = c.WriteMessage(mt, message)
        if err != nil {
            log.Println(err)
            break
        }
    }
}
 
func main() {
    http.HandleFunc("/echo", echo)
    fmt.Printf("Starting server on :8080\n")
    if err := http.ListenAndServe(":8080", nil); err != nil {
        log.Fatal("ListenAndServe: ", err)
    }
}

这段代码实现了一个简单的WebSocket服务器，它将接收到的任何消息回显给客户端。服务器监听8080端口上的/echo路径。代码使用gorilla/websocket库来简化WebSocket的实现。服务器允许跨域请求，在实际应用中应当根据安全策略来配置。