-- Проверки качества данных для tickets

DO $$
DECLARE
    v_batch_id      TEXT      := '{{ run_id }}'::text;
    v_prev_ts       TIMESTAMP;
    v_source_count  BIGINT;
    v_stg_count     BIGINT;
    v_orphan_count  BIGINT;
    v_null_count    BIGINT;
    v_dup_count     BIGINT;
    v_empty_name_count BIGINT;
BEGIN
    -- Опорная метка: максимум src_created_at_ts среди предыдущих батчей
    SELECT max(src_created_at_ts)
    INTO v_prev_ts
    FROM stg.tickets
    WHERE batch_id <> v_batch_id
        OR batch_id IS NULL;

    -- Количество в источнике (новые билеты в том же окне инкремента, что и загрузка)
    SELECT COUNT(*)
    INTO v_source_count
    FROM stg.tickets_ext AS t
    JOIN stg.bookings_ext AS b ON t.book_ref = b.book_ref
    WHERE b.book_date > COALESCE(v_prev_ts, TIMESTAMP '1900-01-01 00:00:00');

    IF v_source_count = 0 THEN
        -- Пустое окно инкремента допустимо: новых данных может не быть.
        -- В этом случае ожидаем, что в текущем batch_id тоже 0 строк.
        SELECT COUNT(*)
        INTO v_stg_count
        FROM stg.tickets
        WHERE batch_id = v_batch_id;

        IF v_stg_count <> 0 THEN
            RAISE EXCEPTION
                'DQ FAILED: источник tickets_ext за окно инкремента пустой, но в stg.tickets есть строки текущего batch_id (batch_id=%): %',
                v_batch_id,
                v_stg_count;
        END IF;

        RAISE NOTICE
            'В источнике tickets_ext нет строк для окна инкремента (book_date > %). Пропускаем DQ проверки (batch_id=%).',
            COALESCE(v_prev_ts, TIMESTAMP '1900-01-01 00:00:00'),
            v_batch_id;
        RETURN;
    END IF;

    -- Количество в STG (текущий батч)
    SELECT COUNT(*)
    INTO v_stg_count
    FROM stg.tickets
    WHERE batch_id = v_batch_id;

    IF v_source_count <> v_stg_count THEN
        RAISE EXCEPTION
            'DQ FAILED: несовпадение количества билетов. Источник: %, STG (batch_id=%): %',
            v_source_count,
            v_batch_id,
            v_stg_count;
    END IF;

    -- Проверка ссылочной целостности: все tickets должны иметь соответствующие bookings в этом же STG
    -- Используем LEFT JOIN вместо NOT EXISTS для лучшей производительности на больших объёмах
    SELECT COUNT(*)
    INTO v_orphan_count
    FROM stg.tickets AS t
    LEFT JOIN stg.bookings AS b ON t.book_ref = b.book_ref
    WHERE t.batch_id = v_batch_id
        AND b.book_ref IS NULL;

    IF v_orphan_count <> 0 THEN
        RAISE EXCEPTION
            'DQ FAILED: найдены tickets без соответствующих bookings (batch_id=%): %',
            v_batch_id,
            v_orphan_count;
    END IF;

    -- Проверка обязательных полей
    SELECT COUNT(*)
    INTO v_null_count
    FROM stg.tickets AS t
    WHERE t.batch_id = v_batch_id
        AND (t.ticket_no IS NULL OR t.book_ref IS NULL);

    IF v_null_count <> 0 THEN
        RAISE EXCEPTION
            'DQ FAILED: найдены tickets с NULL в обязательных полях (batch_id=%): %',
            v_batch_id,
            v_null_count;
    END IF;

    -- Проверка на дубликаты ticket_no
    SELECT COUNT(*) - COUNT(DISTINCT ticket_no)
    INTO v_dup_count
    FROM stg.tickets AS t
    WHERE t.batch_id = v_batch_id;

    IF v_dup_count <> 0 THEN
        RAISE EXCEPTION
            'DQ FAILED: найдены дубликаты ticket_no (batch_id=%): %',
            v_batch_id,
            v_dup_count;
    END IF;

    -- Проверка на пустые passenger_name
    SELECT COUNT(*)
    INTO v_empty_name_count
    FROM stg.tickets AS t
    WHERE t.batch_id = v_batch_id
        AND (t.passenger_name IS NULL OR t.passenger_name = '');

    IF v_empty_name_count <> 0 THEN
        RAISE EXCEPTION
            'DQ FAILED: найдены tickets с пустым именем пассажира (batch_id=%): %',
            v_batch_id,
            v_empty_name_count;
    END IF;

    RAISE NOTICE
        'DQ PASSED: tickets ок (batch_id=%): source=% stg=%',
        v_batch_id,
        v_source_count,
        v_stg_count;
END $$;
