Sqlserver
 sql >> Database >  >> RDS >> Sqlserver

Come leggere più set di risultati restituiti da una stored procedure di SQL Server in R

RODBC potrebbe non disporre di una funzionalità per recuperare più recordset da una stored procedure. Tuttavia, Windows ADO ha il NextRecordSet() metodo. Considerare che R effettui una chiamata dell'interfaccia COM ad ADO utilizzando RDCOMClient libreria (supponendo ovviamente che tu usi R per Windows). Nell'oggetto di connessione di ADO, passi la stessa stringa di connessione che hai fatto in RODBC.

Di seguito vengono recuperati i risultati delle query con GetRows() che restituisce un array bidimensionale tradotto come elenchi nidificati in R.

SQL Server (Proc memorizzato)

CREATE PROCEDURE MultipleResults 
AS    
BEGIN
    SET NOCOUNT ON;
    SELECT * FROM Table1;
    SELECT * FROM Table2;
END

R (Chiamata ADO)

library(RDCOMClient)

conn <- COMCreate("ADODB.Connection")
rst <- COMCreate("ADODB.Recordset")

conn$Open("driver={SQL Server};server=server;database=db;trusted_connection=yes;")

# FIRST QUERY RESULT
rst$Open("MultipleResults", conn)
dfList1 <- rst$GetRows()    
# RETRIEVE COLUMN NAMES
dfnames1 <- vapply(c(0:(rst[['Fields']]$Count()-1)), 
                   function(i) (rst$Fields(i)$Name()), character(1))

# SECOND QUERY RESULT
rst <- rst$NextRecordset()
dfList2 <- rst$GetRows()    
# RETRIEVE COLUMN NAMES
dfnames2 <- vapply(c(0:(rst[['Fields']]$Count()-1)), 
                   function(i) (rst$Fields(i)$Name()), character(1))   
# CLOSE OBJECTS
rst$Close(); conn$Close()

# FREE RESOURCES
rst <-  conn <- NULL
rm(rst, conn)
gc()

# CONVERT NESTED LISTS TO DATAFRAMES    
dfList1 <- lapply(dfList1, function(x) setNames(data.frame(x), dfnames1))    
df1 <- do.call(rbind, dfList1)

dfList2 <- lapply(dfList2, function(x) setNames(data.frame(x), dfnames2))      
df2 <- do.call(rbind, dfList2)