Compare commits
2 Commits
0e531a1876
...
ae4cf5d3c8
| Author | SHA1 | Date | |
|---|---|---|---|
| ae4cf5d3c8 | |||
| 18a1b2b01f |
38
Dockerfile.flask
Normal file
38
Dockerfile.flask
Normal file
@@ -0,0 +1,38 @@
|
|||||||
|
# Flask API 服务 Dockerfile
|
||||||
|
# =========================
|
||||||
|
# 用于构建 Universal Data Fetcher API 服务的 Docker 镜像
|
||||||
|
|
||||||
|
FROM index-base:latest
|
||||||
|
|
||||||
|
# 设置工作目录
|
||||||
|
WORKDIR /app
|
||||||
|
|
||||||
|
# 复制依赖文件
|
||||||
|
COPY requirements.txt .
|
||||||
|
|
||||||
|
# 安装依赖
|
||||||
|
RUN uv pip install --system -r requirements.txt
|
||||||
|
|
||||||
|
# 仅复制除 data 目录外的应用代码
|
||||||
|
COPY . .
|
||||||
|
|
||||||
|
# 创建日志目录
|
||||||
|
RUN mkdir -p /app/logs
|
||||||
|
|
||||||
|
# 设置时区为上海
|
||||||
|
ENV TZ=Asia/Shanghai
|
||||||
|
|
||||||
|
# 暴露 Flask 服务端口
|
||||||
|
EXPOSE 5000
|
||||||
|
|
||||||
|
# 设置环境变量默认值
|
||||||
|
ENV FLASK_APP=core/datasource/flask_server.py
|
||||||
|
ENV FLASK_ENV=production
|
||||||
|
ENV PYTHONUNBUFFERED=1
|
||||||
|
|
||||||
|
# 健康检查
|
||||||
|
HEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=3 \
|
||||||
|
CMD python -c "import requests; requests.get('http://localhost:5000/health')" || exit 1
|
||||||
|
|
||||||
|
# 启动 Flask 服务
|
||||||
|
CMD ["python", "core/datasource/flask_server.py", "--host", "0.0.0.0", "--port", "5000"]
|
||||||
118
build-flask-and-push.sh
Normal file
118
build-flask-and-push.sh
Normal file
@@ -0,0 +1,118 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
# Flask API 服务构建和推送脚本
|
||||||
|
# =============================
|
||||||
|
|
||||||
|
set -e # 遇到错误立即退出
|
||||||
|
|
||||||
|
# 配置变量
|
||||||
|
IMAGE_NAME="${1:-etf-data-fetcher}"
|
||||||
|
REGISTRY="192.168.0.115:5000"
|
||||||
|
TAG="${2:-latest}"
|
||||||
|
FULL_IMAGE_NAME="${REGISTRY}/${IMAGE_NAME}:${TAG}"
|
||||||
|
DOCKERFILE="Dockerfile.flask"
|
||||||
|
|
||||||
|
echo "========================================="
|
||||||
|
echo "Flask API 服务 - 构建和推送脚本"
|
||||||
|
echo "========================================="
|
||||||
|
echo "镜像名称: ${IMAGE_NAME}"
|
||||||
|
echo "标签: ${TAG}"
|
||||||
|
echo "Dockerfile: ${DOCKERFILE}"
|
||||||
|
echo ""
|
||||||
|
|
||||||
|
# 检查 Dockerfile 是否存在
|
||||||
|
if [ ! -f "${DOCKERFILE}" ]; then
|
||||||
|
echo "❌ Dockerfile ${DOCKERFILE} 不存在"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 检查并构建基础镜像(如果不存在)
|
||||||
|
echo "0. 检查基础镜像..."
|
||||||
|
if ! docker images --format "{{.Repository}}:{{.Tag}}" | grep -q "index-base:latest"; then
|
||||||
|
echo " 基础镜像不存在,开始构建..."
|
||||||
|
if [ -f "Dockerfile_base" ]; then
|
||||||
|
docker build --platform linux/arm64 -f Dockerfile_base -t index-base:latest .
|
||||||
|
echo " ✅ 基础镜像构建成功"
|
||||||
|
else
|
||||||
|
echo " ⚠️ 未找到 Dockerfile_base,跳过基础镜像构建"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
echo " ✅ 基础镜像已存在"
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 构建镜像
|
||||||
|
echo ""
|
||||||
|
echo "1. 构建 Flask API 镜像..."
|
||||||
|
echo " 镜像名称: ${IMAGE_NAME}"
|
||||||
|
echo " 平台: linux/arm64"
|
||||||
|
docker build --platform linux/arm64 -f ${DOCKERFILE} -t ${IMAGE_NAME} .
|
||||||
|
|
||||||
|
if [ $? -eq 0 ]; then
|
||||||
|
echo " ✅ 镜像构建成功"
|
||||||
|
else
|
||||||
|
echo " ❌ 镜像构建失败"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 打标签
|
||||||
|
echo ""
|
||||||
|
echo "2. 为镜像打标签..."
|
||||||
|
echo " 标签: ${FULL_IMAGE_NAME}"
|
||||||
|
docker tag ${IMAGE_NAME} ${FULL_IMAGE_NAME}
|
||||||
|
|
||||||
|
if [ $? -eq 0 ]; then
|
||||||
|
echo " ✅ 标签添加成功"
|
||||||
|
else
|
||||||
|
echo " ❌ 标签添加失败"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 推送到私有仓库
|
||||||
|
echo ""
|
||||||
|
echo "3. 推送镜像到私有仓库..."
|
||||||
|
echo " 仓库地址: ${REGISTRY}"
|
||||||
|
docker push ${FULL_IMAGE_NAME}
|
||||||
|
|
||||||
|
if [ $? -eq 0 ]; then
|
||||||
|
echo " ✅ 镜像推送成功"
|
||||||
|
else
|
||||||
|
echo " ❌ 镜像推送失败"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 清理本地镜像(可选)
|
||||||
|
echo ""
|
||||||
|
echo "4. 清理临时镜像..."
|
||||||
|
docker rmi ${IMAGE_NAME} || true
|
||||||
|
|
||||||
|
# 显示最终结果
|
||||||
|
echo ""
|
||||||
|
echo "========================================="
|
||||||
|
echo "🎉 构建和推送完成!"
|
||||||
|
echo "========================================="
|
||||||
|
echo "镜像地址: ${FULL_IMAGE_NAME}"
|
||||||
|
echo ""
|
||||||
|
echo "可以使用以下命令运行容器:"
|
||||||
|
echo ""
|
||||||
|
echo "# 基础运行(仅A股数据)"
|
||||||
|
echo "docker run -d --name ${IMAGE_NAME}-container \\"
|
||||||
|
echo " -p 5000:5000 \\"
|
||||||
|
echo " -v \$(pwd)/.env:/app/.env:ro \\"
|
||||||
|
echo " ${FULL_IMAGE_NAME}"
|
||||||
|
echo ""
|
||||||
|
echo "# 启用SSH隧道(支持港美股)"
|
||||||
|
echo "docker run -d --name ${IMAGE_NAME}-container \\"
|
||||||
|
echo " -p 5000:5000 \\"
|
||||||
|
echo " -v \$(pwd)/.env:/app/.env:ro \\"
|
||||||
|
echo " -v \$(pwd)/hk_ecs.pem:/app/hk_ecs.pem:ro \\"
|
||||||
|
echo " -e SSH_ENABLED=true \\"
|
||||||
|
echo " -e SSH_HOST=8.218.167.69 \\"
|
||||||
|
echo " -e SSH_USERNAME=root \\"
|
||||||
|
echo " -e SSH_KEY_PATH=hk_ecs.pem \\"
|
||||||
|
echo " ${FULL_IMAGE_NAME}"
|
||||||
|
echo ""
|
||||||
|
echo "# 测试API"
|
||||||
|
echo "curl http://localhost:5000/health"
|
||||||
|
echo "curl 'http://localhost:5000/api/v1/ohlcv?code=000300.SH&start=2024-01-01&end=2024-03-31'"
|
||||||
|
echo ""
|
||||||
|
echo "========================================="
|
||||||
80
docker-compose.flask.yml
Normal file
80
docker-compose.flask.yml
Normal file
@@ -0,0 +1,80 @@
|
|||||||
|
# Flask API 服务 Docker Compose 配置
|
||||||
|
# ===================================
|
||||||
|
|
||||||
|
version: '3.8'
|
||||||
|
|
||||||
|
services:
|
||||||
|
etf-data-fetcher:
|
||||||
|
build:
|
||||||
|
context: .
|
||||||
|
dockerfile: Dockerfile.flask
|
||||||
|
image: 192.168.0.115:5000/etf-data-fetcher:latest
|
||||||
|
container_name: etf-data-fetcher
|
||||||
|
|
||||||
|
ports:
|
||||||
|
- "5000:5000"
|
||||||
|
|
||||||
|
environment:
|
||||||
|
# Flask 配置
|
||||||
|
- FLASK_ENV=production
|
||||||
|
- PYTHONUNBUFFERED=1
|
||||||
|
|
||||||
|
# SSH 隧道配置(可选,启用后可获取港美股数据)
|
||||||
|
- SSH_ENABLED=${SSH_ENABLED:-false}
|
||||||
|
- SSH_HOST=${SSH_HOST:-8.218.167.69}
|
||||||
|
- SSH_PORT=${SSH_PORT:-22}
|
||||||
|
- SSH_USERNAME=${SSH_USERNAME:-root}
|
||||||
|
- SSH_KEY_PATH=${SSH_KEY_PATH:-hk_ecs.pem}
|
||||||
|
- SSH_LOCAL_PORT=${SSH_LOCAL_PORT:-1080}
|
||||||
|
|
||||||
|
volumes:
|
||||||
|
# 环境变量配置(必需:TUSHARE_TOKEN)
|
||||||
|
- ./.env:/app/.env:ro
|
||||||
|
|
||||||
|
# SSH 私钥(可选,获取港美股数据需要)
|
||||||
|
- ./hk_ecs.pem:/app/hk_ecs.pem:ro
|
||||||
|
|
||||||
|
# 数据目录(可选,用于持久化缓存)
|
||||||
|
- ./data:/app/data
|
||||||
|
|
||||||
|
# 日志目录
|
||||||
|
- ./logs:/app/logs
|
||||||
|
|
||||||
|
# 健康检查
|
||||||
|
healthcheck:
|
||||||
|
test: ["CMD", "python", "-c", "import requests; requests.get('http://localhost:5000/health')"]
|
||||||
|
interval: 30s
|
||||||
|
timeout: 10s
|
||||||
|
retries: 3
|
||||||
|
start_period: 10s
|
||||||
|
|
||||||
|
# 重启策略
|
||||||
|
restart: unless-stopped
|
||||||
|
|
||||||
|
# 资源限制
|
||||||
|
deploy:
|
||||||
|
resources:
|
||||||
|
limits:
|
||||||
|
cpus: '2.0'
|
||||||
|
memory: 2G
|
||||||
|
reservations:
|
||||||
|
cpus: '0.5'
|
||||||
|
memory: 512M
|
||||||
|
|
||||||
|
# 可选:Nginx 反向代理
|
||||||
|
nginx:
|
||||||
|
image: nginx:alpine
|
||||||
|
container_name: etf-data-fetcher-nginx
|
||||||
|
ports:
|
||||||
|
- "80:80"
|
||||||
|
volumes:
|
||||||
|
- ./nginx.conf:/etc/nginx/nginx.conf:ro
|
||||||
|
depends_on:
|
||||||
|
- etf-data-fetcher
|
||||||
|
restart: unless-stopped
|
||||||
|
profiles:
|
||||||
|
- with-nginx
|
||||||
|
|
||||||
|
networks:
|
||||||
|
default:
|
||||||
|
name: etf-network
|
||||||
@@ -15,6 +15,10 @@ requests[socks]>=2.28.0
|
|||||||
urllib3>=1.26.0
|
urllib3>=1.26.0
|
||||||
pysocks>=1.7.0
|
pysocks>=1.7.0
|
||||||
|
|
||||||
|
# ==================== Web服务 ====================
|
||||||
|
flask>=3.0.0
|
||||||
|
flask-cors>=4.0.0
|
||||||
|
|
||||||
# ==================== 进度条 ====================
|
# ==================== 进度条 ====================
|
||||||
tqdm>=4.65.0
|
tqdm>=4.65.0
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user