xxq250
/
homework-jianmu
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

change

This commit is contained in:
zyyang 2020-12-11 15:19:45 +08:00
parent ad41116841
commit 013d6693db
1 changed files with 23 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
tests/examples/JDBC/taosdemo/src/main/java/com/taosdata/taosdemo/components/TaosDemoCommandLineRunner.java
View File

@ -44,8 +44,23 @@ public class TaosDemoCommandLineRunner implements CommandLineRunner {
} }
// 准备数据 // 准备数据
prepareData(config); prepareData(config);
// 创建数据库 // 创建数据库
createDatabaseTask(config);
// 建表
createTableTask(config);
// 插入
insertTask(config);
// 查询: 1. 生成查询语句, 2. 执行查询
// 删除表
if (config.dropTable) {
superTableService.drop(config.database, config.superTable);
}
System.exit(0);
}
private void createDatabaseTask(JdbcTaosdemoConfig config) {
Map<String, String> databaseParam = new HashMap<>(); Map<String, String> databaseParam = new HashMap<>();
databaseParam.put("database", config.database); databaseParam.put("database", config.database);
databaseParam.put("keep", Integer.toString(config.keep)); databaseParam.put("keep", Integer.toString(config.keep));
@ -55,18 +70,20 @@ public class TaosDemoCommandLineRunner implements CommandLineRunner {
databaseService.dropDatabase(config.database); databaseService.dropDatabase(config.database);
databaseService.createDatabase(databaseParam); databaseService.createDatabase(databaseParam);
databaseService.useDatabase(config.database); databaseService.useDatabase(config.database);
}
// 建表 // 建超级表，三种方式：1. 指定SQL，2. 指定field和tags的个数，3. 默认
// 建超级表，三种方式：1. 指定SQL，2. 指定field和tags的个数，3. 默认 private void createTableTask(JdbcTaosdemoConfig config) {
if (config.doCreateTable) { if (config.doCreateTable) {
superTableService.create(superTableMeta); superTableService.create(superTableMeta);
// 批量建子表 // 批量建子表
subTableService.createSubTable(subTableMetaList, config.numOfThreadsForCreate); subTableService.createSubTable(subTableMetaList, config.numOfThreadsForCreate);
} }
}
// 插入 private int insertTask(JdbcTaosdemoConfig config) {
int numOfThreadsForInsert = 1; int numOfThreadsForInsert = config.numOfThreadsForInsert;
int sleep = 0; int sleep = config.sleep;
if (config.autoCreateTable) { if (config.autoCreateTable) {
// 批量插入，自动建表 // 批量插入，自动建表
dataList.stream().forEach(subTableValues -> { dataList.stream().forEach(subTableValues -> {
@ -79,18 +96,6 @@ public class TaosDemoCommandLineRunner implements CommandLineRunner {
sleep(sleep); sleep(sleep);
}); });
} }
// 批量插入,不使用自动建表
// 查询
// 1. 生成查询语句
// 2. 执行查询
// 删除表
if (config.dropTable) {
superTableService.drop(config.database, config.superTable);
}
System.exit(0);
} }
private void prepareData(JdbcTaosdemoConfig config) { private void prepareData(JdbcTaosdemoConfig config) {