python异步tcp代理服务器

import asyncio



class TcpServer:
    def __init__(self, addr, proxy_addr):
        self.addr = addr
        self.proxy_addr = proxy_addr
        self.server = None

    async def get_connect(self):
        reader, writer = await asyncio.open_connection(*self.proxy_addr)
        return reader, writer

    async def handle_data(self, data, flg):

        return data  # 这里简单返回原数据

    async def handle_rw(self, r, w, flg):
        try:
            while True:
                data = await r.read(1024)  # 读取数据，这里假设每次最多读取1024字节
                if not data:
                    break  # 连接关闭或没有更多数据
                processed_data = await self.handle_data(data, flg)
                w.write(processed_data)  # 发送处理后的数据
                await w.drain()  # 等待数据发送完毕
        except Exception as e:
            print(f"Error: {e}")
        finally:
            w.close()
            await asyncio.sleep(0.1)  # 确保关闭操作完成


    async