Как юзать WebSocket?

Question

[Владимир Грабко]

Добрый день. Я знаю что необходимо сделать пул коннектов var clients map[string]*websocket.Conn

и потом слать сообщения всем в цикле как то так

for _, v := range clients {
    websocket.JSON.Send(v, message)
  }

Но обьяните мне как при подключении клиент может отправить свой login для занесения в пул. Зарание спасибо.

Answer 1

[Andrey K]

https://github.com/onrik/wshub

Answer 2

[Oleg Shevelev]

package main

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

var upgrader = websocket.Upgrader{
    ReadBufferSize:  4096,
    WriteBufferSize: 4096,
    CheckOrigin: func(r *http.Request) bool {
        return true
    },
}

func wsHandler(w http.ResponseWriter, r *http.Request) {
    // вот тут вы можете проверить login/password и прочее или же сделать для этого отдельный метод /login, а тут проверять лишь есть ли у него сессия
    ws, err := upgrader.Upgrade(w, r, nil)
    if _, ok := err.(websocket.HandshakeError); ok {
        http.Error(w, "Not a websocket handshake", 400)
        return
    } else if err != nil {
        return
    }

    ws_list.Append(ws, r) // вот тут вы уже знаете что сокет вам подходит и от какого он пользователя и дальше заносите его в какое-то хранилище которое их хранит и по которому потом рассылаете данные.
}

И опять таки, читайте документацию, https://godoc.org/github.com/gorilla/websocket

Comments

[Владимир Грабко]

я юзал golang.org/x/net/websocket

[Oleg Shevelev]

Владимир Грабко: принцип тот-же.
1. пользователь подключается по HTTP/HTTP2
2. пользователь авторизуется - проверяется GET/POST login/password
3. к соединению привязывается сессия
4. делается Upgrade соединения до WebSockets
5. соединение попадает в хранилище соединений по которым в последствии рассылаются сообщения

Если пользователь уже авторизован и у него есть сессия то пункт 2 и 3 пропускаются.

[Владимир Грабко]

Oleg Shevelev: ну а как это на js реализовать... Пойду ка я наговнокодю ))

[Oleg Shevelev]

Владимир Грабко: на js как и на обычном сайте:
1. сделать ajax запрос GET/POST с передачей логина/пароля
2. сделать коннект к WebSockets с передачей GET/POST параметров с логином/паролем. GET точно можно сделать, можно ли передать POST - надо посмотреть в документации к html5/websockets
3. обычная форма авторизации без всякого ajax, а в websockets потом просто сессией пользоваться

[Владимир Грабко]

Oleg Shevelev: извините но я не понял. Я сделал аякс запрос к методу сохранил его на сервере. А как дальше сделать WebSocket коннект я хз... Я предполагаю что передать данные GET в функции
new WebSocket(); как то так

new WebSocket("ws://localhost:8081/events?login=admin&pass=qwerty");

[Oleg Shevelev]

Владимир Грабко: передача login/password через GET параметры не безопасно. Тем более что вы используете не защищённый SSL коннект.

Вариант 1

var conn;
$.post("/login/", {login: "login", password: "password"}, func(data){
   if (data.status == "OK") {
       conn = new WebSocket("ws://localhost:8081/events/");
   } else {
       alert("не авторизовались")
   }
});

Вариант 2

var conn = new WebSocket("ws://localhost:8081/events/");
    var message = {
        'action': 'login',
        'login': 'login',
        'password': 'password'
    };
    conn.send(JSON.stringify(message)); // <-- возможно можно отправить просто объект, без преобразований.

В первом случае при подключении к WebSockets на сервере вы проверяете есть ли правильная сессия, во втором случае вы принимаете все коннекты, но потом отправляете на соединения сообщения только если по вебсокету пришол логин и пароль и это записалось в сессию.

[Владимир Грабко]

Oleg Shevelev: первый вариант заюзает коннект который добавлен в ws_list.Append ?

[Oleg Shevelev]

Владимир Грабко: Не знаю как ответить, похоже вы вообще не понимаете что делаете.

Прочитайте стандарт html5 по WebSockets, прочитайте документацию к выбранной вами библиотеке, посмотрите примеры и реализации, как в составе библиотеке так и в виде статей на том же Хабре.

[Владимир Грабко]

Oleg Shevelev: я вообще не понимаю. Вы правы. Я тупо прошел урок https://learn.javascript.ru/websockets

[Владимир Грабко]

Oleg Shevelev: я уже разобрался. Благодарю вас за помощь.

[spotifi]

Владимир Грабко:
golang овская библиотека похуже.
gorilla овская библиотека получше.

[Oleg Shevelev]

spotifi: возможно соглашусь, но интересно ваше мнение "почему":)

[spotifi]

Oleg Shevelev: уже и не помню. я для себя провел сравнение и тесты когда выбирал что использовать. вроде Горилла показалась пофункциональней и поудобней. А Голанговская - ограниченной и не столь удобной.

[Oleg Shevelev]

spotifi: тоже сравнивал, горилла показалась более понятной и "объектной" чтоли, проще для последующего использования и кастомизации.