shell 判断与循环、传参选项、--help帮助信息
一、条件选择、判断
1、条件选择if语句
(1)多分支判断
if 判断条件 1 ; then 条件为真的分支代码 elif 判断条件 2 ; then 条件为真的分支代码 elif 判断条件 3 ; then 条件为真的分支代码 else 以上条件都为假的分支代码 fi
逐条件进行判断,第一次遇为“真”条件时,执行其分支,而后结束整个if。
(2)经典案例:
#!/bin/bash
read -p "请输入你的年龄: " age
if [[ $age =~ [^0-9] ]] ;then
echo "输入不合规,请输入整数"
exit 10
### exit命令是Shell内建命令,用于退出当前Shell进程。0为正常退出,非0为异常
elif [ $age -ge 120 ];then
echo "你已经活到人类极限!"
exit 20
elif [ $age -ge 60 -a $age -lt 90 ];then
echo "你已到退休年纪!好好修养!"
exit 20
elif (( "$age" <= "18" ));then
echo "你未成年,好好学习!"
else
echo "你已成年,好好工作!"
fi
2、条件判断 case
(1)用法格式
case $name in; PART1) cmd ;; PART2) cmd ;; *) cmd ;; esac
注意:case 支持glob 风格的通配符:
*: 任意长度任意字符 ?: 任意单个字符 [] :指定范围内的任意单个字符 a|b: a 或 b
(2)案例:
#判断yes or no
#!/bin/bash
read -p "Please input yes or no: " anw
case $anw in
[yY][eE][sS]|[yY])
echo yes
;;
[nN][oO]|[nN])
echo no
;;
*)
echo false
;;
esac
二、四个循环
1、for
(1)用法格式
① for name in 列表 ;do 循环体 done ② for (( exp1; exp2; exp3 )) ;do cmd done
(2)案例1
#求出(1+2+...+n)的总和
sum=0
read -p "Please input a positive integer: " num
if [[ $num =~ [^0-9] ]] ;then
echo "input error"
elif [[ $num -eq 0 ]] ;then
echo "input error"
else
for i in `seq 1 $num` ;do
sum=$[$sum+$i]
done
echo $sum
fi
unset zhi
分析:sum初始值为0,请输入一个数,先判断输入的是否含有除数字以外的字符,有,就报错;没有判断是否为0,不为0进入for循环,i的范围为1~输入的数,每次的循环为sum=sum+i,循环结束,最后输出sum的值。
#求出(1+2+...+100)的总和
for (( i=1,num=0;i<=100;i++ ));do
[ $[i%2] -eq 1 ] && let sum+=i
done
echo sum=$sum
分析:i=1,num=0;当i<=100,进入循环,若i÷2取余=1,则sum=sum+i,i=i+1。
案例2:背景,100w文本数据,拆分成200个(docker不能执行太长的数据量),for循环去分别执行200个文本,输出200个结果
for ((i=0;i<num;i++))
#!/bin/bash
i=0
num=200
for ((i=0;i<num;i++))
{
sql="select pi.person_name,
xxx,
xxx,
xxx,
xxx,
xxx
from xxx as pi
where xxx in
(`cat 100w_$i`)"
docker exec -it postgres sh -c "export PGPASSWORD=xxx; psql -h 127.0.0.1 -d database -U xxx -c \"COPY (${sql}) to stdout (FORMAT CSV, HEADER);\" | tee /tmp/zjz-$i " && docker cp postgres:/tmp/zjz-$i /mnt/hdd1/zjz/result_$i
}
嵌套循环案例
原始数据
# cat class.txt 甲班|jia|abc,% 甲班|jia|def,% 乙班|yi|a,% 乙班|yi|b,% 丙班|bing|bbb,% 丙班|bing|eee,%
想要的效果
甲班|jia|abc,%def,% 乙班|yi|a,%b,% 丙班|bing|bbb,%eee,%
脚本如下
cat class.sh #!/bin/bash content=$(cat class.txt) class=$(echo "$content" |cut -d "|" -f 1 | uniq) for classname in $class;do names=$(echo "$content" | grep "$classname|" | cut -d "|" -f 2 | uniq) for name in $names; do result=$(echo "$content"|grep "^$classname|$name|" | cut -d "|" -f 3) echo "$classname|$name|"$result done done

