Push R-Code zu PL/R-Code in Postgresql-Datenbank

Question

Ich versuche, einen erfolgreichen R-Code mit Rpostgresql in einen PL/R-Code zu konvertieren, um zu vermeiden, push/ziehen Daten in und aus der Postgreql-Datenbank.

-Code ist ein dcast auf einem data.table:

#libs 
library(RPostgreSQL); 
library(data.table); 

# connect 
drv <- dbDriver("PostgreSQL"); 
con <- dbConnect(drv, dbname="postgres", user="postgres"); 

# load 
cli_ranges <- dbGetQuery(con, "SELECT custid, prod_ranges, is_cli from cli_ranges;") 

# DT 
setDT(cli_ranges) 
setkeyv(cli_ranges , c("prod_ranges")) 

# pivot 
cli_ranges.pivoted <- dcast(cli_ranges, custid ~ paste0("is_cli_", prod_ranges), fun=sum, value.var = "is_cli") 

# send back to DB 
dbWriteTable(con, "cli_ranges_pivoted", cli_ranges.pivoted, row.names=F) 

-Code in R arbeitet gut & schnell.

ich jetzt versuchen, den Code in einer PL/R-Funktion zu drücken,

CREATE OR REPLACE FUNCTION public.pivot() 
    RETURNS void AS 
$BODY$ 

[copy/paste R code] 

$BODY$ 
    LANGUAGE plr; 

... aber die letzte Zeile des R-Code (dbWriteTable) wirft:

ERROR: R interpreter expression evaluation error 
DETAIL: Error in (function (classes, fdef, mtable) : 
    unable to find an inherited method for function 'dbWriteTable' for signature '"logical", "character", "data.frame"' 
CONTEXT: In PL/R function pivot 

ändern Die data.table zu einem Datenrahmen (as.data.frame(cli_ranges.pivoted)) funktioniert auch nicht.

Ein Trick könnte sein, den data.table/Frame zurückzukehren, um ein CREATE TABLE cli_ranges_pivoted AS SELECT pivot(); auszuführen, aber ich weiß nicht wirklich, wie die data.frame als Ausgang schieben ...

cli_ranges Tabelle:

custid prod_ranges is_cli 
1  A   1 
1  B   1 
1  C   0 
2  A   1 
2  B   0 
2  C   1 
3  A   0 
3  B   1 
3  C   0 
4  A   1 
...  ...   ... 

Nach dcast (dh Schwenk) datafram ist wie folgt:

custid prod_ranges_A prod_ranges_B prod_ranges_C 
1  1    1    0 
2  1    0    1 
3  0    1    0 
4  1    ... 
... 

Anzahl unterschiedlicher Werte in prod_ranges chang Es ist oft, so kann ich im Voraus die Anzahl der Spalten nach der Schwenkung definieren.

Env: Postgresql 9.5, R 3.3, PL/R 08.03.00.16, Win 10 64bits

Answer 1

könnten Sie mit http://gpdb.docs.pivotal.io/4330/ref_guide/pl_r.html Beispiel beginnen 2 und 3

Oder Sie könnten die UNNEST Funktion versuchen, zu modifizieren, wie Lukas Eklund und Erwind Brandstetter haben in diesem Beitrag (ich verwende Lukas Lösung): Unnest array by one level.

CREATE OR REPLACE FUNCTION unnest_multidim(anyarray) 
RETURNS SETOF anyarray AS 
$BODY$ 
    SELECT array_agg($1[series2.i][series2.x]) FROM 
    (SELECT generate_series(array_lower($1,2),array_upper($1,2)) as x, series1.i 
    FROM 
    (SELECT generate_series(array_lower($1,1),array_upper($1,1)) as i) series1 
    ) series2 
GROUP BY series2.i 
$BODY$ 
LANGUAGE sql IMMUTABLE; 

und dann könnten Sie versuchen, ein Feld zurückgibt, und etwas tun, wie folgt aus:

CREATE OR REPLACE FUNCTION r_norm(n integer, mean float8, std_dev float8) 
RETURNS float8[] 
AS $$ 
x<-rnorm(n,mean,std_dev); 
y<-rnorm(n,mean,std_dev); 
final<-cbind(as.data.frame(x), as.data.frame(y)); 
return(final) 

$$ LANGUAGE 'plr'; 

CREATE TABLE test_norm_var AS SELECT R_output[1] as col1, R_output[2] as col2 FROM unnest_multidim(r_norm(10,0,1)) R_output; 

SELECT col1 FROM test_norm_var; 

EDIT

konnte ich nicht bekommen dbWriteTable die Art und Weise zu arbeiten, um es als beabsichtigt war, PL/R-Funktion ... ABER, könnten Sie versuchen, diese Methode auch

CREATE OR REPLACE FUNCTION pivot() 
    RETURNS VOID as $$ 

library(RPostgreSQL); 
library(data.table); 

drv <- dbDriver("PostgreSQL"); 
con <- dbConnect(drv, dbname ="postgres"); 

fields <- list(custid = "numeric",prod_ranges = "varchar(128)", is_cli = "numeric") 

custid <- c(1,1,1,2,2,2) 
prod_ranges <- c("A","B","C","A","B","C") 
is_cli <- c(1,1,0,1,0,1) 

cli_ranges <- data.frame(custid,prod_ranges,is_cli, stringsAsFactors = default.stringsAsFactors()) 

setDT(cli_ranges) 
setkeyv(cli_ranges , c("prod_ranges")) 

cli_ranges.pivoted <- as.data.frame(dcast(cli_ranges, custid ~ paste0("is_cli_", prod_ranges), fun=sum, value.var = "is_cli")) 

create_query <- paste0("CREATE TABLE cli_ranges (",paste0(colnames(cli_ranges.pivoted), collapse = " numeric, "), 
" numeric) DISTRIBUTED BY (",colnames(cli_ranges)[1],")") 

dbGetQuery(con, create_query); 

values_string <- "(" 
for (i in 1:dim(cli_ranges.pivoted)[1]){ 
for (j in 1:dim(cli_ranges.pivoted)[2]){ 
    if (j != dim(cli_ranges.pivoted)[2]) { 
     values_string <- paste0(values_string,cli_ranges.pivoted[i,j],",") 
    } else { 
     values_string <- paste0(values_string,cli_ranges.pivoted[i,j]) 
    } 
    } 
    if (i != dim(cli_ranges.pivoted)[1]){ 
     values_string <- paste0(values_string,"),(") 
    } else { 
     values_string <- paste0(values_string,")") 
    } 
} 

insert_query <- paste0("INSERT INTO cli_ranges (",paste0(colnames(cli_ranges.pivoted), collapse = ", "), 
") VALUES ", values_string) 

dbGetQuery(con, insert_query); 

$$ LANGUAGE plr;