批量SCP文件的脚本

最近在集群上部署Hadoop时发现在服务器之间传递文件比较麻烦，原因是使用scp每次只能指定一台目标服务器，并且必须指定用户名和目标路径。当集群中存在很多服务器时，分发Hadoop的配置文件就会耗费很大的时间，虽然没有技术挑战性，但却令人乏味。都说程序员是比较懒惰的物种，为了以后减少复制文件时的枯燥，决定写一个批量拷贝文件的脚本，当然是基于SCP命令的。

在我的Hadoop集群中，所有的用户名和密码是一致的，为了减少敲打键盘时出错的可能性，或者将懒惰进行到底，该脚本支持只输入一次用户名即可。考虑到集群中的目录层级也是一致的，也支持目标路径一致，还支持对目录的复制，其余scp的高级特性就不打算实现了。

那下面就对该脚本语法做个简单的介绍。语法比较简单，主要遵循上述偷懒的原则，具体如下：

语法一，以相同的用户名、目标名称复制文件: bscp -ut[r] sourceDir user targetDir host1 host2…

语法二，以相同的用户名、不同的目标名称复制文件: bscp -u[r] sourceDir user host1:targetDir1 host2:target2…

语法三，不同的用户名、相同的目标名称复制文件: bscp -t[r] sourceDir targetDir user1@host1 user2@host2…"

语法四，用户名、目标名称均不同: bscp [-r] sourceDir user1@host1:targetDir1 user2@host2:targetDir2…

相信大家都能够很容易地看懂语法格式，就不再赘述了，下面直接展现代码：

#!/bin/sh

print_usage()
{
echo "语法一，以相同的用户名、目标名称复制文件: bscp -ut[r] sourceDir user targetDir host1 host2…"
echo "语法二，以相同的用户名、不同的目标名称复制文件: bscp -u[r] sourceDir user host1:targetDir1 host2:target2…"
echo "语法三，不同的用户名、相同的目标名称复制文件: bscp -t[r] sourceDir targetDir user1@host1 user2@host2…"
echo "语法四，用户名、目标名称均不同: bscp [-r] sourceDir user1@host1:targetDir1 user2@host2:targetDir2…"
}
argNumber=$#
# 如果没有参数则打印使用方法
if [ $argNumber -le 1 ]; then
print_usage
exit 1
fi

COMMAND=$1
#shift

case $COMMAND in
(--help|-help|-h|help)
print_usage
exit 0
;;
esac

if [ "${COMMAND:0:1}" != "-" ];then
shift
args="$@"
for i in $args
do
scp $COMMAND $i
done
else
length=${#COMMAND}
#echo $length
if [ $length -eq 4 ]; then
shift
if [ $# -le 3 ];then
print_usage
exit 1
fi
source=$1
targetUser=$2
targetDir=$3
shift 3
args="$@"
for i in $args
do
scp -r $source $targetUser@$i:$targetDir
#echo $targetUser
done
elif [ $length -eq 3 ]; then
shift
source=$1
if [ `expr index "$COMMAND" r` -eq 0 ]; then
if [ $# -le 3 ]; then
print_usage
exit 1
fi
targetUser=$2
targetDir=$3
shift 3
args="$@"
for i in $args
do
scp $source $targetUser@$i:$targetDir
done
else
if [ `expr index "$COMMAND" u` -gt 0 ]; then
if [ $# -le 2 ]; then
print_usage
exit 1
fi
targetUser=$2
shift 2
args="$@"
for i in $args
do
scp -r $source $targetUser@$i
done
elif [ `expr index "$COMMAND" t` -gt 0 ]; then
if [ $# -le 2 ]; then
print_usage
exit 1
fi
targetDir=$2
shift 2
args="$@"
for i in $args
do
scp -r $source $i:$targetDir
done
else
print_usage
exit 1
fi
fi
elif [ $length -eq 2 ]; then
shift
source=$1
if [ `expr index "$COMMAND" u` -gt 0 ]; then
if [ $# -le 2 ]; then
print_usage
exit 1
fi
targetUser=$2
shift 2
args="$@"
for i in $args
do
scp $source $targetUser@$i
done
elif [ `expr index "$COMMAND" t` -gt 0 ]; then
if [ $# -le 2 ]; then
print_usage
exit 1
fi
targetDir=$2
shift 2
args="$@"
for i in $args
do
scp $source $i:$targetDir
done
elif [ `expr index "$COMMAND" r` -gt 0 ]; then
if [ $# -le 1 ]; then
print_usage
exit 1
fi
shift 1
args="$@"
for i in $args
do
scp -r $source $i
done
else
print_usage
exit 1
fi
else
print_usage
exit 1
fi
fi

在集群上分别测试了一些情况，对结果还是相当满意的。当然还是希望路过的大神可以指教，让这个小脚本更加完善。