PL/R 函数接受两个表作为参数

Question

我正在努力寻找可以接受两个 postgres 表的 PL/R 函数的示例。 PL/R docs 没有提供任何此类示例。
为了有一个工作示例，让我们考虑在 R 端使用两个 postgres 表的合并。

在 postgres 中有两个表

CREATE TABLE x (a numeric, b text);
CREATE TABLE y (a numeric, d text);
INSERT INTO x VALUES (1, 'a'),(2, 'b');
INSERT INTO y VALUES (2, 'b'),(3, 'c');

我正在寻找替换以下查询

SELECT * FROM x INNER JOIN y ON x.a=y.a;

在 R 中定义的 PL/R 函数为：

my_function = function(x, y){
    merge(x, y, by = "a")
}

我能够调用 PL/R 函数，它接受单个表，但不是两个。

Answer 1

我不认为 postgeql 可以接受真实表格作为参数。

但是还有另一种方法可以做到这一点。您可以将表名作为参数传递。
这是函数

CREATE OR REPLACE FUNCTION merge(t1 text, t2 text)
returns setof x as 
   BEGIN
     execute 'select * from ' || t1 ||' join ' || t2 || ' on t1.a=t2.a';
   END

以上是postgresql的函数，也可以写成R函数。
这是R的代码

1. 我们必须将表 x 的所有值存储到变量 x 中。请参阅下面的代码

   x <- dbGetQuery(con, "SELECT * from sandbox.x") --con 是连接到你的数据库的连接，sandbox 是模式名，x 是表名

2. 将表 y 的值存储到变量 y 中

   y<-dbGetQuery(con, "SELECT * from sandbox.y")

3. 合并两张表

   total <- merge(x,y,by="a")

4. 你也可以写另一个函数来包装合并函数，见下面的代码

   myTotal <- function(x,y) { result <- merge(x,y,by="a") return(result) }

我附上了步骤截图供你参考