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

Question

我正在尝试在 JPA 文件中定义一些 PostgreSQL 函数和触发器import.sql。我使用 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 FUNCTION my_function() RETURNS TRIGGER AS $$ 中的位置 65 开始

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

Answer 1

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

如果你看一下语法定义，hibernate-core/src/main/antlr/sql-stmt.g 就会发现语句结束的定义：

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

NOT_STMT_END
    : ~( ';' )
    ;

这表明语句结束是分号符号，后跟“空格”“制表符”“插入符回车”或“换行符”符号。

因此：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 对正在发生的事情的认识。）