JS UDF, который возвращает структуру для стандартного SQL / BigQuery и создает два столбца - PullRequest
Новая
       15

JS UDF, который возвращает структуру для стандартного SQL / BigQuery и создает два столбца

0 голосов
/ 03 апреля 2019

Я пытаюсь написать пользовательскую функцию для BigQuery, используя Javascript, который возвращает структуру и генерирует два столбца: 

CREATE TEMP FUNCTION exampleFunction(exampleString STRING)
  RETURNS STRUCT<index INT64, latency INT64> LANGUAGE js AS
  LANGUAGE js AS 
"""
    var exampleStruct = {1:100, 2:200}
    return exampleStruct;
""";

Мой запрос будет выглядеть примерно так: 

SELECT
exampleCol,
exampleFunction(stringCol) -- use SELECT AS STRUCT somewhere here? with the aliases “First” and “Second”
FROM
[SOME DATATBASE HERE]

Я хочу, чтобы выходные данные exampleFunction(stringCol) генерировали два столбца (всего три столбца, если мы включим exampleCol). Например, если exampleCol дал нам «НЕЧТО», я бы хотел вернуть столбцы: «НЕЧТО» для exampleCol, 1 для «Первого» и 2 для «Второго». Это что-то, что возможно?

У меня проблемы с возвратом STRUCT из функции JS (не уверен, что мой синтаксис выключен) и с правильным запросом. Для запроса я хочу избежать запуска функции JavaScript дважды. Спасибо!

1 Ответ

1 голос
/ 03 апреля 2019

Ниже приведен пример для BigQuery Standard SQL 

#standardSQL
CREATE TEMP FUNCTION exampleFunction(exampleString STRING)
  RETURNS STRUCT<index INT64, latency INT64> 
  LANGUAGE js AS 
"""
    arr = exampleString.split(':');
    this.index = arr[0];
    this.latency = arr[1];
    return this;
""";
WITH `project.dataset.table` AS (
  SELECT 1 exampleCol, '10:100' stringCol UNION ALL
  SELECT 2, '20:200' UNION ALL
  SELECT 3, '30:456'
)
SELECT exampleCol, exampleFunction(stringCol).*
FROM `project.dataset.table`
-- ORDER BY exampleCol

с результатом 

Row exampleCol  index   latency  
1   1           10      100  
2   2           20      200  
3   3           30      456

Примечание: если вы хотите, чтобы столбцы связывались с First, Second - вы можете либо заменить index и latency с соответственно first, second как в примере ниже 

#standardSQL
CREATE TEMP FUNCTION exampleFunction(exampleString STRING)
  RETURNS STRUCT<first INT64, second INT64> 
  LANGUAGE js AS 
"""
    arr = exampleString.split(':');
    this.first = arr[0];
    this.second = arr[1];
    return this;
""";
SELECT exampleCol, exampleFunction(stringCol).*
FROM `project.dataset.table`

или вы можете использовать подход ниже 

#standardSQL
CREATE TEMP FUNCTION exampleFunction(exampleString STRING)
  RETURNS STRUCT<index INT64, latency INT64> 
  LANGUAGE js AS 
"""
    arr = exampleString.split(':');
    this.index = arr[0];
    this.latency = arr[1];
    return this;
""";
SELECT exampleCol, index AS first, latency AS second   
FROM (
  SELECT exampleCol, exampleFunction(stringCol).*
  FROM `project.dataset.table`
)

с результатом ниже в обоих случаях 

Row exampleCol  first   second   
1   1           10      100  
2   2           20      200  
3   3           30      456
...