fix: 首包转发到wstunnel后端, shutdown延迟调用, macOS兼容

.gitattributes

@@ -1,35 +0,0 @@
-*.7z filter=lfs diff=lfs merge=lfs -text
-*.arrow filter=lfs diff=lfs merge=lfs -text
-*.bin filter=lfs diff=lfs merge=lfs -text
-*.bz2 filter=lfs diff=lfs merge=lfs -text
-*.ckpt filter=lfs diff=lfs merge=lfs -text
-*.ftz filter=lfs diff=lfs merge=lfs -text
-*.gz filter=lfs diff=lfs merge=lfs -text
-*.h5 filter=lfs diff=lfs merge=lfs -text
-*.joblib filter=lfs diff=lfs merge=lfs -text
-*.lfs.* filter=lfs diff=lfs merge=lfs -text
-*.mlmodel filter=lfs diff=lfs merge=lfs -text
-*.model filter=lfs diff=lfs merge=lfs -text
-*.msgpack filter=lfs diff=lfs merge=lfs -text
-*.npy filter=lfs diff=lfs merge=lfs -text
-*.npz filter=lfs diff=lfs merge=lfs -text
-*.onnx filter=lfs diff=lfs merge=lfs -text
-*.ot filter=lfs diff=lfs merge=lfs -text
-*.parquet filter=lfs diff=lfs merge=lfs -text
-*.pb filter=lfs diff=lfs merge=lfs -text
-*.pickle filter=lfs diff=lfs merge=lfs -text
-*.pkl filter=lfs diff=lfs merge=lfs -text
-*.pt filter=lfs diff=lfs merge=lfs -text
-*.pth filter=lfs diff=lfs merge=lfs -text
-*.rar filter=lfs diff=lfs merge=lfs -text
-*.safetensors filter=lfs diff=lfs merge=lfs -text
-saved_model/**/* filter=lfs diff=lfs merge=lfs -text
-*.tar.* filter=lfs diff=lfs merge=lfs -text
-*.tar filter=lfs diff=lfs merge=lfs -text
-*.tflite filter=lfs diff=lfs merge=lfs -text
-*.tgz filter=lfs diff=lfs merge=lfs -text
-*.wasm filter=lfs diff=lfs merge=lfs -text
-*.xz filter=lfs diff=lfs merge=lfs -text
-*.zip filter=lfs diff=lfs merge=lfs -text
-*.zst filter=lfs diff=lfs merge=lfs -text
-*tfevents* filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1,2 @@
+__pycache__/
+*.pyc

server.py

@@ -1,9 +1,7 @@
 import asyncio
 import os
-import subprocess
 import signal
 import sys
-import functools
 
 WSTUNNEL_PORT = 7861
 PROXY_PORT = 7860
@@ -16,7 +14,7 @@ async def health_check(reader, writer):
     writer.close()
 
 
-async def tcp_proxy(reader, writer):
+async def tcp_proxy(reader, writer, initial_data=None):
     try:
         backend_reader, backend_writer = await asyncio.wait_for(
             asyncio.open_connection("127.0.0.1", WSTUNNEL_PORT), timeout=5
@@ -27,6 +25,10 @@ async def tcp_proxy(reader, writer):
         writer.close()
         return
 
+    if initial_data:
+        backend_writer.write(initial_data)
+        await backend_writer.drain()
+
     async def forward(src, dst):
         try:
             while True:
@@ -62,9 +64,7 @@ async def handle_client(reader, writer):
         if is_health_check:
             await health_check(reader, writer)
         else:
-            writer.write(data)
-            await writer.drain()
-            await tcp_proxy(reader, writer)
+            await tcp_proxy(reader, writer, initial_data=data)
     except (ConnectionResetError, BrokenPipeError, OSError, asyncio.TimeoutError):
         pass
     finally:
@@ -105,9 +105,9 @@ async def main():
     loop = asyncio.get_event_loop()
     for sig in (signal.SIGTERM, signal.SIGINT):
         try:
-            loop.add_signal_handler(sig, functools.partial(asyncio.create_task, shutdown()))
+            loop.add_signal_handler(sig, lambda: asyncio.create_task(shutdown()))
         except NotImplementedError:
-            pass
+            signal.signal(sig, lambda n, f: asyncio.run_coroutine_threadsafe(shutdown(), loop))
 
     async with server:
         print(f"proxy server ready on port {PROXY_PORT}", flush=True)