使用python socket实现双向的tcp通信

# coding: utf-8
# tcp stream server
import socket
import logging
import time
import datetime as dt
from threading import Thread, currentThread
# 配置logging
logging.basicConfig(
    level=logging.INFO,
    format="%(asctime)s %(levelname)s %(process)s %(threadName)s |%(message)s",
)


class Server:
    """ socket 服务端 """
    def __init__(self, host='localhost', port=8099):
        self._sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self._sock.bind((host, port))
        self.msg = None

    def read(self, conn: socket.socket = None):
        """ 从tcp连接里面读取数据 """
        while True:
            try:
                data = conn.recv(1024).decode()
            except Exception as e:
                logging.info("recv failed: %s", e)
                return
            logging.info("[R %s]<< %s", currentThread().getName(), data)
            self.msg = data
            time.sleep(1)

    def write(self, conn: socket.socket = None):
        """ 向tcp连接里面写入数据 """
        while True:
            msg = f"{dt.datetime.now()} - {self.msg}"
            logging.info("[W %s]>> %s", currentThread().getName(), msg)
            try:
                conn.send(msg.encode())
            except Exception as e:
                logging.info("send failed: %s", e)
                return
            time.sleep(1)

    def serve(self):
        """ 开启服务 """
        self._sock.listen()
        logging.info("Serving...")
        while True:
            logging.info("Waiting for connection...")
            conn, addr = self._sock.accept()
            logging.info("Recived new conn: %s from %s", conn, addr)
            # 开启读写线程处理当前连接
            Thread(target=self.read, args=(conn, )).start()
            Thread(target=self.write, args=(conn, )).start()


if __name__ == '__main__':
    s = Server()
    s.serve()

# coding: utf-8
# tcp stream client
import socket
import logging
import time
import os
from threading import Thread, currentThread

# 配置logging
logging.basicConfig(
    level=logging.INFO,
    format="%(asctime)s %(levelname)s %(process)s %(threadName)s |%(message)s",
)


class Client:
    """ socket 客户端 """
    def __init__(self, host='localhost', port=8099):
        self._sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self._addr = (host, port)

    def read(self):
        """ 向连接中接受数据 """
        while True:
            try:
                data = self._sock.recv(1024).decode()
            except Exception as e:
                logging.info("recv failed: %s", e)
                return
            logging.info("[R %s]<< %s", currentThread().getName(), data)
            time.sleep(1)

    def write(self):
        """ 向连接中发送随机数 """
        while True:
            msg = os.urandom(4).hex()
            logging.info("[W %s]>> %s", currentThread().getName(), msg)
            try:
                self._sock.send(msg.encode())
            except Exception as e:
                logging.info("send failed: %s", e)
                return
            time.sleep(1)

    def run(self):
        """ 开启连接 """
        self._sock.connect(self._addr)
        logging.info("New connection: %s", self._sock)
        r = Thread(target=self.read)
        r.start()
        w = Thread(target=self.write)
        w.start()
        r.join()
        w.join()


if __name__ == '__main__':
    client = Client()
    client.run()

class Client(Thread):
    pass

if __name__ == '__main__':
    cs = [Client() for _ in range(3)]
    for c in cs:
        c.start()
    for c in cs:
        c.join()

class Server:
    def __init__(self, host='localhost', port=8099):
        # ...
        self.msg = {}

    def read(self, conn: socket.socket = None):
        # ...
        self.msg[conn.fileno()] = data

    def write(self, conn: socket.socket = None):
        # ...
        msg = f"{dt.datetime.now()} - {self.msg.get(conn.fileno())}"

    def serve(self):
        # ...
        while True:
            # ...
            conn, addr = self._sock.accept()
            self.msg[conn.fileno()] = ''
            # ...