run-tests

#!/bin/bash

log_error() {
  echo "$*" 1>&2
}

run_govet() {
    GOVET_ERRORS=$(go vet "$*" 2>&1)
    if [ -n "$GOVET_ERRORS" ]
        then
            log_error "go vet failed for the following reasons:
            $GOVET_ERRORS
please run 'go vet $*' on your changes before committing."
        ps -ef | grep slock | grep -v grep | awk '{print $2}' | xargs kill -HUP > /dev/null 2>&1
        exit 1
    fi
}

run_gotest() {
    GOTEST_OUTPUTS=$(go test "$*" 2>&1)
    GOTEST_ERRORS=$(echo "$GOTEST_OUTPUTS" | grep "FAIL")
    if [ -n "$GOTEST_ERRORS" ]
        then
            log_error "go test failed for the following reasons:
            $GOTEST_OUTPUTS
please run 'go test $*' on your changes before committing."
        ps -ef | grep slock | grep -v grep | awk '{print $2}' | xargs kill -HUP > /dev/null 2>&1
        exit 1
    fi
}

echo "Run go vet"
run_govet "github.com/snower/slock/protocol"
run_govet "github.com/snower/slock/server"
run_govet "github.com/snower/slock/client"

echo "Run go test"
run_gotest "github.com/snower/slock/protocol"
run_gotest "github.com/snower/slock/server"

export PATH=$(pwd):$PATH
WORKLOAD_DIR="/tmp/$(date '+%Y%m%d%H%M%S')"
mkdir $WORKLOAD_DIR
echo "Workload: ${WORKLOAD_DIR}"

echo "Start Slock Server 5657"
DATA_DIR="${WORKLOAD_DIR}/data5657"
mkdir $DATA_DIR
cp .github/data/5657-meta.pb $DATA_DIR/meta.pb
LOG_FILENAME="${DATA_DIR}/slock.log"
slock --port=5657 --data_dir=$DATA_DIR --replset=test > $LOG_FILENAME 2>&1 &
SLOCK_PS=$(ps aux | grep slock | grep 5657 | grep -v grep)
if [ -z "$SLOCK_PS" ]
    then
        SLOCK_ERRORS=$(slock 2>&1)
        log_error "slock start 5657 failed for the following reasons:
                   $SLOCK_ERRORS
please run 'slock' on your changes before committing."
    exit 1
fi
sleep 3

echo "Start Slock Server 5659"
DATA_DIR="${WORKLOAD_DIR}/data5659"
mkdir $DATA_DIR
cp .github/data/5659-meta.pb $DATA_DIR/meta.pb
LOG_FILENAME="${DATA_DIR}/slock.log"
slock --port=5659 --data_dir=$DATA_DIR --replset=test > $LOG_FILENAME 2>&1 &
SLOCK_PS=$(ps aux | grep slock | grep 5659 | grep -v grep)
if [ -z "$SLOCK_PS" ]
    then
        SLOCK_ERRORS=$(slock 2>&1)
        log_error "slock start 5659 failed for the following reasons:
                   $SLOCK_ERRORS
please run 'slock' on your changes before committing."
    exit 1
fi
sleep 3

echo "Start Slock Server 5658"
DATA_DIR="${WORKLOAD_DIR}/data5658"
mkdir $DATA_DIR
cp .github/data/5658-meta.pb $DATA_DIR/meta.pb
LOG_FILENAME="${DATA_DIR}/slock.log"
slock --port=5658 --data_dir=$DATA_DIR --replset=test > $LOG_FILENAME 2>&1 &
SLOCK_PS=$(ps aux | grep slock | grep 5658 | grep -v grep)
if [ -z "$SLOCK_PS" ]
    then
        SLOCK_ERRORS=$(slock 2>&1)
        log_error "slock start 5658 failed for the following reasons:
                   $SLOCK_ERRORS
please run 'slock' on your changes before committing."
    exit 1
fi
sleep 3

run_gotest "github.com/snower/slock/client"

echo "Run Benchmark"
echo

