PostgreSQL / JPA - 在import.sql文件中导入函数?

时间:2018-05-08 15:24:19

标签: postgresql hibernate function jpa triggers

我试图在我的JPA import.sql文件中定义一些PostgreSQL函数和触发器。我使用Hibernate 5.x作为我的底层JPA提供程序。由于我的import.sql文件包含多行命令,因此我在persistence.xml文件中设置了此属性:

<property name="hibernate.hbm2ddl.import_files_sql_extractor" value="org.hibernate.tool.hbm2ddl.MultipleLinesSqlCommandExtractor" />

从那时起,我试图在import.sql中定义我的功能。他们看起来像这样:

DROP FUNCTION IF EXISTS update_total_feedback_count() CASCADE;
CREATE FUNCTION update_total_feedback_count() RETURNS TRIGGER AS
$$
DECLARE
  application_version_id BIGINT := 0;
  app_version_metadata_id BIGINT:= 0;
  application_id BIGINT := 0;
  app_metadata_id BIGINT := 0;
BEGIN
    IF (TG_OP = 'INSERT') THEN
        SELECT INTO application_version_id tbl_application_version.id
            FROM tbl_application_version
            INNER JOIN tbl_feedback ON tbl_feedback.application_version_id = tbl_application_version.id
            WHERE tbl_feedback.id = NEW.id;

        SELECT INTO app_version_metadata_id tbl_application_version.application_version_metadata_id
            FROM tbl_application_version
            WHERE id = application_version_id;

        SELECT INTO app_metadata_id registered_application_metadata_id
            FROM tbl_registered_application
            INNER JOIN tbl_application_version ON tbl_application_version.registered_application_id = tbl_registered_application.id
            WHERE tbl_application_version.id = application_version_id;

        UPDATE tbl_registered_application_metadata SET feedbackcount = (feedbackcount + 1), lastfeedbackdate = NEW.createddate WHERE id = app_metadata_id;
        UPDATE tbl_application_version_metadata SET feedbackcount = (feedbackcount + 1), lastfeedbackdate = NEW.createddate WHERE id = app_version_metadata_id;

        RETURN NEW;
    ELSIF (TG_OP = 'DELETE') THEN
        -- IMPLEMENT THIS TRIGGER

        RETURN NULL;
    END IF;
END;
$$
LANGUAGE 'plpgsql';
ALTER FUNCTION update_total_feedback_count() OWNER TO feedback_tracker;

然而,当我部署我的WAR文件时,我收到一个错误,上面写着:

  

在SQL CREATE中,未终止的美元报价从第65位开始   功能my_function()RETURNS TRIGGER AS $$

所以,显然它在我的函数声明中的$$上死了。有没有解决的办法?我应该以不同方式声明我的功能/触发器吗?我可以在我的persistence.xml文件中设置一个可以解决这个问题的属性吗?

1 个答案:

答案 0 :(得分:3)

hibernate的默认SqlStatementParser实现的问题,它在多行sql命令提取器中使用。

如果你看一下语法定义hibernate-core/src/main/antlr/sql-stmt.g,就会有Statement End的定义:

STMT_END
    : ';' ( '\t' | ' ' | '\r' | '\n' )*
    ;

NOT_STMT_END
    : ~( ';' )
    ;

这表示语句结束是分号符号,后跟“Space”“tab”“carret return”或“new line”符号。

THUS:HIBERNATE的默认实施不支持美元报价。

如果您不想实现自定义hibernate的解析器,则可以使用简单的'引用重写所有函数而无需引用美元。但是你需要小心地逃避'字符。

更新:您可以创建自定义ImportSqlCommandExtractor。例如:将命令与--******分开(注释中的6个星号符号,只是为了使您的文件成为正确的SQL文件,但在注释中使用自定义命令分隔,或者选择您喜欢的任何疯狂组合)然后将它们拆分在简单的实现中

public class ImportSqlCE implements ImportSqlCommandExtractor {

    private static final Logger log = LoggerFactory.getLogger(ImportSqlCE.class);

    @Override
    public String[] extractCommands(Reader reader) {
        try {
            String allCommands = IOUtils.toString(reader);

            return allCommands.split("--******");
        } catch (IOException e) {
            log.error("error reading import commands", e);
            log.info("sengind default empty command set");
            return new String[0];
        }
    }
}

然后配置hibernate以使用它<property name="hibernate.hbm2ddl.import_files_sql_extractor" value="example.ImportSqlCE" />

这样你的import.sql将支持美元引用(即它将简单地忽略任何sql对正在发生的事情的认识。)