Перейти на PRO

Как реализовать WebSocket-соединение в Go и какие ключевые аспекты нужно учитывать?

Тип вопроса: Сети Вероятность: 24% Категория: Golang Разработчик

Ответ

Да, я реализовывал WebSocket-взаимодействие в Go. Чаще всего для этого используется библиотека gorilla/websocket, так как она предоставляет больше гибкости и функциональности по сравнению со стандартной golang.org/x/net/websocket.

Процесс реализации выглядит так:

  1. HTTP Upgrader: WebSocket-соединение начинается как обычный HTTP-запрос, который затем "обновляется" до протокола WebSocket. Для этого используется websocket.Upgrader.
  2. Цикл чтения и записи: После успешного обновления соединения, сервер входит в бесконечный цикл, где он слушает входящие сообщения от клиента (conn.ReadMessage()) и обрабатывает их. Запись клиенту (conn.WriteMessage()) обычно происходит в отдельной горутине для неблокирующей работы.

Ключевые аспекты, которые нужно учитывать:

  • Конкурентность: Объект соединения *websocket.Conn не является потокобезопасным для одновременной записи. Запись должна производиться только из одной горутины. Чтение безопасно производить из другой горутины.
  • Управление соединениями: Необходимо создать пул или менеджер соединений для отслеживания всех активных клиентов, например, для рассылки сообщений всем участникам чата.
  • Обработка ошибок и закрытия: Важно корректно обрабатывать ошибки, возникающие при чтении или записи, и закрывать соединение. gorilla/websocket позволяет установить CloseHandler для выполнения логики при закрытии соединения клиентом.
  • Ping/Pong: Для поддержания соединения активным и обнаружения "мертвых" клиентов используется механизм Ping/Pong. Библиотека может делать это автоматически, если настроить PingHandler.

Пример простого обработчика:

import (
    "log"
    "net/http"
    "github.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{
    ReadBufferSize:  1024,
    WriteBufferSize: 1024,
    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(err)
        return
    }
    defer conn.Close()

    for {
        // Читаем сообщение от клиента
        messageType, p, err := conn.ReadMessage()
        if err != nil {
            log.Println("read error:", err)
            break
        }

        // Отправляем сообщение обратно клиенту (эхо)
        if err := conn.WriteMessage(messageType, p); err != nil {
            log.Println("write error:", err)
            break
        }
    }
}

WebSocket отлично подходит для создания интерактивных приложений: чатов, систем уведомлений в реальном времени, онлайн-игр и дашбордов.

Ответ 18+ 🔞

А, ну, вебсокеты на Go, говоришь? Да, блядь, приходилось, конечно. Все эти твои чаты в реальном времени, дашборды, которые пиздец как тормозят без них.

Так вот, слушай сюда, сука. Стандартная библиотека, golang.org/x/net/websocket — это, конечно, мило, но как-то уж совсем голенько, понимаешь? Все нормальные пацаны берут gorilla/websocket. Там и функционала овердохуища, и гибкость пиздец какая.

Как это, блядь, работает, если по-простому:

  1. Начинается с подставы, ёпта. Клиент стучится к тебе обычным HTTP-запросом, такой весь из себя невинный. А ты ему — раз! — и websocket.Upgrader. Это как сказать: "О, дружок, обычный HTTP? Да пошёл ты нахуй, давай на настоящий протокол переходим!" И соединение апгрейдится до вебсокета.
  2. А дальше — вечный цикл, блядь. Соединение установлено, и сервер входит в петлю, из которой нет выхода. Сидит, как дурак, и ждёт сообщений от клиента (conn.ReadMessage()). А как получил — тут же, или в отдельной горутине, чтобы не тормозить, отвечает (conn.WriteMessage()).

Но тут, сука, подводные ебучки есть, на которые все наступают:

  • Конкурентность — ёперный театр. Объект соединения *websocket.Conn — он не для оргии. Нельзя в него писать одновременно из десяти горутин, он с ума сойдёт. Писать можно только из одной горутины, блядь! Читать — пожалуйста, из другой. А писать — нет, сука, очередь.
  • Клиентов же дохуя. Один подключился, второй, третий... Надо же как-то знать, кто тут у нас вообще есть, чтобы всем рассылку сделать. Для этого заводишь какую-нибудь мапу или пул соединений, мудя. Иначе как ты всем сразу "Новое сообщение!" отошлёшь?
  • Закрываться надо красиво. Клиент может просто взять и отвалиться. Или связь пропала. Надо ловить эти ошибки чтения/записи и аккуратно соединение прибивать. В gorilla даже есть специальный CloseHandler, чтобы уборку сделать, когда клиент сам дверью хлопнул.
  • Ping/Pong — чтоб не сдох. Чтобы понять, жив ли ещё тот чел на том конце провода, или он уже полчаса как выключился, нужны контрольные посылки. Библиотека умеет сама пинговать, если ей PingHandler правильно подсунуть.

Вот, смотри, простейший пример, чтоб понятно было, о чём речь:

import (
    "log"
    "net/http"
    "github.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{
    ReadBufferSize:  1024,
    WriteBufferSize: 1024,
    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(err)
        return
    }
    defer conn.Close() // Главное — не забыть закрыть, а то ресурсы похуекают

    for {
        // Ждём, пока клиент что-то ляпнет
        messageType, p, err := conn.ReadMessage()
        if err != nil {
            log.Println("read error:", err) // Клиент, сука, отвалился
            break
        }

        // И тупо отправляем ему его же сообщение обратно (эхо-сервер, бля)
        if err := conn.WriteMessage(messageType, p); err != nil {
            log.Println("write error:", err) // А тут уже мы обосрались
            break
        }
    }
}

В общем, штука мощная. Для всего, что должно обновляться без этих дурацких перезагрузок страницы — самое то. Чат, онлайн-игра какая-нибудь, или там графики, которые сами ползут. Без вебсокетов — нихуя не real-time, а одно разочарование.

© ХакСобесов, 2026
support@hacksobesov.com