mirror of
https://github.com/featurebasedb/featurebase.git
synced 2026-09-06 00:25:55 +00:00
b82375bca2
- terraform scripts to set up cluster - cloud-formation scripts to set up cluster - set up ingest node with kafka server and datagen - set up second ingest node with molecula-consumer-kafka-static - set up datadog in all nodes (ingest + featurebase) - set up script to execute different queries - only run delete test on schedule
116 lines
3.7 KiB
Bash
Executable file
116 lines
3.7 KiB
Bash
Executable file
#!/bin/bash
|
|
|
|
# Make sure you have already deployed your cloudformation stack
|
|
# before you run this script
|
|
# To run script: ./setupTremorDeleteTestCFT.sh <stack-name> <aws profile> <replica count> <fb instance count> <branch-name>
|
|
|
|
SCRIPT_DIR=$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )
|
|
source $SCRIPT_DIR/utilCluster.sh
|
|
|
|
# store arguments
|
|
STACK_NAME=$1
|
|
AWS_PROFILE=$2
|
|
DEPLOYED_CLUSTER_REPLICA_COUNT=$3
|
|
DEPLOYED_CLUSTER_INSTANCE_COUNT=$4
|
|
BRANCH_NAME=$5
|
|
|
|
# update ebs device name for ingest nodes
|
|
EBS_DEVICE_NAME="/dev/sdb"
|
|
|
|
# get producer ip (assume only 1 instance for now)
|
|
PRODUCER_IPS=$($SCRIPT_DIR/getASGInstanceIPs.sh ${STACK_NAME}-producer-asg ${AWS_PROFILE})
|
|
PRODUCER0=$(echo $PRODUCER_IPS | head -n1 | cut -d " " -f1)
|
|
echo "setting up producer node, producer_ip: $PRODUCER0, EBS device: $EBS_DEVICE_NAME"
|
|
setupProducerNode $PRODUCER0
|
|
|
|
# setup consumer
|
|
CONSUMER_IPS=$($SCRIPT_DIR/getASGInstanceIPs.sh ${STACK_NAME}-consumer-asg ${AWS_PROFILE})
|
|
CONSUMER0=$(echo $CONSUMER_IPS | head -n1 | cut -d " " -f1)
|
|
echo "setting up consumer node, consumer_ip: $CONSUMER0, EBS device: $EBS_DEVICE_NAME"
|
|
setupConsumerNode $CONSUMER0
|
|
|
|
# no ebs device, using nvme instead, we can skip ebs mount in executeGeneralNodeConfigCommands()
|
|
# nvme is used for data nodes
|
|
EBS_DEVICE_NAME="nvme"
|
|
|
|
# get featurebase binary from the current branch
|
|
FB_BINARY=featurebase_linux_amd64
|
|
echo "Downloading FeatureBase binary"
|
|
aws s3 cp s3://molecula-artifact-storage/featurebase/${BRANCH_NAME}/_latest/featurebase_linux_amd64 ./ --profile=$AWS_PROFILE
|
|
|
|
# get the data node ips
|
|
DEPLOYED_DATA_IPS=$($SCRIPT_DIR/getASGInstanceIPs.sh ${STACK_NAME}-asg ${AWS_PROFILE})
|
|
echo "DEPLOYED_DATA_IPS: {"
|
|
echo "${DEPLOYED_DATA_IPS}"
|
|
echo "}"
|
|
DEPLOYED_DATA_IPS_LEN=`echo "$DEPLOYED_DATA_IPS" | wc -l`
|
|
|
|
# poll for autoscaling instance counts, until it matches the desired count
|
|
ASG_NOT_STABLE="true"
|
|
for i in {0..24}
|
|
do
|
|
if [ $DEPLOYED_DATA_IPS_LEN -eq $DEPLOYED_CLUSTER_INSTANCE_COUNT ]
|
|
then
|
|
echo "Autoscaling group is ready after ${i} tries."
|
|
ASG_NOT_STABLE="false"
|
|
break
|
|
fi
|
|
echo "waiting for autoscaling group to reach steady state..."
|
|
sleep 10
|
|
DEPLOYED_DATA_IPS=$($SCRIPT_DIR/getASGInstanceIPs.sh ${STACK_NAME}-asg ${AWS_PROFILE})
|
|
echo "DEPLOYED_DATA_IPS: {"
|
|
echo "${DEPLOYED_DATA_IPS}"
|
|
echo "}"
|
|
DEPLOYED_DATA_IPS_LEN=`echo "$DEPLOYED_DATA_IPS" | wc -l`
|
|
done
|
|
|
|
if [[ "$ASG_NOT_STABLE" == "true" ]]
|
|
then
|
|
echo "Autoscaling group still unstable - giving up"
|
|
exit 1
|
|
fi
|
|
|
|
DATANODE0=$(echo $DEPLOYED_DATA_IPS | head -n1 | cut -d " " -f1)
|
|
echo "using DATANODE0 ${DATANODE0}"
|
|
|
|
# define some env var and print them
|
|
DEPLOYED_CLUSTER_PREFIX=${STACK_NAME}
|
|
echo "Using DEPLOYED_CLUSTER_PREFIX: ${DEPLOYED_CLUSTER_PREFIX}"
|
|
echo "Using DEPLOYED_CLUSTER_REPLICA_COUNT: ${DEPLOYED_CLUSTER_REPLICA_COUNT}"
|
|
|
|
#wait until we can connect to one of the hosts
|
|
for i in {0..24}
|
|
do
|
|
ssh -A -i ~/.ssh/gitlab-featurebase-ci.pem -o StrictHostKeyChecking=no -o ConnectTimeout=10 ec2-user@${DATANODE0} "pwd"
|
|
if [ $? -eq 0 ]
|
|
then
|
|
echo "Cluster is up after ${i} tries."
|
|
break
|
|
fi
|
|
sleep 10
|
|
done
|
|
|
|
ssh -A -i ~/.ssh/gitlab-featurebase-ci.pem -o StrictHostKeyChecking=no -o ConnectTimeout=10 ec2-user@${DATANODE0} "pwd"
|
|
if [ $? -ne 0 ]
|
|
then
|
|
echo "Unable to connect to cluster - giving up"
|
|
exit 1
|
|
fi
|
|
|
|
#setup data nodes
|
|
generateInitialClusterString
|
|
setupDataNodes
|
|
startDataNodes
|
|
|
|
# copy script for test
|
|
scp -i ~/.ssh/gitlab-featurebase-ci.pem -o StrictHostKeyChecking=no ./qa/scripts/delete/runQueries.sh ec2-user@${DATANODE0}:
|
|
|
|
# verify featurebase running
|
|
echo "Verifying featurebase cluster running..."
|
|
curl -s http://${DATANODE0}:10101/status
|
|
if (( $? != 0 ))
|
|
then
|
|
echo "Featurebase cluster not running"
|
|
exit 1
|
|
fi
|
|
echo "FeatureBase Cluster running."
|