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

Question

我试图在我的 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 文件时，出现如下错误:

Unterminated dollar quote started at position 65 in SQL CREATE FUNCTION my_function() RETURNS TRIGGER AS $$

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

Answer 1

hibernate 的默认 SqlStatementParser 实现的问题，用于多行 sql 命令提取器。

如果你看语法定义hibernate-core/src/main/antlr/sql-stmt.g有语句结束的定义:

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

NOT_STMT_END
    : ~( ';' )
    ;

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

因此: 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 感知。)