Sobes.tech
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/websocket library 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.

Have you ever had to write a WebSocket server? — Golang - sobes.tech