优化linux建表,初始化整体业务
This commit is contained in:
137
init_mysql.sh
Normal file
137
init_mysql.sh
Normal file
@@ -0,0 +1,137 @@
|
||||
#!/bin/bash
|
||||
set -e
|
||||
|
||||
# ============================================================
|
||||
# Weibo-HotSign MySQL 初始化脚本(需要 root 权限)
|
||||
# 用途: 建库、建表、创建业务用户并授权
|
||||
# 使用方式: chmod +x init_mysql.sh && ./init_mysql.sh
|
||||
# 只需运行一次,后续用 setup_linux.sh 做业务配置
|
||||
# ============================================================
|
||||
|
||||
# ===================== 请在此处填写配置 =====================
|
||||
|
||||
MYSQL_ROOT_PASSWORD="" # ← 填写 MySQL root 密码
|
||||
|
||||
# 业务用户配置(setup_linux.sh 中要保持一致)
|
||||
BIZ_USER="weibo"
|
||||
BIZ_PASSWORD="123456" # ← 填写业务用户密码
|
||||
BIZ_HOST="%" # 允许连接的来源,% 表示任意
|
||||
|
||||
MYSQL_HOST="127.0.0.1"
|
||||
MYSQL_PORT="3306"
|
||||
DB_NAME="weibo_hotsign"
|
||||
|
||||
# ===================== 配置结束 =====================
|
||||
|
||||
PROJECT_DIR="$(cd "$(dirname "$0")" && pwd)"
|
||||
|
||||
RED='\033[0;31m'
|
||||
GREEN='\033[0;32m'
|
||||
YELLOW='\033[1;33m'
|
||||
NC='\033[0m'
|
||||
|
||||
info() { echo -e "${GREEN}[✓]${NC} $1"; }
|
||||
warn() { echo -e "${YELLOW}[!]${NC} $1"; }
|
||||
error() { echo -e "${RED}[✗]${NC} $1"; exit 1; }
|
||||
|
||||
echo ""
|
||||
echo "========================================"
|
||||
echo " Weibo-HotSign MySQL 初始化(root)"
|
||||
echo "========================================"
|
||||
echo ""
|
||||
|
||||
# 检查 mysql 客户端
|
||||
command -v mysql >/dev/null 2>&1 || error "未找到 mysql 客户端"
|
||||
|
||||
# 构建 root 连接命令
|
||||
ROOT_CMD="mysql -h${MYSQL_HOST} -P${MYSQL_PORT} -uroot"
|
||||
if [ -n "$MYSQL_ROOT_PASSWORD" ]; then
|
||||
ROOT_CMD="$ROOT_CMD -p${MYSQL_ROOT_PASSWORD}"
|
||||
fi
|
||||
|
||||
# 测试 root 连接
|
||||
if $ROOT_CMD -e "SELECT 1" >/dev/null 2>&1; then
|
||||
info "MySQL root 连接成功"
|
||||
else
|
||||
error "MySQL root 连接失败,请检查 root 密码"
|
||||
fi
|
||||
|
||||
# ------ 1. 创建数据库 ------
|
||||
|
||||
echo ""
|
||||
echo "--- 创建数据库 ---"
|
||||
|
||||
$ROOT_CMD -e "CREATE DATABASE IF NOT EXISTS ${DB_NAME} DEFAULT CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci;"
|
||||
info "数据库 ${DB_NAME} 已创建"
|
||||
|
||||
# ------ 2. 创建业务用户并授权 ------
|
||||
|
||||
echo ""
|
||||
echo "--- 创建业务用户 ---"
|
||||
|
||||
# 创建用户(如果已存在则忽略)
|
||||
$ROOT_CMD -e "CREATE USER IF NOT EXISTS '${BIZ_USER}'@'${BIZ_HOST}' IDENTIFIED BY '${BIZ_PASSWORD}';" 2>/dev/null || true
|
||||
# 授权
|
||||
$ROOT_CMD -e "GRANT ALL PRIVILEGES ON ${DB_NAME}.* TO '${BIZ_USER}'@'${BIZ_HOST}';"
|
||||
$ROOT_CMD -e "FLUSH PRIVILEGES;"
|
||||
info "用户 ${BIZ_USER} 已创建并授权 ${DB_NAME} 库"
|
||||
|
||||
# ------ 3. 建表 ------
|
||||
|
||||
echo ""
|
||||
echo "--- 创建表结构 ---"
|
||||
|
||||
$ROOT_CMD ${DB_NAME} < "${PROJECT_DIR}/init-db.sql"
|
||||
info "所有表创建完成"
|
||||
|
||||
# ------ 4. 创建测试用户(可选) ------
|
||||
|
||||
echo ""
|
||||
echo "--- 检查测试用户 ---"
|
||||
|
||||
USER_COUNT=$($ROOT_CMD -N -e "SELECT COUNT(*) FROM ${DB_NAME}.users" 2>/dev/null)
|
||||
if [ "$USER_COUNT" = "0" ]; then
|
||||
# 检查 bcrypt 是否可用
|
||||
if python3 -c "import bcrypt" 2>/dev/null; then
|
||||
HASHED_PW=$(python3 -c "import bcrypt; print(bcrypt.hashpw(b'Admin123!', bcrypt.gensalt(12)).decode())")
|
||||
USER_ID=$(python3 -c "import uuid; print(str(uuid.uuid4()))")
|
||||
$ROOT_CMD ${DB_NAME} -e "
|
||||
INSERT INTO users (id, username, email, hashed_password, is_active)
|
||||
VALUES ('${USER_ID}', 'admin', 'admin@example.com', '${HASHED_PW}', 1);
|
||||
"
|
||||
info "测试用户已创建: admin / Admin123!"
|
||||
else
|
||||
warn "bcrypt 未安装,跳过测试用户创建(运行 setup_linux.sh 安装依赖后可手动创建)"
|
||||
fi
|
||||
else
|
||||
info "已有 ${USER_COUNT} 个用户,跳过"
|
||||
fi
|
||||
|
||||
# ------ 5. 验证 ------
|
||||
|
||||
echo ""
|
||||
echo "--- 验证业务用户权限 ---"
|
||||
|
||||
BIZ_CMD="mysql -h${MYSQL_HOST} -P${MYSQL_PORT} -u${BIZ_USER}"
|
||||
if [ -n "$BIZ_PASSWORD" ]; then
|
||||
BIZ_CMD="$BIZ_CMD -p${BIZ_PASSWORD}"
|
||||
fi
|
||||
|
||||
TABLE_COUNT=$($BIZ_CMD -N -e "SELECT COUNT(*) FROM information_schema.TABLES WHERE TABLE_SCHEMA='${DB_NAME}'" 2>/dev/null)
|
||||
if [ "$TABLE_COUNT" -gt 0 ] 2>/dev/null; then
|
||||
info "业务用户 ${BIZ_USER} 可正常访问 ${DB_NAME}(${TABLE_COUNT} 张表)"
|
||||
else
|
||||
error "业务用户权限验证失败"
|
||||
fi
|
||||
|
||||
echo ""
|
||||
echo "========================================"
|
||||
echo " MySQL 初始化完成"
|
||||
echo "========================================"
|
||||
echo ""
|
||||
echo " 数据库: ${DB_NAME}"
|
||||
echo " 业务用户: ${BIZ_USER}"
|
||||
echo " 表数量: ${TABLE_COUNT}"
|
||||
echo ""
|
||||
echo " 下一步: ./setup_linux.sh"
|
||||
echo ""
|
||||
Reference in New Issue
Block a user