'commit'

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("增加成功！");

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