Веб-сокеты Python и gtk - запутались в очереди asyncio
Я новичок в асинхронном программировании на python, и я пытаюсь написать сценарий, который запускает сервер websocket, прослушивает сообщения, а также отправляет сообщения, когда определенные события (например, нажатие клавиши 's') запускаются в окне gtk . Вот что у меня есть на данный момент:
import gi
gi.require_version('Gtk', '3.0')
from gi.repository import Gtk
import asyncio
import websockets
import threading
ws = None
async def consumer_handler(websocket, path):
global ws
ws = websocket
await websocket.send("Hello client")
while True:
message = await websocket.recv()
print("Message from client: " + message)
def keypress(widget,event):
global ws
if event.keyval == 115 and ws: #s key pressed, connection open
asyncio.get_event_loop().create_task(ws.send("key press"))
print("Message to client: key press")
def quit(widget):
Gtk.main_quit()
window = Gtk.Window(Gtk.WindowType.TOPLEVEL)
window.connect("destroy", quit)
window.connect("key_press_event", keypress)
window.show()
start_server = websockets.serve(consumer_handler, 'localhost', 8765)
asyncio.get_event_loop().run_until_complete(start_server)
wst = threading.Thread(target=asyncio.get_event_loop().run_forever)
wst.daemon = True
wst.start()
Gtk.main()
А вот и веб-страница клиента:
<!DOCTYPE html>
<html>
<head>
<title>Websockets test page</title>
<meta charset = "UTF-8" />
<script>
var exampleSocket = new WebSocket("ws://localhost:8765");
function mylog(msg) {
document.getElementById("log").innerHTML += msg + "<br/>";
}
function send() {
mylog("Message to server: Hello server");
exampleSocket.send("Hello server");
}
exampleSocket.onopen = function (event) {
mylog("Connection opened");
};
exampleSocket.onmessage = function (event) {
mylog("Message from server: " + event.data);
}
</script>
</head>
<body>
<p id = "log"></p>
<input type = "button" value = "Send message" onclick = "send()"/>
</body>
</html>
Запустите код python, затем загрузите веб-страницу в браузере, теперь все сообщения, отправляемые браузером, отображаются в stdout python, пока все хорошо. Но если вы нажмете клавишу «s» в окне gtk, python не отправит сообщение, пока другое сообщение не будет получено из браузера (от нажатия кнопки «Отправить сообщение»). Я думал, что await websocket.recv() должен был возвращать управление циклу событий до тех пор, пока не будет получено сообщение? Как мне заставить его отправлять сообщения, пока он ожидает получения?
Ответы 1
But if you hit the 's' key in the gtk window, python doesn't send the message until another message is received from the browser
Проблема в этой строке:
asyncio.get_event_loop().create_task(ws.send("key press"))
Поскольку цикл событий asyncio и основной цикл GTK выполняются в разных потоках, вам нужно использовать run_coroutine_threadsafe для отправки сопрограммы в asyncio. Что-то вроде:
asyncio.run_coroutine_threadsafe(ws.send("key press"), loop)
create_task добавляет сопрограмму в очередь выполняемых сопрограмм, но не может разбудить цикл событий, поэтому ваша сопрограмма запускается только тогда, когда в asyncio происходит что-то еще. Кроме того, create_task не является потокобезопасным, поэтому его вызов, когда сам цикл событий изменяет очередь выполнения, может повредить его структуры данных. run_coroutine_threadsafe не имеет ни одной из этих проблем, он организует активизацию цикла событий как можно скорее и использует мьютекс для защиты структур данных цикла событий.