Отправка сообщения Websocket конкретному клиенту в Go (с использованием Gorilla) - PullRequest
Новая
       32

Отправка сообщения Websocket конкретному клиенту в Go (с использованием Gorilla)

0 голосов
/ 21 января 2019

Я очень новичок в Go и обнаружил, что работаю с сокетами в качестве моего первого проекта.Это излишний вопрос, но я не смог понять, как отправить обновление веб-сокета определенному клиенту в Go (используя Gorilla).

Основная проблема, которую я пытаюсь решить, - создание заголовка с использованиемвеб-сокеты и поисковая система, как ES / Lucene.Я сохранил несколько индексов в своей поисковой системе и обернул их в Go.Когда я начал работать с использованием веб-сокетов в Go, я обнаружил почти все примеры, демонстрирующие механизм вещания.Когда я попытался разобраться в этом и попытался изменить пример, приведенный в github repo Гориллы, основываясь на примерах, приведенных в этом потоке и в этом ответе , яКажется, я не понимаю connections и как это вписывается в client.go

В идеале, я бы хотел увидеть эту работу -

  • Соединение разъема междуклиент и сервер установлены
  • Когда клиент отправляет входные данные через сокет, сервер выбирает его и добавляет в канал (канал Go)
  • Оболочка индексирования проверяет этот канал, икогда есть что выбрать, индекс извлекается и записывается обратно в сокет

Как сервер может однозначно идентифицировать Client?

Я использовал примеры, приведенные наGithub Гориллы репо

Из моей кодовой базы hub.go имеет следующее 

type Hub struct {
    // Registered clients.
    clients map[*Client]bool

    // Inbound messages from the clients.
    broadcast chan []byte

    // Register requests from the clients.
    register chan *Client

    // Unregister requests from clients.
    unregister chan *Client

    connections map[string]*connection
}

func newHub() *Hub {
    return &Hub{
        broadcast:  make(chan []byte),
        register:   make(chan *Client),
        unregister: make(chan *Client),
        clients:    make(map[*Client]bool),
        connection: make(map[*Client]bool), // is this alright?
    }
}

func (h *Hub) run() {
    for {
        select {
        case client := <-h.register:
            h.clients[client] = true
        case client := <-h.unregister:
            if _, ok := h.clients[client]; ok {
                delete(h.clients, client)
                close(client.send)
            }
        case message := <-h.broadcast:
            for client := range h.connections {
                select {
                case client.send <- message:
                default:
                    close(client.send)
                    delete(h.connections, client)
                }
            }
        }
    }
}

, и я не уверен, что я должен добавить к client.go

type Client struct {
    // unique ID for each client
    // id string

    // Hub object
    hub *Hub

    // The websocket connection.
    conn *websocket.Conn

    // Buffered channel of outbound messages.
    send chan []byte

    // connection --> (what should the connection property be?)
    connection string
}

Обратите внимание - я будудобавление поля Id в структуре Client.Как я могу продолжить отсюда?

1 Ответ

0 голосов
/ 21 января 2019

Пример чата показывает, как реализовать трансляцию.Пример чата не является хорошей отправной точкой для приложения, если широковещательная рассылка не требуется.

Чтобы отправить сообщение на определенное соединение через веб-сокет, просто напишите в соединение, используя NextWriter или WriteMessage .Эти методы не поддерживают одновременные записи, поэтому вам может потребоваться использовать мьютекс или групповую процедуру для обеспечения единственной записи.

Простой способ найти конкретный *websocket.Connection состоит в передаче *websocket.Connection коду, которыйэто нужноЕсли приложению необходимо связать другое состояние с соединением, определите тип для хранения этого состояния и передайте указатель на него: 

type Client struct {
    conn *websocket.Conn
    mu sync.Mutex
    ...
}

Hub можно изменить, чтобы отправлять сообщения определенному соединению, но это обходной путь, если трансляция не нужна.Вот как это сделать:

Добавить поле идентификатора к клиенту: 

 ID idType // replace idType with int, string, or whatever you want to use

Изменить поле концентратора Gorilla с connections map[*connection]bool на connections map[idType]*connection.

Определить сообщениетип, содержащий данные сообщения и идентификатор целевого клиента: 

type message struct {
   idType ID
   data []byte
}

Замените поле широковещания концентратора на: 

   send chan message

Измените концентратор для цикла на: 

for {
    select {
    case client := <-h.register:
        h.clients[client.ID] = client
    case client := <-h.unregister:
        if _, ok := h.clients[client.ID]; ok {
            delete(h.clients, client.ID)
            close(client.send)
        }
    case message := <-h.send:
        if client, ok := h.clients[message.ID]; ok {
            select {
            case client.send <- message.data:
            default:
                close(client.send)
                delete(h.connections, client)
            }
        }
    }

Отправляйте сообщения определенному клиенту, создав message с соответствующим идентификатором: 

   hub.send <- message{ID: targetID, data: data}
...