使用Python3.xSocketUDP广播消息（测试大数据流式处理）

使用Python Socket UDP广播消息（测试大数据流式处理）

以下代码基于Python 3.8.x；在Window 10环境下测试无误。

Server

发送广播消息；

from socket import *SERV = socket(AF_INET, SOCK_STREAM)
HOST = &＃39;127.0.0.1&＃39;# port
p = input(&＃39;set port(default 9999):&＃39;)
PORT = int(p) if p else 9999SERV.bind((HOST, PORT))
SERV.listen(5)print(&＃39;server startup, port: %d ..&＃39; % PORT)while True:print(&＃39;waiting ..&＃39;)sock, addr = SERV.accept()print(&＃39;[%s:%d]: ready ..&＃39; % (addr[0], addr[1]))try:while True:data = input(&＃39;[%s:%d]: &＃39; % (addr[0], addr[1]))sock.send((data + &＃39;\n&＃39;).encode(&＃39;utf8&＃39;))except:print(&＃39;disconnected ..&＃39;)


注意：如果是Java客户端，一定要带上\n这个标记。

Server（可指定端口，同时启动多个端口）

from socket import *SERV = socket(AF_INET, SOCK_STREAM)
HOST = &＃39;127.0.0.1&＃39;# port
p = input(&＃39;set port(default 9999):&＃39;)
PORT = int(p) if p else 9999SERV.bind((HOST, PORT))
SERV.listen(5)print(&＃39;server startup, port: %d ..&＃39; % PORT)while True:print(&＃39;waiting ..&＃39;)sock, addr = SERV.accept()print(&＃39;[%s:%d]: ready ..&＃39; % (addr[0], addr[1]))try:while True:data = input(&＃39;[%s:%d]: &＃39; % (addr[0], addr[1]))sock.send((data + &＃39;\n&＃39;).encode(&＃39;utf8&＃39;))except:print(&＃39;disconnected ..&＃39;)


Client（非必须，测试使用）

用于接收消息；

实际场景为Flink或者Spark Streaming接收数据，最优方案为MQ，但测试环境部署较为麻烦，所以用Socket替代。

from socket import *HOST = &＃39;127.0.0.1&＃39;
PORT = 9999sock = socket(AF_INET, SOCK_DGRAM)
sock.bind((HOST, PORT))while True:try:data = sock.recv(1024)print(data.decode(&＃39;utf8&＃39;))except Exception as e:print(e)