36 lines
2.2 KiB
PL/PgSQL
36 lines
2.2 KiB
PL/PgSQL
SET search_path TO atlas, public;
|
|
|
|
-- 重试拥有独立代次；旧轮询或迟到结果不得影响新的生图请求。
|
|
ALTER TABLE media_ai_edits ADD COLUMN generation_attempt INTEGER NOT NULL DEFAULT 1 CHECK (generation_attempt >= 1);
|
|
ALTER TABLE media_ai_edits DROP CONSTRAINT media_ai_edits_suite_shape;
|
|
ALTER TABLE media_ai_edits ADD CONSTRAINT media_ai_edits_suite_shape CHECK (
|
|
(suite_id IS NULL AND suite_shot_id IS NULL AND suite_position IS NULL) OR
|
|
(suite_id IS NOT NULL AND product_id IS NOT NULL AND suite_shot_id IS NOT NULL AND suite_position IS NOT NULL AND suite_position BETWEEN 0 AND 19));
|
|
ALTER TABLE product_image_suites ADD CONSTRAINT image_suite_input_references CHECK (
|
|
coalesce(jsonb_typeof(input -> 'reference_asset_ids') = 'array' AND jsonb_array_length(input -> 'reference_asset_ids') BETWEEN 1 AND 8, false));
|
|
|
|
CREATE FUNCTION guard_image_suite_snapshot() RETURNS TRIGGER LANGUAGE plpgsql AS $$
|
|
BEGIN
|
|
IF NEW.input IS DISTINCT FROM OLD.input OR NEW.product_id IS DISTINCT FROM OLD.product_id
|
|
OR NEW.variant_id IS DISTINCT FROM OLD.variant_id OR NEW.purpose IS DISTINCT FROM OLD.purpose THEN
|
|
RAISE EXCEPTION '套图输入快照不可改写' USING ERRCODE = '23514';
|
|
END IF;
|
|
IF NEW.shots IS DISTINCT FROM OLD.shots AND (OLD.purpose <> 'plan' OR OLD.status NOT IN ('queued', 'running')) THEN
|
|
RAISE EXCEPTION '已固化的出图方案不可改写' USING ERRCODE = '23514';
|
|
END IF;
|
|
RETURN NEW;
|
|
END $$;
|
|
CREATE TRIGGER product_image_suite_snapshot BEFORE UPDATE ON product_image_suites
|
|
FOR EACH ROW EXECUTE FUNCTION guard_image_suite_snapshot();
|
|
|
|
CREATE FUNCTION guard_image_suite_reference_slot() RETURNS TRIGGER LANGUAGE plpgsql AS $$
|
|
BEGIN
|
|
IF NOT EXISTS (SELECT 1 FROM product_image_suites WHERE workspace_id = NEW.workspace_id AND id = NEW.suite_id
|
|
AND input -> 'reference_asset_ids' ->> NEW.position = NEW.asset_id::text) THEN
|
|
RAISE EXCEPTION '底图槽位与套图输入快照不一致' USING ERRCODE = '23514';
|
|
END IF;
|
|
RETURN NEW;
|
|
END $$;
|
|
CREATE TRIGGER product_image_suite_reference_slot BEFORE INSERT OR UPDATE ON product_image_suite_references
|
|
FOR EACH ROW EXECUTE FUNCTION guard_image_suite_reference_slot();
|