java scriptrunner,java使用ScriptRunner执行sql文件

/**

* 使用ScriptRunner执行sql文件

*/

public class ExecuteSql {

private static final Logger LOGGER = LoggerFactory.getLogger(ExecuteSql.class);

@Autowired

private DataSource dataSource; // 获取数据库连接对象

public static void main(String[] args) {

try {

executeSqlFile("/sql/myTest.sql");

LOGGER.info("【执行sql文件】执行成功");

} catch (Exception e) {

e.printStackTrace();

LOGGER.info("【执行sql文件】执行出错" + e);

}

}

private void executeSqlFile(String sqlFileName) throws Exception {

if (StringUtils.isEmpty(sqlFileName)) {

return;

}

Exception error = null;

// 获取连接对象

Connection conn = dataSource.getConnection();

try {

// 设置不自动提交

conn.setAutoCommit(false);

// 初始化ScriptRunner

ScriptRunner runner = new ScriptRunner(conn);

// 设置不自动提交

runner.setAutoCommit(false);

// true，遇见错误会停止执行，打印并抛出异常，捕捉异常，并进行回滚，保证在一个事务内执行；

// false，遇见错误不会停止，会继续执行，会打印异常信息，并不会抛出异常，当前方法无法捕捉异常无法进行回滚操作，无法保证在一个事务内执行；

runner.setStopOnError(true);

// true则获取整个脚本并执行；

// false则按照自定义的分隔符每行执行；

runner.setSendFullScript(false);

// 定义命令间的分隔符

runner.setDelimiter(";");

runner.setFullLineDelimiter(false);

// 设置是否输出日志，null不输出日志，不设置自动将日志输出到控制台

runner.setLogWriter(null);

LOGGER.info("【执行sql文件】读取sql文件");

// 读取文件

Resource resource = new ClassPathResource(sqlFileName);

File file = resource.getFile();

LOGGER.info("【执行sql文件】开始执行sql文件");

// 如果有多个sql文件，可以写多个runner.runScript(xxx),

runner.runScript(new InputStreamReader(new FileInputStream(file), StandardCharsets.UTF_8));

// 执行

conn.commit();

} catch (Exception e) {

LOGGER.info("【执行sql文件】执行出错，开始回滚");

conn.rollback();

error = e;

} finally {

close(conn);

}

if (error != null) {

throw error;

}

}

/*

* @描述：关闭连接

* @创建人：吴翔龙

* @创建时间：2019/2/20

* @param：[conn]

* @return：void

*/

private static void close(Connection conn) {

LOGGER.info("【执行sql文件】关闭数据库连接");

try {

if (conn != null) {

conn.close();

}

} catch (Exception e) {

if (conn != null) {

conn = null;

}

}

}

}