[英]Python asyncio: closing a socket and releasing an await sock_read()
I'm writing a small multi-user game.我正在编写一个小型多用户游戏。 Users are logged on via console or sockets.用户通过控制台或 sockets 登录。 I want to be able to kick out other users.我希望能够踢出其他用户。
I use asyncio
and wait for user input by calling await loop.sock_recv(sock, 256)
.我使用asyncio
并通过调用await loop.sock_recv(sock, 256)
来等待用户输入。 Now, if some other user (say, from the console) closes the socket, the event crashes since select.select
seems to have problems.现在,如果其他用户(例如,从控制台)关闭套接字,则事件会崩溃,因为select.select
似乎有问题。
How can I kill a connection and release the sock_recv()
?如何终止连接并释放sock_recv()
?
Attached is a small(ish) MWE.附件是一个小的(ish)MWE。 It creates listening sockets and will accept a connection on port 4000. After that you can kill a connection by entering "x"
on the console.它会创建监听 sockets 并将接受端口 4000 上的连接。之后,您可以通过在控制台上输入"x"
来终止连接。 logoff()
is my attempt to kill the connection. logoff()
是我试图终止连接的尝试。
import asyncio
import socket
import sys
import threading
# ------ console input -------------------------------------------------------
async def _ainput(loop):
fut = loop.create_future()
def _run():
line = sys.stdin.readline().strip()
loop.call_soon_threadsafe(fut.set_result, line)
threading.Thread(target=_run, daemon=True).start()
return await fut
async def console_input_loop(loop):
while True:
inp = (await _ainput(loop)).strip()
print(f"[{inp.strip()}]")
if inp == "x":
logoff()
# ------ socket input --------------------------------------------------------
alive = True
async def socket_input_loop(loop, sock):
print(f"New connection")
global alive
while alive:
try:
inp = await loop.sock_recv(sock, 256)
except ConnectionResetError:
break
print(inp)
print("shutting down")
sock.shutdown(socket.SHUT_RDWR)
sock.close()
print(f"Connection closed")
alive = True
listen_addr = ('', 4000)
async def _run_server(loop, server):
server.bind(listen_addr)
server.listen(8)
server.setblocking(False)
while loop.is_running():
global sock
sock = (await loop.sock_accept(server))[0]
loop.create_task(socket_input_loop(loop, sock))
async def run_server4(loop):
await _run_server(loop, socket.socket(socket.AF_INET, socket.SOCK_STREAM))
async def run_server6(loop):
await _run_server(loop, socket.socket(socket.AF_INET6, socket.SOCK_STREAM))
def async_driver():
loop = asyncio.get_event_loop()
loop.create_task(console_input_loop(loop))
loop.create_task(run_server4(loop))
loop.create_task(run_server6(loop))
loop.run_forever()
print()
def logoff():
global alive
alive = False
#loop = asyncio.get_event_loop()
#try:
# key = loop._selector.get_key(sock.fileno())
#except KeyError:
# pass
#else:
# mask, (reader, writer) = key.events, key.data
# #loop._add_callback(reader)
# the next line is needed, otherwise we get:
# r, w, x = select.select(r, w, w, timeout)
# OSError: [WinError 10038] An operation was attempted on something that is not a socket
#loop.remove_reader(sock.fileno())
sock.shutdown(socket.SHUT_RDWR)
sock.close()
#reader._run()
async_driver()
I get this crash:我得到这个崩溃:
Traceback (most recent call last):
File "C:\xx7.py", line 91, in <module>
async_driver()
File "C:\xx7.py", line 69, in async_driver
loop.run_forever()
File "C:\Users\chris\Anaconda3\lib\asyncio\base_events.py", line 528, in run_forever
self._run_once()
File "C:\Users\chris\Anaconda3\lib\asyncio\base_events.py", line 1728, in _run_once
event_list = self._selector.select(timeout)
File "C:\Users\chris\Anaconda3\lib\selectors.py", line 323, in select
r, w, _ = self._select(self._readers, self._writers, [], timeout)
File "C:\Users\chris\Anaconda3\lib\selectors.py", line 314, in _select
r, w, x = select.select(r, w, w, timeout)
OSError: [WinError 10038] An operation was attempted on something that is not a socket
The solution was simple: just shutdown()
without close()
.解决方案很简单:只需shutdown()
而不使用close()
。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.