Я открываю веб-сокет для отправки обновлений баланса клиента, но после {'message': 'accepted' веб-сокет закрывается, и я не могу отправлять какие-либо данные.
@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
