Alasql UDF在分组聚合中的正确实践：解决undefined参数问题

interface Cat {
  name: string
  age: number
  color: string
}

const cats: Cat[] = [
  { name: 'Whiskers', age: 3, color: 'white' },
  { name: 'Shadow', age: 5, color: 'black' },
  { name: 'Mittens', age: 2, color: 'white' },
  { name: 'Oreo', age: 4, color: 'black' },
];

// 尝试定义一个自定义求和函数
import { sum } from 'lodash';
alasql.fn.mySum = function(xs) {
  // 在此处，如果缺少return，Alasql可能无法正确识别并传递聚合数据
  sum(xs); // 即使lodash.sum(xs)执行了，函数本身也未返回任何值
};

const sql = `
  SELECT color, mySum(age) AS totalAge
  FROM ?
  GROUP BY color
`;

// 执行查询时，mySum函数内部的xs参数会是undefined
// 这导致lodash.sum(undefined)失败或返回NaN
alasql(sql, [cats]).then(res => {
  console.log("不正确的UDF结果:", res);
});

import { sum } from 'lodash';

// 正确定义一个自定义求和函数
alasql.fn.mySum = function(xs) {
  // 关键：明确使用return语句返回聚合结果
  return sum(xs);
};

const sql = `
  SELECT color, mySum(age) AS totalAge
  FROM ?
  GROUP BY color
`;

// 执行查询
alasql(sql, [cats]).then(res => {
  console.log("正确的UDF结果:", res);
});

import alasql from 'alasql';
import { sum } from 'lodash'; // 引入lodash的sum函数

// 1. 定义数据模型
interface Cat {
  name: string;
  age: number;
  color: string;
}

// 2. 准备样本数据
const cats: Cat[] = [
  { name: 'Whiskers', age: 3, color: 'white' },
  { name: 'Shadow', age: 5, color: 'black' },
  { name: 'Mittens', age: 2, color: 'white' },
  { name: 'Oreo', age: 4, color: 'black' },
  { name: 'Ginger', age: 6, color: 'orange' },
  { name: 'Smokey', age: 1, color: 'black' },
];

// 3. 注册正确的用户自定义聚合函数
alasql.fn.mySum = function(xs: number[]): number {
  if (!Array.isArray(xs)) {
    console.warn("mySum received non-array input:", xs);
    return 0; // 或者抛出错误，根据业务需求处理
  }
  return sum(xs);
};

// 4. 构建SQL查询
const sqlQuery = `
  SELECT 
    color, 
    mySum(age) AS totalAgeByColor,
    COUNT(*) AS numberOfCats
  FROM ?
  GROUP BY color
  ORDER BY totalAgeByColor DESC
`;

// 5. 执行查询并打印结果
alasql(sqlQuery, [cats]).then((results: any[]) => {
  console.log("--- Alasql UDF 分组聚合结果 ---");
  console.table(results);
  /*
  预期输出 (大致):
  [
    { color: 'black', totalAgeByColor: 10, numberOfCats: 3 },
    { color: 'orange', totalAgeByColor: 6, numberOfCats: 1 },
    { color: 'white', totalAgeByColor: 5, numberOfCats: 2 }
  ]
  */
}).catch((error: any) => {
  console.error("Alasql 查询执行失败:", error);
});

// 6. 演示不正确的UDF（仅作对比，实际开发中应避免）
alasql.fn.myBrokenSum = function(xs: number[]): number {
  sum(xs); // 缺少return
  return 0; // 即使这里加了return，如果sum(xs)在之前执行，但没有被return，也会有问题
};

const brokenSqlQuery = `
  SELECT color, myBrokenSum(age) AS totalAgeByColor
  FROM ?
  GROUP BY color
`;

alasql(brokenSqlQuery, [cats]).then((results: any[]) => {
  console.log("\n--- 不正确的UDF 分组聚合结果 (可能不符合预期或出错) ---");
  console.table(results);
  // 实际结果可能显示totalAgeByColor为0 (因为myBrokenSum最后返回0) 或 NaN (如果lodash.sum(undefined)返回NaN)
  // 如果alasql内部机制在检测到UDF无返回值时，直接传递undefined给xs，则lodash.sum(undefined)会是NaN
}).catch((error: any) => {
  console.error("不正确的Alasql 查询执行失败:", error);
});