2、while
(1)用法格式
while 循环控制条件 ;do 循环 done
循环控制条件;进入循环之前,先做一次判断;每一次循环之后会再次做判断;条件为“true” ,则执行一次循环;直到条件测试状态为“false” 终止循环
(2)特殊用法(遍历文件的每一行):
while read line; do控制变量初始化 循环体 done < /PATH/FROM/SOMEFILE 或cat /PATH/FROM/SOMEFILE | while read line; do 循环体 done
依次读取/PATH/FROM/SOMEFILE文件中的每一行,且将行赋值给变量line
(3)案例:
function add_tag() {
...
}
function add_node() {
...
}
while true; do
read -rp "Is this a fresh machine? Enter yes to perform initialization, or no to skip initialization: " res
res=$(echo "$res" | tr '[:upper:]' '[:lower:]')
if [[ $res == "yes" || $res == "y" ]]; then
cd /mnt/deploy/scripts
py=$(whereis /usr/bin/python* | head -1 | awk '{print $2}')
$py /mnt/deploy/scripts/createfiles.py
/usr/bin/bash /mnt/deploy/scripts/initialize_"$(hostname -i)".sh
add_node
exit 0
elif [[ $res == "no" || $res == "n" ]]; then
add_node
exit 0
else
echo "Invalid input, please enter yes or no."
fi
done
while常见形式:(重定向、循环)
while read line
do
…
done < file
实战案例(一个txt文本,两行为一组,有四列逗号相隔的文件,处理后如下)
1 500xxxxxxxxxxxxxxx 陈宕 R5xxxxxxxxxxxxxx7110018 1 500xxxxxxxxxxxxxxxx 陈铭 R500xxxxxxxxxxxx8070005 2 5ddllllllllll0036 李业盛 Rdsjdjkdsvlsdvlsdkv069 2 5ddllllllllll0036 李业盛 Rdsjdjkdsvlsdvlsdkv069 3 6ddllllllllll0036 张业 Rdsjdjkdsvlsdvlsdkv069 3 6ddllllllllll0036 张业 Rdsjdjkdsvlsdvlsdkv069
需求:
1、以第一列为序号建分组,如第x组;
2、以2、3、4列建下级子目录500xxxxxxxxxxxxxxx_陈宕_R5xxxxxxxxxxxxxx7110018;
3、将与第四列同名的*.bmp文件移到对应的目录下;
4、不成功的行,做标记。
#!/bin/bash
echo > ./info.log
rm -rf 第*
num=0
num_info=0
is_no_num=0
########a b c d 分别代表txt文本中的四列数据
while read a b c d ##### while 读取每一行以空格为分界线
do
let num_info=${num_info}+1
if [ $a -eq $num ];then
mkdir /root/group2/第${a}组
mkdir /root/group2/第${a}组/$info
mkdir /root/group2/第${a}组/${b}_${c}_${d}
mv /root/group2/${d}.bmp /root/group2/第${a}组/${b}_${c}_${d}
mv /root/group2/${d_up}.bmp /root/group2/第${a}组/$info
else
let is_no=$is_no_num+1
if [ $is_no -eq $num_info ];then
echo $is_no_num >>./info.log ###没成功的行号做标记
fi
is_no_num=$num_info
num=$a
info=${b}_${c}_${d}
d_up=$d
continue
fi
done < output.txt
补充:while和getopts的用法
while getopts ":abc:d:" opt; do
case $opt in
a)
# 处理 -a 选项
echo "Option -a"
;;
b)
# 处理 -b 选项
echo "Option -b"
;;
c)
# 处理 -c 选项,$OPTARG 表示选项的参数
echo "Option -c with argument '$OPTARG'"
;;
d)
# 处理 -d 选项,$OPTARG 表示选项的参数
echo "Option -d with argument '$OPTARG'"
;;
\?)
# 处理未知选项
echo "Invalid option: -$OPTARG"
;;
:)
# 处理缺少参数的情况
echo "Option -$OPTARG requires an argument."
;;
esac
done
在getopts命令中,冒号(:)的作用是用于处理错误和定义选项是否需要参数。
-
处理错误: 如果选项列表中的开头有一个冒号,那么
getopts会在发现无效选项时不会输出错误信息,而是将错误信息存储在$OPTARG变量中,这样你可以自己处理错误。如果选项列表中的开头没有冒号,getopts会默认输出错误信息到标准错误输出。 -
定义选项是否需要参数: 冒号还用于指示哪些选项需要参数。如果选项列表中的字母后面跟着一个冒号,表示该选项需要参数。如果字母后面没有冒号,表示该选项不需要参数。如果选项需要参数但未提供参数,或者无法识别的选项,
getopts将执行 case 语句中的:分支。
args=$(getopt -o hlbr: --long help,list,backup,restore: -n "$0" -- "$@")
if [ $? != 0 ]; then
usage
fi
eval set -- "${args}"
while true
do
case "$1" in
-r | --restore)
# 如果有增量备份参数,则接受三个参数;否则接受两个参数
if [[ $@ =~ ^-r|--restore ]]; then
shift # 移除 --restore 参数
mc_restore $@; break
else
usage
break;
fi ;;
-b | --backup) mc_backup; break; ;;
-l | --list) show_data_list ;;
-h | --help) usage ;;
--) shift; break ;;
*) echo "Unexpected option: $1 - this should not happen."
usage ;;
esac
done
3、until 循环
(1)用法
unitl 循环条件 ;do 循环 done
进入条件:循环条件为true ;退出条件:循环条件为false;刚好和while相反,所以不常用,用while就行。
案例一
#监控xiaoming用户,登录就杀死
until pgrep -u xiaoming &> /dev/null ;do
sleep 0.5
done
pkill -9 -u xiaoming
分析:每隔0.5秒扫描,直到发现xiaoming用户登录,杀死这个进程,退出脚本,用于监控用户登录。
案例二
直到es运行之后再修改默认密码
wait_for_elasticsearch() {
until curl -s http://localhost:9200 >/dev/null; do
echo "Waiting for Elasticsearch..."
sleep 2
done
}
change_elastic_password() {
curl -u elastic:changeme -X PUT 'http://localhost:9200/_xpack/security/user/elastic/_password?pretty' \
-H 'Content-Type: application/json' \
-d @- <<EOF
{
"password": "$es_password"
}
EOF
}
4、select 循环与菜单
(1)用法
select variable in list do 循环体命令 done
- ① select 循环主要用于创建菜单,按数字顺序排列的示菜单项将显示在标准错误上,并显示PS3 提示符,等待用户输入
- ② 用户输入菜单列表中的某个数字,执行相应的命令
- ③ 用户输入被保存在内置变量 REPLY 中
- ④ select 是个无限循环,因此要记住用 break 命令退出循环,或用 exit 按 命令终止脚本。也可以按 ctrl+c退出循环
- ⑤ select 和 经常和 case 联合使用
- ⑥ 与for循环类似,可以省略 in list, 此时使用位置参量
(2)示例
#生成菜单,并显示选中的价钱
PS3="Please choose the menu: "
select menu in mifan huimian jiaozi babaozhou quit
do
case $REPLY in
1|4)
echo "the price is 15"
;;
2|3)
echo "the price is 20"
;;
5)
break
;;
*)
echo "no the option"
esac
done
#!/bin/bash
echo "What is your favourite OS?"
select name in "Linux" "Windows" "Mac OS" "UNIX" "Android"
do
case $name in
"Linux")
echo "Linux是一个类UNIX操作系统,它开源免费,运行在各种服务器设备和嵌入式设备。"
break
;;
"Windows")
echo "Windows是微软开发的个人电脑操作系统,它是闭源收费的。"
break
;;
"Mac OS")
echo "Mac OS是苹果公司基于UNIX开发的一款图形界面操作系统,只能运行与苹果提供的硬件之上。"
break
;;
"UNIX")
echo "UNIX是操作系统的开山鼻祖,现在已经逐渐退出历史舞台,只应用在特殊场合。"
break
;;
"Android")
echo "Android是由Google开发的手机操作系统,目前已经占据了70%的市场份额。"
break
;;
*)
echo "输入错误,请重新输入"
esac
done
分析:PS3是select的提示符,自动生成菜单,选择5 break退出循环。
补充:菜单实现参数选项
# ========== 参数处理 ==========
case "$1" in
--backup|-b)
backup_databases "$2"
;;
--restore|-r)
restore_databases "$2"
;;
*)
show_help
;;
esac
小工具代码
#!/bin/bash
# ========== 配置 ==========
SOCKET="/var/run/mysqld/mysqld.sock"
DATA_PATH="/healsci/docker/volumes/mysqlm_data/_data"
BACKUP_DIR="/root/mysql_backups"
MYSQL_USER="root"
MYSQL_PASS="bigdata"
MYSQL_HOST="127.0.0.1"
MYSQL_PORT="3306"
EXCLUDE="mysql|sys|performance_schema|information_schema"
TIMESTAMP=$(date +%Y%m%d_%H%M%S)
LOG_FILE="/var/log/mysql_backup_restore.log"
mkdir -p "$BACKUP_DIR"
touch "$LOG_FILE"
log() {
echo "[$(date +'%F %T')] $*" | tee -a "$LOG_FILE"
}
# ========== 时间格式化函数 ==========
format_duration() {
local total_seconds=$1
local hours=$(( total_seconds / 3600 ))
local minutes=$(( (total_seconds % 3600) / 60 ))
local seconds=$(( total_seconds % 60 ))
printf "%02d小时%02d分钟%02d秒" "$hours" "$minutes" "$seconds"
}
# ========== 备份函数 ==========
backup_databases() {
local target_db=$1
local counter=1
SECONDS=0
if [[ -n "$target_db" ]]; then
log "备份单个数据库:$target_db"
mysqldump -S "$SOCKET" --column-statistics=0 \
--triggers --routines --events \
--databases "$target_db" > "$BACKUP_DIR/${target_db}_${TIMESTAMP}.sql"
if [[ $? -eq 0 ]]; then
log "备份成功:$BACKUP_DIR/${target_db}_${TIMESTAMP}.sql"
else
log "备份失败:$target_db"
fi
else
log "开始备份全部用户数据库..."
databases=$(du -sh "$DATA_PATH"/*/ \
| sort -hr \
| grep -Ev "/($EXCLUDE)/$" \
| awk -F'/' '{print $(NF-1)}')
for db in $databases; do
log "[$counter] 正在备份数据库:$db"
mysqldump -S "$SOCKET" --column-statistics=0 \
--triggers --routines --events \
--databases "$db" > "$BACKUP_DIR/${db}_${TIMESTAMP}.sql"
if [[ $? -eq 0 ]]; then
log "[$counter] 备份完成:$BACKUP_DIR/${db}_${TIMESTAMP}.sql"
else
log "[$counter] 备份失败:$db"
fi
((counter++))
done
fi
duration=$(( SECONDS ))
log "总共耗时:$(format_duration $duration) 进行备份操作"
}
# ========== 恢复函数 ==========
restore_databases() {
local target_db=$1
local counter=1
SECONDS=0
if [[ -n "$target_db" ]]; then
latest_file=$(ls -t "$BACKUP_DIR"/${target_db}_*.sql 2>/dev/null | head -n 1)
if [[ ! -f "$latest_file" ]]; then
log "没有找到 ${target_db} 的备份文件"
return 1
fi
log "恢复单个数据库:$target_db(文件:$latest_file)"
pv "$latest_file" | mysql -h "$MYSQL_HOST" -P "$MYSQL_PORT" -u "$MYSQL_USER" -p"$MYSQL_PASS"
[[ $? -eq 0 ]] && log "导入成功:$target_db" || log "导入失败:$target_db"
else
log "开始恢复全部数据库..."
for file in "$BACKUP_DIR"/*.sql; do
dbname=$(basename "$file" | sed -E 's/^([^_]+)_.*/\1/')
log "[$counter] 正在导入数据库:$dbname(文件:$file)"
pv "$file" | mysql -h "$MYSQL_HOST" -P "$MYSQL_PORT" -u "$MYSQL_USER" -p"$MYSQL_PASS"
[[ $? -eq 0 ]] && log "[$counter] 导入成功:$dbname" || log "[$counter] 导入失败:$dbname"
((counter++))
done
fi
duration=$(( SECONDS ))
log "总共耗时:$(format_duration $duration) 进行恢复操作"
}
# ========== 帮助函数 ==========
show_help() {
echo "用法:"
echo " $0 --backup [dbname] # 备份指定数据库或所有数据库"
echo " $0 --restore [dbname] # 导入指定数据库或全部数据库"
echo ""
echo "示例:"
echo " $0 --backup # 备份全部"
echo " $0 --backup pbm_app # 仅备份 pbm_app"
echo " $0 --restore # 恢复全部"
echo " $0 --restore pbm_app # 仅恢复 pbm_app"
}
# ========== 参数处理 ==========
case "$1" in
--backup|-b)
backup_databases "$2"
;;
--restore|-r)
restore_databases "$2"
;;
*)
show_help
;;
esac
补充:打印--help帮助信息
# 主程序入口
if [ $# -ne 1 ]; then
usage
fi
case "$1" in
save)
save_images
;;
push)
load_and_push_images
;;
*)
usage
;;
esac

