旧的代码为单线程,只能同时接收数据或同时发送数据。
单线程代码如下所示:
import socket
def send_msg(udp_socket):
"""获取键盘数据,并将其发送给对方"""
# 1. 从键盘输入数据
msg = input("\n请输入要发送的数据:")
# 2. 输入对方的ip地址
dest_ip = input("\n请输入对方的ip地址:")
# 3. 输入对方的port
dest_port = int(input("\n请输入对方的port:"))
# 4. 发送数据
udp_socket.sendto(msg.encode("utf-8"), (dest_ip, dest_port))
def recv_msg(udp_socket):
"""接收数据并显示"""
# 1. 接收数据
recv_msg = udp_socket.recvfrom(1024)
# 2. 解码
recv_ip = recv_msg[1]
recv_msg = recv_msg[0].decode("utf-8")
# 3. 显示接收到的数据
print(">>>%s:%s" % (str(recv_ip), recv_msg))
def main():
# 1. 创建套接字
udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
# 2. 绑定本地信息
udp_socket.bind(("", 7890))
while True:
# 3. 选择功能
print("="*30)
print("1:发送消息")
print("2:接收消息")
print("="*30)
op_num = input("请输入要操作的功能序号:")
# 4. 根据选择调用相应的函数
if op_num == "1":
send_msg(udp_socket)
elif op_num == "2":
recv_msg(udp_socket)
else:
print("输入有误,请重新输入...")
if __name__ == "__main__":
main()
单线程修改为多线程的注意点:
1.因为在程序运行期间,接受数据和发送数据可以一直运行,所以需要在两个函数中加入While Ture循环。
2.在主线程常见两个字线程,循环调用两个函数,从而同时实现发送和接收数据的功能。
3.需要注意定义的函数中有参数,因此创建多线程对象的时候需要arg或wargs传参,传递udp套接字。
4.传递套接字之前需要创建套接字和绑定本地信息。
5.在发送和接受两个函数中在加入一个死循环,避免一直输入对方ip地址和端口号,造成麻烦。
只要循环不退出,默认只给一个对象发送数据/接收数据。
6.udp是全双工的,可以同时收发数据。
多线程代码如下所示:
import socket
import threading
def send_msg(udp_socket):
"""获取键盘数据,并将其发送给对方"""
while True:
# 1. 从键盘输入数据
# 2. 输入对方的ip地址
dest_ip = input("\n请输入对方的ip地址:")
dest_port = int(input("\n请输入对方的port:"))
while True:
# 3. 输入对方的port
msg = input("\n请输入要发送的数据:")
if msg:
# 4. 发送数据
udp_socket.sendto(msg.encode("utf-8"), (dest_ip, dest_port))
else: # 回车默认重新输入收到人信息
break
def recv_msg(udp_socket):
"""接收数据并显示"""
while True:
# 1. 接收数据
recv_msg = udp_socket.recvfrom(1024)
# 2. 解码
recv_ip = recv_msg[1]
recv_msg = recv_msg[0].decode("utf-8")
# 3. 显示接收到的数据
print(">>>%s:%s" % (str(recv_ip), recv_msg))
def main():
# 1. 创建套接字
udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
# 2. 绑定本地信息
udp_socket.bind(("", 7890))
send_msg_t = threading.Thread(target=send_msg,args=(udp_socket,))
recv_msg_t = threading.Thread(target=recv_msg, args=(udp_socket,))
send_msg_t.start()
recv_msg_t.start()
if __name__ == "__main__":
main()