Я открываю веб-сокет для отправки обновлений баланса клиента, но после {'message': 'accepted' веб-сокет закрывается, и я не могу отправлять какие-либо данные.
def get_contract_listener(access_token: str, callback: Callable[[int | None], Awaitable[None]]) -> None | Watch:
logger.debug('get contract listener token=%s', access_token)
contract = db.collection('Contract').where(filter=FieldFilter('access_token', '==', access_token)).get()
if not contract:
logger.debug('contract not found')
return
async def wrap(document: DocumentSnapshot):
await callback((document.to_dict() or {}).get('last_saldo'))
return cast(DocumentReference, contract[0].reference).on_snapshot(lambda doc, _, __: run(wrap(doc[0])))
@router.websocket('/wait_update')
async def api_websocket_wait_update(socket: WebSocket):
await socket.accept()
token = socket.cookies.get('access_token')
if token is None:
return JSONResponse({'status': 'fail', 'detail': 'no access token'}, 401)
async def callback(balance: int | None) -> None:
assert listener is not None
await socket.send_json({'message': 'updated', 'balance': balance})
listener.unsubscribe()
listener = get_contract_listener(token, callback)
if listener is None:
await socket.send_json({'message': 'failed', 'detail': 'contract not found'})
await socket.close(1007)
return
await socket.send_json({'message': 'accepted'})
JS:
function create_socket() {
const socket = new WebSocket('/api/contract/wait_update');
socket.addEventListener('open', () => {
console.log('open websocket connection');
});
socket.addEventListener('message', (e) => {
const data = JSON.parse(e.data);
if (data.message == 'failed') {
console.error(`error from websocket: ${data.detail}`);
return;
}
console.log(`message from websocket: ${e.data}`);
if (data.message == 'updated') {
console.log('balance updated');
// update logic ...
}
});
socket.addEventListener('close', () => {
console.log('close websocket connection');
});
}
логи из js:
open websocket connection
message from websocket: {"message":"accepted"}
Подробнее здесь: https://stackoverflow.com/questions/798 ... out-asking
