Вы можете просто вставить счетчик в таблицу компаний:
CREATE TABLE companies (
id SERIAL PRIMARY KEY,
name TEXT,
product_id INT DEFAULT 0
);
CREATE TABLE products (
company INT REFERENCES companies(id),
product_id INT,
PRIMARY KEY (company, product_id),
name TEXT
);
INSERT INTO companies (id, name) VALUES (1, 'Acme Corporation');
INSERT INTO companies (id, name) VALUES (2, 'Umbrella Corporation');
Затем используйте UPDATE ... RETURNING, чтобы получить следующий идентификатор продукта для данной компании:
> INSERT INTO products VALUES (1, (UPDATE companies SET product_id = product_id+1 WHERE id=$1 RETURNING product_id), 'Anvil');
ERROR: syntax error at or near "companies"
LINE 1: INSERT INTO products VALUES (1, (UPDATE companies SET produc...
^
О нет! Кажется, вы не можете (начиная с PostgreSQL 9.1) использовать UPDATE ... RETURNING в качестве подзапроса.
Хорошая новость в том, что это не проблема! Просто создайте хранимую процедуру, которая выполняет этап приращения / возврата:
CREATE FUNCTION next_product_id(company INT) RETURNS INT
AS $$
UPDATE companies SET product_id = product_id+1 WHERE id=$1 RETURNING product_id
$$ LANGUAGE 'sql';
Теперь вставка - кусок пирога:
INSERT INTO products VALUES (1, next_product_id(1), 'Anvil');
INSERT INTO products VALUES (1, next_product_id(1), 'Dynamite');
INSERT INTO products VALUES (2, next_product_id(2), 'Umbrella');
INSERT INTO products VALUES (1, next_product_id(1), 'Explosive tennis balls');
Обязательно используйте один и тот же идентификатор компании как в значении продукта, так и в аргументе next_product_id(company INT)
.