如何用Snowflake JavaScript UDF统计数组中两个日期间的元素数量

原有代码核心错误点 #

• 数组下标起始错误：JavaScript数组下标从0开始，你设置i初始值为1，直接跳过了第一个元素"2017-01-01"，导致统计少算第一条数据。
• 初始计数错误：count初始值设为1，相当于默认存在1个符合条件的元素，就算遍历没有匹配项也会返回1，和统计逻辑完全不符，初始值应该设为0。
• 数组越界风险：for循环判断条件仅写了a[i]<=DATE2/a[i]<=DATE1，没有增加i < a.length的边界判断，当输入日期大于数组最后一个元素时，会出现越界访问undefined的问题，导致比较逻辑异常。
• 类型定义不合理：入参定义为String、返回值定义为double，和实际使用场景不匹配，入参用DATE、返回值用NUMBER更稳妥，避免字符串格式不统一导致的比较错误。

修复后代码 #

create or replace function workdays_between(date1 DATE, date2 DATE)
 RETURNS NUMBER
  LANGUAGE JAVASCRIPT
  as 
  $$
  // 日期转YYYY-MM-DD格式字符串和数组元素匹配
  const formatDate = (d) => d.toISOString().split('T')[0];
  const d1Str = formatDate(DATE1);
  const d2Str = formatDate(DATE2);
  const start = d1Str < d2Str ? d1Str : d2Str;
  const end = d1Str > d2Str ? d1Str : d2Str;
  const workdays = ["2017-01-01","2017-01-02","2017-01-03","2017-01-06","2017-01-09","2017-01-10"];
  let count = 0;

  for (let i = 0; i < workdays.length; i++) {
    const current = workdays[i];
    if (current > end) break; // 数组为升序，超过上限直接终止循环提升效率
    if (current >= start) count++;
  }
  // 兼容原有逻辑：date1大于date2时返回负数
  return d1Str < d2Str ? count : -count;
  $$

测试验证 #

可以用以下语句验证效果：

• 统计2017-01-01到2017-01-03的符合条件的元素：select workdays_between('2017-01-01'::DATE, '2017-01-03'::DATE); 返回值为3，符合预期
• 反向输入日期测试：select workdays_between('2017-01-03'::DATE, '2017-01-01'::DATE); 返回值为-3，和你原有逻辑一致

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