以下是一个示例PostgreSQL函数,使用IF语句并返回新插入记录的ID:
CREATE OR REPLACE FUNCTION insert_and_return_id(
p_column1_type p_column1_datatype,
p_column2_type p_column2_datatype
) RETURNS INTEGER AS
$$
DECLARE
new_id INTEGER;
BEGIN
-- 插入新记录
INSERT INTO your_table_name (column1, column2)
VALUES (p_column1_type, p_column2_type)
RETURNING id INTO new_id;
-- 使用IF语句判断插入是否成功
IF new_id IS NOT NULL THEN
RETURN new_id;
ELSE
RETURN -1; -- 或者返回其他错误代码
END IF;
END;
$$
LANGUAGE plpgsql;
在上面的代码中,你需要将p_column1_type
和p_column2_type
替换为你的实际列名和数据类型,并将your_table_name
替换为你的实际表名。
要使用这个函数,你可以调用它并传递相应的参数:
SELECT insert_and_return_id('column1_value', 'column2_value');
此函数将尝试将新记录插入到表中,并返回新插入记录的ID。如果插入成功,它将返回ID值;否则,它将返回-1或其他错误代码。