Shell 脚本中 date 命令常用技巧与实战

动态参数与脚本调度

编写定时任务脚本时，通常需要既支持手动指定日期，又能在无参数时使用当前日期。这种写法兼顾了灵活性与自动化。

#!/bin/sh
if [[ $# -eq 1 ]]; then
    dt="${1}"
else
    dt=$(date +"%Y%m%d")
fi

dt_start=$(date -d "${dt}" +"%Y-%m-%d")
dt_end=$(date -d "1 day ${dt_start}" +"%Y-%m-%d")

echo "start druid ingest ${dt_start}"
sh process.sh para1 "${dt_start}" "${dt_end}"
echo "end druid ingest ${dt_start}"

连续时间戳生成逻辑

根据开始和结束时间生成连续的时间戳序列，这在数据回溯场景中非常有用。网上常见的思路是把年月日格式转成时间戳比大小，但更直观的做法是把年月日格式直接转成整数比较。

#!/bin/sh
# $1 是开始时间，$2 是结束时间
startdate=$(date -d "$1" +"%Y%m%d")
tempdate=${startdate}
enddate=$(date -d "$2" +"%Y%m%d")

# 将时间戳转成整数比大小
while [ $((tempdate)) -le $((enddate)) ]; do
    echo $tempdate
    # 20180801
    # echo $(date -d "$tempdate" +"%F")
    # 2018-08-01
    tempdate=$(date -d "1 day $tempdate" +"%Y%m%d")
    # 20180802
done

实际场景：MySQL 分表处理

下面给出一个用于 MySQL 分表的脚本示例。之所以要分到小时级别，是为了防止一次性将内存占满。在指定时间段内生成连续时间戳，并对每天的 24 小时分别处理。

注意：生产环境中请替换为实际的数据库连接信息，切勿硬编码密码。

#!/bin/sh

handleDay() {
    day=$1
    year_month=${day:0:6}
    echo ${year_month}
    
    hours=(0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23)
    for i in ${hours[@]}; do
        echo "${i}hour start"
        # 替换为实际连接配置
        mysql -h remote_host -u user -p password -D database_name -e "insert into table_${year_month} select * from table_ori where day=${day} and hour=${i}"
        sleep 3
    done
}

# 主函数：$1 是开始时间，$2 是结束时间
startdate=$(date -d "$1" +"%Y%m%d")
tempdate=${startdate}
enddate=$(date -d "$2" +"%Y%m%d")

while [ $((tempdate)) -le $((enddate)) ]; do
    echo $tempdate
    handleDay ${tempdate}
    tempdate=$(date -d "1 day $tempdate" +"%Y%m%d")
done

只取年月部分

有时候只需要年份和月份，例如 201810。同样可以通过判断参数是否存在来决定使用传入值还是当前时间。

#!/bin/sh
if [ ! -z "$1" ]; then
    current_month=$1
else
    current_month=$(date -d today +"%Y%m")
fi

以上这些技巧涵盖了日常数据处理中最常见的日期操作场景。在实际编写脚本时，记得处理好引号引用和变量扩展，避免因为空格或特殊字符导致脚本报错。