#!/usr/bin/env bash
# Quantifies what component scanning costs, by adding classes to the scanned package
# and changing nothing else.
#
#   baseline    : the application as committed
#   +5000 plain : 5000 classes with no annotations -- pure scan cost
#   +5000 @Component : the same 5000, now bean definitions -- scan + define + instantiate
#   +5000 @Component, lazy : lazy init removes the instantiation, not the scan
#
# Each variant is a full rebuild. Budget about ten minutes.
set -uo pipefail
source "$(dirname "$0")/env.sh"
cd "$ROOT"
RUNS="${RUNS:-3}"

measure() {   # measure <label> <extra java args...>
  local label="$1"; shift
  local started=() parse="" inst="" steps=""
  for i in $(seq 1 "$RUNS"); do
    "$ROOT/scripts/stop.sh"
    "$JAVA_HOME/bin/java" -Dstartup.tracking=buffering "$@" -jar "$JAR" > /tmp/scan-app.log 2>&1 &
    local pid=$!
    for _ in $(seq 1 240); do curl -fs -o /dev/null localhost:8080/actuator/health && break; sleep 0.5; done
    started+=("$(grep -ho 'in [0-9.]* seconds' /tmp/scan-app.log | head -1 | awk '{print $2}')")
    if [ "$i" = "1" ]; then
      curl -s 'localhost:8080/diag/startup?top=1' > /tmp/scan-diag.json
      read -r parse inst steps <<<"$(python3 - <<'PY'
import json
d = json.load(open('/tmp/scan-diag.json'))
p = {r['step']: r for r in d['phasesBySelfTime']}
print(p.get('spring.context.config-classes.parse', {}).get('selfMs', 0),
      p.get('spring.beans.instantiate', {}).get('selfMs', 0),
      d['recordedSteps'])
PY
)"
    fi
    kill -9 $pid 2>/dev/null; wait $pid 2>/dev/null
  done
  printf '%-26s | %-18s | %9s | %9s | %6s\n' \
    "$label" "$(IFS=,; echo "${started[*]}")" "$parse" "$inst" "$steps"
}

hdr() {
  printf '%-26s | %-18s | %9s | %9s | %6s\n' \
    "variant" "Started in (s)" "parse ms" "instMs" "steps"
  printf -- '---------------------------+--------------------+-----------+-----------+-------\n'
}

echo "runs per variant: $RUNS   (first run of each also queried for phase self times)"
echo
hdr

./scripts/gen-bulk.sh 0 plain >/dev/null 2>&1 || true
rm -rf src/main/java/com/ankurm/startup/bulk
mvn -B -q -DskipTests package > /tmp/scan-build.log 2>&1
measure "baseline"

./scripts/gen-bulk.sh 5000 plain > /dev/null
mvn -B -q -DskipTests package > /tmp/scan-build.log 2>&1
measure "+5000 plain classes"

./scripts/gen-bulk.sh 5000 component > /dev/null
mvn -B -q -DskipTests package > /tmp/scan-build.log 2>&1
measure "+5000 @Component"
measure "+5000 @Component, lazy" -Dspring.profiles.active=lazy

rm -rf src/main/java/com/ankurm/startup/bulk
mvn -B -q -DskipTests package > /tmp/scan-build.log 2>&1
echo
echo "bulk package removed; jar rebuilt at baseline."
true