Python多任务编程-07多线程版udp聊天程序

旧的代码为单线程,只能同时接收数据或同时发送数据。

单线程代码如下所示:

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()
相关推荐
blammmp22 分钟前
Java:数据结构-枚举
java·开发语言·数据结构
何曾参静谧34 分钟前
「C/C++」C/C++ 指针篇 之 指针运算
c语言·开发语言·c++
暗黑起源喵40 分钟前
设计模式-工厂设计模式
java·开发语言·设计模式
WaaTong1 小时前
Java反射
java·开发语言·反射
Troc_wangpeng1 小时前
R language 关于二维平面直角坐标系的制作
开发语言·机器学习
努力的家伙是不讨厌的1 小时前
解析json导出csv或者直接入库
开发语言·python·json
Envyᥫᩣ1 小时前
C#语言:从入门到精通
开发语言·c#
云空1 小时前
《Python 与 SQLite:强大的数据库组合》
数据库·python·sqlite
童先生1 小时前
Go 项目中实现类似 Java Shiro 的权限控制中间件?
开发语言·go
lulu_gh_yu1 小时前
数据结构之排序补充
c语言·开发语言·数据结构·c++·学习·算法·排序算法