yueqian-ai
143 lines
3.9 KiB
Bash
143 lines
3.9 KiB
Bash
#!/bin/sh
|
||
set -e
|
||
|
||
DM_PATH=/home/dmdba/dmdbms
|
||
DM_DATA_DIR=/home/dmdba/data/DAMENG
|
||
|
||
INIT_SQL=/home/dmdba/initdata/init-qmodel.sql
|
||
FIRST_RUN_FLAG=/var/run/qmodel_init.done
|
||
PORT=${PORT_NUM:-5236}
|
||
|
||
# ---------- 检查是否已经初始化 ----------
|
||
check_is_init() {
|
||
if [ -d "${DM_DATA_DIR}" ]; then
|
||
DATABASE_ALREADY_EXISTS=true
|
||
fi
|
||
}
|
||
|
||
# ---------- 初始化数据库 ----------
|
||
db_init() {
|
||
echo "[dm8] init db ..."
|
||
mkdir -p "${DM_DATA_DIR}"
|
||
chown -R dmdba "${DM_DATA_DIR}"
|
||
cd "${DM_PATH}/bin"
|
||
|
||
# 初始化参数，参考你原来的需求
|
||
INIT_ARGS="PATH=/home/dmdba/data DB_NAME=DAMENG PORT_NUM=${PORT} PAGE_SIZE=16 CHARSET=1 LENGTH_IN_CHAR=1 CASE_SENSITIVE=${CASE_SENSITIVE:-0}"
|
||
|
||
if [ -n "${SYSDBA_PWD}" ]; then
|
||
INIT_ARGS="${INIT_ARGS} SYSDBA_PWD=${SYSDBA_PWD}"
|
||
fi
|
||
if [ -n "${SYSAUDITOR_PWD}" ]; then
|
||
INIT_ARGS="${INIT_ARGS} SYSAUDITOR_PWD=${SYSAUDITOR_PWD}"
|
||
fi
|
||
|
||
echo "[dm8] dminit ${INIT_ARGS}"
|
||
gosu dmdba "${DM_PATH}/bin/dminit" ${INIT_ARGS}
|
||
echo "[dm8] db init done."
|
||
}
|
||
|
||
# ---------- 等待 dmserver TCP 端口可用（参考老版本逻辑） ----------
|
||
wait_dm_ready() {
|
||
echo "[dm8] wait dmserver tcp on 127.0.0.1:${PORT} ..."
|
||
|
||
i=0
|
||
max=6000 # 最多等 6000 秒
|
||
|
||
while [ $i -lt $max ]; do
|
||
# 使用 /dev/tcp 检测端口是否可连
|
||
if (echo >"/dev/tcp/127.0.0.1/${PORT}") >/dev/null 2>&1; then
|
||
echo "[dm8] dmserver tcp ready."
|
||
return 0
|
||
fi
|
||
i=`expr $i + 1`
|
||
sleep 1
|
||
done
|
||
|
||
echo "[dm8] WARN: dmserver tcp not ready after ${max}s, continue anyway."
|
||
return 1
|
||
}
|
||
|
||
# ---------- 首次启动时导入 init-qmodel.sql ----------
|
||
run_init_sql_once() {
|
||
# 已经执行过了就直接跳过
|
||
if [ -f "${FIRST_RUN_FLAG}" ]; then
|
||
echo "[dm8] qModel init already done, skip."
|
||
return 0
|
||
fi
|
||
|
||
# 没有 SQL 文件就不导入，避免每次都尝试
|
||
if [ ! -f "${INIT_SQL}" ]; then
|
||
echo "[dm8] ${INIT_SQL} not found, skip qModel init."
|
||
mkdir -p "$(dirname "${FIRST_RUN_FLAG}")"
|
||
touch "${FIRST_RUN_FLAG}"
|
||
return 0
|
||
fi
|
||
|
||
# 必要环境变量没配也跳过
|
||
if [ -z "${QMODEL_USER}" ] || [ -z "${QMODEL_PWD}" ] || [ -z "${SYSDBA_PWD}" ]; then
|
||
echo "[dm8] QMODEL_USER / QMODEL_PWD / SYSDBA_PWD not set, skip qModel init."
|
||
mkdir -p "$(dirname "${FIRST_RUN_FLAG}")"
|
||
touch "${FIRST_RUN_FLAG}"
|
||
return 0
|
||
fi
|
||
|
||
# 仅用 TCP 端口检测等待服务就绪
|
||
wait_dm_ready
|
||
|
||
echo "[dm8] start create business user and import init-qmodel.sql ..."
|
||
|
||
# 为了防止“用户已存在”等导致整个容器退出，这里临时关掉 set -e
|
||
set +e
|
||
|
||
# 1) SYSDBA 创建业务用户并授权
|
||
gosu dmdba "${DM_PATH}/bin/disql" "SYSDBA/${SYSDBA_PWD}@127.0.0.1:${PORT}" <<EOF
|
||
create user "${QMODEL_USER}" identified by "${QMODEL_PWD}" hash with SHA512 salt;
|
||
grant "PUBLIC","SOI","DBA" to "${QMODEL_USER}";
|
||
commit;
|
||
EOF
|
||
|
||
# 2) 业务用户执行初始化脚本
|
||
gosu dmdba "${DM_PATH}/bin/disql" "${QMODEL_USER}/${QMODEL_PWD}@127.0.0.1:${PORT}" <<EOF
|
||
set define off;
|
||
set CHAR_CODE UTF8;
|
||
\`${INIT_SQL}
|
||
set define on;
|
||
EOF
|
||
|
||
# 恢复 set -e
|
||
set -e
|
||
|
||
mkdir -p "$(dirname "${FIRST_RUN_FLAG}")"
|
||
touch "${FIRST_RUN_FLAG}"
|
||
echo "[dm8] qModel init finished."
|
||
}
|
||
|
||
# ---------- 主流程 ----------
|
||
|
||
check_is_init
|
||
if [ -z "${DATABASE_ALREADY_EXISTS}" ]; then
|
||
db_init
|
||
else
|
||
echo "[dm8] db already exists, skip dminit."
|
||
fi
|
||
|
||
# 安装服务（修复你原脚本里的那一处 `DmAPService}` 的 bug）
|
||
if [ ! -f "${DM_PATH}/bin/DmAPService" ]; then
|
||
"${DM_PATH}/script/root/dm_service_installer.sh" -s "${DM_PATH}/bin/DmAPService"
|
||
fi
|
||
|
||
if [ ! -f "${DM_PATH}/bin/DmServiceDMSERVER" ]; then
|
||
"${DM_PATH}/script/root/dm_service_installer.sh" -t dmserver -p "DMSERVER" -dm_ini "${DM_DATA_DIR}/dm.ini"
|
||
fi
|
||
|
||
# 启动服务
|
||
gosu dmdba "${DM_PATH}/bin/DmAPService" start
|
||
gosu dmdba "${DM_PATH}/bin/DmServiceDMSERVER" start
|
||
|
||
# 首次导入任务放到后台，不阻塞主进程
|
||
run_init_sql_once &
|
||
|
||
# 保持容器前台进程：沿用原脚本 tail 日志
|
||
exec gosu dmdba tail -f /home/dmdba/dmdbms/log/DmServiceDMSERVER.log
|