Initial commit: 000-platform project skeleton

sdk/http_client.py

@@ -0,0 +1,83 @@
+"""HTTP客户端封装"""
+import os
+from typing import Optional, Any, Dict
+import httpx
+
+from .trace import get_trace_id
+
+
+class PlatformHttpClient:
+    """平台HTTP客户端
+    
+    自动传递trace_id和API Key
+    
+    使用示例:
+        client = PlatformHttpClient(base_url="https://api.example.com")
+        response = await client.get("/users/1")
+    """
+    
+    def __init__(
+        self,
+        base_url: Optional[str] = None,
+        api_key: Optional[str] = None,
+        timeout: float = 30.0
+    ):
+        self.base_url = base_url or os.getenv("PLATFORM_URL", "")
+        self.api_key = api_key or os.getenv("PLATFORM_API_KEY", "")
+        self.timeout = timeout
+    
+    def _get_headers(self, extra_headers: Optional[Dict[str, str]] = None) -> Dict[str, str]:
+        """获取请求头"""
+        headers = {}
+        
+        # 添加trace_id
+        trace_id = get_trace_id()
+        if trace_id:
+            headers["X-Trace-ID"] = trace_id
+        
+        # 添加API Key
+        if self.api_key:
+            headers["X-API-Key"] = self.api_key
+        
+        # 合并额外header
+        if extra_headers:
+            headers.update(extra_headers)
+        
+        return headers
+    
+    async def request(
+        self,
+        method: str,
+        path: str,
+        **kwargs
+    ) -> httpx.Response:
+        """发送HTTP请求"""
+        headers = self._get_headers(kwargs.pop("headers", None))
+        
+        async with httpx.AsyncClient(
+            base_url=self.base_url,
+            timeout=self.timeout
+        ) as client:
+            response = await client.request(
+                method=method,
+                url=path,
+                headers=headers,
+                **kwargs
+            )
+            return response
+    
+    async def get(self, path: str, **kwargs) -> httpx.Response:
+        """GET请求"""
+        return await self.request("GET", path, **kwargs)
+    
+    async def post(self, path: str, **kwargs) -> httpx.Response:
+        """POST请求"""
+        return await self.request("POST", path, **kwargs)
+    
+    async def put(self, path: str, **kwargs) -> httpx.Response:
+        """PUT请求"""
+        return await self.request("PUT", path, **kwargs)
+    
+    async def delete(self, path: str, **kwargs) -> httpx.Response:
+        """DELETE请求"""
+        return await self.request("DELETE", path, **kwargs)