教程-使用Unity进行可视分析(6)

from websocket_server import WebsocketServer
import asyncio

# 新连接
def new_client(client, server):
   	print("New client connected and was given id %d" % client['id'])
   	server.send_message_to_all("Hey all, a new client has joined us")

# 断开连接
# Called for every client disconnecting
def client_left(client, server):
   	print("Client(%d) disconnected" % client['id'])

# Websocket 服务器收到消息
# Called when a client sends a message
def message_received(client, server, message):
   	if len(message) > 200:
       	message = message[:200]+'..'
   	print("Client(%d) said: %s" % (client['id'], message))
   	server.send_message(client, "got your message")
   	server.send_message_to_all("Client(%d) said: %s" % (client['id'], message))

   	pyc(message, 9090)
   	# 直接给hololens发websocket消息不好用，因为websocket使用域名连接。这里加了一段TCP


# 转发给hololens，用TCP
# hololens那头最多接收256字符长度信息，但不够的话hololesn无法读取显示，需要填充
# 多线程部分参考了zhihu设计，连接在最下方
def pyc(msg, p):
   	message = msg
   	while(len(message) < 256):
       	message += '^'
   	# loop = asyncio.get_event_loop()
   	# loop.run_until_complete(tcp_echo_clientP(message, p))
   	# loop.close()
   	loop = asyncio.new_event_loop()
   	asyncio.set_event_loop(loop)
   	loop.run_until_complete(tcp_echo_clientP(message, p, loop))

# 异步通信实现无间断的TCP通信
# 指定了hololens的IP地址，换了的话记得修改
async def tcp_echo_clientP(message, port, loop):
   	reader, writer = await asyncio.open_connection('192.168.1.141', port)

   	print('Send: %r' % message)
   	writer.write(message.encode())

   	data = await reader.read(64)
   	print('Received: %r' % data.decode())

   	print('Close the socket')
   	writer.close()


PORT = 8181
server = WebsocketServer(PORT, '192.168.1.116')
 # websocket server地址。改成服务器当前IP。
server.set_fn_new_client(new_client)
server.set_fn_client_left(client_left)
server.set_fn_message_received(message_received)
server.run_forever()