Funções adicionais de tabela simples da linguagem R

Para entender melhor os conceitos do R AE, considere os exemplos a seguir.

Exemplo de função de tabela 1

Este exemplo cria uma função que retorna o comprimento total das colunas de entrada de caracteres e a soma total de todas as colunas de entrada numéricas para cada linha de entrada.

Código

Digite o seguinte código no arquivo " /tmp/sampleudtf.R:

nz.fun <- function () {
while(getNext()) {
tl <- 0
ts <- 0
for (i in seq(inputColumnCount())) {
x <- getInputColumn(i-1)
if (is.character(x)) {
tl <- tl + nchar(x)
}
if (is.numeric(x)) {
ts <- ts + as.double(x)
}
}
setOutputInt32(0, tl)
setOutputDouble(1, ts)
outputResult()
}
}

Compilação

Compile o arquivo da seguinte forma:

/nz/export/ae/utilities/bin/compile_ae --language r --version 3 \
--template compile --user nz --db dev /tmp/sampleudtf.R

Registro

Registre o AE da seguinte forma:

/nz/export/ae/utilities/bin/register_ae --language r --version 3 --user nz \
--db dev --template udtf --sig 'rtotal(VARCHAR(1024),DOUBLE)' \
--return 'TABLE(length INT4, sum DOUBLE)' --exe sampleudtf.R

Executando

Execute o UDTF da seguinte forma:

SELECT * FROM TABLE WITH FINAL(rtotal('text', 1));
LENGTH | SUM
--------+-----
4 | 1
(1 row)

Exemplo de função de tabela 2

Este exemplo mostra várias linhas de saída. Como o comprimento total das colunas de caracteres é um

Código

Digite o seguinte código no arquivo " /tmp/sampleudtf.R:

nz.fun <- function () {
while(getNext()) {
tl <- 0
ts <- 0
for (i in seq(inputColumnCount())) {
x <- getInputColumn(i-1)
if (is.character(x)) {
tl <- tl + nchar(x)
}
if (is.numeric(x)) {
ts <- ts + as.double(x)
}
}
setOutputString(0, 'total length of character columns')
setOutputDouble(1, tl)
outputResult()
setOutputString(0, 'total sum of numerics')
setOutputDouble(1, ts)
outputResult()
}
}

integer, e a soma total de numérico é do tipo double, escolha o tipo de dados mais flexível para a segunda coluna de saída com base nos tipos de dados das entradas, nesse caso double. O código R é um pouco diferente para este exemplo.

Compilação

Compile o arquivo da seguinte forma:

/nz/export/ae/utilities/bin/compile_ae --language r \
--version 3 --template compile --user nz --db dev \
/tmp/sampleudtf.R

Registro

Registre o AE da seguinte forma. Devido à diferença na saída, é necessário especificar uma assinatura de saída diferente durante o registro.

/nz/export/ae/utilities/bin/register_ae --language r \
--version 3 --user nz --db dev --template udtf \
--sig 'rtotal(VARCHAR(1024),DOUBLE)' \
--return 'TABLE(name VARCHAR(64), value DOUBLE)' \
--exe sampleudtf.R

Executando

Quando você chama essa função, a seguinte saída é produzida:

SELECT * FROM TABLE WITH FINAL(rtotal('text', 1));
NAME | VALUE
-----------------------------------+-------
total length of character columns | 4
total sum of numerics | 1
(2 rows)

Exemplo de função de tabela 3

Ao alterar o comando de registro do exemplo anterior, é possível criar uma função de assinatura de entrada dinâmica usando VARARGS.

As etapas do código R e da compilação são as mesmas do exemplo anterior.

Registro

Registre o AE da seguinte forma:

/nz/export/ae/utilities/bin/register_ae --language r --version 3 \
--user nz --db dev --template udtf --sig 'rtotal(VARARGS)' \
--exe sampleudtf.R --return 'TABLE(name VARCHAR(64), value DOUBLE)'

Executando

Agora você pode passar qualquer número de colunas para o R:

SELECT * FROM TABLE WITH FINAL(rtotal('text', 1, 'second text',
2, 3, 4, 'the last text'));
NAME | VALUE
-----------------------------------+-------
total length of character columns | 28
total sum of numerics | 10
(2 rows)