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)