fix: ручной UPSERT вместо ON CONFLICT — не требует индекса (v1.2.39)
ON CONFLICT требовал уникальный индекс idx_runs_op_uid, который не всегда создавался (мультистейтмент SQL в psycopg2). → save_run падал молча → история пустая. Новый подход: UPDATE по op_uid, если rowcount==0 → INSERT. Не зависит от индексов, работает всегда.
This commit is contained in:
+1
-1
@@ -32,7 +32,7 @@ from routes.api_scenario_defs import bp_defs as api_scenario_defs_bp
|
||||
|
||||
# Версия — показывается в топбаре UI. Меняется при КАЖДОМ изменении кода.
|
||||
# Нужна для фильтрации истории (пользователь видит только записи своей версии).
|
||||
VERSION = "1.2.38"
|
||||
VERSION = "1.2.39"
|
||||
|
||||
# Flask-приложение с Jinja2-шаблонами из папки templates/
|
||||
app = Flask(__name__, template_folder="templates", static_folder="static")
|
||||
|
||||
@@ -84,12 +84,6 @@ ALTER TABLE runs ADD COLUMN IF NOT EXISTS app_version VARCHAR(16);
|
||||
ALTER TABLE runs ADD COLUMN IF NOT EXISTS scenario_run_id INTEGER;
|
||||
ALTER TABLE runs ADD COLUMN IF NOT EXISTS step_number INTEGER;
|
||||
ALTER TABLE runs ADD COLUMN IF NOT EXISTS instance_meta JSONB;
|
||||
|
||||
-- Уникальный индекс на op_uid (частичный — только не-NULL).
|
||||
-- Нужен для UPSERT: сценарий вызывает save_run дважды (RUNNING → OK),
|
||||
-- второй вызов обновляет существующую строку вместо INSERT дубликата.
|
||||
CREATE UNIQUE INDEX IF NOT EXISTS idx_runs_op_uid
|
||||
ON runs (op_uid) WHERE op_uid IS NOT NULL;
|
||||
"""
|
||||
|
||||
# ── DDL для scenario_runs (запуски сценариев) ──
|
||||
@@ -218,11 +212,6 @@ def init_db():
|
||||
cur.execute(SCHEMA_SQL)
|
||||
cur.execute(MIGRATION_SQL)
|
||||
cur.execute(SCENARIO_RUNS_SQL)
|
||||
|
||||
# Индекс для UPSERT в save_run — отдельным запросом для надёжности
|
||||
# (мультистейтмент SQL в psycopg2 может выполнить не все операторы)
|
||||
cur.execute("CREATE UNIQUE INDEX IF NOT EXISTS idx_runs_op_uid ON runs (op_uid) WHERE op_uid IS NOT NULL")
|
||||
|
||||
conn.commit()
|
||||
cur.close()
|
||||
print("[DB] Schema initialized", flush=True)
|
||||
|
||||
+21
-15
@@ -51,30 +51,36 @@ def save_run(client_id, stand, user_email, svc_id, svc_name, op_name, svc_op_id,
|
||||
|
||||
try:
|
||||
cur = conn.cursor()
|
||||
|
||||
# Ручной UPSERT (UPDATE → INSERT если не затронуто строк).
|
||||
# Не используем ON CONFLICT — не требует индекса, надёжнее.
|
||||
# Сценарий вызывает save_run дважды (RUNNING → OK) — второй вызов
|
||||
# обновляет существующую строку вместо создания дубликата.
|
||||
params_json = json.dumps(params) if params else None
|
||||
stages_json = json.dumps(stages) if stages else None
|
||||
meta_json = json.dumps(instance_meta) if instance_meta else None
|
||||
|
||||
if op_uid:
|
||||
cur.execute("""
|
||||
UPDATE runs SET
|
||||
status = %s, duration_sec = %s, error_log = %s,
|
||||
stages = %s, svc_name = %s, instance_meta = %s
|
||||
WHERE op_uid = %s
|
||||
""", (status, duration_sec, error_log, stages_json, svc_name, meta_json, op_uid))
|
||||
|
||||
if not op_uid or cur.rowcount == 0:
|
||||
# Нет op_uid ИЛИ UPDATE не затронул строк → INSERT
|
||||
cur.execute("""
|
||||
INSERT INTO runs (client_id, stand, user_email, svc_id, svc_name,
|
||||
op_name, svc_op_id, op_uid, instance_uid, display_name,
|
||||
status, duration_sec, error_log, params, stages, app_version,
|
||||
scenario_run_id, step_number, instance_meta)
|
||||
VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s)
|
||||
ON CONFLICT (op_uid) WHERE op_uid IS NOT NULL DO UPDATE SET
|
||||
status = EXCLUDED.status,
|
||||
duration_sec = EXCLUDED.duration_sec,
|
||||
error_log = EXCLUDED.error_log,
|
||||
stages = EXCLUDED.stages,
|
||||
svc_name = EXCLUDED.svc_name,
|
||||
instance_meta = EXCLUDED.instance_meta
|
||||
""", (
|
||||
client_id, stand, user_email, svc_id, svc_name, op_name, svc_op_id,
|
||||
op_uid, instance_uid, display_name, status, duration_sec,
|
||||
error_log,
|
||||
# params и stages — JSONB: сериализуем в JSON-строку
|
||||
json.dumps(params) if params else None,
|
||||
json.dumps(stages) if stages else None,
|
||||
app_version,
|
||||
scenario_run_id, step_number,
|
||||
# instance_meta — тоже JSONB
|
||||
json.dumps(instance_meta) if instance_meta else None,
|
||||
error_log, params_json, stages_json, app_version,
|
||||
scenario_run_id, step_number, meta_json,
|
||||
))
|
||||
conn.commit()
|
||||
cur.close()
|
||||
|
||||
Reference in New Issue
Block a user