Senior
Have you ever had to write a WebSocket server?
sobes.tech AI
Answer from AI
Yes, I have written a WebSocket server in Go. Key points in the implementation:
- Used the
gorilla/websocketlibrary for convenient protocol handling. - Managed connection handling through separate goroutines.
- Implemented a client pool for message broadcasting.
- Properly handled connection closures and errors.
Example of a simple WebSocket server in Go:
package main
import (
"log"
"net/http"
"github.com/gorilla/websocket"
)
var upgrader = websocket.Upgrader{
CheckOrigin: func(r *http.Request) bool { return true },
}
func wsHandler(w http.ResponseWriter, r *http.Request) {
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Println("Upgrade error:", err)
return
}
defer conn.Close()
for {
mt, message, err := conn.ReadMessage()
if err != nil {
log.Println("Read error:", err)
break
}
log.Printf("Received: %s", message)
err = conn.WriteMessage(mt, message) // echo
if err != nil {
log.Println("Write error:", err)
break
}
}
}
func main() {
http.HandleFunc("/ws", wsHandler)
log.Println("Server started on :8080")
log.Fatal(http.ListenAndServe(":8080", nil))
}
This server accepts connections, reads messages, and sends them back (echo). In real projects, client management, authorization, error handling, and scaling are added.