message_queues ={}
client_socket_fd_map={}defstart_socket_select_server():
sock=socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR,1)
sock.bind(('0.0.0.0', 8002))
sock.listen(5)print("WebSocket 服务器启动成功 监听IP", ('127.0.0.1', 8002))
sock.setblocking(False)
inputs=[sock, ]
outputs=[]whileTrue:
readable, writeable, exceptional=select.select(inputs, outputs, inputs)#print('select finish, inputs size:%d, outputs size:%d' % (len(inputs), len(outputs)))
for s inreadable:if s issock:
conn, client_addr=s.accept()print("new connection from", client_addr)
conn.setblocking(False)
inputs.append(conn)
message_queues[conn]=queue.Queue()else:if s not inoutputs:#第一次 表示 websocket的握手
data = s.recv(1024)ifdata:print('received [%s] from %s' %(data, s.getpeername()[0]))#message_queues[s].put(data)
headers=get_headers(data)
response_tpl= "HTTP/1.1 101 Switching Protocols\r\n"\"Upgrade: websocket\r\n"\"Connection: Upgrade\r\n"\"Sec-WebSocket-Accept: %s\r\n"\"WebSocket-Location: ws://%s%s\r\n\r\n"sha1=hashlib.sha1()
magic_string= "258EAFA5-E914-47DA-95CA-C5AB0DC85B11"value= headers['Sec-WebSocket-Key'] +magic_string
sha1.update(value.encode('utf-8'))
ac=base64.b64encode(sha1.digest())
response_str= response_tpl %(
ac.decode('utf-8'), headers['Host'], headers['url'])
s.send(bytes(response_str, encoding='utf-8'))
# 这里将文件描述符返回给浏览器 浏览器可以在接下来的http请求中带上这个参数 服务端就可以向这个文件描述符中写入信息返回给指定浏览器
fileno_dict_str= '{"type":1, "body":%s}' %s.fileno()
message_queues[s].put(fileno_dict_str)if s not inoutputs:
outputs.append(s)
client_socket_fd_map[s.fileno()]=selse:#表示客户端已经断开
print("~~~~~~~~~~~client [%s] closed" %s)if s inoutputs:
outputs.remove(s)delmessage_queues[s]delclient_socket_fd_map[s.fileno()]
inputs.remove(s)
s.close()else:#websocket 通信
data = s.recv(8096)ifdata:pass
else:#表示客户端已经断开
print("-------------client [%s] closed" %s)if s inoutputs:
outputs.remove(s)delmessage_queues[s]delclient_socket_fd_map[s.fileno()]
inputs.remove(s)
s.close()for s inwriteable:try:
next_msg=message_queues[s].get_nowait()exceptqueue.Empty:pass
else:
send_msg(s, next_msg)