如何在不阻塞主线程的情况下实现UDP

import socket
import threading

def receive_data(sock):
    while True:
        data, addr = sock.recvfrom(1024)
        # 处理接收到的数据
        print("Received data:", data.decode())

def send_data(sock):
    while True:
        message = input("Enter message to send: ")
        sock.sendto(message.encode(), ('127.0.0.1', 12345))

def main():
    # 创建UDP套接字
    sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    # 设置套接字为非阻塞模式
    sock.setblocking(False)
    # 绑定套接字到指定的IP地址和端口
    sock.bind(('127.0.0.1', 12345))

    # 创建接收数据的线程
    receive_thread = threading.Thread(target=receive_data, args=(sock,))
    receive_thread.start()

    # 创建发送数据的线程
    send_thread = threading.Thread(target=send_data, args=(sock,))
    send_thread.start()

if __name__ == '__main__':
    main()