mirror of
https://github.com/EvernodeXRPL/sashimono.git
synced 2026-04-29 15:38:00 +00:00
Bash script to manage test vm clusters (#40)
This commit is contained in:
Chalith Desaman
1
test/vm-cluster/.gitignore
vendored
Normal file
1
test/vm-cluster/.gitignore
vendored
Normal file
@@ -0,0 +1 @@
|
||||
config.json
|
||||
266
test/vm-cluster/cluster.sh
Executable file
266
test/vm-cluster/cluster.sh
Executable file
@@ -0,0 +1,266 @@
|
||||
#!/bin/bash
|
||||
# Sashimono cluster management script.
|
||||
|
||||
# Usage examples:
|
||||
# ./cluster.sh select contract
|
||||
# ./cluster.sh create 1
|
||||
# ./cluster.sh create
|
||||
|
||||
# Command modes:
|
||||
# select - Sets the currently active contract from the list of contracts defined in config.json file.
|
||||
# create - Create new sashimono hotpocket instance in each node.
|
||||
# initiate - Initiate sashimono hotpocket instance with configs.
|
||||
# start - Start sashimono hotpocket instance.
|
||||
# stop - Stop sashimono hotpocket instance.
|
||||
# destroy - Destroy sashimono hotpocket instance.
|
||||
|
||||
mode=$1
|
||||
|
||||
if [ "$mode" == "select" ] || [ "$mode" == "create" ] || [ "$mode" == "initiate" ] || [ "$mode" == "start" ] || [ "$mode" == "stop" ] || [ "$mode" == "destroy" ]; then
|
||||
echo "mode: $mode"
|
||||
else
|
||||
echo "Invalid command."
|
||||
echo " Expected: select <contract name> | create [N] | initiate [N] | start [N] | stop [N] | destroy [N]"
|
||||
echo " <N>: Required node no. [N]: Optional node no."
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# jq command is used for json manipulation.
|
||||
if ! command -v jq &>/dev/null; then
|
||||
sudo apt-get install -y jq
|
||||
fi
|
||||
|
||||
configfile=config.json
|
||||
if [ ! -f $configfile ]; then
|
||||
# Create default config file.
|
||||
echo '{"selected":"contract","contracts":[{"name":"contract","sshuser":"root","sshpass":"<ssh password>","owner_pubkey":"ed.....","contract_id":"<uuid>","image":"<docker image key>","hosts":{"host1_ip":{}},"config":{}}]}' | jq . >$configfile
|
||||
fi
|
||||
|
||||
if [ $mode == "select" ]; then
|
||||
selectedcont=$2
|
||||
if [ "$selectedcont" == "" ] || [ "$selectedcont" == "null" ]; then
|
||||
echo "Please specify contract name to select."
|
||||
exit 1
|
||||
fi
|
||||
continfo=$(jq -r ".contracts[] | select(.name == \"$selectedcont\")" $configfile)
|
||||
if [ "$continfo" == "" ] || [ "$continfo" == "null" ]; then
|
||||
echo "No configuration found for selected contract '"$selectedcont"'"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Set the 'selected' field value on cluster config file.
|
||||
jq ".selected = \"$selectedcont\"" $configfile >$configfile.tmp && mv $configfile.tmp $configfile
|
||||
echo "Selected '"$selectedcont"'"
|
||||
exit 0
|
||||
fi
|
||||
|
||||
selectedcont=$(jq -r '.selected' $configfile)
|
||||
if [ "$selectedcont" == "" ] || [ "$selectedcont" == "null" ]; then
|
||||
echo "No contract selected."
|
||||
exit 1
|
||||
fi
|
||||
|
||||
continfo=$(jq -r ".contracts[] | select(.name == \"$selectedcont\")" $configfile)
|
||||
if [ "$continfo" == "" ] || [ "$continfo" == "null" ]; then
|
||||
echo "No configuration found for selected contract '"$selectedcont"'"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Read ssh user and password and set contract directory based on username.
|
||||
sshuser=$(echo $continfo | jq -r '.sshuser')
|
||||
sshpass=$(echo $continfo | jq -r '.sshpass')
|
||||
if [ "$sshuser" == "" ] || [ "$sshuser" == "null" ]; then
|
||||
echo "sshuser not specified."
|
||||
exit 1
|
||||
fi
|
||||
|
||||
shopt -s expand_aliases
|
||||
alias sshskp='ssh -o StrictHostKeychecking=no'
|
||||
if [ "$sshpass" != "" ] && [ "$sshpass" != "null" ]; then
|
||||
alias sshskp='sshpass -p $sshpass ssh -o StrictHostKeychecking=no'
|
||||
fi
|
||||
|
||||
hosts=$(echo $continfo | jq -r '.hosts')
|
||||
hostaddrs=($(echo $hosts | jq -r 'keys[]'))
|
||||
|
||||
# Check if second arg (nodeid) is a number or not.
|
||||
# If it's a number then reduce 1 from it to get zero-based node index.
|
||||
if ! [[ $2 =~ ^[0-9]+$ ]]; then
|
||||
let nodeid=-1
|
||||
else
|
||||
let nodeid=$2-1
|
||||
fi
|
||||
|
||||
if [ $mode == "create" ]; then
|
||||
# Read owner pubkey, contract id and image
|
||||
ownerpubkey=$(echo $continfo | jq -r '.owner_pubkey')
|
||||
if [ "$ownerpubkey" = "" ] || [ "$ownerpubkey" = "null" ]; then
|
||||
echo "owner_pubkey not specified."
|
||||
exit 1
|
||||
fi
|
||||
|
||||
contractid=$(echo $continfo | jq -r '.contract_id')
|
||||
if [ "$contractid" == "" ] || [ "$contractid" == "null" ]; then
|
||||
echo "contract_id not specified."
|
||||
exit 1
|
||||
fi
|
||||
|
||||
image=$(echo $continfo | jq -r '.image')
|
||||
if [ "$image" == "" ] || [ "$image" == "null" ]; then
|
||||
echo "image not specified."
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Create an instance for given host.
|
||||
function createinstance() {
|
||||
hostaddr=$1
|
||||
command="sashi json '{\"type\":\"create\",\"owner_pubkey\":\"$ownerpubkey\",\"contract_id\":\"$contractid\",\"image\":\"$image\"}'"
|
||||
output=$(sshskp $sshuser@$hostaddr $command | tr '\0' '\n')
|
||||
# If an output received consider updating the json file.
|
||||
if [ ! "$output" = "" ]; then
|
||||
content=$(echo $output | jq -r '.content')
|
||||
echo $output
|
||||
# Update the json if no error.
|
||||
if [ ! "$content" == "" ] && [ ! "$content" == "null" ] && [[ ! "$content" =~ ^[a-zA-Z]+_error$ ]]; then
|
||||
jq "(.contracts[] | select(.name == \"$selectedcont\") | .hosts.\"$hostaddr\") |= $content" $configfile >$configfile.tmp && mv $configfile.tmp $configfile
|
||||
fi
|
||||
fi
|
||||
}
|
||||
|
||||
if [ $nodeid = -1 ]; then
|
||||
for hostaddr in "${hostaddrs[@]}"; do
|
||||
createinstance $hostaddr &
|
||||
done
|
||||
wait
|
||||
else
|
||||
hostaddr=${hostaddrs[$nodeid]}
|
||||
createinstance $hostaddr
|
||||
fi
|
||||
exit 0
|
||||
fi
|
||||
|
||||
if [ $mode == "initiate" ]; then
|
||||
# Initiate the instance of given host.
|
||||
function initiateinstance() {
|
||||
hostaddr=$1
|
||||
peers=$2
|
||||
unl=$3
|
||||
config=$(echo $continfo | jq -r ".config")
|
||||
containername=$(echo $continfo | jq -r ".hosts.\"$hostaddr\".name")
|
||||
peerport=$(echo $continfo | jq -r ".hosts.\"$hostaddr\".peer_port")
|
||||
selfpeer="\"$hostaddr:$peerport\""
|
||||
# Remove self peer from the peers.
|
||||
updatedpeers=$(echo $peers | sed "s/\($selfpeer,\|,$selfpeer\|$selfpeer\)//g")
|
||||
# Update the in memory config with received peers and unl.
|
||||
updatedconfig=$(echo $config | jq ".mesh.known_peers = $updatedpeers" | jq ".contract.unl = $unl")
|
||||
command="sashi json '{\"type\":\"initiate\",\"container_name\":\"$containername\",\"config\":$updatedconfig}'"
|
||||
sshskp $sshuser@$hostaddr $command
|
||||
}
|
||||
|
||||
# Read each hosts config and construct cluster unl and peers.
|
||||
peers=""
|
||||
unl=""
|
||||
for hostaddr in "${hostaddrs[@]}"; do
|
||||
hostinfo=$(echo $continfo | jq -r ".hosts.\"$hostaddr\"")
|
||||
pubkey=$(echo $hostinfo | jq -r '.pubkey')
|
||||
ip=$(echo $hostinfo | jq -r '.ip')
|
||||
peerport=$(echo $hostinfo | jq -r '.peer_port')
|
||||
|
||||
if [ "$hostinfo" == "" ] || [ "$hostinfo" == "null" ] ||
|
||||
[ "$pubkey" == "" ] || [ "$pubkey" == "null" ] ||
|
||||
[ "$ip" == "" ] || [ "$ip" == "null" ] ||
|
||||
[ "$peerport" == "" ] || [ "$peerport" == "null" ]; then
|
||||
echo "Host info is empty for $hostaddr"
|
||||
exit 1
|
||||
fi
|
||||
peers+="\"$hostaddr:$peerport\","
|
||||
unl+="\"$pubkey\","
|
||||
done
|
||||
|
||||
# Remove trainling comma(,) and add square brackets for the lists.
|
||||
peers="[${peers%?}]"
|
||||
unl="[${unl%?}]"
|
||||
|
||||
if [ $nodeid = -1 ]; then
|
||||
for hostaddr in "${hostaddrs[@]}"; do
|
||||
initiateinstance $hostaddr $peers $unl &
|
||||
done
|
||||
wait
|
||||
else
|
||||
hostaddr=${hostaddrs[$nodeid]}
|
||||
initiateinstance $hostaddr $peers $unl
|
||||
fi
|
||||
exit 0
|
||||
fi
|
||||
|
||||
if [ $mode == "start" ]; then
|
||||
# Start instance of given host.
|
||||
function startinstance() {
|
||||
hostaddr=$1
|
||||
containername=$(echo $continfo | jq -r ".hosts.\"$hostaddr\".name")
|
||||
command="sashi json '{\"type\":\"start\",\"container_name\":\"$containername\"}'"
|
||||
sshskp $sshuser@$hostaddr $command
|
||||
}
|
||||
|
||||
if [ $nodeid = -1 ]; then
|
||||
for hostaddr in "${hostaddrs[@]}"; do
|
||||
startinstance $hostaddr &
|
||||
done
|
||||
wait
|
||||
else
|
||||
hostaddr=${hostaddrs[$nodeid]}
|
||||
startinstance $hostaddr
|
||||
fi
|
||||
exit 0
|
||||
fi
|
||||
|
||||
if [ $mode == "stop" ]; then
|
||||
# Stop instance of given host.
|
||||
function stopinstance() {
|
||||
hostaddr=$1
|
||||
containername=$(echo $continfo | jq -r ".hosts.\"$hostaddr\".name")
|
||||
command="sashi json '{\"type\":\"stop\",\"container_name\":\"$containername\"}'"
|
||||
sshskp $sshuser@$hostaddr $command
|
||||
}
|
||||
|
||||
if [ $nodeid = -1 ]; then
|
||||
for hostaddr in "${hostaddrs[@]}"; do
|
||||
stopinstance $hostaddr &
|
||||
done
|
||||
wait
|
||||
else
|
||||
hostaddr=${hostaddrs[$nodeid]}
|
||||
stopinstance $hostaddr
|
||||
fi
|
||||
exit 0
|
||||
fi
|
||||
|
||||
if [ $mode == "destroy" ]; then
|
||||
# Destroy instance of given host.
|
||||
function destroyinstance() {
|
||||
hostaddr=$1
|
||||
containername=$(echo $continfo | jq -r ".hosts.\"$hostaddr\".name")
|
||||
command="sashi json '{\"type\":\"destroy\",\"container_name\":\"$containername\"}'"
|
||||
output=$(sshskp $sshuser@$hostaddr $command | tr '\0' '\n')
|
||||
# If an output received consider updating the json file.
|
||||
if [ ! "$output" = "" ]; then
|
||||
content=$(echo $output | jq -r '.content')
|
||||
echo $output
|
||||
# Update the json if no error.
|
||||
if [ ! "$content" == "" ] && [ ! "$content" == "null" ] && [[ ! "$content" =~ ^[a-zA-Z]+_error$ ]]; then
|
||||
jq "(.contracts[] | select(.name == \"$selectedcont\") | .hosts.\"$hostaddr\") |= {}" $configfile >$configfile.tmp && mv $configfile.tmp $configfile
|
||||
fi
|
||||
fi
|
||||
}
|
||||
|
||||
if [ $nodeid = -1 ]; then
|
||||
for hostaddr in "${hostaddrs[@]}"; do
|
||||
destroyinstance $hostaddr &
|
||||
done
|
||||
wait
|
||||
else
|
||||
hostaddr=${hostaddrs[$nodeid]}
|
||||
destroyinstance $hostaddr
|
||||
fi
|
||||
exit 0
|
||||
fi
|
||||
Reference in New Issue
Block a user