diff --git a/src/main/java/Tools/MaxKb/Test.java b/src/main/java/Tools/MaxKb/TestMaxKB.java similarity index 67% rename from src/main/java/Tools/MaxKb/Test.java rename to src/main/java/Tools/MaxKb/TestMaxKB.java index d2b78aec..6bd49521 100644 --- a/src/main/java/Tools/MaxKb/Test.java +++ b/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("增加成功!"); } } \ No newline at end of file