Как установить значение в имени столбца переменной?

Question

Как установить значение в имени столбца переменной? В некотором контексте я пишу функцию, которая будет использоваться в качестве триггера, который устанавливает столбец переменной в постоянное значение. Для того, чтобы использовать следующим образом:

CREATE TRIGGER always_6_trigger 
    BEFORE INSERT 
    ON table 
    FOR EACH ROW 
    EXECUTE PROCEDURE always_6('col1'); 

выше приведет следующих строк, имеющих все col1 значение 6. Так, например:

INSERT INTO table (col1, col2) VALUES (6, 2), (null, 9), (null, 10), (7, 2); 

приведет ли:

| col1 | col2 | 
--------------- 
| 6 | 2 | 
| 6 | 9 | 
| 6 | 10 | 
| 6 | 2 | 

Или если используется следующий пусковой механизм:

CREATE TRIGGER always_6_trigger 
    BEFORE INSERT 
    ON table 
    FOR EACH ROW 
    EXECUTE PROCEDURE always_6('col2'); 

И та же вставка:

INSERT INTO table (col1, col2) VALUES (6, 2), (null, 9), (null, 10), (7, 2); 

таблица будет выглядеть следующим образом:

| col1 | col2 | 
--------------- 
| 6 | 6 | 
| null | 6 | 
| null | 6 | 
| 7 | 6 | 

Как бы написать функцию always_6?

Редактировать: Чтобы лучше объяснить случай использования, постоянным значением будет current_setting('user_id') (или что-то подобное). И название столбца будет таким, как author_id и user_id. Мысль о том, что пользователь никогда не может добавить данные, которые не были их собственными.

Answer 1

Вы можете определить вашу функцию для производства dynamically generated SQL.

С помощью команды EXECUTE принимает строку в качестве входных данных и выполняет его как SQL, так что это будет выглядеть примерно так:

EXECUTE FORMAT('UPDATE mytable SET %I='constantvalue' WHERE condition', colname); 

Здесь я использовал функцию FORMAT подготовить строку со значением colname замещенного в котором будет указано имя столбца. condition - это некоторые действующие предложения WHERE для выбора записи для обновления.

Если значение colname может поступать из внешнего источника (то есть с предоставленных пользователем данных), тогда вам нужно быть очень осторожным, чтобы проверить его заранее, иначе вы можете создать вектор инъекции SQL.

Answer 2

Вы можете использовать динамический запрос для этого с условной проверкой переданного значения параметра входному параметру хранимой процедуры.

Answer 3

Я думаю, что я упростил задачу, основанную на первоначальном описании, но что-то вроде этой работы? Вы не можете передать параметр функции триггера, но вы упомянули, что параметр был результатом другой функции current_setting('user_id'), так что можно объединить две концепции вместе?

CREATE OR REPLACE FUNCTION always_6() 
    RETURNS trigger AS 
$BODY$ 
DECLARE 
    current_user_id varchar; 
BEGIN 

    current_user_id := current_setting('user_id'); 

    if current_user_id = 'test1' then 
    new.col_1 := 6; 
    elsif current_user_id = 'test2' then 
    new.col_2 := 6; 
    end if; 

    return NEW; 
END; 
$BODY$ 
    LANGUAGE plpgsql VOLATILE 
    COST 100;