optimization bat

.container/build_docker.bat

@@ -1,22 +1,145 @@
 @echo off
+setlocal enabledelayedexpansion
+
 echo 在Windows上构建Docker镜像...
 
+REM 设置配置变量
+set CACHE_DIR=.docker-cache\pip
+set BUILD_ARGS=--build-arg BUILDKIT_INLINE_CACHE=1
+set COMPOSE_FILE=docker-compose.yml
+
+REM 解析命令行参数
+set CLEAN_CACHE=0
+set REBUILD=0
+set SERVICE=
+
+:parse_args
+if "%~1"=="" goto :end_parse_args
+if /i "%~1"=="--clean" (
+    set CLEAN_CACHE=1
+    shift
+    goto :parse_args
+)
+if /i "%~1"=="--rebuild" (
+    set REBUILD=1
+    shift
+    goto :parse_args
+)
+if /i "%~1"=="--service" (
+    set SERVICE=%~2
+    shift
+    shift
+    goto :parse_args
+)
+if /i "%~1"=="--help" (
+    echo 用法: build_docker.bat [选项]
+    echo 选项:
+    echo   --clean     清理缓存目录
+    echo   --rebuild   强制重新构建镜像
+    echo   --service   指定要构建的服务名称
+    echo   --help      显示此帮助信息
+    exit /b 0
+)
+shift
+goto :parse_args
+:end_parse_args
+
+REM 检查Docker是否安装
+where docker >nul 2>nul
+if %ERRORLEVEL% NEQ 0 (
+    echo 错误: Docker未安装
+    echo 请先安装Docker Desktop: https://docs.docker.com/desktop/install/windows-install/
+    pause
+    exit /b 1
+)
+
+REM 检查Docker是否运行
+docker info >nul 2>nul
+if %ERRORLEVEL% NEQ 0 (
+    echo 错误: Docker未运行
+    echo 请启动Docker Desktop应用程序
+    pause
+    exit /b 1
+)
+
+REM 检查docker-compose.yml文件是否存在
+if not exist "%COMPOSE_FILE%" (
+    echo 错误: 未找到%COMPOSE_FILE%文件
+    echo 请确保在正确的目录中运行此脚本
+    pause
+    exit /b 1
+)
+
+REM 检查Docker Compose命令
+where docker-compose >nul 2>nul
+if %ERRORLEVEL% EQU 0 (
+    set COMPOSE_CMD=docker-compose
+) else (
+    echo 尝试使用新的docker compose命令...
+    docker compose version >nul 2>nul
+    if %ERRORLEVEL% EQU 0 (
+        set COMPOSE_CMD=docker compose
+    ) else (
+        echo 错误: 未找到Docker Compose命令
+        echo 请确保Docker Desktop已正确安装
+        pause
+        exit /b 1
+    )
+)
+
 REM 设置Docker BuildKit环境变量
 set DOCKER_BUILDKIT=1
 set COMPOSE_DOCKER_CLI_BUILD=1
 
 echo 启用Docker BuildKit加速构建...
 
+REM 清理缓存（如果指定）
+if %CLEAN_CACHE% EQU 1 (
+    echo 清理缓存目录...
+    if exist "%CACHE_DIR%" rmdir /s /q "%CACHE_DIR%"
+)
+
 REM 创建缓存目录
-if not exist ".docker-cache\pip" mkdir .docker-cache\pip
+if not exist "%CACHE_DIR%" (
+    echo 创建缓存目录...
+    mkdir "%CACHE_DIR%"
+)
+
+REM 添加构建时间标记
+for /f "tokens=2 delims==" %%a in ('wmic OS Get localdatetime /value') do set "dt=%%a"
+set "YEAR=%dt:~0,4%"
+set "MONTH=%dt:~4,2%"
+set "DAY=%dt:~6,2%"
+set "HOUR=%dt:~8,2%"
+set "MINUTE=%dt:~10,2%"
+set "BUILD_TIME=%YEAR%%MONTH%%DAY%_%HOUR%%MINUTE%"
+set "BUILD_ARGS=%BUILD_ARGS% --build-arg BUILD_TIME=%BUILD_TIME%"
 
 REM 构建Docker镜像
-docker-compose build --build-arg BUILDKIT_INLINE_CACHE=1
+echo 开始构建Docker镜像...
+
+if "%SERVICE%"=="" (
+    if %REBUILD% EQU 1 (
+        echo 强制重新构建所有服务...
+        %COMPOSE_CMD% build --no-cache %BUILD_ARGS%
+    ) else (
+        %COMPOSE_CMD% build %BUILD_ARGS%
+    )
+) else (
+    if %REBUILD% EQU 1 (
+        echo 强制重新构建服务 %SERVICE%...
+        %COMPOSE_CMD% build --no-cache %BUILD_ARGS% %SERVICE%
+    ) else (
+        echo 构建服务 %SERVICE%...
+        %COMPOSE_CMD% build %BUILD_ARGS% %SERVICE%
+    )
+)
 
 if %ERRORLEVEL% EQU 0 (
     echo Docker镜像构建成功！
+    echo 构建时间: %BUILD_TIME%
     echo 可以使用以下命令启动容器：
-    echo docker-compose up -d
+    echo %COMPOSE_CMD% up -d
 ) else (
     echo Docker镜像构建失败，请检查错误信息。
 )