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);

我仔细看了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实现中完成，我们将返回我们需要的Collection / Object

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);
    }

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

2 个答案: