1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51
| package main
import ( "log" "net/http"
"github.com/gorilla/websocket" ) var upgrader = websocket.Upgrader{ ReadBufferSize: 1024, WriteBufferSize: 1024, CheckOrigin: func(r *http.Request) bool { return true // 允许所有源,生产环境中应根据需要设置更严格的CORS策略 }, } func serveWs(w http.ResponseWriter, r *http.Request) { conn, err := upgrader.Upgrade(w, r, nil) if err != nil { log.Println("Failed to set up WebSocket connection:", err) return } defer conn.Close() for { // 读取客户端发送的消息 msgType, msg, err := conn.ReadMessage() if err != nil { log.Println("Failed to read WebSocket message:", err) break } log.Printf("Received message from client: %s", msg) err = conn.WriteMessage(msgType, []byte("receive message successfully")) if err != nil { log.Println("Failed to send WebSocket message:", err) break } } } func main() { http.HandleFunc("/ws", serveWs) log.Println("Starting WebSocket server on :8080...") err := http.ListenAndServe(":8080", nil) if err != nil { log.Fatal("Error starting server:", err) } }
|