三、循环里的一些用法
1、循环控制语句
(1)语法
continue [N]:提前结束第N层的本轮循环,而直接进入下一轮判断;最内层为第1层 break [N]:提前结束第N层循环,最内侧为第1层
例:while CONDTITON1; do CMD1 if CONDITION2; then continue / break fi CMD2 done
(2)案例:
#①求(1+3+...+49+53+...+100)的和
#!/bin/bash
sum=0
for i in {1..100} ;do
[ $i -eq 51 ] && continue
[ $[$i%2] -eq 1 ] && { let sum+=i;let i++; }
done
echo sum=$sum
分析:做1+2+...+100的循环,当i=51时,跳过这次循环,但是继续整个循环,结果为:sum=2449
#②求(1+3+...+49)的和
#!/bin/bash
sum=0
for i in {1..100} ;do
[ $i -eq 51 ] && break
[ $[$i%2] -eq 1 ] && { let sum+=i;let i++; }
done
echo sum=$sum
分析:做1+2+...+100的循环,当i=51时,跳出整个循环,结果为:sum=625
2、循环控制shift命令
(1)作用
用于将参数列表list左移指定次数,最左端的那个参数就从列表中删除,其后边的参数继续进入循环
(2)案例:
#①创建指定的多个用户
#!/binbash
if [ $# -eq 0 ] ;then
echo "Please input a arg(eg:`basename $0` arg1)"
exit 1
else
while [ -n "$1" ];do
useradd $1 &> /dev/null
shift
done
fi
分析:如果没有输入参数(参数的总数为0),提示错误并退出;反之,进入循环;若第一个参数不为空字符,则创建以第一个参数为名的用户,并移除第一个参数,将紧跟的参数左移作为第一个参数,直到没有第一个参数,退出。
#②打印直角三角形的字符
#!/binbash
while (( $# > 0 ))
do
echo "$*"
shift
done
3、返回值结果
true 永远返回成功结果 : null command ,什么也不干,返回成功结果 false 永远返回错误结果
创建无限循环
while true ;do 循环体 done
4、循环中可并行执行,使脚本运行更快
(1)用法
for name in 列表 ;do
{
循环体
}&
done
wait
(2)实例:
#搜寻自己指定ip(子网掩码为24的)的网段中,UP的ip地址
read -p "Please input network (eg:192.168.0.0): " net
echo $net |egrep -o "\<(([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])\.){3}([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])\>"
[ $? -eq 0 ] || ( echo "input error";exit 10 )
IP=`echo $net |egrep -o "^([0-9]{1,3}\.){3}"`
for i in {1..254};do
{
ping -c 1 -w 1 $IP$i &> /dev/null && \
echo "$IP$i is up"
}&
done
wait
分析:请输入一个IP地址例192.168.37.234,如果格式不是0.0.0.0 则报错退出;正确则进入循环,IP变量的值为192.168.37. i的范围为1-254,并行ping 192.168.37.1-154,ping通就输出此IP为UP。直到循环结束。
四、信号捕获trap
1、用法格式
trap ' 触发指令' 信号,自定义进程收到系统发出的指定信号后,将执行触发指令,而不会执行原操作 trap '' 信号,忽略信号的操作 trap '-' 信号,恢复原信号的操作 trap -p,列出自定义信号操作
信号可以3种表达方法:信号的数字2、全名SIGINT、缩写INT
2、常用信号
1) SIGHUP: 无须关闭进程而让其重读配置文件 2) SIGINT: 中止正在运行的进程;相当于Ctrl+c 3) SIGQUIT: 相当于ctrl+\ 9) SIGKILL: 强制杀死正在运行的进程 15) SIGTERM :终止正在运行的进程(默认为15) 18) SIGCONT :继续运行 19) SIGSTOP :后台休眠 9 信号,强制杀死,捕获不住
3、案例
#①打印0-9,ctrl+c不能终止
#!/bin/bash
trap 'echo press ctrl+c' 2
for ((i=0;i<10;i++));do
sleep 1
echo $i
done
分析:i=0,当i<10,每休眠1秒,i+1,捕获2信号,并执行echo press ctrl+c
#②打印0-3,ctrl+c不能终止,3之后恢复,能终止
#!/bin/bash
trap '' 2
trap -p
for ((i=0;i<3;i++));do
sleep 1
echo $i
done
trap '-' SIGINT
for ((i=3;i<10;i++));do
sleep 1
echo $i
done
分析:i=0,当i<3,每休眠1秒,i+1,捕获2信号;i>3时,解除捕获2信号。
五、生成连续数字(实用)
1、$ echo {1..10}
1 2 3 4 5 6 7 8 9 10
2、 awk 'BEGIN { for (i=0; i<=20; i++) printf("%02d ", i) }'
00 01 02 03 04 05 06 07 08 09 10 11 12 13 14 15 16 17 18 19 20
3、seq -w 1 10
01 02 03 04 05 06 07 08 09 10
4、 printf "%02d " {1..10}
01 02 03 04 05 06 07 08 09 10
5、seq -f "%03g" 0 10
000 001 002 003 004 005 006 007 008 009 010
运用1
for i in `echo {10..20}`;do mkdir 第$i组;done
$ ls
第10组 第11组 第12组 第13组 第14组 第15组 第16组 第17组 第18组 第19组 第20组

浙公网安备 33010602011771号