PostgreSQL触发并传递参数

时间:2010-06-10 22:25:05

标签: postgresql stored-procedures triggers

这是一个多部分问题。

我有一张类似于:

的表格
CREATE TABLE sales_data (
  Company character(50),
  Contract character(50),
  top_revenue_sum integer,
  top_revenue_sales integer,
  last_sale timestamp) ;

我想为这个表中的新插入创建一个触发器,如下所示:

CREATE OR REPLACE FUNCTION add_contract()
RETURNS VOID
DECLARE
    myCompany   character(50),
    myContract  character(50),

BEGIN
    myCompany   = TG_ARGV[0];
    myContract  = TG_ARGV[1];

    IF (TG_OP = 'INSERT') THEN
        EXECUTE 'CREATE TABLE salesdata_' || $myCompany || '_' || $myContract || ' (
            sale_amount integer,
            updated TIMESTAMP not null,
            some_data varchar(32),
            country varchar(2)
        ) ;'
        EXECUTE 'CREATE TRIGGER update_sales_data BEFORE INSERT OR DELETE ON salesdata_'
            || $myCompany || '_' || $myContract || ' FOR EACH ROW EXECUTE update_sales_data( '
            || $myCompany || ',' || $myContract || ', revenue);' ;
    END IF;
END;
$add_contract$ LANGUAGE plpgsql;

CREATE TRIGGER add_contract AFTER INSERT ON sales_data FOR EACH ROW EXECUTE add_contract() ;

基本上,每次我在sales_data中插入一个新行时,我想生成一个新表,其中表的名称将被定义为“salesdata_Company_Contract”

所以我的第一个问题是如何将公司和合同数据传递给触发器,以便将其传递给add_contract()存储过程?

从我的存储过程中,您将看到我还想在将新数据插入salesdata_Company_Contract表时更新原始的sales_data表。此触发器将执行以下操作:

CREATE OR REPLACE FUNCTION update_sales_data() RETURNS trigger as $update_sales_data$
DECLARE
    myCompany character(50) NOT NULL,
    myContract character(50) NOT NULL,
    myRevenue integer NOT NULL

BEGIN
    myCompany = TG_ARGV[0] ;
    myContract = TG_ARGV[1] ;
    myRevenue = TG_ARGV[2] ;

    IF (TG_OP = 'INSERT') THEN
        UPDATE sales_data SET
            top_revenue_sales = top_revenue_sales + 1,
            top_revenue_sum = top_revenue_sum + $myRevenue,
            updated = now()
        WHERE
            Company = $myCompany AND
            Contract = $myContract ;
    ELSIF (TG_OP = 'DELETE') THEN
        UPDATE sales_data SET
            top_revenue_sales = top_revenue_sales - 1,
            top_revenue_sum = top_revenue_sum - $myRevenue,
            updated = now()
        WHERE
            Company = $myCompany AND
            Contract = $myContract ;
    END IF;
END;
$update_sales_data$ LANGUAGE plpgsql;

当然,这将要求我在这些存储过程和触发器中传递几个参数,我不确定(a)这是否可行,或(b)实用,或(c)最佳实践我们应该把这个逻辑放到我们的其他软件中,而不是让数据库为我们做这项工作。

为了保持我们的桌面大小,因为我们每天会有数十万笔交易,我们决定使用公司和合同字符串作为表名自身的一部分来划分我们的数据,所以他们都非常体积小;我们的文件IO速度更快,我们觉得我们的性能会更好。

感谢您的任何想法或指示。

我的想法,既然我已经完成了所有这些,那么我们可能需要编写存储过程,我们将插入数据作为参数传递,并从我们的其他软件调用,并让存储过程执行插入“sales_data”然后创建另一个表。然后,有第二个存储过程将新数据插入salesdata_Company_Contract表,其中表名作为参数传递给存储过程,并再次让存储过程执行插入,然后更新主sales_data表。

你会采取什么方法?

1 个答案:

答案 0 :(得分:3)

看起来您只需要引用进入表格的NEW或OLD参数。可以在优秀的Postgresql文档here中找到示例。将您的值引用为NEW.Company和NEW.Contract。