moobench/frameworks/OpenTelemetry-python/functions.sh at 8f08af774aaa1f1d2bafe669d9981e0fb5f2a31b · FeritBaturAltinn/moobench · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
export RECURSION_DEPTH="${RECURSION_DEPTH:-10}"

# helper to inject filename
function updateConfigFilename {
    local filename=$1
    grep -v "output_filename" "$CONFIG_TEMPLATE" > "$CONFIG_FILE"
    echo "output_filename = $filename" >> "$CONFIG_FILE"
}

function get_os_path {
    local raw_path=$1
    if command -v cygpath &>/dev/null; then cygpath -w "$raw_path"; else echo "$raw_path"; fi
}

function runNoInstrumentation {
    local k=$1
    local i=$2

    # use the global RESULTS_DIR directly (Fixes the /output read-only error)
    local RAW_CSV="${RAWFN}-${i}-${RECURSION_DEPTH}-${k}.csv"
    local CSV_FILE=$(get_os_path "$RAW_CSV")
    local LOG_FILE="${RESULTS_DIR}/output-raw-${i}-${RECURSION_DEPTH}-${k}.txt"

    echo " # Running Config $k: ${TITLE[$k]} (Iter $i)"

    updateConfigFilename "$CSV_FILE"
    export ENABLE_OTEL="false"

    python3 "$MOOBENCH_BIN_PY" "$CONFIG_FILE" > "$LOG_FILE" 2>&1
}

function runOpenTelemetryNoExport {
    local k=$1
    local i=$2

    local RAW_CSV="${RAWFN}-${i}-${RECURSION_DEPTH}-${k}.csv"
    local CSV_FILE=$(get_os_path "$RAW_CSV")
    local LOG_FILE="${RESULTS_DIR}/output-raw-${i}-${RECURSION_DEPTH}-${k}.txt"

    echo " # Running Config $k: ${TITLE[$k]} (Iter $i)"
    updateConfigFilename "$CSV_FILE"

    export ENABLE_OTEL="true"
    export OTEL_TRACES_EXPORTER="none"
    export OTEL_METRICS_EXPORTER="none"
    export OTEL_LOGS_EXPORTER="none"

    python3 "$MOOBENCH_BIN_PY" "$CONFIG_FILE" > "$LOG_FILE" 2>&1
}

function runOpenTelemetryZipkin {
    local k=$1
    local i=$2

    local RAW_CSV="${RAWFN}-${i}-${RECURSION_DEPTH}-${k}.csv"
    local CSV_FILE=$(get_os_path "$RAW_CSV")
    local LOG_FILE="${RESULTS_DIR}/output-raw-${i}-${RECURSION_DEPTH}-${k}.txt"

    startZipkin
    echo " # Running Config $k: ${TITLE[$k]} (Iter $i)"
    updateConfigFilename "$CSV_FILE"

    export ENABLE_OTEL="true"
    export OTEL_SERVICE_NAME="moobench-python"
    export OTEL_TRACES_EXPORTER="zipkin"
    export OTEL_EXPORTER_ZIPKIN_ENDPOINT="http://localhost:9411/api/v2/spans"
    export OTEL_METRICS_EXPORTER="none"
    export OTEL_LOGS_EXPORTER="none"

    python3 "$MOOBENCH_BIN_PY" "$CONFIG_FILE" > "$LOG_FILE" 2>&1

    stopBackgroundProcess
}

function executeBenchmark {
   for index in $MOOBENCH_CONFIGURATIONS
   do
      case $index in
         0) runNoInstrumentation 0 $i ;;
         1) runOpenTelemetryNoExport 1 $i ;;
         2) runOpenTelemetryZipkin 2 $i ;;
      esac
      sleep 1
   done
}