DB2自定义函数如何支持接收行对象以适配多字段主键？

完全可行，DB2支持将**行类型（ROW Type）**作为函数输入参数，通过行对象可以提取任意数量的主键字段完成验证，完美适配多字段主键的场景。

改造思路与示例代码 #

1. 针对特定表的复合主键场景（推荐） #

如果你的验证逻辑是针对某张有复合主键的表，可以直接使用表的%ROWTYPE作为参数类型，这样能直接提取表的所有字段（包括主键）：

假设存在表EMP，复合主键为EMP_ID(VARCHAR(8))和DEPT_ID(INTEGER)，改造后的函数如下：

CREATE OR REPLACE FUNCTION CHECK_VALID(EMP_ROW EMP%ROWTYPE) 
RETURNS BOOLEAN 
LANGUAGE SQL 
SPECIFIC SQL23562233244578_EMP
RETURN 
  -- 示例：验证复合主键是否存在于EMP表中
  EXISTS (
    SELECT 1 FROM EMP 
    WHERE EMP_ID = EMP_ROW.EMP_ID AND DEPT_ID = EMP_ROW.DEPT_ID
  );

2. 通用型多主键场景 #

如果需要适配多种不同结构的复合主键，可以先自定义一个通用行类型，再基于该类型创建函数：

-- 先定义通用主键行类型，可根据实际业务扩展字段
CREATE TYPE GENERIC_PK_ROW AS ROW (
  PK1 VARCHAR(8),
  PK2 INTEGER,
  PK3 DATE
);

-- 基于通用行类型创建验证函数
CREATE OR REPLACE FUNCTION CHECK_VALID(PK_ROW GENERIC_PK_ROW) 
RETURNS BOOLEAN 
LANGUAGE SQL 
SPECIFIC SQL23562233244578_GENERIC
RETURN 
  -- 示例：验证所有主键字段非空
  PK_ROW.PK1 IS NOT NULL AND PK_ROW.PK2 IS NOT NULL AND PK_ROW.PK3 IS NOT NULL;

关键注意事项 #

• 函数重载：可以为不同表的行类型创建多个重载的CHECK_VALID函数，DB2会根据传入的行对象类型自动匹配对应的函数
• 类型兼容性：使用表%ROWTYPE时，传入的行对象必须来自对应表，否则会触发类型不匹配错误
• 逻辑适配：原单字段验证逻辑需要调整为多字段逻辑，比如联合主键的存在性检查、多字段格式校验等
• 权限配置：确保函数拥有访问对应表、行类型的必要权限

内容的提问来源于stack exchange，提问作者DB2fan