# client.py
from socket import *
client = socket(AF_INET, SOCK_STREAM)
client.connect(('127.0.0.1', 8081))
while True:
msg = input('msg>: ').strip()
if not msg:
continue
client.send(msg.encode('utf-8'))
data = client.recv(1024)
print(data.decode('utf-8'))
# server.py
from socket import *
server = socket(AF_INET, SOCK_STREAM)
server.bind(('127.0.0.1', 8081))
server.listen(5)
server.setblocking(False)
print('start...')
rlist = []
wlist = []
while True:
try:
# Non-blocking IO accept没有client来连时,抛出信号异常BlockingIOError
# 捕获之后 while 就会一直循环,负荷状态
conn, address = server.accept()
rlist.append(conn)
print('rlist: %s' % rlist)
except BlockingIOError:
# 没有连接来,接住BlockingIOError,做其他事情:在conn基础上接收与发送数据
# 收消息
del_rlist = []
for conn in rlist:
try:
data = conn.recv(1024)
if not data:
del_rlist.append(conn)
continue
# conn.send 也是IO操作,可优化
# conn.send(data.upper())
wlist.append((conn, data.upper()))
except BlockingIOError:
# 没有收到消息,跳过就行
continue
except Exception:
conn.close()
del_rlist.append(conn)
for conn in del_rlist:
rlist.remove(conn)
# 发消息
del_wlist = []
for item in wlist:
conn, data = item
try:
# 需求是发过一次就删除
conn.send(data)
del_wlist.append(item)
except BlockingIOError:
pass
for item in del_wlist:
wlist.remove(item)