Shell 脚本编程

L0 能跑           —— 逐条命令堆一起
L1 可控           —— set -euo pipefail：出错即停、失败可见
L2 可复用         —— 函数化、参数化、可被 cron 与人同时调用
L3 可维护         —— 日志输出、幂等、trap 清理、可 dry-run

#!/usr/bin/env bash
set -euo pipefail
# -e  任何命令失败立即退出（默认是继续执行下一条——灾难之源）
# -u  使用未定义变量直接报错（把 "$typo" 从"空字符串"变成"显式错误"）
# -o pipefail  管道中任何一段失败都算整体失败（默认只看最后一段）

IFS=$'\n\t'        # 词分割只认换行与 Tab——文件名带空格不再炸循环

# trap：脚本退出前清理（临时目录、锁文件）
TMPDIR=$(mktemp -d)
trap 'rm -rf "$TMPDIR"' EXIT

# 陷阱 1：未加引号的变量会按空格分词
rm -rf "$TARGET_DIR"      # 正确
rm -rf $TARGET_DIR        # TARGET_DIR="/data/my app" 时变成删两个目录

# 陷阱 2：[ ] 与 [[ ]] 的区别
if [ "$name" = "prod" ]; then ... fi        # [ 是命令：空变量必须加引号
if [[ $name == "prod" && -f $cfg ]]; then   # [[ 是关键字：可免引号、支持 && 与模式匹配

# 陷阱 3：命令替换的退出码丢失
COUNT=$(cat counts | wc -l)      # cat 失败时 COUNT=0，错误被吞
COUNT=$(wc -l < counts)          # 更好：少一个进程；配合 -o pipefail 才能感知 cat 失败

#!/usr/bin/env bash
# cleanup_logs.sh — 清理过期应用日志（crontab: 30 2 * * *）
set -euo pipefail
IFS=$'\n\t'

LOG_DIRS=(
    "/var/log/myapp"
    "/var/log/nginx"
)
RETAIN_DAYS=30

log() { echo "[$(date '+%F %T')] $*"; }

total_freed=0
for dir in "${LOG_DIRS[@]}"; do
    # -u 用 -u：目录不存在只跳过不报错（配合 -e 不至于中断整个清理）
    [[ -d "$dir" ]] || { log "跳过不存在的目录: $dir"; continue; }

    # -print0/-xargs -0：文件名含空格/换行都安全
    freed=$(find "$dir" -type f -name '*.log*' -mtime +"$RETAIN_DAYS" -print0 |
            xargs -0 -r du -cb 2>/dev/null | tail -1 | cut -f1 || echo 0)
    find "$dir" -type f -name '*.log*' -mtime +"$RETAIN_DAYS" -print0 |
        xargs -0 -r rm -f

    log "清理 $dir：释放 $((freed / 1024 / 1024)) MB"
    total_freed=$((total_freed + freed))
done

log "共释放 $((total_freed / 1024 / 1024)) MB"

# 兜底：清完还超阈值就告警（对接监控见巡检脚本）
usage=$(df --output=pcent / | tail -1 | tr -d ' %')
if (( usage > 85 )); then
    log "告警：根分区仍使用 ${usage}%" >&2
    exit 1            # cron 会捕获非零退出码；配合监控按失败告警
fi

#!/usr/bin/env bash
# healthcheck.sh — 系统健康巡检（可人跑也可 cron 跑）
set -uo pipefail          # 注意：这里不用 -e——单项检查失败应该继续查其他项

RED='\033[0;31m'; GREEN='\033[0;32m'; YELLOW='\033[1;33m'; NC='\033[0m'
FAILED=0

log_ok()   { echo -e "${GREEN}[OK]${NC} $1"; }
log_warn() { echo -e "${YELLOW}[WARN]${NC} $1"; }
log_fail() { echo -e "${RED}[FAIL]${NC} $1"; FAILED=$((FAILED + 1)); }

check_cpu() {
    local load cores threshold
    load=$(awk '{print $1}' /proc/loadavg)
    cores=$(nproc)
    threshold=$(awk "BEGIN{print $cores * 0.8}")
    if awk "BEGIN{exit !($load > $threshold)}"; then
        log_warn "CPU 负载过高: $load (阈值: $threshold)"
    else
        log_ok "CPU 负载正常: $load / ${cores}核"
    fi
}

check_memory() {
    local usage
    usage=$(free | awk '/Mem/{printf "%.1f", $3/$2*100}')
    if awk "BEGIN{exit !($usage > 90)}"; then
        log_fail "内存使用率过高: ${usage}%"
    elif awk "BEGIN{exit !($usage > 80)}"; then
        log_warn "内存使用率偏高: ${usage}%"
    else
        log_ok "内存使用率正常: ${usage}%"
    fi
}

check_disk() {
    local usage
    usage=$(df -h / | awk 'NR==2{gsub(/%/,""); print $5}')
    if (( usage > 90 )); then
        log_fail "磁盘使用率过高: ${usage}%"
    elif (( usage > 80 )); then
        log_warn "磁盘使用率偏高: ${usage}%"
    else
        log_ok "磁盘使用率正常: ${usage}%"
    fi
}

check_services() {
    local svc
    for svc in nginx docker sshd; do
        if systemctl is-active --quiet "$svc" 2>/dev/null; then
            log_ok "$svc 运行中"
        else
            log_fail "$svc 未运行"
        fi
    done
}

echo "===== 系统健康检查 $(date '+%F %T') ====="
check_cpu
check_memory
check_disk
check_services
echo "===== 完成：${FAILED} 项失败 ====="

# 退出码语义：0 全绿；1 有 FAIL——cron/CI 按退出码决定告警
exit $(( FAILED > 0 ? 1 : 0 ))

#!/usr/bin/env bash
# deploy.sh <产物包> — 滚动部署到 HOSTS
set -euo pipefail
IFS=$'\n\t'

ARTIFACT="${1:?用法: $0 <产物包.tar.gz>}"      # :? 缺参即报错退出
HOSTS=(web1 web2 web3 web4 web5)
APP_DIR=/opt/myapp
DEPLOY_LOCK=/tmp/myapp-deploy.lock

# 防并发：同一时刻只允许一个部署（flock 文件锁）
exec 200>"$DEPLOY_LOCK"
flock -n 200 || { echo "已有部署在进行" >&2; exit 1; }

[[ -f "$ARTIFACT" ]] || { echo "产物不存在: $ARTIFACT" >&2; exit 1; }

for host in "${HOSTS[@]}"; do
    echo "===== 部署 $host ====="

    # 上传（scp 失败即整体失败——-e 保证）
    scp -q "$ARTIFACT" "$host:/tmp/myapp-new.tar.gz"

    # 远程执行：注意引号嵌套——单引号内的 $ 是远端展开
    ssh -o BatchMode=yes "$host" bash -s <<'REMOTE'
        set -euo pipefail
        tar -xzf /tmp/myapp-new.tar.gz -C /opt/myapp --strip-components=1
        sudo systemctl restart myapp
        sleep 2
        curl -fsS http://localhost:8080/healthz > /dev/null   # 健康检查不过即失败
        echo "远端部署完成"
REMOTE

    echo "$host OK"
done

echo "全部完成：${#HOSTS[@]} 台"