sql >> Databáze >  >> RDS >> PostgreSQL

Jak napsat omezení týkající se maximálního počtu řádků v postgresql?

Quassnoi má pravdu; spouštěč by byl nejlepší způsob, jak toho dosáhnout.

Zde je kód:

CREATE OR REPLACE FUNCTION enforce_photo_count() RETURNS trigger AS $$
DECLARE
    max_photo_count INTEGER := 10;
    photo_count INTEGER := 0;
    must_check BOOLEAN := false;
BEGIN
    IF TG_OP = 'INSERT' THEN
        must_check := true;
    END IF;

    IF TG_OP = 'UPDATE' THEN
        IF (NEW.owner != OLD.owner) THEN
            must_check := true;
        END IF;
    END IF;

    IF must_check THEN
        -- prevent concurrent inserts from multiple transactions
        LOCK TABLE photos IN EXCLUSIVE MODE;

        SELECT INTO photo_count COUNT(*) 
        FROM photos 
        WHERE owner = NEW.owner;

        IF photo_count >= max_photo_count THEN
            RAISE EXCEPTION 'Cannot insert more than % photos for each user.', max_photo_count;
        END IF;
    END IF;

    RETURN NEW;
END;
$$ LANGUAGE plpgsql;


CREATE TRIGGER enforce_photo_count 
    BEFORE INSERT OR UPDATE ON photos
    FOR EACH ROW EXECUTE PROCEDURE enforce_photo_count();

Zahrnul jsem zamykání tabulky, abych se vyhnul situacím, kdy by dvě souběžné tansakce započítaly uživateli fotografie, viděli, že aktuální počet je o 1 pod limitem, a poté se obě vloží, což by způsobilo překročení limitu o 1. Pokud se vás to netýká, bylo by nejlepší odstranit zamykání, protože se může stát úzkým hrdlem s mnoha vložkami/aktualizacemi.



  1. Úvod do auto_explain:Jak automaticky zaznamenávat pomalé plány dotazů Postgres

  2. Jak zvýšit výjimku v PL/SQL?

  3. Migrace sítě s nulovými výpadky s MySQL Galera Cluster pomocí Relay Node

  4. Výběr druhého řádku tabulky pomocí rownum