|
|
|
|
|
#!/bin/bash
|
|
|
|
|
|
# ==============================================================================
|
|
|
|
|
|
# MySQL 分区表自动扩展脚本（按月分区）
|
|
|
|
|
|
# 适用表：t_stock_daily_trade（已按 trade_date 按月分区）
|
|
|
|
|
|
# 执行时机：每月最后一天凌晨2点（确保下一个月分区已创建）
|
|
|
|
|
|
# ==============================================================================
|
|
|
|
|
|
|
|
|
|
|
|
# -------------------------- 配置区（用户需根据实际情况修改） --------------------------
|
|
|
|
|
|
DB_HOST="localhost"
|
|
|
|
|
|
DB_PORT="3306"
|
|
|
|
|
|
DB_USER="root"
|
|
|
|
|
|
DB_PASS="your_password"
|
|
|
|
|
|
DB_NAME="a_stock_db"
|
|
|
|
|
|
TARGET_TABLE="t_stock_daily_trade" # 目标分区表
|
|
|
|
|
|
# ----------------------------------------------------------------------------------
|
|
|
|
|
|
|
|
|
|
|
|
# 日志函数
|
|
|
|
|
|
log() {
|
|
|
|
|
|
echo "[$(date +'%Y-%m-%d %H:%M:%S')] $1" >> /var/log/auto_add_mysql_partition.log
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
# 1. 计算下一个月的关键日期（如当前12月，下一个月是次年1月）
|
|
|
|
|
|
# 下一个月的1号（用于分区边界：VALUES LESS THAN (TO_DAYS('下下月1号'))）
|
|
|
|
|
|
next_month_first_day=$(date -d "$(date +%Y-%m-01) +1 month" +%Y-%m-01)
|
|
|
|
|
|
# 下下月的1号（分区边界：小于下下月1号，即包含整个下一个月）
|
|
|
|
|
|
next_next_month_first_day=$(date -d "$next_month_first_day +1 month" +%Y-%m-01)
|
|
|
|
|
|
# 分区名（格式：p202501）
|
|
|
|
|
|
partition_name="p$(date -d "$next_month_first_day" +%Y%m)"
|
|
|
|
|
|
|
|
|
|
|
|
log "INFO: 开始检查MySQL分区！下一个月：$next_month_first_day，分区名：$partition_name，边界：$next_next_month_first_day"
|
|
|
|
|
|
|
|
|
|
|
|
# 2. 检查分区是否已存在
|
|
|
|
|
|
check_sql="SELECT COUNT(*) FROM INFORMATION_SCHEMA.PARTITIONS
|
|
|
|
|
|
WHERE TABLE_SCHEMA = '$DB_NAME'
|
|
|
|
|
|
AND TABLE_NAME = '$TARGET_TABLE'
|
|
|
|
|
|
AND PARTITION_NAME = '$partition_name';"
|
|
|
|
|
|
|
|
|
|
|
|
# 执行查询并获取结果（过滤掉表头和空行）
|
|
|
|
|
|
partition_count=$(mysql -h$DB_HOST -P$DB_PORT -u$DB_USER -p$DB_PASS -D$DB_NAME -N -s -e "$check_sql")
|
|
|
|
|
|
|
|
|
|
|
|
if [ "$partition_count" -eq 1 ]; then
|
|
|
|
|
|
log "INFO: 分区已存在！无需创建，分区名：$partition_name"
|
|
|
|
|
|
exit 0
|
|
|
|
|
|
fi
|
|
|
|
|
|
|
|
|
|
|
|
# 3. 创建新分区
|
|
|
|
|
|
add_partition_sql="ALTER TABLE $TARGET_TABLE
|
|
|
|
|
|
ADD PARTITION (
|
|
|
|
|
|
PARTITION $partition_name
|
|
|
|
|
|
VALUES LESS THAN (TO_DAYS('$next_next_month_first_day'))
|
|
|
|
|
|
COMMENT '$(date -d "$next_month_first_day" +%Y年%m月)数据'
|
|
|
|
|
|
);"
|
|
|
|
|
|
|
|
|
|
|
|
if mysql -h$DB_HOST -P$DB_PORT -u$DB_USER -p$DB_PASS -D$DB_NAME -e "$add_partition_sql"; then
|
|
|
|
|
|
log "SUCCESS: 分区创建完成！分区名：$partition_name，边界：$next_next_month_first_day"
|
|
|
|
|
|
else
|
|
|
|
|
|
log "ERROR: 分区创建失败！SQL：$add_partition_sql"
|
|
|
|
|
|
exit 1
|
|
|
|
|
|
fi
|
