check_linux.sh

#!/usr/bin/env bash

set -e

trap Cleanup EXIT

producer_bin=$1
consumer_bin=$2
asapo_tool_bin=$3
network_type=$4

data_source=detector

beamtime_id1=asapo_test1

beamtime_id2=asapo_test2

monitor_database_name=db_test
proxy_address=127.0.0.1:8400

beamline1=test1
beamline2=test2
receiver_root_folder=/tmp/asapo/receiver/files
facility=test_facility
year=2019
receiver_folder1=${receiver_root_folder}/${facility}/gpfs/${beamline1}/${year}/data/${beamtime_id1}
receiver_folder2=${receiver_root_folder}/${facility}/gpfs/${beamline2}/${year}/data/${beamtime_id2}

Cleanup() {
    echo cleanup
    set +e
    if [[ $network_type == "fabric" ]]; then
      #nomad stop receiver
      #nomad run receiver_tcp.nmd
      cp /receiver_tcp.json /receiver.json
      supervisorctl restart receiver
      while true
      do
        sleep 1
        curl --silent 127.0.0.1:8400/asapo-discovery/v0.1/asapo-receiver?protocol=v0.6 --stderr - | grep localhost  || continue
        echo recevier started
        break
      done
    fi
    rm -rf ${receiver_root_folder}
    echo "db.dropDatabase()" | mongo ${beamtime_id1}_${data_source}
    echo "db.dropDatabase()" | mongo ${beamtime_id2}_${data_source}

    set +e
    influx_out=`influx -database ${monitor_database_name} -execute "drop series from statistics, RequestsRate"`
    influx_status=$?
    echo "Influx output: ${influx_out}"
    if [ $influx_status -ne 0 ]; then
        if [[ $influx_out == *"401"* ]]; then
            echo "Ignoring auth error from influxdb"
        else
            echo "influxdb failed to delete test database '${monitor_database_name}'"
            exit $influx_status
        fi
    fi
}

if [[ $network_type == "fabric" ]]; then
    #nomad stop receiver
    #nomad run receiver_fabric.nmd
    cp /receiver_fabric.json /receiver.json
    supervisorctl restart receiver
    while true
    do
      sleep 1
      curl --silent 127.0.0.1:8400/asapo-discovery/v0.1/asapo-receiver?protocol=v0.6 --stderr - | grep localhost  || continue
      echo recevier started
      break
    done
fi

token1=`$asapo_tool_bin token -endpoint http://localhost:8400/asapo-authorizer -secret admin_token.key -types read $beamtime_id1`
token2=`$asapo_tool_bin token -endpoint http://localhost:8400/asapo-authorizer -secret admin_token.key -types read $beamtime_id2`

echo "Start producers"
mkdir -p ${receiver_folder1}
mkdir -p ${receiver_folder2}
$producer_bin localhost:8400 ${beamtime_id1} 100 1000 4 0 100 &
$producer_bin localhost:8400 ${beamtime_id2} 100 900 4 0 100 &
#producerid=`echo $!`

echo "Start consumers in $network_type mode"
$consumer_bin ${proxy_address} ${receiver_folder1} ${beamtime_id1} 2 $token1 12000 0 | tee /dev/stderr consumer_1.out
$consumer_bin ${proxy_address} ${receiver_folder2} ${beamtime_id2} 2 $token2 12000 0 | tee /dev/stderr consumer_2.out

grep "from memory buffer: 1000" consumer_1.out
grep -i "Using connection type: $network_type" consumer_1.out

grep "from memory buffer: 900" consumer_2.out
grep -i "Using connection type: $network_type" consumer_2.out