sql >> Databasteknik >  >> RDS >> PostgreSQL

PostgreSQL-sekvens baserad på en annan kolumn

Inga problem! Vi kommer att göra två tabeller, things och stuff . stuff kommer att vara tabellen du beskriver i din fråga, och things är den det hänvisar till:

CREATE TABLE things (
    id serial primary key,
    name text
);

CREATE TABLE stuff (
    id integer references things,
    seq integer NOT NULL,
    notes text,
    primary key (id, seq)
);

Sedan ställer vi in ​​things upp med en trigger som skapar en ny sekvens varje gång en rad skapas:

CREATE FUNCTION make_thing_seq() RETURNS trigger
    LANGUAGE plpgsql
    AS $$
begin
  execute format('create sequence thing_seq_%s', NEW.id);
  return NEW;
end
$$;

CREATE TRIGGER make_thing_seq AFTER INSERT ON things FOR EACH ROW EXECUTE PROCEDURE make_thing_seq();

Nu kommer vi att sluta med thing_seq_1 , thing_seq_2 , etc, etc...

Nu ytterligare en trigger på stuff så att den använder rätt sekvens varje gång:

CREATE FUNCTION fill_in_stuff_seq() RETURNS trigger
    LANGUAGE plpgsql
    AS $$
begin
  NEW.seq := nextval('thing_seq_' || NEW.id);
  RETURN NEW;
end
$$;

CREATE TRIGGER fill_in_stuff_seq BEFORE INSERT ON stuff FOR EACH ROW EXECUTE PROCEDURE fill_in_stuff_seq();

Det kommer att säkerställa att när rader går in i stuff , id kolumnen används för att hitta rätt sekvens för att anropa nextval på.

Här är en demonstration:

test=# insert into things (name) values ('Joe');
INSERT 0 1
test=# insert into things (name) values ('Bob');
INSERT 0 1
test=# select * from things;
 id | name
----+------
  1 | Joe
  2 | Bob
(2 rows)

test=# \d
              List of relations
 Schema |     Name      |   Type   |  Owner
--------+---------------+----------+----------
 public | stuff         | table    | jkominek
 public | thing_seq_1   | sequence | jkominek
 public | thing_seq_2   | sequence | jkominek
 public | things        | table    | jkominek
 public | things_id_seq | sequence | jkominek
(5 rows)

test=# insert into stuff (id, notes) values (1, 'Keychain');
INSERT 0 1
test=# insert into stuff (id, notes) values (1, 'Pet goat');
INSERT 0 1
test=# insert into stuff (id, notes) values (2, 'Family photo');
INSERT 0 1
test=# insert into stuff (id, notes) values (1, 'Redundant lawnmower');
INSERT 0 1
test=# select * from stuff;
 id | seq |        notes
----+-----+---------------------
  1 |   1 | Keychain
  1 |   2 | Pet goat
  2 |   1 | Family photo
  1 |   3 | Redundant lawnmower
(4 rows)

test=#


  1. 4 Out-of-the-box SQL-datakonverteringsmetoder och användningsfall

  2. Konverteringen misslyckades när varchar-värdet 'simple' konverterades till datatyp int

  3. Anslut Metabase med MySQL för datautforskning

  4. Hur gör man en uppdatering + går med i PostgreSQL?