【Python】シングルスレッドで複数通信を処理！selectを使ったイベント駆動プログラミングの基本

import time
import threading
import socket
import select

PORT_NUM = 12345
stop_application = False

def create_server(ip, port):
    server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    server.bind((ip, port))
    server.listen()
    return server

def create_client(ip, port):
    client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    client.connect((ip, port))
    return client

# クライアント側から通信のきっかけを作るタイマー処理
def tm_callback(con):
    if not stop_application:
        snd = bytearray([5, 6, 7, 8])
        con.send(snd)

# サーバー側スレッド処理
def th1_func():
    global stop_application
    server = create_server('127.0.0.1', PORT_NUM)
    (con, client) = server.accept()
    
    # 監視対象のリストを作成（今回は接続されたソケットのみ）
    fd_set = set()
    fd_set.add(con)
    
    while True:
        try:
            if stop_application:
                con.close()
                break
            
            # select関数でデータの受信を監視
            r, w, x = select.select(list(fd_set), [], [], 10)
            
            for fd in r:
                data = fd.recv(1024)
                print(f"Server received: {data}")
                time.sleep(0.1)
                snd = bytearray([1, 2, 3, 4])
                fd.send(snd)  # 応答を返す
                
        except Exception:
            con.close()
            break

# クライアント側スレッド処理
def th2_func():
    global stop_application
    con = create_client('127.0.0.1', PORT_NUM)
    
    fd_set = set()
    fd_set.add(con)
    
    # 1秒後に通信のきっかけを送信
    tm = threading.Timer(1, tm_callback, args=[con,])
    tm.start()
    
    while True:
        try:
            if stop_application:
                con.close()
                break
                
            r, w, x = select.select(list(fd_set), [], [], 10)
            
            for fd in r:
                data = fd.recv(1024)
                print(f"Client received: {data}")
                time.sleep(0.1)
                snd = bytearray([9, 10, 11, 12])
                fd.send(snd)  # 応答を返す
                
        except Exception:
            con.close()
            break

# メイン処理
def main_func():
    global stop_application
    
    # サーバーとクライアントのスレッドを起動
    thread1 = threading.Thread(target=th1_func)
    thread1.start()

    thread2 = threading.Thread(target=th2_func)
    thread2.start()
    
    while True:
        try:
            time.sleep(1)
        except KeyboardInterrupt:
            # Ctrl+Cで安全に停止
            print("stop")
            stop_application = True
            thread1.join()
            thread2.join()
            break

if __name__ == '__main__':
    main_func()

import select

fd_set = set()
fd_set.add(con) # 監視したいソケットを追加

r, w, x = select.select(list(fd_set), [], [], 10)

for fd in r:
    data = fd.recv(1024) 
    fd.send(snd)