Как корректно создавать ws соединение?

Question

[Олег Ведерников]

Как корректно создавать ws соединение?
В учебных целях начал ознакамливаться aiohtt.
Делаю пет проект где на фронте испольую VueJs, на бэке aiohttp соответственно.
Столкнулся с задачей как правильно сделать оповещения в режиме реального времени, к примеру, как это реализовано в ВК, когда приходят новые сообщения/добавляются друзья/прочее.

На фронте, если пользователь авторизирован, то автоматически устанавливается ws соединение с сервером
ws = new WebSocket(URL);
На бэке есть обработчик ws:

async def websocket_handler(request):
    ws = web.WebSocketResponse()
    await ws.prepare(request)
    request.app['websockets']['user_id'] = ws

    async for msg in ws:
        pass   # корректна ли эта инструкция, чтобы при потери соединения выполнялась строчка ниже?
    request.app['websockets'].remove(ws)

    return ws

Так же есть несколько эндпоинтов(new_message, new_friend, etc), которые должны посылать в этот вебсокет какую-то информацию

async def new_message(request):
    # тут валидируем и сохраняем сообщение

    # отправляем данные в сокет
    ws = request.app['websockets']['receiver_id']:
    ws.send_str('new message detail')

Ожидается, что в какой бы части приложения клиент не находился, ему всегда будут залетать оповещения о новых событиях, т.к. вебсокет открывается раз, когда пользователь авторизируется, и не закрывается, пока не будет потеряна связь.
Правильно или плохо так делать? Почему?
Как будет вести себя приложение, если к примеру одновременно будет открыто 10к соединений?

Answer 1

[Сергей Горностаев]

В целом всё правильно, только await ws.send_str('new message detail') и вместо pass стоит добавить

if msg.type == WSMsgType.ERROR:
    какая-нибудь реакция на ошибку

Как будет вести себя приложение, если к примеру одновременно будет открыто 10к соединений?

Для этого aiohttp и создавался.

Comments

[Олег Ведерников]

Спасибо, что откликнулись. Хотел бы ещё кое-что дополнительно спросить.
Если не планируется передача каких-либо данных с клиента по вебсокету, а только их получение, то есть ли тогда смысл проверять тип соощения?

if msg.type == WSMsgType.ERROR:
    какая-нибудь реакция на ошибку

Саму же отправку данных делать дёргая отдельные эндпоинты, как было написано выше async def new_message(request)
Или всё такие лучше на клиенте всё посылать в сокет, кроме login/logout/change_settings/etc, а на сервере сделать что-то наподобие этого

async def websocket_handler(request):
    ws = web.WebSocketResponse()
    await ws.prepare(request)
    request.app['websockets']['user_id'] = ws

    async for msg in ws:
        if msg.type == WSMsgType.TEXT:
            data = json.loads(msg.data)
            if data.action == 'new_message':
                # вызываем корутину создания нового сообщения
            if data.action == 'new_friend':
                # вызываем корутину добавления нового друга

            # тут будет код оповещения необходимого пользователя
        elif msg.type == WSMsgType.ERROR:
            # process error
    request.app['websockets'].remove(ws)

    return ws

[Сергей Горностаев]

Олег Ведерников, потому, что клиенты не всегда видут себя так, как ожидаешь. И лучше заметить непредвиденное поведение.

Оба варианта имеют право на жизнь. Хотя, лично я предпочёл бы растащить функционал под эндпоинтам, а через websocket'ы передавать ограниченный набор сигналов.

[Олег Ведерников]

Сергей Горностаев, Спасибо за помощь)