154_account_spark_shadow.sql 1.7 KB

123456789101112131415161718192021222324252627282930
  1. -- 154_account_spark_shadow.sql
  2. ALTER TABLE accounts
  3. ADD COLUMN IF NOT EXISTS parent_account_id BIGINT,
  4. ADD COLUMN IF NOT EXISTS quota_dimension VARCHAR(20) NOT NULL DEFAULT 'global';
  5. -- 幂等加约束:维度合法 + 禁自指 + parent⟺非global 维度一致(评审 P1-d)
  6. DO $$ BEGIN
  7. IF NOT EXISTS (SELECT 1 FROM pg_constraint WHERE conname = 'chk_accounts_quota_dimension') THEN
  8. ALTER TABLE accounts ADD CONSTRAINT chk_accounts_quota_dimension
  9. CHECK (quota_dimension IN ('global','spark')) NOT VALID;
  10. END IF;
  11. IF NOT EXISTS (SELECT 1 FROM pg_constraint WHERE conname = 'chk_accounts_parent_dimension') THEN
  12. ALTER TABLE accounts ADD CONSTRAINT chk_accounts_parent_dimension
  13. CHECK ((parent_account_id IS NULL AND quota_dimension = 'global')
  14. OR (parent_account_id IS NOT NULL AND quota_dimension <> 'global')) NOT VALID;
  15. END IF;
  16. IF NOT EXISTS (SELECT 1 FROM pg_constraint WHERE conname = 'chk_accounts_parent_not_self') THEN
  17. ALTER TABLE accounts ADD CONSTRAINT chk_accounts_parent_not_self
  18. CHECK (parent_account_id IS NULL OR parent_account_id <> id) NOT VALID;
  19. END IF;
  20. IF NOT EXISTS (SELECT 1 FROM pg_constraint WHERE conname = 'fk_accounts_parent_account_id') THEN
  21. ALTER TABLE accounts ADD CONSTRAINT fk_accounts_parent_account_id
  22. FOREIGN KEY (parent_account_id) REFERENCES accounts(id) ON DELETE RESTRICT NOT VALID;
  23. END IF;
  24. END $$;
  25. ALTER TABLE accounts VALIDATE CONSTRAINT chk_accounts_quota_dimension;
  26. ALTER TABLE accounts VALIDATE CONSTRAINT chk_accounts_parent_dimension;
  27. ALTER TABLE accounts VALIDATE CONSTRAINT chk_accounts_parent_not_self;
  28. ALTER TABLE accounts VALIDATE CONSTRAINT fk_accounts_parent_account_id;