基于上篇:分布式处理数据:server,提供client调用代码
import asyncio
import websockets
#async 为异步调用,可以理解为线程,但不等于线程,一公有10个活要干,10个人干,
# 就相当于10个线程活动,但是明明这些活我一个人就能干,那我用一个人干10个活就
# 好了,就是携程,await 为挂起,意思是这个活不用我干了,我就去干别的,把这个
# 活挂起来,等下下次要干就继续干这个
# 向服务器端认证,用户名密码通过才能退出循环
async def auth_system(websocket):
while True:
cred_text = input("please enter your username and password and type(account:password:send or recv): ")
await websocket.send(cred_text)
response_str = await websocket.recv()
if "congratulation" in response_str:
return True
# 向服务器端发送认证后的消息,拿取数据
async def recv_msg(websocket):
while True:
#接收的脚本
_text="request recv data"
await websocket.send(_text)
recv_text =await websocket.recv()
print(recv_text,type(recv_text))
if str(recv_text)=="exit":
#关闭连接
await websocket.close(reason="user exit")
return
# 向服务器端发送认证后的消息,传输数据给服务端
async def send_msg(websocket):
while True:
# 发送的脚本
_text = "request send data"
await websocket.send(_text)
recv_text = await websocket.recv()
print(recv_text)
#这里是发送数据给服务端,因此这里没有数据的时候,要写个关闭连接的方法
# #关闭连接await websocket.close(reason="user exit")
# 客户端主逻辑
async def main_logic():
async with websockets.connect('ws://localhost:5678') as websocket:
await auth_system(websocket)
#让服务端存储数据用这个,认证的时候发送root:123456:send,仔细看服务端代码认证的代码
#如果需要自定义服务端队列传的认证为root:123456:自定义队列名称:send
# await send_msg(websocket)
#从服务端拿取数据用这个,和上面的存不能一起用发送root:123456:recv,仔细看服务端代码
# 如果需要自定义服务端队列传的认证为root:123456:自定义队列名称:recv
await recv_msg(websocket)
loop=asyncio.get_event_loop()
loop.run_until_complete(main_logic())
# loop.run_forever()
因上篇服务端提供了默认队列,客户端调用 send_msg 只需要发送:root:123456:send,服务端会默认给个队列,客户端调用 recv_msg 只需要发送:root:123456:recv,就会从默认队列里面拿数据,如果需自定义队列,可以在调用send_msg的时候,发送root:123456:自定义名称:send,在拿取自定义队列数据的时候发送:toot:123456:自定义名称:recv,注意,自定义名称一定要和发送send的时候一致,才能拿到自定义队列的数据,注意send_msg和recv_msg不能一起调用