sql-server - R 的存储过程参数

Question

我正在尝试使用传递给存储过程的参数在 R-Services 中运行一个简单的 R sum，但我不知道该怎么做，也找不到一个很好的例子。这是我到目前为止所拥有的:

IF OBJECT_ID ( 'TEST', 'P' ) IS NOT NULL   
    DROP PROCEDURE TEST;
GO

CREATE PROCEDURE TEST @a int = 0, @b int = 0 AS
BEGIN
    EXEC sp_execute_external_script
    @language = N'R'
    ,@script = N'print(sum(@a, @b))' -- how to pass params here?
    ,@input_data_1 = N'@a'
    return @a + @b;
END

EXEC dbo.TEST @a = 2, @b = 3
GO

我的问题是如何将存储过程中读取的变量(@a 和 @b)传递给 R 脚本？

Answer 1

考虑使用@params将SQL Server参数传递给R变量，如MS docs所示。请务必返回一个数据帧对象，并为需要传递到 InputDataSet 数据帧对象中的 R 脚本的 SQL Server 查询保留 @input_data_1(因此此处为空)。

CREATE PROCEDURE myProc (@a int, @b int)
AS
    EXEC sp_execute_external_script    
      @language = N'R'    
    , @script = N' OutputDataSet <- data.frame(SumRes = sum(a_r, b_r));'    
    , @input_data_1 = N'   ;' 
    , @params = N' @a_r int, @b_r int'  
    , @a_r = @a
    , @b_r = @b
    WITH RESULT SETS (([SumResult] int NOT NULL)); 

然后调用程序:

EXEC dbo.myProc @a = 2, @b = 3