diff --git a/base/Resources/mysql/initialize.bat b/base/Resources/mysql/initialize.bat new file mode 100644 index 0000000..f99ddc9 --- /dev/null +++ b/base/Resources/mysql/initialize.bat @@ -0,0 +1,453 @@ +@echo off +chcp 65001 >nul +setlocal enabledelayedexpansion + +:: version: 0.13 + +:: ===================================================================== 主逻辑顺序 + +:: 初始化脚本. +call :INITIALIZE +:: 调用主函数. +goto :main >nul +exit /b 0 + +:INITIALIZE + set "SCRIPT_PATH=%~dp0" + set "SCRIPT_NAME=%~nx0" + set "SCRIPT_NAME_WITHOUT_EXT=%~n0" + set "LOG_LEVEL_THRESHOLD=DEBUG" + + call :get_current_time + set "CALL_TIME=!CURRENT_TIME!" + call :LOG_INFO "Startup %SCRIPT_NAME%: %CALL_TIME%" + + set "SHOW_TITLE=true" + for %%a in (%*) do (if /i "%%a"=="/notitle" set set "SHOW_TITLE=false") + if "!SHOW_TITLE!"=="true" (title %SCRIPT_NAME%) + +exit /b 0 + +:CHECK_IS_ADMIN + net session >nul 2>&1 + if %errorLevel% neq 0 ( + exit /b 1 + ) +exit /b 0 + +:MD52String +setlocal +set "input=%~1" +for /f "delims=" %%i in (' + powershell -Command "[System.BitConverter]::ToString((New-Object -TypeName System.Security.Cryptography.MD5CryptoServiceProvider).ComputeHash([System.Text.Encoding]::UTF8.GetBytes('!input!'))).Replace('-','').ToLower()" +') do set "md5=%%i" +endlocal & set "md5=%md5%" +exit /b 0 + +:MD5Hash2File +setlocal +set "input_text=%~1" +set "return_var=%~2" +where md5sum >nul 2>nul +if errorlevel 1 ( + echo Error: md5sum command not found in PATH + endlocal + exit /b 1 +) +set "temp_file=%TEMP%\md5_%RANDOM%.txt" +echo !input_text! > "!temp_file!" +for /f "tokens=1" %%i in ('md5sum "!temp_file!" 2^>nul') do ( + set "hash_result=%%i" +) +if "!hash_result!"=="" ( + echo Error: Failed to calculate MD5 hash + del "!temp_file!" 2>nul + endlocal + exit /b 2 +) +del "!temp_file!" 2>nul +endlocal & set "%return_var%=%hash_result%" +exit /b 0 + +:GET_TIME_STAMP + setLocal + set "level=%~1" + if /i "%level%"=="ms" ( + for /f %%i in ('powershell -command "[math]::Truncate([decimal]((Get-Date).ToUniversalTime() - [datetime]'1970-01-01').TotalMilliseconds)"') do set timestamp_ms=%%i + if %errorlevel% == 0 ( + set "timestamp=!timestamp_ms!" + ) + ) + if /i "%level%"=="s" ( + for /f %%i in ('powershell -command "[math]::Truncate([decimal]((Get-Date).ToUniversalTime() - [datetime]'1970-01-01').TotalSeconds)"') do set timestamp_s=%%i + if %errorlevel% == 0 ( + set "timestamp=!timestamp_s!" + ) + ) + endLocal & set "timestamp=%timestamp%" +exit /b 0 + +:GET_DATE_TIME + for /f "tokens=2-4 delims=/ " %%a in ('date /t') do (set __date=%%a-%%b-%%c) + for /f "tokens=1-2 delims=/:" %%a in ('time /t') do (set __time=%%a%%b) + set "CURRENT_DATE_TIME=%__date%_%__time%" +exit /b 0 + +:GET_CURRENT_TIME + for /f "tokens=2 delims==" %%a in ('wmic OS Get localdatetime /value') do set "datetime=%%a" + set "CURRENT_TIME=%datetime:~0,4%-%datetime:~4,2%-%datetime:~6,2% %datetime:~8,2%:%datetime:~10,2%:%datetime:~12,2%" +exit /b 0 + +:GET_CURRENT_DATE + for /f "tokens=2 delims==" %%a in ('wmic OS Get localdatetime /value') do set "datetime=%%a" + set "CURRENT_DATE=%datetime:~0,4%%datetime:~4,2%%datetime:~6,2%" +exit /b 0 + +:WAIT + set /a "SECOND=%~1" + timeout /t %SECOND% /nobreak >nul +exit /b 0 + +:LOG + setLocal + set "LOG_LEVEL=%~1" + set LOG_MESSAGE= + shift + :log__params_loop + if not "%~1"=="" ( + set "LOG_MESSAGE=%LOG_MESSAGE% %~1" + shift + goto log__params_loop + ) + call :get_log_level_num "%LOG_LEVEL%" "CALLER_LEVEL_NUM" + call :get_log_level_num "%LOG_LEVEL_THRESHOLD%" "THRESHOLD_LEVEL_NUM" + if %CALLER_LEVEL_NUM% lss %THRESHOLD_LEVEL_NUM% ( + endLocal + exit /b 0 + ) + call :GET_CURRENT_TIME + if /i "%LOG_LEVEL%"=="GROUP" echo. + echo [%CURRENT_TIME%] - [%LOG_LEVEL%] - %LOG_MESSAGE% + endLocal +exit /b 0 +:get_log_level_num + setLocal + set "level=%~1" + set "num=0" + if /i "%level%"=="TRACE" set "num=1" + if /i "%level%"=="DEBUG" set "num=2" + if /i "%level%"=="INFO" set "num=3" + if /i "%level%"=="WARN" set "num=4" + if /i "%level%"=="GROUP" set "num=5" + if /i "%level%"=="ERROR" set "num=6" + endLocal & set "%~2=%num%" +exit /b 0 + +:LOG_ERROR + call :LOG ERROR %* && exit /b 0 + +:LOG_GROUP + call :LOG GROUP "===============================" %* "===============================" +exit /b 0 + +:LOG_WARN + call :LOG WARN %* && exit /b 0 + +:LOG_INFO + call :LOG INFO %* && exit /b 0 + +:LOG_DEBUG + call :LOG DEBUG %* && exit /b 0 + +:LOG_DEBUG + call :LOG DEBUG %* && exit /b 0 + +:LOG_TRACE + call :LOG TRACE %* && exit /b 0 + +:GET_STRING_LENGTH + setlocal + set "input_str=%~1" + set "str_length=%~2" + :get_string_length_loop + if defined input_str ( + set /a str_length+=1 + set "input_str=!input_str:~1!" + goto get_string_length_loop + ) + endlocal & ( + if not "%~2"=="" ( + set "%~2=%str_length%" + ) else ( + exit /b %str_length% + ) + ) +exit /b 0 + +:GET_STRING_LENGTH_WITHOUT_SPACES + setlocal + set "input_str=%~1" + set "result_var=%~2" + set "tmp_char=!input_str!" + set /a str_length=0 + for %%p in ( 4096 2048 1024 512 256 128 64 32 16 8 4 2 1 ) do ( + if not "!tmp_char:~%%p!"=="" ( + set "tmp_char=!tmp_char:~%%p!" + set /a str_length+=%%p + ) + ) + if not "!tmp_char!"=="" set /a str_length+=1 + call :LOG_TRACE Length=%str_length% String=%input_str% + endlocal & ( + if not "%~2"=="" ( + set "%~2=%str_length%" + ) else ( + exit /b %str_length% + ) + ) +exit /b 0 + +:RANDOM_STR + setlocal + set "len=%~1" + set "varname=%~2" + if "!varname!"=="" set "varname=RAND_STR" + set "chars=ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789" + set "random_str=" + call :GET_STRING_LENGTH "%chars%" "charts_length" + for /l %%i in (1,1,!len!) do ( + set /a rand_index=!random! %% !charts_length! + call set "char=%%chars:~!rand_index!,1%%" + set "random_str=!random_str!!char!" + ) + endlocal & set "%varname%=%random_str%" +exit /b 0 + +:GET_SINGLE_PID_BY_PORT + setlocal + call :RandomStr single_pid_tmp + call :LOG_INFO netstat -ano ^| grep -E ":%1 " ^| grep "LISTENING" ^| awk "{print $NF}" ^| head -n 1 + netstat -ano | grep -E ":%1 " | grep "LISTENING" | awk "{print $NF}" | head -n 1 > %single_pid_tmp%.txt + set /p pid=<%single_pid_tmp%.txt + call :LOG_INFO pid: %pid% + del %single_pid_tmp%.txt + endlocal & set pid=%pid% +exit /b 0 + +:GET_USED_PID_BY_PORT + setlocal + set "PORT=%~1" + set /a "UNIQUE_COUNT=0" + + if "!PORT!"=="" ( + call :log_error ERROR: Port number required + endlocal & exit /b 0 + ) + + set "TEMP_FILE=%TEMP%\port_%PORT%_%RANDOM%_%TIME::=_%.tmp" + set "PID_LIST_FILE=%TEMP%\pids_%PORT%_%RANDOM%.tmp" + + call :LOG_INFO Checking for processes using port !PORT!... + netstat -ano | grep -E ":!PORT! " > "!TEMP_FILE!" + + if !errorlevel! neq 0 ( + call :LOG_INFO No process found using port !PORT! + del "!TEMP_FILE!" 2>nul + endlocal & exit /b 0 + ) + + echo. > "!PID_LIST_FILE!" + + echo. + call :LOG_GROUP Processes using port !PORT! + for /f "tokens=5" %%a in (!TEMP_FILE!) do ( + set "PID=%%a" + :: grep -E "^^!PID! $" "!PID_LIST_FILE!" >nul + findstr /x "!PID!" "!PID_LIST_FILE!" >nul + if !errorlevel! neq 0 ( + echo !PID! >> "!PID_LIST_FILE!" + set /a "UNIQUE_COUNT+=1" + call :LOG_INFO [!UNIQUE_COUNT!] PID: !PID! + set "PROCESS_NAME=" + set "SESSION_NAME=" + set "SESSION_NUM=" + set "MEM_USAGE=" + + for /f "tokens=1,2,3,4,5*" %%b in ('tasklist /fi "PID eq !PID!" /fo table ^| findstr /i "!PID!"') do ( + if not "%%b"=="=" if not "%%b"=="Image" ( + call :LOG_INFO "PROCESS_NAME=%%b" + call :LOG_INFO "PID_VALUE=%%c" + call :LOG_INFO "SESSION_NAME=%%d" + call :LOG_INFO "SESSION_NUM=%%e" + call :LOG_INFO "MEM_USAGE=%%f" + ) + ) + + if not "!PROCESS_NAME!"=="" ( + call :LOG_INFO Process: !PROCESS_NAME! + call :LOG_INFO Memory: !MEM_USAGE! + ) + echo. + ) + ) + + if !UNIQUE_COUNT! gtr 0 ( + call :LOG_GROUP PIDs using port !PORT! + :: type "!PID_LIST_FILE!" + call :LOG_INFO Total: !UNIQUE_COUNT! process + ) + + del "!TEMP_FILE!" 2>nul + del "!PID_LIST_FILE!" 2>nul + +endlocal & exit /b %UNIQUE_COUNT% + + +:LISTEN_PORT_LIST + call :listen_port_list__default_set + set all_ports_up=1 + for %%i in (%PORT_LIST%) do ( + for /f "tokens=1,2 delims=:" %%a in ("%%i") do ( + set port=%%a + set service=%%b + netstat -an | find ":%%a" > nul + if !errorlevel! neq 0 ( + set all_ports_up=0 + call :LOG_DEBUG "!service! Server (Port: %%a) isn't ready..." + ) + ) + ) + + if !all_ports_up! equ 1 ( + call :DISPLAY_SUCCESS && exit /b 0 + ) + + set /a COUNTER+=CHECK_INTERVAL + if !COUNTER! geq %MAX_WAIT% ( + call :LOG_ERROR "Services did not start within !COUNTER! seconds" && exit /b 1 + ) + + call :LOG_DEBUG "Waiting for services... (!COUNTER!/%MAX_WAIT% seconds)" + echo. + call :WAIT %CHECK_INTERVAL% + goto LISTEN_PORT_LIST + :listen_port_list__default_set + set CHECK_INTERVAL=5 + set MAX_WAIT=90 + set COUNTER=0 +exit /b 0 + +:DISPLAY_SUCCESS + call :WAIT 3 + echo. + echo ======================================== + echo %SERVER_HUMP% Services Successfully Started + echo ======================================== + for %%i in (%PORT_LIST%) do ( + for /f "tokens=1,2 delims=:" %%a in ("%%i") do ( + echo - %%b Server - Port %%a is listening + ) + ) + echo ======================================== + echo. +exit /b 0 + +:AUTOSTART + setlocal + set DESCRIPTION=%~1 + set SCRIPT_PATH=%~2 + set SCRIPT_FILE=%~3 + + set PARAMS= + shift /3 + :autostart__params_loop + if not "%~1"=="" ( + set "PARAMS=%PARAMS% %~3" + shift + goto autostart__params_loop + ) + + call :GET_STRING_LENGTH "%SCRIPT_PATH%" "len" + set /a "last_pos=len-1" + for /f "delims=" %%c in ("!SCRIPT_PATH:~%last_pos%,1!") do ( + if not "%%c"=="\" ( + set "SCRIPT_PATH=%SCRIPT_PATH%\" + ) + ) + + call :START "%DESCRIPTION%" "%SCRIPT_PATH%" "%SCRIPT_FILE%" %PARAMS% +exit /b 0 + +:START + setlocal + set DESCRIPTION=%~1 + set SCRIPT_PATH=%~2 + set SCRIPT_FILE=%~3 + + set PARAMS= + shift /3 + :start__params_loop + if not "%~1"=="" ( + set "PARAMS=%PARAMS% %~3" + shift + goto start__params_loop + ) + + call :LOG_DEBUG "%SCRIPT_PATH%%SCRIPT_FILE%" + if exist "%SCRIPT_PATH%%SCRIPT_FILE%" ( + echo. + call :LOG_INFO "Starting %DESCRIPTION%..." + cd /d "%SCRIPT_PATH%" + start "%DESCRIPTION% Startup" /B cmd /c "%SCRIPT_FILE% %PARAMS%" + if errorlevel 1 ( + call :log_error %DESCRIPTION% start Failed. + pause && exit /b 1 + ) + ) else ( + call :log_error Startup file not found: %SCRIPT_PATH%\%SCRIPT_FILE% + pause && exit /b 1 + ) + endlocal +exit /b 0 + +:Exit + set "exit_code=%~1" + set "error_msg=%~2" + if "!exit_code!"=="" set "exit_code=0" + if not "!error_msg!"=="" ( + call :LOG_ERROR An error occurred^(!exit_code!^): !error_msg! + ) + call :Cleanup +exit /b !exit_code! + +:Cleanup + call :LOG_INFO Cleaning up resources ... +goto :EOF + + + +:: ===================================================================== 主逻辑起点 +:: 主函数. +:main +setlocal +call :LOG_INFO "脚本开始执行" + +call :LOG_GROUP 检查权限 +call :CHECK_IS_ADMIN +if %errorLevel% neq 0 ( + call :Exit "1" "请使用管理员身份运行此脚本!" + pause + exit /b 1 +) + +call :LOG_GROUP 系统库初始化 +cd "%SCRIPT_PATH%" +call :LOG_INFO SCRIPT_PATH %SCRIPT_PATH% +call :AUTOSTART MySQL初始化 %SCRIPT_PATH% initialize-and-start-mysql.bat +call :WAIT 10 + +call :LOG_GROUP 基本库初始化 +call :AUTOSTART "MySQL默认库初始化" "%SCRIPT_PATH%" "import-data-into-mysql.bat" + +endlocal +goto :Exit 0