如何优化运维Shell脚本

优化运维 Shell 脚本可以从可读性、健壮性、性能、安全性和可维护性几个维度入手。下面是一套比较系统的优化思路和实战建议。

一、脚本结构优化

1. 明确脚本头部（Shebang & 环境）

#!/usr/bin/env bash
set -euo pipefail

• set -e：命令失败立即退出
• set -u：使用未定义变量报错
• set -o pipefail：管道中任意命令失败都算失败

2. 统一日志与输出

LOG_FILE="/var/log/ops.log"

log() {
  echo "[$(date '+%F %T')] $*" | tee -a "$LOG_FILE"
}

3. 函数化拆分

避免“一整坨代码”：

check_env() { :; }
backup_data() { :; }
main() {
  check_env
  backup_data
}
main "$@"

二、健壮性优化（防坑）

1. 输入参数校验

if [ $# -ne 2 ]; then
  echo "Usage: $0 <src> <dst>"
  exit 1
fi

2. 路径和变量加引号

rm -rf "$dir"/*

防止空格、空变量导致灾难。

3. 锁机制（防止重复执行）

exec 200>/var/lock/myscript.lock
flock -n 200 || exit 1

三、性能优化

1. 减少子进程

❌ 慢：

for i in $(cat list.txt); do
  echo $i
done

✅ 快：

while read -r line; do
  echo "$line"
done < list.txt

2. 避免频繁 awk/grep 调用

❌：

cat file | grep a | awk '{print $1}'

✅：

awk '/a/{print $1}' file

3. 批量处理而非循环

• 用 xargs
• 用 rsync 替代循环 scp

四、安全优化

1. 避免明文密码

source /etc/ops/secret.env

2. 最小权限

• 用普通用户运行
• sudo 精确授权

3. 危险命令保护

[ -z "$TARGET" ] && exit 1
rm -rf "$TARGET"

五、可维护性优化

1. 配置与逻辑分离

CONFIG_FILE=/etc/ops/config.conf
source "$CONFIG_FILE"

2. 统一退出码

EXIT_OK=0
EXIT_ERR=1

3. 加注释但不冗余

说明“为什么”，而不是“做什么”。

六、示例：优化前后对比

❌ 原始脚本

for i in `cat ip.txt`
do
ping $i
done

✅ 优化后

#!/usr/bin/env bash
set -euo pipefail

while read -r ip; do
  ping -c1 -W1 "$ip" >/dev/null && echo "$ip OK"
done < ip.txt

七、推荐工具辅助优化

• shellcheck：静态检查
• bashate：风格检查
• shfmt：格式化

如果你有具体脚本，可以直接贴出来，我可以帮你逐行优化并解释原因。