benchmark_node.sh

#!/bin/bash

# Constants used along the program
readonly PROGNAME=$(basename $0)
readonly PROGDIR=$(readlink -m $(dirname $0))
readonly ARGS="$@"

start_benchmark() {
    local readonly NODE_ID=$1
    local readonly DATADIR=$2
    local readonly NETWORKID=$3
    local readonly PORT=$4
    local readonly IPC_PATH=$5
    local readonly BOOTNODES=$6
    local readonly KEYSTORE="keystore"
    local readonly JS_SCRIPT_PATH=$7
    local readonly OUTPUT_FILE=$8
    local readonly ROLE=$9
    local readonly TEST_TIMEOUT=${10}
    local readonly ETHASH_DIR=${11}
    local readonly ETHASH_CACHE_DIR="$ETHASH_DIR/cache"
    local readonly ETHASH_DAG_DIR="$ETHASH_DIR/dag"

    # if [[ "$ROLE" = "miner" ]]; then
    #    printf "Node is miner...\n"
    #    extra_option=" --mine --minerthreads 1 "
    # fi
    # js_cmd="js $JS_SCRIPT_PATH"
    # extra_option=$(eval echo $extra_option)
    # js_cmd=$(eval echo $js_option)

    local extra_option=""
    local js_cmd=""
    
    if [[ "$ROLE" = "miner" ]]; then
        printf "$ROLE is miner...\n"
        extra_option="--mine --minerthreads 1"
    fi
    
    if [[ "$ROLE" = "client" ]]; then
        printf "$ROLE is client...\n"
        js_cmd="js $JS_SCRIPT_PATH"
    fi
    
    
    js_cmd=$(eval echo $js_cmd)
    extra_option=$(eval echo $extra_option)

    printf "Node $NODE_ID"
    
    echo "$OUTPUT_FILE"
    nohup timeout -s SIGUSR1 "$TEST_TIMEOUT" ./geth_with_catch.sh \
        "$NODE_ID" \
        "--datadir" "$DATADIR" \
        "--keystore" "$KEYSTORE" \
        "--ipcpath" "$IPC_PATH" \
        "--port" "$PORT" \
        "--networkid" "$NETWORKID" \
        "--bootnodes" "$BOOTNODES" \
        "--metrics" \
        "--ethash.cachedir" "$ETHASH_CACHE_DIR" \
        "--ethash.dagdir" "$ETHASH_DAG_DIR" \
        "--ethash.dagsondisk" "1" \
        "$extra_option" \
        "$js_cmd" \
        >> $OUTPUT_FILE 2>&1 &
}


read_chainid() {
    grep -E -o '"chainId"\s*:\s*[0-9]+' conf/genesis_block.json | grep -E -o '[0-9]+'
}


main() { 
    # check arguments
    if [ $# -lt 5 ]; then
        printf "Usage: `basename "$0"` <role_list> <first-node-index> <bootnodes> <timeout-interval> <tx-interval>\n"
        exit 1
    fi
    
    local readonly FIRST_NODE_INDEX=$2
    local readonly ROLE_LIST="$1"
    local readonly NODES_AMOUNT=$(echo $ROLE_LIST | jq "length")
    
    local readonly BASE_ETHASH_DIR="$HOME/ethash"
    local readonly BASE_DATADIR="ethtest-datadir-"
    local readonly OUTPUT_DIR="logs"
    local readonly JS_SCRIPTS_DIR="js-scripts"
    local readonly BOOTNODES=$3
    local readonly TIMEOUT_INTERVAL=$4
    local readonly TX_INTERVAL=$5

    # start geth node in background
    NETWORKID=$(read_chainid)

    printf "Let's start $ROLE_LIST $NODES_AMOUNT nodes\n"

    # Get the run number to create different logs
    RUN=0 
    while [[ -a "$OUTPUT_DIR/node-0-$RUN.out" ]];
    do
        RUN=$((RUN+1))
    done

    for node in $(seq 0 $(($NODES_AMOUNT - 1))); do
        # build datadir and output file string
        local readonly DATADIR=$BASE_DATADIR$node
        local readonly ETHASH_DIR=$BASE_ETHASH_DIR$node
        local readonly OUTPUT_FILE="$OUTPUT_DIR/node-$node-$RUN.out"
        local readonly ROLE=$(echo $ROLE_LIST | jq -r ".[$node]")
        local readonly JS_SCRIPT_PATH="$JS_SCRIPTS_DIR/node-$node.js"
        local readonly IPC_PATH="$HOME/geth-$node.ipc"

        echo $ROLE > $OUTPUT_DIR/role-$node.out

        printf "Starting node $node with role $ROLE...\n"
        
        PORT=$((30300 + $node))

        printf "conf = {};
            conf.accountIndex = $(($FIRST_NODE_INDEX + $node));
            conf.txDelay = $TX_INTERVAL;
            " | cat - sendTransactions.js > "$JS_SCRIPT_PATH"

        start_benchmark \
            "$node" \
            "$DATADIR" \
            "$NETWORKID" \
            "$PORT" \
            "$IPC_PATH" \
            "$BOOTNODES" \
            "$JS_SCRIPT_PATH" \
            "$OUTPUT_FILE" \
            "$ROLE" \
            "$TIMEOUT_INTERVAL" \
            "$ETHASH_DIR" &
    
        # printf "Node started. Output in $OUTPUT_FILE\n"
    done
}


main $ARGS