提问人：auf keinsten 提问时间：12/13/2021 最后编辑：Jim Jonesauf keinsten 更新时间：12/13/2021 访问量：54
如何在Postgres-custom-function中参数化表和列，如果值存在，则选择PK，否则插入它并返回PK？
How do I parameterize table & column in a Postgres-custom-function, selecting PK if value exists, otherwise insert it and return PK anyways?
问：
尝试做我在标题中指定的事情，我已经让 upsert-functionalities 工作了，但是当我尝试参数化它时，我只是超出了我的深度，无法调试它。
我的疑问：
CREATE OR REPLACE FUNCTION custom_upsert(target_value_input text,
target_table_input text,
target_column_input text,
OUT pk_output int)
LANGUAGE plpgsql AS
$func$
BEGIN
LOOP
execute 'SELECT id '
' FROM ' || target_table_input ||
' WHERE ' || target_column_input || ' = ' || target_value_input ||
' INTO pk_output';
EXIT WHEN FOUND;
execute 'INSERT INTO ' || target_table_input || 'AS o ( ' || target_column_input || ' )'
' VALUES ( ' || target_value_input || ' ) '
' ON CONFLICT ( ' || target_column_input || ' ) DO NOTHING '
' RETURNING o.id'
' INTO pk_output';
EXIT WHEN FOUND;
END LOOP;
END
$func$;
现在，当我尝试使用该函数时，我得到：
ERROR: syntax error at or near "INTO"
LINE 1: ...module WHERE artifact_id = artifact_id_example_1 INTO pk_ou...
^
QUERY: SELECT id FROM maven_module WHERE artifact_id = artifact_id_example_1 INTO pk_output
CONTEXT: PL/pgSQL function custom_upsert(text,text,text) line 4 at EXECUTE
让我感到困惑的是，这种语法在非参数化版本中工作正常：
https://dbfiddle.uk/?rdbms=postgres_14&fiddle=765389a746d3a392bc646fbedb7ed3b3
我的参数化尝试：
https://dbfiddle.uk/?rdbms=postgres_14&fiddle=1bffab45d8a9587342a7c3253ea35fc8
https://dbfiddle.uk/?rdbms=postgres_14&fiddle=de6ba235aa21dae33b922f8fddac3b63
提前非常感谢您，第一次发帖，所以如果我在提问时有什么不同的事情，我很高兴得到反馈
编辑：这是我的函数调用：
-- should return pk of existing artifact_id
SELECT custom_upsert('artifact_id_example_1', 'maven_module', 'artifact_id');
-- should return pk of new artifact_id
SELECT custom_upsert('artifact_id_example_2', 'maven_module', 'artifact_id');
答：
0赞
Jim Jones
12/13/2021
#1
不要像那样连接字符串。函数 format（） 使你的生活更轻松（更安全），例如
EXECUTE format('INSERT INTO %1$I AS o (%2$I)
VALUES (%3$L) ON CONFLICT (%2$I) DO NOTHING RETURNING o.id',
target_table_input,
target_column_input,
target_value_input) INTO pk_output;
%I将用双引号括起来标识符，这在表或列区分大小写且包含特殊字符时非常方便。%L将用单引号括起来1$和 是调用中提供的变量位置，如果一个变量被多次使用，这将非常方便。2$3$format()
演示：db<>fiddle
评论
1赞
auf keinsten
12/13/2021
再次尝试，不仅是为了解决方案，还因为很好的解释，帮助了我很多
评论
execute 'INSERT INTO ' || target_column_input || ...target_column_inputSELECT id INTO pk_output FROM ...