update:自动构建

Author: openrz

.dockerignore

@@ -0,0 +1,61 @@
+# Git相关
+.git
+.gitignore
+
+# Python相关
+__pycache__
+*.pyc
+*.pyo
+*.pyd
+.Python
+env
+pip-log.txt
+pip-delete-this-directory.txt
+.tox
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.log
+.git
+.mypy_cache
+.pytest_cache
+.hypothesis
+
+# 虚拟环境
+venv/
+env/
+ENV/
+
+# IDE相关
+.vscode/
+.idea/
+*.swp
+*.swo
+*~
+
+# 操作系统相关
+.DS_Store
+.DS_Store?
+._*
+.Spotlight-V100
+.Trashes
+ehthumbs.db
+Thumbs.db
+
+# 日志文件
+logs/*.log
+
+# 临时文件
+*.tmp
+*.temp
+
+# Docker相关
+Dockerfile
+.dockerignore
+
+# 文档
+README*.md
+LICENSE 

.github/workflows/docker-image.yml

@@ -0,0 +1,62 @@
+name: Docker Image CI
+
+on:
+  push:
+    tags:
+      - 'v*.*.*'  # 只在以 v 开头的标签推送时触发，例如 v1.0.0
+  workflow_dispatch:
+
+jobs:
+  release:
+    name: Release Docker images
+    runs-on: ubuntu-latest
+    permissions:
+      packages: write
+      contents: write
+      id-token: write
+      issues: write
+    steps:
+      - name: Check Disk Space
+        run: |
+          df -h
+          docker system df
+
+      - name: Clean up Docker resources
+        run: |
+          docker system prune -af
+          docker builder prune -af
+
+      - name: Checkout code
+        uses: actions/checkout@v4
+
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v3
+
+      - name: Login to GitHub Container Registry
+        uses: docker/login-action@v3
+        with:
+          registry: ghcr.io
+          username: ${{ github.actor }}
+          password: ${{ secrets.TOKEN }}
+
+      - name: Extract version from tag
+        id: get_version
+        run: |
+          if [[ "$GITHUB_REF" =~ ^refs/tags/v([0-9]+\.[0-9]+\.[0-9]+)$ ]]; then
+            echo "VERSION=${BASH_REMATCH[1]}" >> $GITHUB_ENV
+            echo "IS_VERSION=true" >> $GITHUB_ENV
+          else
+            echo "VERSION=latest" >> $GITHUB_ENV
+            echo "IS_VERSION=false" >> $GITHUB_ENV
+          fi
+
+      # 构建 MCP Endpoint Server 镜像
+      - name: Build and push mcp-endpoint-server
+        uses: docker/build-push-action@v6
+        with:
+          context: .
+          file: Dockerfile
+          push: true
+          tags: |
+            ${{ env.IS_VERSION == 'true' && format('ghcr.io/{0}:{1},ghcr.io/{0}:latest', github.repository, env.VERSION) || format('ghcr.io/{0}:latest', github.repository) }}
+          platforms: linux/amd64,linux/arm64

Dockerfile

@@ -0,0 +1,29 @@
+# 第一阶段：构建Python依赖
+FROM python:3.10-slim AS builder
+
+# 设置工作目录
+WORKDIR /app
+
+# 复制依赖文件
+COPY requirements.txt .
+
+# 安装Python依赖
+RUN pip install --no-cache-dir -r requirements.txt
+
+# 第二阶段：运行阶段
+FROM python:3.10-slim
+
+# 设置工作目录
+WORKDIR /app
+
+# 从构建阶段复制Python依赖
+COPY --from=builder /usr/local/lib/python3.10/site-packages /usr/local/lib/python3.10/site-packages
+
+# 创建必要的目录
+RUN mkdir -p logs data
+
+# 复制应用代码
+COPY . .
+
+# 启动命令
+CMD ["python", "main.py"]

src/handlers/websocket_handler.py

@@ -4,7 +4,6 @@
 """
 
 import json
-import asyncio
 from typing import Optional, Any
 from urllib.parse import parse_qs, urlparse
 from websockets.server import WebSocketServerProtocol

src/server.py

@@ -26,6 +26,11 @@ async def lifespan(app: FastAPI):
     """应用生命周期管理"""
     # 启动时
     logger.info("MCP Endpoint Server 正在启动...")
+    logger.info(f"======================================================")
+    logger.info(
+        f"接口地址: http://{config.get('server', 'host', '127.0.0.1')}:{config.getint('server', 'port', 8004)}/mcp_endpoint/health?key={config.get('server', 'key', '')}"
+    )
+    logger.info("=======上面的地址是websocket协议地址，请勿用浏览器访问=======")
     yield
     # 关闭时
     logger.info("MCP Endpoint Server 已关闭")
@@ -63,10 +68,15 @@ async def root():
 
 
 @app.get("/mcp_endpoint/health")
-async def health_check():
+async def health_check(key: str = None):
     """健康检查"""
+    # 验证key参数
+    expected_key = config.get("server", "key", "")
+    if not key or key != expected_key:
+        return {"status": "key_error"}
+
     stats = connection_manager.get_connection_stats()
-    return {"status": "healthy", "connections": stats}
+    return {"status": "success", "connections": stats}
 
 
 @app.websocket("/mcp_endpoint/mcp/")