如何在 Java 中运行 .sql 脚本（从文件）并使用 Spring 返回结果集？

Question

如何在 Java 中运行 .sql 脚本（从文件）并使用 Spring 返回

ResultSet

？

我有一个程序，可以在数据库上运行

SQL

查询，返回

ResultSet

，稍后我会在类中处理并使用数据。我目前正在将

JDBC

与 Java 程序内的脚本一起使用。

StringBuilder query = new StringBuilder("some script on multiple lines");
PreparedStatement statement = connection.prepareStatement(query.toString());
ResultSet resultSet = statement.executeQuery();

我想将 Java 程序外部的 SQL 查询移至 .sql 文件，但我想保留

executeQuery

语句中的所有程序逻辑。这意味着我想让查询返回一个结果集。

我查看了几种方法，例如使用

ScriptRunner

、使用 Spring

JdbcTestUtils.executeSqlScript

或使用

BufferReader

读取 .sql 文件，然后将字符串传递给我的语句。

ScriptRunner

和Spring

JdbcTestUtils.executeSqlScript

似乎没有返回ResultSet，或者我找不到正确的实现。我想远离

BufferReader

方法，因为它需要文本解析和很多异常需要处理。

ScriptRunner scriptRunner = new ScriptRunner(connection, true, true);
scriptRunner.runScript(new FileReader("script.sql"));

runScript

方法返回void。

Spring

的实现也是如此：

MysqlDataSource ds = new MysqlDataSource();
ds.setServerName("host");
ds.setUser("user");
ds.setPassword("password");

JdbcTemplate jdbcTemplate = new JdbcTemplate(ds);

Resource resource = new ClassPathResource("script.sql");
JdbcTestUtils.executeSqlScript(jdbcTemplate, resource, true);

我彻底查看了

Spring

api，但找不到与我想要的类似的东西。 有没有办法从文件加载脚本然后运行它，以便它使用

ResultSet

返回

Spring

？我更喜欢使用

Spring

，因为它是积极维护的。

Answer 1

我找到了一种使用 Spring 来做到这一点的方法：

MysqlDataSource ds = new MysqlDataSource();
ds.setServerName("host");
ds.setUser("user");
ds.setPassword("password");

JdbcTemplate jdbcTemplate = new JdbcTemplate(ds);

BufferedReader in = new BufferedReader(new FileReader("script.sql"));
LineNumberReader fileReader = new LineNumberReader(in);
String query = JdbcTestUtils.readScript(fileReader);

现在我们将使用

jdbcTemplate.query

使用我们读取的 .sql 脚本来查询数据库。所需的

ResultSet

将作为参数传递给

extractData

实现的

ResultSetExtractor

或

mapRow

实现的

RowMapper

。因此，

ResultSet

的处理将在

extractData

或

mapRow

实现中完成，我们将返回我们需要的集合/对象

List<YourClass> result = jdbcTemplate.query(query, new RowMapper<YourClass>() {
            @Override
            public YourClass mapRow(ResultSet rs, int i) throws SQLException {
                // processing of the ResultSet
                return result;
            }
        });

或

YourClass object = jdbcTemplate.query(query, new ResultSetExtractor<YourClass>() {
            @Override
            public YourClass extractData(ResultSet rs) throws SQLException, DataAccessException {
                // processing of the ResultSet
                return result;
            }
        });

当然，使用最后一个实现，您可以返回任何您想要的对象，甚至是对象的集合。

Answer 2

我会使用属性文件来存储sql查询。

例如：

person-save=insert into person values....
person-get-all=select * from person....

这样您就可以从文件中读取语句

在上下文中：

<util:properties id="sqlProps" location="classpath:sqlProps.properties" />

在 DAO 中：

@Autowired
@Qualifier("sqlProps")
private Properties sqlProps;

...

String query = sqlProps.getProperty("person-get-all");
PreparedStatement statement = connection.prepareStatement(query);
ResultSet resultSet = statement.executeQuery();

如果你想执行一个文件中的所有语句，你必须自己实现。只需循环遍历属性并一一执行它们即可。您将需要一些标志来确定它是选择还是更新/删除语句。 循环属性：

Enumeration e = props.propertyNames();

    while (e.hasMoreElements()) {
      String key = (String) e.nextElement();
      String query= props.getProperty(key);
    }

Answer 3

String sqlQuery = IOUtils.toString((new ClassPathResource("me.sql")).getInputStream(), Charset.defaultCharset());