feat: spread tool
This commit is contained in:
parent
9f85dfc443
commit
c968b9be27
2 changed files with 75 additions and 0 deletions
5
servers.txt
Normal file
5
servers.txt
Normal file
|
@ -0,0 +1,5 @@
|
|||
link
|
||||
keymaker
|
||||
apoc
|
||||
kid
|
||||
mouse
|
70
tools/spread.sh
Executable file
70
tools/spread.sh
Executable file
|
@ -0,0 +1,70 @@
|
|||
#!/bin/bash
|
||||
|
||||
# spread command over multiple servers
|
||||
|
||||
SPREAD_SERVERS=${SPREAD_SERVERS:-}
|
||||
SPREAD_SERVERS_FILE=${SPREAD_SERVERS_FILE:-servers.txt}
|
||||
SPREAD_USER=${SPREAD_USER:-$USER}
|
||||
SPREAD_COMMAND=${SPREAD_COMMAND:-}
|
||||
|
||||
while getopts "s:f:u:" opt; do
|
||||
case $opt in
|
||||
s)
|
||||
SPREAD_SERVERS="$SPREAD_SERVERS $OPTARG"
|
||||
;;
|
||||
f)
|
||||
SPREAD_SERVERS_FILE="$SPREAD_SERVERS_FILE $OPTARG"
|
||||
;;
|
||||
u)
|
||||
SPREAD_USER="$OPTARG"
|
||||
;;
|
||||
\?)
|
||||
echo "Invalid option: -$OPTARG" >&2
|
||||
;;
|
||||
esac
|
||||
done
|
||||
shift $((OPTIND - 1))
|
||||
|
||||
if [ -n "$1" ]; then
|
||||
SPREAD_COMMAND="$@"
|
||||
fi
|
||||
|
||||
if [ -f $SPREAD_SERVERS_FILE ]; then
|
||||
SPREAD_SERVERS="$SPREAD_SERVERS $(cat $SPREAD_SERVERS_FILE)"
|
||||
fi
|
||||
|
||||
if [ -z "$SPREAD_SERVERS" ]; then
|
||||
echo "No servers specified"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
if [ -z "$SPREAD_COMMAND" ]; then
|
||||
echo "No command specified"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
_ssh() {
|
||||
ssh \
|
||||
-l $SPREAD_USER \
|
||||
-o ConnectTimeout=5 \
|
||||
$1 \
|
||||
"$SPREAD_COMMAND"
|
||||
}
|
||||
|
||||
for server in $SPREAD_SERVERS; do
|
||||
# while loop so we get:: $SPEAD_USER@$server: <output>
|
||||
case ${SPREAD_LOG_STYLE:-prefixed} in
|
||||
prefixed)
|
||||
_ssh $server | while read -r line; do
|
||||
printf "%s@%-10s: %s\n" "$SPREAD_USER" "$server" "$line"
|
||||
done
|
||||
;;
|
||||
header)
|
||||
printf "\n%s@%s\n---\n" "$SPREAD_USER" "$server"
|
||||
_ssh $server
|
||||
;;
|
||||
*)
|
||||
_ssh $server
|
||||
;;
|
||||
esac
|
||||
done
|
Loading…
Reference in a new issue