269 lines
8.8 KiB
Bash
269 lines
8.8 KiB
Bash
#!/bin/bash
|
||
|
||
# =====================================================
|
||
# 校园思政新闻平台数据库重新初始化脚本
|
||
# 版本: 1.0.0
|
||
# 说明: 该脚本会删除现有数据库，重新创建并初始化所有数据
|
||
# 注意: 执行前请确保MySQL服务已启动，并且有足够的权限
|
||
# =====================================================
|
||
|
||
# 设置颜色输出
|
||
RED='\033[0;31m'
|
||
GREEN='\033[0;32m'
|
||
YELLOW='\033[1;33m'
|
||
BLUE='\033[0;34m'
|
||
NC='\033[0m' # No Color
|
||
|
||
# 数据库配置（优先使用环境变量）
|
||
DB_HOST="${DB_HOST:-localhost}"
|
||
DB_PORT="${DB_PORT:-3306}"
|
||
DB_USER="${DB_USER:-root}"
|
||
DB_PASSWORD="${DB_PASSWORD:-123456}"
|
||
DB_NAME="${DB_NAME:-school_news}"
|
||
|
||
# 脚本目录
|
||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||
|
||
# 日志文件
|
||
LOG_FILE="$SCRIPT_DIR/reInit.log"
|
||
|
||
# 打印带颜色的消息
|
||
print_message() {
|
||
local color=$1
|
||
local message=$2
|
||
echo -e "${color}${message}${NC}"
|
||
echo "$(date '+%Y-%m-%d %H:%M:%S') - $message" >> "$LOG_FILE"
|
||
}
|
||
|
||
# 检查MySQL连接
|
||
check_mysql_connection() {
|
||
print_message $BLUE "检查MySQL连接..."
|
||
|
||
if command -v mysql &> /dev/null; then
|
||
if mysql -h"$DB_HOST" -P"$DB_PORT" -u"$DB_USER" -p"$DB_PASSWORD" -e "SELECT 1;" &> /dev/null; then
|
||
print_message $GREEN "MySQL连接成功"
|
||
return 0
|
||
else
|
||
print_message $RED "MySQL连接失败，请检查配置"
|
||
return 1
|
||
fi
|
||
else
|
||
print_message $RED "MySQL客户端未安装或不在PATH中"
|
||
return 1
|
||
fi
|
||
}
|
||
|
||
# 备份现有数据库（如果存在）
|
||
backup_database() {
|
||
if mysql -h"$DB_HOST" -P"$DB_PORT" -u"$DB_USER" -p"$DB_PASSWORD" -e "USE $DB_NAME;" &> /dev/null; then
|
||
print_message $YELLOW "发现现有数据库，正在备份..."
|
||
local backup_file="$SCRIPT_DIR/backup_${DB_NAME}_$(date +%Y%m%d_%H%M%S).sql"
|
||
mysqldump -h"$DB_HOST" -P"$DB_PORT" -u"$DB_USER" -p"$DB_PASSWORD" "$DB_NAME" > "$backup_file"
|
||
if [ $? -eq 0 ]; then
|
||
print_message $GREEN "数据库备份成功: $backup_file"
|
||
else
|
||
print_message $YELLOW "数据库备份失败，继续执行..."
|
||
fi
|
||
fi
|
||
}
|
||
|
||
# 删除现有数据库
|
||
drop_database() {
|
||
print_message $YELLOW "删除现有数据库..."
|
||
mysql -h"$DB_HOST" -P"$DB_PORT" -u"$DB_USER" -p"$DB_PASSWORD" -e "DROP DATABASE IF EXISTS $DB_NAME;"
|
||
if [ $? -eq 0 ]; then
|
||
print_message $GREEN "数据库删除成功"
|
||
else
|
||
print_message $RED "数据库删除失败"
|
||
exit 1
|
||
fi
|
||
}
|
||
|
||
# 执行初始化脚本
|
||
execute_init_script() {
|
||
print_message $BLUE "开始执行数据库初始化..."
|
||
|
||
# 切换到脚本目录
|
||
cd "$SCRIPT_DIR"
|
||
|
||
# 执行initAll.sql
|
||
mysql -h"$DB_HOST" -P"$DB_PORT" -u"$DB_USER" -p"$DB_PASSWORD" < initAll.sql
|
||
|
||
if [ $? -eq 0 ]; then
|
||
print_message $GREEN "数据库初始化成功"
|
||
else
|
||
print_message $RED "数据库初始化失败"
|
||
exit 1
|
||
fi
|
||
}
|
||
|
||
# 执行敏感词导入
|
||
import_sensitive_words() {
|
||
print_message $BLUE "开始导入敏感词数据..."
|
||
|
||
# 切换到敏感词脚本目录
|
||
local sensitive_dir="$SCRIPT_DIR/sensitiveData"
|
||
if [ ! -d "$sensitive_dir" ]; then
|
||
print_message $YELLOW "敏感词脚本目录不存在: $sensitive_dir"
|
||
return 0
|
||
fi
|
||
|
||
# 检查Shell脚本是否存在
|
||
if [ -f "$sensitive_dir/importSensitiveWords.sh" ]; then
|
||
print_message $BLUE "使用Shell脚本导入敏感词"
|
||
cd "$sensitive_dir"
|
||
|
||
# 导出数据库配置环境变量
|
||
export DB_HOST="$DB_HOST"
|
||
export DB_PORT="$DB_PORT"
|
||
export DB_USER="$DB_USER"
|
||
export DB_PASSWORD="$DB_PASSWORD"
|
||
export DB_NAME="$DB_NAME"
|
||
export AUTO_CONFIRM=true
|
||
|
||
bash importSensitiveWords.sh -y
|
||
|
||
if [ $? -eq 0 ]; then
|
||
print_message $GREEN "敏感词数据导入成功"
|
||
else
|
||
print_message $YELLOW "敏感词数据导入失败，但不影响系统运行"
|
||
fi
|
||
|
||
cd "$SCRIPT_DIR"
|
||
return 0
|
||
fi
|
||
|
||
# 兼容旧的Python脚本（如果Shell脚本不存在）
|
||
if [ -f "$sensitive_dir/writeWord.py" ]; then
|
||
print_message $YELLOW "使用Python脚本导入敏感词（建议使用Shell版本）"
|
||
|
||
# 检查conda是否可用
|
||
if ! command -v conda &> /dev/null; then
|
||
print_message $YELLOW "conda命令未找到，跳过敏感词导入"
|
||
return 0
|
||
fi
|
||
|
||
# 检查schoolNewsCrawler环境是否存在
|
||
if ! conda env list | grep -q "schoolNewsCrawler"; then
|
||
print_message $YELLOW "conda环境 'schoolNewsCrawler' 不存在，跳过敏感词导入"
|
||
return 0
|
||
fi
|
||
|
||
cd "$sensitive_dir"
|
||
conda run -n schoolNewsCrawler python writeWord.py -y
|
||
|
||
if [ $? -eq 0 ]; then
|
||
print_message $GREEN "敏感词数据导入成功"
|
||
else
|
||
print_message $YELLOW "敏感词数据导入失败，但不影响系统运行"
|
||
fi
|
||
|
||
cd "$SCRIPT_DIR"
|
||
return 0
|
||
fi
|
||
|
||
# 如果两个脚本都不存在
|
||
print_message $YELLOW "敏感词导入脚本不存在，跳过导入"
|
||
print_message $YELLOW "提示: 可以使用 importSensitiveWords.sh (推荐) 或 writeWord.py"
|
||
}
|
||
|
||
# 验证初始化结果
|
||
verify_initialization() {
|
||
print_message $BLUE "验证初始化结果..."
|
||
|
||
# 检查数据库是否存在
|
||
if mysql -h"$DB_HOST" -P"$DB_PORT" -u"$DB_USER" -p"$DB_PASSWORD" -e "USE $DB_NAME;" &> /dev/null; then
|
||
print_message $GREEN "数据库创建成功"
|
||
else
|
||
print_message $RED "数据库创建失败"
|
||
exit 1
|
||
fi
|
||
|
||
# 检查表数量
|
||
local table_count=$(mysql -h"$DB_HOST" -P"$DB_PORT" -u"$DB_USER" -p"$DB_PASSWORD" -D"$DB_NAME" -e "SHOW TABLES;" | wc -l)
|
||
print_message $GREEN "创建了 $table_count 张表"
|
||
|
||
# 检查默认用户
|
||
local user_count=$(mysql -h"$DB_HOST" -P"$DB_PORT" -u"$DB_USER" -p"$DB_PASSWORD" -D"$DB_NAME" -e "SELECT COUNT(*) FROM tb_sys_user WHERE username='admin';" | tail -n 1)
|
||
if [ "$user_count" -gt 0 ]; then
|
||
print_message $GREEN "默认用户创建成功"
|
||
else
|
||
print_message $YELLOW "默认用户创建失败"
|
||
fi
|
||
|
||
# 检查敏感词数量
|
||
local sensitive_count=$(mysql -h"$DB_HOST" -P"$DB_PORT" -u"$DB_USER" -p"$DB_PASSWORD" -D"$DB_NAME" -e "SELECT COUNT(*) FROM tb_sensitive_word;" 2>/dev/null | tail -n 1)
|
||
if [ -n "$sensitive_count" ] && [ "$sensitive_count" -gt 0 ]; then
|
||
print_message $GREEN "敏感词数据: $sensitive_count 个"
|
||
else
|
||
print_message $YELLOW "敏感词数据: 0 个 (可能未导入或表不存在)"
|
||
fi
|
||
}
|
||
|
||
# 显示初始化信息
|
||
show_initialization_info() {
|
||
print_message $BLUE "====================================================="
|
||
print_message $GREEN "数据库初始化完成！"
|
||
print_message $BLUE "====================================================="
|
||
print_message $YELLOW "数据库信息:"
|
||
print_message $YELLOW " 数据库名: $DB_NAME"
|
||
print_message $YELLOW " 主机: $DB_HOST"
|
||
print_message $YELLOW " 端口: $DB_PORT"
|
||
print_message $YELLOW " 用户: $DB_USER"
|
||
print_message $BLUE "====================================================="
|
||
print_message $YELLOW "默认账户信息:"
|
||
print_message $YELLOW " 用户名: admin"
|
||
print_message $YELLOW " 密码: admin123"
|
||
print_message $YELLOW " 角色: 管理员"
|
||
print_message $BLUE "====================================================="
|
||
print_message $YELLOW "系统功能:"
|
||
print_message $YELLOW " - 用户管理"
|
||
print_message $YELLOW " - 权限管理"
|
||
print_message $YELLOW " - 资源管理"
|
||
print_message $YELLOW " - 学习管理"
|
||
print_message $YELLOW " - AI智能体"
|
||
print_message $YELLOW " - 系统配置"
|
||
print_message $YELLOW " - 文件管理"
|
||
print_message $YELLOW " - 敏感词过滤"
|
||
print_message $YELLOW " - 消息通知"
|
||
print_message $BLUE "====================================================="
|
||
}
|
||
|
||
# 主函数
|
||
main() {
|
||
print_message $BLUE "====================================================="
|
||
print_message $BLUE "校园思政新闻平台数据库重新初始化脚本"
|
||
print_message $BLUE "版本: 1.1.0"
|
||
print_message $BLUE "====================================================="
|
||
|
||
# 检查MySQL连接
|
||
if ! check_mysql_connection; then
|
||
exit 1
|
||
fi
|
||
|
||
# 备份现有数据库
|
||
# backup_database
|
||
|
||
# 删除现有数据库
|
||
# drop_database
|
||
|
||
# 执行初始化脚本
|
||
execute_init_script
|
||
|
||
# 导入敏感词数据
|
||
|
||
# 验证初始化结果
|
||
verify_initialization
|
||
|
||
# 显示初始化信息
|
||
show_initialization_info
|
||
import_sensitive_words
|
||
|
||
print_message $GREEN "初始化完成！"
|
||
}
|
||
|
||
# 脚本入口
|
||
if [ "${BASH_SOURCE[0]}" == "${0}" ]; then
|
||
main "$@"
|
||
fi
|