'commit'

src/main/java/Tools/MaxKb/TestMaxKB.java

@@ -5,9 +5,10 @@ import com.jfinal.kit.PropKit;
 import com.jfinal.plugin.activerecord.Db;
 import com.jfinal.plugin.activerecord.Record;
 
+import java.time.LocalDateTime;
 import java.util.UUID;
 
-public class Test {
+public class TestMaxKB {
 
     /**
      * 获取模型
@@ -37,7 +38,7 @@ public class Test {
      * @param sourceApplicationName 源应用名称
      * @param targetApplicationName 目标应用名称
      */
-    public static void addApplication(String sourceApplicationName, String targetApplicationName) {
+    public static String addApplication(String sourceApplicationName, String targetApplicationName) {
         //获取应用
         Record record = getApplication(sourceApplicationName);
         if (record == null) {
@@ -52,6 +53,8 @@ public class Test {
         record.set("name", targetApplicationName);
         //增加
         Db.save("application", "id", record);
+
+        return record.getStr("id");
     }
 
     /**
@@ -70,7 +73,7 @@ public class Test {
      *
      * @param sourceDatasetName 数据集名称
      */
-    public static void addDataSet(String sourceDatasetName, String targetDatasetName) {
+    public static String addDataSet(String sourceDatasetName, String targetDatasetName) {
         //获取数据集
         Record record = getDataSet(sourceDatasetName);
         if (record == null) {
@@ -86,6 +89,7 @@ public class Test {
         record.set("desc", targetDatasetName);
         //增加
         Db.save("dataset", "id", record);
+        return record.getStr("id");
     }
 
 
@@ -101,15 +105,25 @@ public class Test {
         String model_id = record.getStr("id");
         System.out.println(model_id);
 
-        //增加应用
-        String sourceApplicationName = "黄海的一个应用";
-        String targetApplicationName = "XXX的拷贝应用";
-        addApplication(sourceApplicationName, targetApplicationName);
-        System.out.println("增加成功！");
+        //拷贝生成应用
+        String sourceApplicationName = "应用【模板】";
+        String targetApplicationName = "XXX的应用";
+        String application_id = addApplication(sourceApplicationName, targetApplicationName);
+
+        //拷贝生成知识库
+        String sourceDataSetName = "知识库【模板】";
+        String targetDataSetName = "XXX的知识库";
+        String dataset_id = addDataSet(sourceDataSetName, targetDataSetName);
 
-        //每个人一个应用，每个人一个知识库
-        String sourceDataSetName = "长春教育知识库";
-        String targetDataSetName = "XXX的拷贝知识库";
-        addDataSet(sourceDataSetName, targetDataSetName);
+        //应用与知识库关联
+        record = new Record();
+        record.set("create_time", LocalDateTime.now());
+        record.set("update_time", LocalDateTime.now());
+        record.set("id", UUID.randomUUID());
+        record.set("application_id", UUID.fromString(application_id));
+        record.set("dataset_id", UUID.fromString(dataset_id));
+        Db.save("application_dataset_mapping", "id", record);
+
+        System.out.println("增加成功！");
     }
 }