init: 导入团队知识库内容
This commit is contained in:
yueqian-ai
@@ -0,0 +1,157 @@
|
||||
#!/bin/bash
|
||||
set -e
|
||||
|
||||
DB_PATH=${DB_PATH:-"/home/dmdba/data"}
|
||||
INSTANCE_NAME=${INSTANCE_NAME:-"DMSERVER"}
|
||||
DB_NAME=${DB_NAME:-"DAMENG"}
|
||||
DMDB_INSTALL_PATH=${DMDB_INSTALL_PATH:-"/home/dmdba/dmdb"}
|
||||
INIT_PARAMS=""
|
||||
PORT_NUM=${PORT_NUM:-"5236"}
|
||||
TIME_ZONE=${TIME_ZONE:-"+08:00"}
|
||||
BUFFER=${BUFFER:-"8000"}
|
||||
PAGE_CHECK=${PAGE_CHECK:-"3"}
|
||||
PAGE_SIZE=${PAGE_SIZE:-"8"}
|
||||
LOG_SIZE=${LOG_SIZE:-"4096"}
|
||||
EXTENT_SIZE=${EXTENT_SIZE:-"16"}
|
||||
CHARSET=${CHARSET:-"0"}
|
||||
USE_DB_NAME=${USE_DB_NAME:-"1"}
|
||||
AUTO_OVERWRITE=${AUTO_OVERWRITE:-"0"}
|
||||
BLANK_PAD_MODE=${BLANK_PAD_MODE:-"0"}
|
||||
DPC_MODE=${DPC_MODE:-"0"}
|
||||
CASE_SENSITIVE=${CASE_SENSITIVE:-"y"}
|
||||
|
||||
OTHER_PARAMS=${OTHER_PARAMS:-""}
|
||||
|
||||
SYSDBA_PWD=${SYSDBA_PWD:-""}
|
||||
SYSAUDITOR_PWD=${SYSAUDITOR_PWD:-""}
|
||||
|
||||
# ===== 新增：一次性执行标记 & 日志路径 =====
|
||||
FIRST_RUN_FLAG="/var/run/dm8_first_run.done" # 首次执行标记
|
||||
INIT_LOG="/init.log" # 初始化日志文件
|
||||
|
||||
function init_db() {
|
||||
if [ -z "$SYSDBA_PWD" ]; then
|
||||
echo "SYSDBA_PWD is empty, please set it in environment variables"
|
||||
exit 1
|
||||
fi
|
||||
if [ -z "$SYSAUDITOR_PWD" ]; then
|
||||
echo "SYSAUDITOR_PWD is empty, please set it in environment variables"
|
||||
exit 1
|
||||
fi
|
||||
# 判断DB_PATH文件夹内是否存在文件
|
||||
if [ -d "$DB_PATH" ]; then
|
||||
if [ "$(ls -A $DB_PATH)" ]; then
|
||||
echo "DB_PATH is not empty, please check it"
|
||||
exit 1
|
||||
fi
|
||||
else
|
||||
echo "DB_PATH is not exist, create it"
|
||||
mkdir -p $DB_PATH
|
||||
chown -R dmdba $DB_PATH
|
||||
fi
|
||||
INIT_PARAMS="$INIT_PARAMS PATH=$DB_PATH SYSDBA_PWD=$SYSDBA_PWD SYSAUDITOR_PWD=$SYSAUDITOR_PWD INSTANCE_NAME=$INSTANCE_NAME"
|
||||
INIT_PARAMS="$INIT_PARAMS PORT_NUM=$PORT_NUM DB_NAME=$DB_NAME TIME_ZONE=$TIME_ZONE BUFFER=$BUFFER PAGE_CHECK=$PAGE_CHECK PAGE_SIZE=$PAGE_SIZE"
|
||||
INIT_PARAMS="$INIT_PARAMS LOG_SIZE=$LOG_SIZE EXTENT_SIZE=$EXTENT_SIZE CHARSET=$CHARSET USE_DB_NAME=$USE_DB_NAME"
|
||||
INIT_PARAMS="$INIT_PARAMS AUTO_OVERWRITE=$AUTO_OVERWRITE BLANK_PAD_MODE=$BLANK_PAD_MODE DPC_MODE=$DPC_MODE CASE_SENSITIVE=$CASE_SENSITIVE"
|
||||
INIT_PARAMS="$INIT_PARAMS $OTHER_PARAMS"
|
||||
echo "Initializing database..."
|
||||
echo "Initializing database with parameters:"
|
||||
echo $INIT_PARAMS
|
||||
sudo -u dmdba /home/dmdba/dmdb/bin/dminit $INIT_PARAMS
|
||||
echo "Database initialized"
|
||||
}
|
||||
function start_dmap() {
|
||||
echo "Starting DmAPService..."
|
||||
sudo -u dmdba /home/dmdba/dmdb/bin/dmap dmap_ini=/home/dmdba/dmdb/bin/dmap.ini &
|
||||
echo "DmAPService started"
|
||||
}
|
||||
|
||||
# 创建一个函数，用来修改文件的权限
|
||||
function modify_db_permissions() {
|
||||
echo "Modifying $DB_PATH permissions..."
|
||||
chown -R dmdba $DB_PATH
|
||||
echo "$DB_PATH permissions modified"
|
||||
}
|
||||
|
||||
function check_initialized() {
|
||||
# 判断 $DB_PATH/$DB_NAME/dm.ini 是否存在
|
||||
if [ -f "$DB_PATH/$DB_NAME/dm.ini" ]; then
|
||||
echo "Database already initialized"
|
||||
modify_db_permissions
|
||||
else
|
||||
echo "Database not initialized"
|
||||
init_db
|
||||
fi
|
||||
}
|
||||
|
||||
# ===== 新增：等待 TCP 就绪的小函数 =====
|
||||
wait_tcp_ready() {
|
||||
local waited=0
|
||||
local timeout=600
|
||||
echo "Waiting for dmserver TCP on 127.0.0.1:${PORT_NUM} ..."
|
||||
# /dev/tcp 存在即可用；若镜像不支持，可改用 nc -z
|
||||
while ! (echo >"/dev/tcp/127.0.0.1/${PORT_NUM}") >/dev/null 2>&1; do
|
||||
sleep 2
|
||||
waited=$((waited+2))
|
||||
if [ $waited -ge $timeout ]; then
|
||||
echo "ERROR: TCP 127.0.0.1:${PORT_NUM} not ready in ${timeout}s"
|
||||
return 1
|
||||
fi
|
||||
done
|
||||
echo "TCP ready."
|
||||
return 0
|
||||
}
|
||||
|
||||
# ===== 新增：首次执行的后台任务（按你的两段 disql 顺序）=====
|
||||
post_boot_first_run() {
|
||||
# 已执行过则直接退出
|
||||
if [ -f "$FIRST_RUN_FLAG" ]; then
|
||||
return 0
|
||||
fi
|
||||
|
||||
# 等 TCP 就绪（服务启动完成）
|
||||
if ! wait_tcp_ready; then
|
||||
echo "WARN: dmserver TCP not ready; skip first-run init."
|
||||
return 0
|
||||
fi
|
||||
|
||||
# 准备日志
|
||||
if [ ! -f "$INIT_LOG" ]; then
|
||||
echo "init database at $(date +'%Y-%m-%d %H:%M:%S')" > "$INIT_LOG"
|
||||
fi
|
||||
|
||||
echo "Creating business user & grant ..." | tee -a "$INIT_LOG"
|
||||
# 第一步：SYSDBA 执行建用户与授权
|
||||
/home/dmdba/dmdb/bin/disql SYSDBA/${SYSDBA_PWD}@localhost:${PORT_NUM} <<EOF >> "$INIT_LOG" 2>&1
|
||||
create user "${QMODEL_USER}" identified by "${QMODEL_PWD}" hash with SHA512 salt;
|
||||
grant "PUBLIC","SOI","DBA" to "${QMODEL_USER}";
|
||||
COMMIT;
|
||||
EOF
|
||||
|
||||
echo "Importing initial data ..." | tee -a "$INIT_LOG"
|
||||
# 第二步：业务用户导入数据（确保 /home/dmdba/initdata/init-qmodel.sql 可读）
|
||||
/home/dmdba/dmdb/bin/disql ${QMODEL_USER}/${QMODEL_PWD}@localhost:${PORT_NUM} <<EOF >> "$INIT_LOG" 2>&1
|
||||
set define off;
|
||||
set CHAR_CODE UTF8;
|
||||
\`/home/dmdba/initdata/init-qmodel.sql
|
||||
set define on;
|
||||
EOF
|
||||
|
||||
# 落标记：仅首次执行
|
||||
touch "$FIRST_RUN_FLAG"
|
||||
echo "First-run initialization finished." | tee -a "$INIT_LOG"
|
||||
}
|
||||
|
||||
cd $DMDB_INSTALL_PATH/bin
|
||||
# 检查DB是否初始化，如果没有初始化则执行初始化
|
||||
check_initialized
|
||||
|
||||
# 启动 DmAPServer
|
||||
start_dmap
|
||||
|
||||
# ===== 新增：后台启动首次执行任务（不阻塞主进程）=====
|
||||
post_boot_first_run &
|
||||
|
||||
# 启动数据库实例（前台）
|
||||
echo "Starting DmServer..."
|
||||
exec sudo -u dmdba /home/dmdba/dmdb/bin/dmserver path=$DB_PATH/$DB_NAME/dm.ini
|
||||
Reference in New Issue
Block a user