带For Loop的PostgreSQL mod_10_int函数转Snowflake SQL求助

原PostgreSQL函数示例（基于常见Luhn校验mod10实现） #

CREATE OR REPLACE FUNCTION mod_10_int(p_num integer)
RETURNS integer AS $$
DECLARE
    v_num text := CAST(p_num AS text);
    v_sum integer := 0;
    v_len integer := LENGTH(v_num);
    v_pos integer;
    v_digit integer;
BEGIN
    FOR v_pos IN REVERSE 1..v_len LOOP
        v_digit := CAST(SUBSTRING(v_num, v_pos, 1) AS integer);
        IF (v_len - v_pos) % 2 = 1 THEN
            v_digit := v_digit * 2;
            IF v_digit > 9 THEN
                v_digit := v_digit - 9;
            END IF;
        END IF;
        v_sum := v_sum + v_digit;
    END LOOP;
    RETURN (10 - (v_sum % 10)) % 10;
END;
$$ LANGUAGE plpgsql;

Snowflake 完整转换实现（修复循环问题） #

Snowflake的SQL UDF不支持循环逻辑，因此采用JavaScript UDF实现，贴合原函数的循环逻辑：

CREATE OR REPLACE FUNCTION mod_10_int(p_num INT)
RETURNS INT
LANGUAGE JAVASCRIPT
AS $$
    let numStr = p_num.toString();
    let sum = 0;
    const len = numStr.length;
    
    // 反向遍历数字字符串，对应原PostgreSQL的REVERSE循环
    for (let pos = len - 1; pos >= 0; pos--) {
        let digit = parseInt(numStr.charAt(pos));
        // 匹配原函数的奇偶位置判断逻辑
        if ((len - 1 - pos) % 2 === 1) {
            digit *= 2;
            if (digit > 9) digit -= 9;
        }
        sum += digit;
    }
    
    return (10 - (sum % 10)) % 10;
$$;

核心差异与修复说明 #

• 循环语法差异：PostgreSQL PL/pgSQL支持FOR ... IN REVERSE语法，Snowflake的JavaScript UDF采用原生JS的反向for循环实现相同逻辑。
• 索引转换：原函数用1-based字符串位置，JS中字符串是0-based，因此调整位置计算逻辑(len -1 - pos) %2 ===1，确保奇偶位判断和原函数完全一致。
• UDF类型选择：Snowflake SQL UDF不支持循环，必须用JavaScript/Python UDF或存储过程，这里选JS UDF是因为调用方式和原函数一致，无需额外存储过程调用步骤。

测试验证 #

用相同输入验证结果一致性：

-- 测试输入：79927398713
SELECT mod_10_int(79927398713); -- 预期输出：3

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