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

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语句中的所有程序逻辑。这意味着我想让查询返回一个ResultSet。

我研究了几种方法，如使用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);

我仔细查看了Springapi，但找不到与我想要的类似的东西。有没有办法从文件中加载脚本，然后运行它，使其使用Spring返回ResultSet？我更喜欢使用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将作为参数传递给ResultSetExtractor实现的extractData或RowMapper实现的mapRow。因此，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();

如果你想执行一个文件中的所有语句，你必须自己实现。只需循环遍历属性并逐个执行即可。您将需要一些标志来确定它的select或update/delete语句。环路特性：

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

Answer 3

您可以使用commons-io:

<dependency>
    <groupId>commons-io</groupId>
    <artifactId>commons-io</artifactId>
    <version>2.11.0</version>
</dependency>

import org.apache.commons.io.IOUtils;
import org.springframework.core.io.ClassPathResource;
//...
String sqlQuery = IOUtils.toString((new ClassPathResource("me.sql")).getInputStream(), Charset.defaultCharset());