自动初始化库
This commit is contained in:
yuejiajun
parent
871162f265
commit
528f28c5b0
453
base/Resources/mysql/initialize.bat
Normal file
453
base/Resources/mysql/initialize.bat
Normal file
@ -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
|
||||||
Loading…
x
Reference in New Issue
Block a user