重构基本完成

app/api/common/test_api.py

@@ -0,0 +1,42 @@
+import asyncio
+import subprocess
+from fastapi import APIRouter
+from fastapi.responses import StreamingResponse
+
+
+test = APIRouter()
+
+
+async def generate_data():
+    for i in range(1, 10):  # 生成 5 行数据
+        await asyncio.sleep(1)  # 等待 1 秒
+        yield f"data: This is line {i}\n\n"  # 返回 SSE 格式的数据
+
+
+def run_command(command):
+    """执行命令并实时打印每一行输出"""
+    # 启动子进程
+    with subprocess.Popen(
+            command,
+            shell=True,
+            stdout=subprocess.PIPE,
+            stderr=subprocess.PIPE,
+            universal_newlines=True,  # 确保输出以字符串形式返回而不是字节
+            bufsize=1,  # 行缓冲
+    ) as process:
+        # 使用iter逐行读取stdout和stderr
+        for line in process.stdout:
+            yield f"stdout: {line.strip()} \n"
+
+        for line in process.stderr:
+            yield f"stderr: {line.strip()} \n"
+
+        # 等待进程结束并获取返回码
+        return_code = process.wait()
+        if return_code != 0:
+            print(f"Process exited with non-zero code: {return_code}")
+
+
+@test.get("/stream")
+async def stream_response():
+    return StreamingResponse(run_command(["ping", "-n", "10", "127.0.0.1"]), media_type="text/plain")