diff --git a/src/main/java/Tools/MaxKb/TestMaxKB.java b/src/main/java/Tools/MaxKb/TestMaxKB.java index 5bb90b0f..9089df80 100644 --- a/src/main/java/Tools/MaxKb/TestMaxKB.java +++ b/src/main/java/Tools/MaxKb/TestMaxKB.java @@ -6,6 +6,7 @@ import com.jfinal.plugin.activerecord.Db; import com.jfinal.plugin.activerecord.Record; import java.time.LocalDateTime; +import java.util.List; import java.util.UUID; public class TestMaxKB { @@ -23,17 +24,20 @@ public class TestMaxKB { // String model_id = record.getStr("id"); // System.out.println(model_id); - //1、拷贝生成应用 + //1、删除代码生成的应用程序与知识库 + MaxKbUtil.delCodeGenerateApplicationDataSet(); + + //2、拷贝生成应用 String sourceApplicationName = "应用【模板】"; String targetApplicationName = "XXX的应用"; String application_id = MaxKbUtil.addApplication(sourceApplicationName, targetApplicationName); - //2、拷贝生成知识库 + //3、拷贝生成知识库 String sourceDataSetName = "知识库【模板】"; String targetDataSetName = "XXX的知识库"; String dataset_id = MaxKbUtil.addDataSet(sourceDataSetName, targetDataSetName); - //3、应用与知识库关联 + //4、应用与知识库关联 Record record = new Record(); record.set("create_time", LocalDateTime.now()); record.set("update_time", LocalDateTime.now()); @@ -42,7 +46,7 @@ public class TestMaxKB { record.set("dataset_id", UUID.fromString(dataset_id)); Db.save("application_dataset_mapping", "id", record); - //4、生成应用的访问Token + //5、生成应用的访问Token MaxKbUtil.addAccessToken(application_id); System.out.println("增加成功!"); diff --git a/src/main/java/Tools/MaxKb/Util/MaxKbUtil.java b/src/main/java/Tools/MaxKb/Util/MaxKbUtil.java index 9bbdf3e8..e3862918 100644 --- a/src/main/java/Tools/MaxKb/Util/MaxKbUtil.java +++ b/src/main/java/Tools/MaxKb/Util/MaxKbUtil.java @@ -8,6 +8,7 @@ import com.jfinal.plugin.activerecord.dialect.PostgreSqlDialect; import com.jfinal.plugin.hikaricp.HikariCpPlugin; import java.time.LocalDateTime; +import java.util.List; import java.util.UUID; public class MaxKbUtil { @@ -132,4 +133,27 @@ public class MaxKbUtil { Db.save("application_access_token", "application_id", record); } + /** + * 删除代码生成的数据 + */ + public static void delCodeGenerateApplicationDataSet(){ + //通过SQL查询出所有desc=【代码生成】的数据 + String sql = "select * from application where \"desc\"='【代码生成】'"; + List list = Db.find(sql); + for (Record record : list) { + String application_id = record.getStr("id"); + //删除应用与知识库的关联 + sql = "delete from application_dataset_mapping where application_id=?"; + Db.update(sql, UUID.fromString(application_id)); + + sql="delete from application_access_token where application_id=?"; + Db.update(sql, UUID.fromString(application_id)); + + //删除应用 + sql = "delete from application where id=?"; + Db.update(sql, UUID.fromString(application_id)); + } + sql = "delete from dataset where \"desc\"='【代码生成】'"; + Db.update(sql); + } }