sql >> Databasteknik >  >> RDS >> PostgreSQL

Generisk utlösare för att begränsa infogningar baserat på antal

Jag har gjort en liknande typ av generiska triggers. Den svåraste delen är att få värdeinmatningen i NEW post baserat på kolumnnamnet.

Jag gör det på följande sätt:

  • konvertera NEW data till array;
  • hitta attnum i kolumnen och använd den som ett index för arrayen.

Det här tillvägagångssättet fungerar så länge det inte finns några kommatecken i data :( Jag vet inte om andra sätt att konvertera NEW eller OLD variabler in i arrayen av värden.

Följande funktion kan hjälpa:

CREATE OR REPLACE FUNCTION impose_maximum() RETURNS trigger AS $impose_maximum$
DECLARE
  _sql  text;
  _cnt  int8;
  _vals text[];
  _anum int4;
  _im   record;

BEGIN
 _vals := string_to_array(translate(trim(NEW::text), '()', ''), ',');

 FOR _im IN SELECT * FROM imposed_maximums WHERE table_name = TG_TABLE_NAME LOOP
  SELECT attnum INTO _anum FROM pg_catalog.pg_attribute a
    JOIN pg_catalog.pg_class t ON t.oid = a.attrelid
   WHERE t.relkind = 'r' AND t.relname = TG_TABLE_NAME
     AND NOT a.attisdropped AND a.attname = _im.column_group;

  _sql := 'SELECT count('||quote_ident(_im.column_count)||')'||
          ' FROM '||quote_ident(_im.table_name)||
          ' WHERE '||quote_ident(_im.column_group)||' = $1';

  EXECUTE _sql INTO _cnt USING _vals[_anum];

  IF _cnt > CAST(_im.max_size AS int8) THEN
    RAISE EXCEPTION 'Maximum of % hit for column % in table %(%=%)',
      _im.max_size, _im.column_count,
      _im.table_name, _im.column_group, _vals[_anum];
  END IF;
 END LOOP;

 RETURN NEW;
END; $impose_maximum$ LANGUAGE plpgsql;

Denna funktion kommer att kontrollera alla villkor som definierats för en given tabell.



  1. Hur man itererar genom PostgreSQL jsonb-matrisvärden i syfte att matcha i en fråga

  2. Django-enhetstest misslyckas för flera Postgres-scheman

  3. Den här sidan kan inte nås efter att du har flyttat WordPress-sidan

  4. Hur kan jag ha flera vanliga tabelluttryck i en enda SELECT-sats?