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连接。