ScriptRunner执行SQL脚本

全网营销 2023-2-5 21:40 6415人围观脚本

明天有人征询我ScriptRunner的相关题目，我也没研讨过，就百度了一些材料起头写一个小测试demo出来，碰到了一些题目，也处理了一些题目，我们接下来看吧。

实在还可以改成MySQL，大概其他DB，样例今朝是Oracle的demo。

一、pom文件增加引入

需要引入oracle以及mybatis的包，具体以下。

<dependency>
<groupId>com.oracle</groupId>
<artifactId>ojdbc8</artifactId>
<version>12.2.0.1.0</version>
</dependency>
<dependency>
<groupId>org.mybatis.spring.boot</groupId>
<artifactId>mybatis-spring-boot-starter</artifactId>
<version>1.1.1</version>
</dependency>

二、编写测试履行的SQL剧本

留意看下面的每条SQL的最初，我写了&&标记，这个是为了针对ScriptRunner的，&&标记暗示每条SQL的竣事，留意不能用分号，由于ScriptRunner也支持declare剧本和存储进程，这时辰就会出现紊乱了，是以每一条SQL完成要用分歧于分号的标记来区分，是以挑选了&&。

剧本我我们别离测试了，删除表，建立表，插入表数据，查询表数据，以及Declare语句块，并将成果插入到履行成果的表中，以便存储剧本的履行成果。

drop table test_script_runer&&
drop table test_script_runer_result&&
create table test_script_runer
(id number(22,0) not null,
num number(22,0) not null)&&
create table test_script_runer_result
(count number(22,0) not null)&&
insert into TEST_SCRIPT_RUNER values (1,2)&&
insert into TEST_SCRIPT_RUNER values (2,3)&&
insert into TEST_SCRIPT_RUNER values (3,3)&&
insert into TEST_SCRIPT_RUNER values (4,3)&&
insert into TEST_SCRIPT_RUNER values (5,3)&&
insert into TEST_SCRIPT_RUNER values (6,3)&&
insert into TEST_SCRIPT_RUNER values (7,3)&&
SELECT id||','||num FROM TEST_SCRIPT_RUNER&&
declare
vn_count number;
begin
SELECT count(1) into vn_count FROM TEST_SCRIPT_RUNER;
insert into test_script_runer_result values (vn_count);
end;&&

三、java实现

新增TestScriptRunner.java，留意其中的scriptRunner.setDelimiter("&&")这一句，就是与上面的SQL中的&&标记对应的，你还可以变动成其他的，别的假如点窜成分号的话，就不可以在sql文件中写存储进程之类的了，只能写SQL。

package com.buxuesong.account;

import org.apache.ibatis.io.Resources;
import org.apache.ibatis.jdbc.ScriptRunner;
import org.springframework.boot.CommandLineRunner;

import java.io.*;
import java.nio.charset.Charset;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

public class TestScriptRunner implements CommandLineRunner {

public static void main(String[] args) throws Exception {
TestScriptRunner test = new TestScriptRunner();
test.run("");
}

@Override
public void run(String... args) throws Exception {
try {
Connection conn = getConnection();
ScriptRunner scriptRunner = new ScriptRunner(conn);
// 设备字符集,否则中文乱码插入毛病
Resources.setCharset(Charset.forName("UTF-8"));
scriptRunner.setEscapeProcessing(false);
scriptRunner.setRemoveCRs(true);
scriptRunner.setSendFullScript(false);
scriptRunner.setAutoCommit(true);
// 分隔符，还未考证具体功用
scriptRunner.setFullLineDelimiter(false);
// 每条号令间的分隔符，留意这个不倡议用分号分隔
// 由于SQL剧本中可以写存储进程，中心存在分号，致使存储进程履行失利
scriptRunner.setDelimiter("&&");
// 从class目录下间接读取
List<String> files = getFiles("/Users/XuesongBu/Documents/NoSQLBooster/");
for (String file : files) {
InputStream inputStream = new FileInputStream(new File(file));
Reader reader = new InputStreamReader(inputStream, "utf-8");
scriptRunner.runScript(reader);
}
scriptRunner.closeConnection();
conn.close();
System.out.println("sql剧本履行终了");
} catch (Exception e) {
System.out.println("sql剧本履行发生异常");
e.printStackTrace();
}
}

// 获得某个目录下一切间接下级文件，不包括目录下的子目录的下的文件，所以不用递归获得
public List<String> getFiles(String path) {
List<String> files = new ArrayList<String>();
File file = new File(path);
File[] tempList = file.listFiles();

for (int i = 0; i < tempList.length; i++) {
if (tempList[i].isFile() && tempList[i].getName().endsWith("sql")) {
files.add(tempList[i].toString());
}
}
return files;
}

// 获得数据库Connection
private Connection getConnection() throws ClassNotFoundException, SQLException {
Class.forName("oracle.jdbc.OracleDriver");
//需要点窜成你的数据库url，以及账号密码
return DriverManager.getConnection(
"ORACLE_URL",
"USER_NAME", "PASSWORD");
}
}

四、测试成果

drop table test_script_runer

drop table test_script_runer_result

create table test_script_runer
(id number(22,0) not null,
num number(22,0) not null)

create table test_script_runer_result
(count number(22,0) not null)

insert into TEST_SCRIPT_RUNER values (1,2)

insert into TEST_SCRIPT_RUNER values (2,3)

insert into TEST_SCRIPT_RUNER values (3,3)

insert into TEST_SCRIPT_RUNER values (4,3)

insert into TEST_SCRIPT_RUNER values (5,3)

insert into TEST_SCRIPT_RUNER values (6,3)

insert into TEST_SCRIPT_RUNER values (7,3)

SELECT id||','||num FROM TEST_SCRIPT_RUNER

ID||','||NUM
1,2
2,3
3,3
4,3
5,3
6,3
7,3
declare
vn_count number;
begin
SELECT count(1) into vn_count FROM TEST_SCRIPT_RUNER;
insert into test_script_runer_result values (vn_count);
end;

sql剧本履行终了

Process finished with exit code 0