Files
ranking/backend/scripts/bootstrap_ranking.sh
2025-12-12 22:44:12 -03:00

110 lines
3.2 KiB
Bash
Executable File

#!/usr/bin/env bash
set -euo pipefail
# Escolhe binário do Python (funciona no host ou dentro do container)
if [ -x "./venv/bin/python" ]; then
VENV_PY="./venv/bin/python"
elif command -v python3 >/dev/null 2>&1; then
VENV_PY="python3"
else
VENV_PY="python"
fi
# Configs
DSN_CANDIDATES=()
if [ -n "${ORACLE_LOCAL_DSN:-}" ]; then
DSN_CANDIDATES+=("${ORACLE_LOCAL_DSN}")
fi
DSN_CANDIDATES+=("127.0.0.1:1521/XEPDB1" "oracle18c:1521/XEPDB1" "mqapilc_oracle18c:1521/XEPDB1")
API_URL=${API_URL:-"http://localhost:8000"}
echo "[1/5] Aguardando Oracle..."
DSN_HOST=""
for candidate in "${DSN_CANDIDATES[@]}"; do
echo " Testando ${candidate}..."
if $VENV_PY - <<PY >/dev/null 2>&1
import cx_Oracle
conn=cx_Oracle.connect("local123","local123","${candidate}")
cur=conn.cursor();cur.execute("select 1 from dual");cur.fetchone()
cur.close();conn.close()
PY
then
DSN_HOST="${candidate}"
break
fi
sleep 2
done
if [ -z "$DSN_HOST" ]; then
echo "ERRO: não foi possível conectar ao Oracle em nenhum dos DSN candidatos."
exit 1
fi
echo "Oracle OK em ${DSN_HOST}."
echo "[2/5] Conferindo contagem atual..."
COUNT=$($VENV_PY - <<PY
import cx_Oracle
conn=cx_Oracle.connect("local123","local123","${DSN_HOST}")
cur=conn.cursor();cur.execute("select count(*) from tb_ranking_consultor")
print(cur.fetchone()[0])
cur.close();conn.close()
PY
)
echo " Registros na TB_RANKING_CONSULTOR: ${COUNT}"
MAX_B=$($VENV_PY - <<PY
import cx_Oracle
conn=cx_Oracle.connect("local123","local123","${DSN_HOST}")
cur=conn.cursor();cur.execute("select nvl(max(componente_b),0) from tb_ranking_consultor")
print(cur.fetchone()[0])
cur.close();conn.close()
PY
)
if [ "${COUNT}" -eq 0 ]; then
echo "[3/5] Rodando job de ranking (A/C/D)..."
curl -s -X POST "${API_URL}/api/v1/ranking/processar?limpar=false" >/dev/null
# Espera concluir
while true; do
STATUS=$(curl -s "${API_URL}/api/v1/ranking/status")
running=$($VENV_PY - <<'PY'
import os, json
data=json.loads(os.environ["STATUS"])
print(data.get("running", False))
PY
STATUS="$STATUS")
echo " Status: ${STATUS}"
if [ "$running" != "True" ] && [ "$running" != "true" ]; then
break
fi
sleep 20
done
NEED_B="yes"
elif [ "${MAX_B}" -eq 0 ]; then
echo "[3/5] Tabela já tem dados, mas componente B nunca foi preenchido (max=0)."
NEED_B="yes"
else
echo "[3/5] Tabela já possui dados e componente B preenchido (max=${MAX_B}). Pulando cálculo pesado."
NEED_B="no"
fi
if [ "${NEED_B}" = "yes" ]; then
echo "[4/5] Rodando Componente B (PPG)..."
ORACLE_LOCAL_DSN=${DSN_HOST} PYTHONUNBUFFERED=1 $VENV_PY scripts/popular_componente_b.py
else
echo "[4/5] Componente B já presente, sem necessidade de recálculo."
fi
echo "[5/5] Contagens finais:"
$VENV_PY - <<PY
import cx_Oracle
conn=cx_Oracle.connect("local123","local123","${DSN_HOST}")
cur=conn.cursor()
cur.execute("select count(*) from tb_ranking_consultor"); print("total",cur.fetchone()[0])
cur.execute("select count(*) from tb_ranking_consultor where componente_b>0"); print("B>0",cur.fetchone()[0])
cur.execute("select max(componente_b) from tb_ranking_consultor"); print("max B",cur.fetchone()[0])
cur.execute("select max(pontuacao_total) from tb_ranking_consultor"); print("max total",cur.fetchone()[0])
cur.close();conn.close()
PY
echo "Pronto."