Sqlserver
 sql >> Base de données >  >> RDS >> Sqlserver

Comment lire plusieurs ensembles de résultats renvoyés par une procédure stockée SQL Server dans R

RODBC peut ne pas avoir de fonctionnalité pour récupérer plusieurs jeux d'enregistrements à partir d'une procédure stockée. Cependant, Windows ADO a le NextRecordSet() méthode. Envisagez de demander à R d'effectuer un appel d'interface COM vers ADO en utilisant le RDCOMClient bibliothèque (en supposant bien sûr que vous utilisez R pour Windows). Dans l'objet de connexion d'ADO, vous transmettez la même chaîne de connexion que dans RODBC.

Ci-dessous récupère les résultats de la requête avec GetRows() qui renvoie un tableau à deux dimensions traduit en listes imbriquées dans R.

SQL Server (Proc stocké)

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

R (Appel 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)