同时计算最大值和总和

时间:2019-05-10 13:55:04

标签: sql postgresql plpgsql postgresql-9.1

对于我的项目,我正在创建一些自定义的集合函数,这些函数用于应用差异隐私。 到目前为止,我能够实现MAX和MIN函数,但会增加一些Laplace噪声。 我在使用SUM_LAPLACE聚合时遇到了麻烦。 我的上限(灵敏度)应为该列中MAX值的绝对值。

如何同时计算最大值和和,然后将两个输出值传递给函数?

CREATE OR REPLACE FUNCTION calculateSum(real,real,OUT real, OUT real) AS $$
DECLARE 
   sumValue real := 0;
       max_v real;
BEGIN
   IF $1 IS NULL THEN
    sumValue := sumValue + $2;
   ELSIF $2 IS NULL THEN
    sumValue := sumValue + $1;
   ELSIF $2 IS NULL AND $1 IS NULL THEN
    sumValue := sumValue;
   ELSE
    sumValue := $1 + $2;
   END IF;
       max_v = searchmaximumvalue($1,$2);
END;
$$ LANGUAGE plpgsql;

CREATE OR REPLACE FUNCTION addLaplacianNoiseSum(real) RETURNS real AS $$
DECLARE
    epsilon real := 1.2;
    sensivity real := (epsilon * 2) + ($1/2);
    laplaceDistribution real;
 BEGIN
 laplaceDistribution := sensivity / (epsilon);
 RETURN  $1 + laplaceDistribution;

 END;
 $$ LANGUAGE plpgsql;

CREATE AGGREGATE SUM_LAPLACE(real)
(
SFUNC = calculateSum,
STYPE = real,
FINALFUNC = addLaplacianNoiseSum
);

函数 searchmaximumvalue($ 1,$ 2)可以正常工作。 我想从第一个函数返回sum和max_v并将它们传递给下面的第二个函数。 我该怎么办?

1 个答案:

答案 0 :(得分:1)

demo:db<>fiddle

您可以创建自己的返回类型:

CREATE TYPE my_type AS (sum real, max_v real);

可以在SFUNC中使用:

CREATE OR REPLACE FUNCTION calculateSum(my_type, real) RETURNS my_type  -- returns my_type
AS $$
DECLARE 
   sumValue real := 0;
   max_v real;
   output my_type; -- new variable of my_type
BEGIN
   /* A LOT OF CODE HERE */

   output.sum := sumValue;
   output.max_v := max_v;

   RETURN output;
END;
$$ LANGUAGE plpgsql;

当然还有您的FINALFUNC的输入内容:

CREATE OR REPLACE FUNCTION addLaplacianNoiseSum(my_type) RETURNS real AS $$

FINALFUNC中使用它:

$1.max_v
$1.sum

您的聚合看起来像

CREATE AGGREGATE SUM_LAPLACE(real) (
    SFUNC = calculateSum,
    STYPE = my_type,         -- return type == my_type
    FINALFUNC = addLaplacianNoiseSum
);