# 文件路径：scripts/async_requests.py
# 异步并发调用多个 API，汇总结果

import asyncio
import aiohttp      # pip install aiohttp --break-system-packages
import json
import sys
import time

API_KEY  = "your_api_key"
BASE_URL = "https://api.example.com/v1"

async def fetch_one(session: aiohttp.ClientSession,
                    endpoint: str, payload: dict) -> dict:
    """异步发出单个 POST 请求"""
    url     = BASE_URL + endpoint
    headers = {"Authorization": f"Bearer {API_KEY}"}
    try:
        async with session.post(url, json=payload,
                                headers=headers, timeout=aiohttp.ClientTimeout(total=30)) as resp:
            resp.raise_for_status()
            return {"endpoint": endpoint, "status": "ok", "data": await resp.json()}
    except Exception as e:
        return {"endpoint": endpoint, "status": "error", "message": str(e)}

async def fetch_all(requests: list) -> list:
    """
    并发发送所有请求，等待全部完成

    参数：
        requests: 列表，每项为 (endpoint, payload) 元组
    """
    async with aiohttp.ClientSession() as session:
        tasks = [
            fetch_one(session, endpoint, payload)
            for endpoint, payload in requests
        ]
        # gather 并发执行所有 task，return_exceptions=True 防止一个失败影响其他
        return await asyncio.gather(*tasks, return_exceptions=True)

def main():
    # 定义需要并发调用的请求列表
    requests = [
        ("/summarize", {"text": "runoob 是技术学习平台", "lang": "zh"}),
        ("/keywords",  {"text": "RUNOOB 提供各类编程教程", "top_n": 5}),
        ("/sentiment", {"text": "这个平台的教程质量很高"}),
    ]

    t0      = time.perf_counter()
    results = asyncio.run(fetch_all(requests))
    elapsed = time.perf_counter() - t0

    print(f"并发完成 {len(results)} 个请求，总耗时 {elapsed:.2f} 秒")
    print(json.dumps(results, ensure_ascii=False, indent=2))

if __name__ == "__main__":
    main()

# 文件路径：scripts/async_file_reader.py
import asyncio
import aiofiles   # pip install aiofiles --break-system-packages
import os

async def read_file(file_path: str) -> dict:
    """异步读取单个文件"""
    try:
        async with aiofiles.open(file_path, encoding="utf-8") as f:
            content = await f.read()
        return {
            "file":    os.path.basename(file_path),
            "chars":   len(content),
            "lines":   content.count("\n"),
            "status":  "ok"
        }
    except Exception as e:
        return {"file": file_path, "status": "error", "message": str(e)}

async def read_all_files(file_paths: list) -> list:
    """并发读取所有文件"""
    tasks = [read_file(fp) for fp in file_paths]
    return await asyncio.gather(*tasks)

if __name__ == "__main__":
    import glob, json
    # 读取上传目录中的所有 .txt 文件
    files = glob.glob("/mnt/user-data/uploads/*.txt")
    if not files:
        print("未找到 .txt 文件")
    else:
        results = asyncio.run(read_all_files(files))
        print(json.dumps(results, ensure_ascii=False, indent=2))

# 文件路径：scripts/async_timeout.py
import asyncio

async def slow_task(name: str, delay: float) -> str:
    """模拟一个耗时任务"""
    await asyncio.sleep(delay)
    return f"{name} 完成"

async def run_with_timeout():
    tasks = {
        "快速任务": slow_task("快速任务", 0.5),
        "慢速任务": slow_task("慢速任务", 5.0),
    }
    results = {}
    for name, coro in tasks.items():
        try:
            # 每个任务最多等待 2 秒
            result = await asyncio.wait_for(coro, timeout=2.0)
            results[name] = {"status": "ok", "result": result}
        except asyncio.TimeoutError:
            results[name] = {"status": "timeout", "message": "超过 2 秒未完成"}
    return results

if __name__ == "__main__":
    import json
    output = asyncio.run(run_with_timeout())
    print(json.dumps(output, ensure_ascii=False, indent=2))