PostgreSQL
 sql >> Database >  >> RDS >> PostgreSQL

Come scrivo i dati da R alle tabelle PostgreSQL con una chiave primaria a incremento automatico?

Dal thread nel commento di hrbrmstr, ho trovato un trucco per farlo funzionare.

Nella postgresqlWriteTable nel RPostgreSQL pacchetto, è necessario sostituire la linea

sql4 <- paste("COPY", postgresqlTableRef(name), "FROM STDIN")

con

sql4 <- paste(
  "COPY ", 
  postgresqlTableRef(name), 
  "(", 
  paste(postgresqlQuoteId(names(value)), collapse = ","), 
  ") FROM STDIN"
)

Nota che la virgoletta delle variabili (non incluse nell'hack originale) è necessaria per passare i nomi delle colonne con distinzione tra maiuscole e minuscole.

Ecco uno script per farlo:

body_lines <- deparse(body(RPostgreSQL::postgresqlWriteTable))
new_body_lines <- sub(
  'postgresqlTableRef(name), "FROM STDIN")', 
  'postgresqlTableRef(name), "(", paste(shQuote(names(value)), collapse = ","), ") FROM STDIN")', 
  body_lines,
  fixed = TRUE
)
fn <- RPostgreSQL::postgresqlWriteTable
body(fn) <- parse(text = new_body_lines)
while("RPostgreSQL" %in% search()) detach("package:RPostgreSQL")
assignInNamespace("postgresqlWriteTable", fn, "RPostgreSQL")