数据库初始化脚本

数据库初始化脚本，可以作为一个学习范例。

@echo off

set choice=
set username=
set pwd=
set dbname=
set schema=
set logfile=
set /p choice=请选择您要运行的数据库脚本：创建自定义函数(U),初始化数据(I),所有的脚本(A)
set choice=%choice:~0,1%

:getdb
echo 请输入数据库名称
set /p dbname=数据库名称：
set dbname=%dbname:~0,50%
if not "%dbname%"=="~0,50" goto getusername
echo 数据库名称不能为空
goto getdb

:getusername
echo 请输入数据库用户名(确认该用户拥有执行脚本的所需权限)
set /p username=用户名：
set username=%username:~0,50%
if not "%username%"=="~0,50" goto getpassword
echo 用户名不能为空
goto getusername

:getpassword
echo 请输入数据库用户密码
set /p pwd=密码：
set pwd=%pwd:~0,50%
if not "%pwd%"=="~0,50" goto getschema
echo 用户密码不能为空
goto getpassword

:getschema
echo 请输入数据库 SCHEMA
set /p schema=密码：
set schema=%schema:~0,50%
if not "%schema%"=="~0,50" goto getlogfile
echo SCHEMA不能为空
goto getschema

:getlogfile
echo 请输入脚本执行日志文件名
set /p logfile=日志文件：
set logfile=%logfile:~0,50%
if not "%logfile%"=="~0,50" goto docheck
echo 日志文件名不能为空
goto getlogfile

:docheck
echo 你输入的数据库相关信息为:
echo 数据库:%dbname%
echo 用 户 名:%username%
echo 密    码:%pwd%
echo SCHEMA :%schema%
echo 日志文件:%logfile%
set check=
set /p check=请确认信息是否正确:正确(Y),有错误(N)
set check=%check:~0,1%
if /i "%check%"=="Y" goto connectdb
goto dobad

:dobad
set action=
set /p action=信息错误:重新输入(M),放弃本次执行(F)
set action=%action:~0,1%
if /i "%action%"=="M" goto getdb
goto end

:dochoice
if /i "%choice%"=="U" goto createudf
if /i "%choice%"=="I" goto initdata
if /i "%choice%"=="A" goto doall
goto end

:connectdb
echo 开始执行脚本.......
db2 connect to %dbname% user %username% using %pwd%
db2 set current schema %schema%
goto dochoice

:createudf
call createudf.bat
goto disconnect

:initdata
call initdata.bat
goto disconnect

:doall
call initdata.bat
goto disconnect

:disconnect
db2 connect reset
goto end

:end
echo 脚本执行完毕
pause

#脚本