如何在BigQuery UDF中使用平方根函数？JS sqrt()报错排查

你遇到的ReferenceError: sqrt is not defined其实是个很常见的小坑——在BigQuery的JavaScript运行环境里，平方根函数不是全局可直接调用的。JavaScript本身的平方根功能是挂在Math对象下的方法，得写成Math.sqrt(x)才行，直接写sqrt(x)的话，系统会认为你在调用一个没定义的函数，自然就报错了。

另外还有个细节要提醒你：你定义的UDF返回类型是INT64，但Math.sqrt()返回的是浮点数（比如算25的平方根会得到5.0而不是5）。虽然BigQuery会自动把这个浮点数转成整数，但如果输入的数不是完全平方数（比如10），结果会被直接截断成整数（比如3）。如果需要保留小数精度的话，建议把返回类型改成FLOAT64。

修正后的代码示例 #

场景1：只处理完全平方数，保持返回INT64 #

CREATE TEMPORARY FUNCTION sqt(x INT64) 
RETURNS INT64 
LANGUAGE js AS """
  return Math.sqrt(x);
""";

WITH table1 AS(
  SELECT 25 AS x UNION ALL 
  SELECT 100 AS x UNION ALL 
  SELECT 625 AS x
)
SELECT x, sqt(x) AS square_root FROM table1;

场景2：需要保留小数精度，返回FLOAT64 #

CREATE TEMPORARY FUNCTION sqt(x INT64) 
RETURNS FLOAT64 
LANGUAGE js AS """
  return Math.sqrt(x);
""";

WITH table1 AS(
  SELECT 25 AS x UNION ALL 
  SELECT 100 AS x UNION ALL 
  SELECT 625 AS x UNION ALL
  SELECT 10 AS x -- 加个非完全平方数测试
)
SELECT x, sqt(x) AS square_root FROM table1;

额外小技巧 #

如果遇到非完全平方数时，你想返回最接近的整数（四舍五入），可以用Math.round()把结果包起来：

CREATE TEMPORARY FUNCTION sqt(x INT64) 
RETURNS INT64 
LANGUAGE js AS """
  return Math.round(Math.sqrt(x));
""";

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