BENCH_CMD="go run tools/benchmark/main.go --port=5659 --client=4 --conc=16 --count=300000"
BENCH_OUTPUTS=$(which timeout > /dev/null 2>&1 && timeout 120 $BENCH_CMD 2>&1 || $BENCH_CMD 2>&1)
BENCH_SUCCEDS=$(echo "$BENCH_OUTPUTS" | grep "Succed")
if [ -z "$BENCH_SUCCEDS" ]
    then
        log_error "slock benchmark multi key single lock failed for the following reasons:
                   $BENCH_OUTPUTS
please run 'go run benchmarks/benchmark/benchmark.go' on your changes before committing."

    ps -ef | grep slock | grep -v grep | awk '{print $2}' | xargs kill -HUP > /dev/null 2>&1
    ps -ef | grep benchmark | grep -v grep | awk '{print $2}' | xargs kill -HUP > /dev/null 2>&1
    exit 1
fi

echo "
Benchmark Multi Key Single Lock Info:
$BENCH_OUTPUTS
"

BENCH_CMD="go run tools/benchmark/main.go --port=5659 --key=1 --client=4 --conc=16 --count=100000"
BENCH_OUTPUTS=$(which timeout > /dev/null 2>&1 && timeout 120 $BENCH_CMD 2>&1 || $BENCH_CMD 2>&1)
BENCH_SUCCEDS=$(echo "$BENCH_OUTPUTS" | grep "Succed")
if [ -z "$BENCH_SUCCEDS" ]
    then
        log_error "slock benchmark single key multi lock failed for the following reasons:
                   $BENCH_OUTPUTS
please run 'go run benchmarks/benchmark/benchmark.go' on your changes before committing."

    ps -ef | grep slock | grep -v grep | awk '{print $2}' | xargs kill -HUP > /dev/null 2>&1
    ps -ef | grep benchmark | grep -v grep | awk '{print $2}' | xargs kill -HUP > /dev/null 2>&1
    exit 1
fi

echo "
Benchmark Single Key Multi Lock Info:
$BENCH_OUTPUTS
"

BENCH_CMD="go run tools/benchmark/main.go --port=5659 --client=4 --conc=64 --count=100000 --timeout_flag=4096"
BENCH_OUTPUTS=$(which timeout > /dev/null 2>&1 && timeout 120 $BENCH_CMD 2>&1 || $BENCH_CMD 2>&1)
BENCH_SUCCEDS=$(echo "$BENCH_OUTPUTS" | grep "Succed")
if [ -z "$BENCH_SUCCEDS" ]
    then
        log_error "slock benchmark multi key multi lock require ack failed for the following reasons:
                   $BENCH_OUTPUTS
please run 'go run benchmarks/benchmark/benchmark.go' on your changes before committing."

    ps -ef | grep slock | grep -v grep | awk '{print $2}' | xargs kill -HUP > /dev/null 2>&1
    ps -ef | grep benchmark | grep -v grep | awk '{print $2}' | xargs kill -HUP > /dev/null 2>&1
    exit 1
fi

echo "
Benchmark Multi Key Single Lock Require Ack Info:
$BENCH_OUTPUTS
"

BENCH_CMD="go run tools/benchmark/main.go --port=5659 --client=4 --conc=16 --count=100000 --data_length=4096"
BENCH_OUTPUTS=$(which timeout > /dev/null 2>&1 && timeout 120 $BENCH_CMD 2>&1 || $BENCH_CMD 2>&1)
BENCH_SUCCEDS=$(echo "$BENCH_OUTPUTS" | grep "Succed")
if [ -z "$BENCH_SUCCEDS" ]
    then
        log_error "slock benchmark multi key multi lock with data failed for the following reasons:
                   $BENCH_OUTPUTS
please run 'go run benchmarks/benchmark/benchmark.go' on your changes before committing."

    ps -ef | grep slock | grep -v grep | awk '{print $2}' | xargs kill -HUP > /dev/null 2>&1
    ps -ef | grep benchmark | grep -v grep | awk '{print $2}' | xargs kill -HUP > /dev/null 2>&1
    exit 1
fi

echo "
Benchmark Multi Key Single Lock With Data Info:
$BENCH_OUTPUTS
"

BENCH_CMD="go run tools/benchmark/main.go --port=5659 --client=4 --conc=64 --count=50000  --timeout_flag=4096 --data_length=4096"
BENCH_OUTPUTS=$(which timeout > /dev/null 2>&1 && timeout 120 $BENCH_CMD 2>&1 || $BENCH_CMD 2>&1)
BENCH_SUCCEDS=$(echo "$BENCH_OUTPUTS" | grep "Succed")
if [ -z "$BENCH_SUCCEDS" ]
    then
        log_error "slock benchmark multi key multi lock require ack and with data failed for the following reasons:
                   $BENCH_OUTPUTS
