今天寫了一個小工具，用shell實現從一臺機器往其他多臺機器複製檔案或者在其他多臺機器執行相同命令用的。

目錄結構如下，

[[email protected]1 ~]# cd tools/
[[email protected]1 tools]# ls
copy_files.sh  main.sh  run_commands.sh  servers

其中，servers裡面維護要執行的節點列表，hostname或ip均可； copy_files.sh是複製檔案的指令碼，run_commands是執行命令的指令碼，main.sh是主入口，用法如下，

[root@cent
 
-1 tools]# ./main.sh
usage: main.sh [-e command-name] | [-c file-name]
  example: $(basename $0) -e ssh-keygen
  example: $(basename $0) -c /etc/hosts

“-e”表示執行一個命令，後面接命令；”-c”表示複製一個檔案，後面接檔案全路徑。
舉例如下，
1 複製檔案

[root@cent-1 tools]# ./main.sh -c /etc/hosts
Copy files to cent-2...

2 執行命令

[[email protected]
 
1 tools]# cat servers
cent-1
cent-2
[[email protected]1 tools]# ./main.sh -e ssh-keygen
Run commands on cent-1...
Enter file in which to save the key (/root/.ssh/id_rsa):
Generating public/private rsa key pair.
/root/.ssh/id_rsa already exists.
Overwrite (y/n)?
Run commands on cent-2...
Enter file in which to
 
 save the key (/root/.ssh/id_rsa):
Generating public/private rsa key pair.
/root/.ssh/id_rsa already exists.
Overwrite (y/n)?

具體程式碼實現如下，

[[email protected]1 tools]# cat main.sh
#!/bin/bash

if [ "$#" != "2" ]; then
  echo "usage: $(basename $0) [-e command-name] | [-c file-name]"
  echo '  example: $(basename $0) -e ssh-keygen'
  echo '  example: $(basename $0) -c /etc/hosts'
  exit 1
fi

if [ "$1" == "-e" ]; then
  sh run_commands.sh $2
fi

if [ "$1" == "-c" ]; then
  sh copy_files.sh $2
fi


[[email protected]1 tools]# cat copy_files.sh
#!/bin/bash

if [ "$#" != "1" ]; then
  echo "usage: $(basename $0) <file-name>"
  echo "  example: $(basename $0) /etc/hosts"
  exit 1
fi

SRC_PATH=$PWD/servers
SOURCE_FILE=$1
SOURCE_FILE_BAK=$1.bak

for srv in $(cat $SRC_PATH);do
  echo "Copy files to $srv..."
  ssh $srv "mv $SOURCE_FILE $SOURCE_FILE_BAK"
  rsync $SOURCE_FILE $srv:$SOURCE_FILE
done


[[email protected]1 tools]# cat run_commands.sh
#!/bin/bash

if [ "$#" != "1" ]; then
  echo "usage: $(basename $0) <command-name>"
  echo "  example: $(basename $0) ssh-keygen"
  exit 1
fi

SRC_PATH=$PWD/servers
COMMAND=$1

for srv in $(cat $SRC_PATH);do
  echo "Run commands on $srv..."
  ssh $srv "$COMMAND"
done