我使用以下 Java 代码来运行 SQL 脚本:
@SpringBootApplication
public class Application implements CommandLineRunner {
@Autowired
DataSource ds;
@Value("classpath:test.sql")
Resource resource;
public static void main(String[] args) {
SpringApplication.run(Application.class, args);
}
@Override
public void run(String... args) throws Exception {
try {
@Cleanup
Connection c = ds.getConnection();
ScriptUtils.executeSqlScript(c, resource);
} catch (Exception e) {
e.printStackTrace();
}
}
}
我使用的 Spring boot 版本是 2.1.7.RELEASE,PostgreSQL 通过以下方式在本地 docker 镜像中运行:
docker run --rm --name pg-docker -e POSTGRES_PASSWORD=PASSWORD -d -p 5432:5432 -v postgresql:/var/lib/postgresql/data postgres:11.3
我使用的 SQL 脚本是:
CREATE OR REPLACE FUNCTION totalRecords ()
RETURNS integer AS $total$
declare
total integer;
BEGIN
SELECT count(*) into total FROM COMPANY;
RETURN total;
END;
$total$ LANGUAGE plpgsql;
我认为这是完全有效的并且在 PgAdmin 中运行正常。
不幸的是,我收到了来自 Java 的以下异常:
Caused by: org.postgresql.util.PSQLException: Unterminated dollar quote started at position 62 in SQL CREATE OR REPLACE FUNCTION totalRecords () RETURNS integer AS $total$ declare total integer. Expected terminating $$
at org.postgresql.core.Parser.checkParsePosition(Parser.java:1274)
at org.postgresql.core.Parser.parseSql(Parser.java:1173)
at org.postgresql.core.Parser.replaceProcessing(Parser.java:1125)
at org.postgresql.core.CachedQueryCreateAction.create(CachedQueryCreateAction.java:41)
at org.postgresql.core.QueryExecutorBase.createQueryByKey(QueryExecutorBase.java:314)
at org.postgresql.jdbc.PgStatement.executeCachedSql(PgStatement.java:289)
at org.postgresql.jdbc.PgStatement.executeWithFlags(PgStatement.java:274)
at org.postgresql.jdbc.PgStatement.execute(PgStatement.java:269)
at com.zaxxer.hikari.pool.ProxyStatement.execute(ProxyStatement.java:95)
at com.zaxxer.hikari.pool.HikariProxyStatement.execute(HikariProxyStatement.java)
at org.springframework.jdbc.datasource.init.ScriptUtils.executeSqlScript(ScriptUtils.java:488)
... 9 more
怎么了 ?请问我该如何修复它?或者如果 ScriptUtils 无法做到这一点,如何从 Java 可靠地运行 PostgreSQL 脚本?
终于我能够解决自己的问题了。不确定我的解决方案是否是最好的想法。问题出在 ScriptUtils 中 - 它通过 ScriptUtils.DEFAULT_STATMENT_SEPARATOR 确定语句结束,即“;”。不幸的是,FUNCTION 命令包含各种“;”这是声明本身的一部分。所以解决方案是声明并使用另一个不干扰 PostgreSQL 语法的分隔符。就我而言,我使用了两个分号 - “;;”。所以我的声明如下:
CREATE OR REPLACE FUNCTION totalRecords ()
RETURNS integer AS $total$
declare
total integer;
BEGIN
SELECT count(*) into total FROM COMPANY;
RETURN total;
END;
$total$ LANGUAGE plpgsql;;
调用它的Java代码更改为:
ScriptUtils.executeSqlScript(c, new EncodedResource(resource, "UTF-8"), false, false, ScriptUtils.DEFAULT_COMMENT_PREFIX, ";;",
ScriptUtils.DEFAULT_BLOCK_COMMENT_START_DELIMITER, ScriptUtils.DEFAULT_BLOCK_COMMENT_END_DELIMITER);
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)