please run 'go run benchmarks/benchmark/benchmark.go' on your changes before committing."

    ps -ef | grep slock | grep -v grep | awk '{print $2}' | xargs kill -HUP > /dev/null 2>&1
    ps -ef | grep benchmark | grep -v grep | awk '{print $2}' | xargs kill -HUP > /dev/null 2>&1
    exit 1
fi

echo "
Benchmark Multi Key Single Require Ack And Lock With Data Info:
$BENCH_OUTPUTS
"

BENCH_CMD="go run tools/benchmark/main.go --client=4 --conc=16 --count=200000"
BENCH_OUTPUTS=$(which timeout > /dev/null 2>&1 && timeout 120 $BENCH_CMD 2>&1 || $BENCH_CMD 2>&1)
BENCH_SUCCEDS=$(echo "$BENCH_OUTPUTS" | grep "Succed")
if [ -z "$BENCH_SUCCEDS" ]
    then
        log_error "slock benchmark transparency multi key single lock failed for the following reasons:
                   $BENCH_OUTPUTS
please run 'go run benchmarks/benchmark/benchmark.go' on your changes before committing."

    ps -ef | grep slock | grep -v grep | awk '{print $2}' | xargs kill -HUP > /dev/null 2>&1
    ps -ef | grep benchmark | grep -v grep | awk '{print $2}' | xargs kill -HUP > /dev/null 2>&1
    exit 1
fi

echo "
Benchmark Transparency Multi Key Single Lock Info:
$BENCH_OUTPUTS
"

BENCH_CMD="go run tools/benchmark/main.go --port=5659 --mode=stream --client=4 --conc=16 --count=1000000"
BENCH_OUTPUTS=$(which timeout > /dev/null 2>&1 && timeout 120 $BENCH_CMD 2>&1 || $BENCH_CMD 2>&1)
BENCH_SUCCEDS=$(echo "$BENCH_OUTPUTS" | grep "Succed")
if [ -z "$BENCH_SUCCEDS" ]
    then
        log_error "slock benchmark parallel lock failed for the following reasons:
                   $BENCH_OUTPUTS
please run 'go run benchmarks/benchmark/benchmark.go' on your changes before committing."

    ps -ef | grep slock | grep -v grep | awk '{print $2}' | xargs kill -HUP > /dev/null 2>&1
    ps -ef | grep benchmark | grep -v grep | awk '{print $2}' | xargs kill -HUP > /dev/null 2>&1
    exit 1
fi

echo "
Benchmark Parallel Lock Info:
$BENCH_OUTPUTS
"

REDIS_BENCHMARK_CMD=`which redis-benchmark`
if [ -x $REDIS_BENCHMARK_CMD ]; then
  echo "Redis Benchmark Info:"
  timeout 15s $REDIS_BENCHMARK_CMD -h 127.0.0.1 -p 5659 -n 100000 -c 64 -t set,get,incr

  echo "Redis Benchmark Transparency Info:"
  timeout 15s $REDIS_BENCHMARK_CMD -h 127.0.0.1 -p 5658 -n 50000 -c 64 -t set,get,incr
fi

sleep 3
SLOCK_INFOS=$(slock info --port=5658 2>&1)
echo "Slock 5658 Info:
$SLOCK_INFOS
"

SLOCK_INFOS=$(slock info --port=5657 2>&1)
echo "Slock 5657 Info:
$SLOCK_INFOS
"

SLOCK_INFOS=$(slock info --port=5659 2>&1)
echo "Slock 5659 Info:
$SLOCK_INFOS
"

ps aux | grep slock | grep -v grep
echo

ps -ef | grep slock | grep -v grep | awk '{print $2}' | xargs kill -HUP > /dev/null 2>&1
ps -ef | grep benchmark | grep -v grep | awk '{print $2}' | xargs kill -HUP > /dev/null 2>&1

sleep 2

echo "Slock Log:"
cat $LOG_FILENAME | grep -v protocol | grep -v connection | grep -v "Transparency client"
echo

exit 0