211 lines
8.9 KiB
Bash
211 lines
8.9 KiB
Bash
#!/bin/bash
|
||
|
||
# 检查环境变量配置文件
|
||
if [ ! -f "./.env" ]; then
|
||
echo "❌ 配置文件 .env 不存在，请先创建配置文件"
|
||
echo "💡 可以复制 .env.example 为 .env 并修改配置"
|
||
exit 1
|
||
fi
|
||
|
||
echo "📁 使用 .env 配置文件，Docker Compose 将自动加载环境变量"
|
||
|
||
# 规范化 .env 的换行符为 LF，避免 CRLF 导致的 $'\r' 报错
|
||
if command -v dos2unix >/dev/null 2>&1; then
|
||
dos2unix -q ./.env || true
|
||
else
|
||
sed -i 's/\r$//' ./.env 2>/dev/null || perl -pi -e 's/\r$//' ./.env 2>/dev/null || true
|
||
fi
|
||
|
||
# 将 .env 中的变量导入当前 Shell，供脚本内使用
|
||
set -a
|
||
. ./.env
|
||
set +a
|
||
|
||
# 校验关键环境变量是否已加载
|
||
missing_vars=()
|
||
[ -z "${PULSAR_ADMIN_USER}" ] && missing_vars+=(PULSAR_ADMIN_USER)
|
||
[ -z "${PULSAR_ADMIN_PASSWORD}" ] && missing_vars+=(PULSAR_ADMIN_PASSWORD)
|
||
[ -z "${SERVER_IP}" ] && missing_vars+=(SERVER_IP)
|
||
[ -z "${PULSAR_BROKER_PORT}" ] && missing_vars+=(PULSAR_BROKER_PORT)
|
||
[ -z "${PULSAR_ADMIN_PORT}" ] && missing_vars+=(PULSAR_ADMIN_PORT)
|
||
if [ ${#missing_vars[@]} -ne 0 ]; then
|
||
echo "❌ 以下环境变量未设置: ${missing_vars[*]}"
|
||
echo "💡 请在 .env 中补充这些变量后重试"
|
||
exit 1
|
||
fi
|
||
|
||
echo "=== Docker 镜像仓库登录 ==="
|
||
|
||
docker login --username=yunjichaobiao --password hub.jishe.com registry.cn-qingdao.aliyuncs.com
|
||
|
||
echo "=== 采集端一键启动脚本 ==="
|
||
|
||
# 检查htpasswd是否安装
|
||
if ! command -v htpasswd &> /dev/null; then
|
||
echo "❌ htpasswd 未安装，正在安装..."
|
||
if command -v apt &> /dev/null; then
|
||
sudo apt update && sudo apt install -y apache2-utils
|
||
elif command -v yum &> /dev/null; then
|
||
sudo yum install -y httpd-tools
|
||
else
|
||
echo "❌ 无法自动安装htpasswd，请手动安装后重试"
|
||
exit 1
|
||
fi
|
||
fi
|
||
|
||
# 创建认证配置
|
||
echo "🔐 创建Pulsar基本认证配置..."
|
||
mkdir -p ./pulsar/auth
|
||
htpasswd -cmb ./pulsar/auth/.htpasswd ${PULSAR_ADMIN_USER} ${PULSAR_ADMIN_PASSWORD}
|
||
|
||
echo "✅ 认证配置完成 - 用户名: ${PULSAR_ADMIN_USER}, 密码: ${PULSAR_ADMIN_PASSWORD}"
|
||
|
||
# 创建数据目录
|
||
echo "📁 创建数据目录..."
|
||
sudo mkdir -p /mnt/dockerdata/appservice/{pulsar,redis,mysql,iotdb,adminapi}
|
||
sudo mkdir -p /mnt/dockerdata/appservice/pulsar/{zookeeper-data,bookkeeper-data,pulsar-data}
|
||
sudo mkdir -p /mnt/dockerdata/appservice/{redis/{outdata,conf},mysql/{conf,log,data},iotdb/{conf,data,logs},adminapi/{conf,logs}}
|
||
sudo chown -R 10000:0 /mnt/dockerdata/appservice/pulsar/
|
||
|
||
# 创建本地目录的软链接（如果不存在）
|
||
echo "🔗 创建本地目录链接..."
|
||
mkdir -p ./pulsar/{zookeeper-data,bookkeeper-data,pulsar-data,auth}
|
||
mkdir -p ./redis/{outdata,conf}
|
||
mkdir -p ./mysql/{conf,log,data}
|
||
mkdir -p ./iotdb/{conf,data,logs}
|
||
mkdir -p ./adminapi/{conf,logs}
|
||
|
||
# 修复MySQL目录权限
|
||
echo "🔧 修复MySQL目录权限..."
|
||
sudo chown -R 1000:1000 ./mysql/data
|
||
sudo chown -R 1000:1000 ./mysql/log
|
||
sudo chmod -R 755 ./mysql/data
|
||
sudo chmod -R 755 ./mysql/log
|
||
|
||
sudo chmod -R 755 ./adminapi/logs
|
||
|
||
# 第一步：启动IoTDB独立服务获取配置
|
||
echo "🚀 第一步：启动IoTDB独立服务获取配置..."
|
||
docker compose -f docker-compose-iotdb-standalone.yml up -d
|
||
|
||
# 等待IoTDB启动完成
|
||
echo "⏳ 等待IoTDB服务启动完成..."
|
||
sleep 10
|
||
|
||
# 第二步：拷贝配置目录
|
||
echo "📁 第二步：拷贝IoTDB配置目录..."
|
||
docker cp iotdb-standalone:/iotdb/conf ./iotdb
|
||
echo "✅ IoTDB配置目录拷贝完成"
|
||
|
||
# 第三步：修改时间戳精度配置
|
||
echo "⚙️ 第三步：修改IoTDB时间戳精度配置..."
|
||
if [ -f "./iotdb/conf/iotdb-system.properties" ]; then
|
||
# 检查是否已经包含配置
|
||
if ! grep -q "timestamp_precision=ns" ./iotdb/conf/iotdb-system.properties; then
|
||
# 添加换行和说明
|
||
echo "" >> ./iotdb/conf/iotdb-system.properties
|
||
echo "# ms 毫秒 us 微秒 ns 纳秒" >> ./iotdb/conf/iotdb-system.properties
|
||
echo "timestamp_precision=ns" >> ./iotdb/conf/iotdb-system.properties
|
||
echo "✅ 时间戳精度配置已添加：timestamp_precision=ns"
|
||
else
|
||
echo "ℹ️ 时间戳精度配置已存在"
|
||
fi
|
||
else
|
||
echo "❌ 配置文件不存在，请检查配置目录"
|
||
fi
|
||
|
||
# 第四步：停止IoTDB独立服务
|
||
echo "🛑 第四步：停止IoTDB独立服务..."
|
||
docker compose -f docker-compose-iotdb-standalone.yml down
|
||
|
||
# 第五步：清理IoTDB数据目录，保留配置目录
|
||
echo "🧹 第五步：清理IoTDB数据目录，保留配置目录..."
|
||
sudo rm -rf ./iotdb/data/*
|
||
sudo rm -rf ./iotdb/logs/*
|
||
echo "✅ IoTDB数据目录已清理，配置目录已保留"
|
||
|
||
# MySQL配置检查
|
||
echo "⚙️ 检查MySQL配置..."
|
||
if [ ! -d "./mysql/init" ]; then
|
||
echo "❌ MySQL初始化脚本目录不存在，请检查"
|
||
exit 1
|
||
fi
|
||
|
||
if [ ! -f "./mysql/init/init-db.sql" ]; then
|
||
echo "❌ MySQL初始化脚本不存在，请检查"
|
||
exit 1
|
||
fi
|
||
|
||
echo "✅ MySQL配置检查通过"
|
||
|
||
sleep 10
|
||
|
||
# 启动主服务
|
||
echo "第六步：🚀 启动 部署主脚本 ..."
|
||
|
||
docker compose -f docker-compose-init.yml up -d
|
||
|
||
# 等待IoTDB启动完成
|
||
echo "⏳ 等待IoTDB服务启动完成..."
|
||
sleep 10
|
||
|
||
# 设置IoTDB密码
|
||
echo "第七步：🔐 设置 IoTDB"
|
||
echo "ALTER USER root SET PASSWORD '${IOTDB_ROOT_PASSWORD}';"
|
||
docker exec -i iotdb-standalone /bin/bash << EOF
|
||
# 等待IoTDB完全启动
|
||
echo "等待IoTDB服务就绪..."
|
||
sleep 10
|
||
|
||
echo "IoTDB服务已就绪，开始创建表模型数据库和设置root账号密码..."
|
||
|
||
# 使用非交互式方式创建表模型数据库
|
||
echo "CREATE DATABASE jisheiotdata;USE jisheiotdata;CREATE TABLE IF NOT EXISTS CTWingAepReceiveMessageEntity( time TIMESTAMP TIME, IoTDataType STRING TAG, DeviceType STRING TAG, DeviceAddress STRING TAG, PlatformTenantId STRING FIELD, ProductId STRING FIELD, ServiceId STRING FIELD, IoTPlatformDeviceOpenInfo STRING FIELD, MessageType STRING FIELD, Protocol STRING FIELD, RawMessage STRING FIELD, ReceivedPayload STRING FIELD, ReceivedTimestamps INT64 FIELD, IMSI STRING FIELD, IMEI STRING FIELD ) COMMENT 'CTWingAepReceiveMessageEntity';CREATE TABLE IF NOT EXISTS DeviceTelemetryPacketTaskInfo( time TIMESTAMP TIME, IoTDataType STRING TAG, DeviceType STRING TAG, DeviceAddress STRING TAG, IoTPlatformProductId STRING FIELD, IoTPlatformDeviceOpenInfo STRING FIELD, IoTPlatformProductName STRING FIELD, IoTPlatformAccountId STRING FIELD, AccountPhoneNumber STRING FIELD, MessageType STRING FIELD, IssueRawMessage STRING FIELD, IssuePayload STRING FIELD, ResponseRawMessage STRING FIELD, ResponsePayload STRING FIELD, TelemetrySource INT32 FIELD, IoTPlatform INT32 FIELD ) COMMENT 'DeviceTelemetryPacketTaskInfo';CREATE TABLE IF NOT EXISTS OneNETReceiveMessageEntity( time TIMESTAMP TIME, IoTDataType STRING TAG, DeviceType STRING TAG, DeviceAddress STRING TAG, ProductId STRING FIELD, IoTPlatformDeviceOpenInfo STRING FIELD, MessageType STRING FIELD, Protocol STRING FIELD, RawMessage STRING FIELD, IsEncrypted BOOLEAN FIELD, PlaintextMessage STRING FIELD, ReceivedPayload STRING FIELD, ReceivedTimestamps INT64 FIELD ) COMMENT 'OneNETReceiveMessageEntity';" | ./start-cli.sh -h iotdb-standalone -sql_dialect table
|
||
|
||
# 使用非交互式方式设置密码
|
||
echo "ALTER USER root SET PASSWORD '${IOTDB_ROOT_PASSWORD}';" | ./start-cli.sh -h iotdb-standalone
|
||
|
||
echo "IoTDB密码设置完成"
|
||
EOF
|
||
|
||
echo "✅ IoTDB密码设置完成 - 用户名: root, 密码: ${IOTDB_ROOT_PASSWORD}"
|
||
|
||
echo ""
|
||
echo "🎉 Pulsar 基本认证集群启动完成！"
|
||
echo ""
|
||
echo "📋 连接信息："
|
||
echo " Pulsar Broker: ${SERVER_IP}:$PULSAR_BROKER_PORT}"
|
||
echo " Pulsar Admin: http://${SERVER_IP}:$PULSAR_ADMIN_PORT}"
|
||
echo " IoTDB: ${SERVER_IP}:${IOTDB_PORT} (root/${IOTDB_ROOT_PASSWORD})"
|
||
echo " Redis: ${SERVER_IP}:6379"
|
||
echo " MySQL: ${SERVER_IP}:$MYSQL_PORT}"
|
||
echo " Admin API: http://${SERVER_IP}:$ADMIN_API_PORT}"
|
||
echo " Admin UI: http://${SERVER_IP}:$ADMIN_WEB_PORT}"
|
||
echo " 用户名: ${PULSAR_ADMIN_USER}"
|
||
echo " 密码: ${PULSAR_ADMIN_PASSWORD}"
|
||
echo ""
|
||
echo "🔧 常用命令："
|
||
echo " 查看状态: docker compose ps"
|
||
echo " 查看日志: docker compose logs -f"
|
||
echo " 停止服务: docker compose down"
|
||
echo ""
|
||
echo "🧪 测试连接："
|
||
echo " Pulsar: curl -u $PULSAR_ADMIN_USER}:$PULSAR_ADMIN_PASSWORD} http://$SERVER_IP}:$PULSAR_ADMIN_PORT}/admin/v2/clusters"
|
||
echo " Redis: redis-cli -h ${SERVER_IP} -p 6379 ping"
|
||
echo " MySQL: mysql -h ${SERVER_IP} -P ${MYSQL_PORT} -uroot -p${MYSQL_ROOT_PASSWORD} -e 'SELECT 1'"
|
||
echo " IoTDB: docker exec -it iotdb-standalone ./start-cli.sh -h iotdb-standalone -u root -p Lixiao@1980"
|
||
echo ""
|
||
echo "🎯 等待服务启动完成..."
|
||
echo "⏳ 请等待3分钟让所有服务完全初始化成功"
|
||
sleep 180
|
||
echo "🎯 服务初始化完成..."
|
||
echo "⏳ 然后等待1分钟让所有服务完全停止 docker compose -f docker-compose-init.yml down"
|
||
|
||
docker compose -f docker-compose-init.yml down
|
||
|
||
sleep 60
|
||
echo "🎯 初始化服务已经完成停止......"
|
||
echo "⏳ 最后再通过 启动最终生产环境，移除初始化依赖，等待服务最终启动完成。"
|
||
|
||
docker compose up -d
|