服务端代码 test_socket_consumer.py,先启动服务端
from function_scheduling_distributed_framework import task_deco, BrokerEnum
@task_deco('10.0.126.147:5691', broker_kind=BrokerEnum.TCP) #可以写TCP, 也可以写UDP那就是udp socket协议。
def f(x):
print(x * 10)
if __name__ == '__main__':
f.consume()
#for i in range(10):
# f.push(f'hello {i}')
客户端代码,可以启动多个客户端脚本。
import time
from test_socket_consumer import f
for i in range(1000000):
time.sleep(1)
# print(i)
f.push(i)
这样就完成了客户端和服务端的交互了。不需要亲自手动写一大堆socket连接代码了,这样如果大量开发tcp任务时候开发效率变快很多。