'commit'

src/main/java/Tools/MaxKb/Impl/MaxKbImpl.java

@@ -1,15 +1,20 @@
-package Tools.MaxKb.Util;
+package Tools.MaxKb.Impl;
 
+import Tools.MaxKb.Model.MaxKbModel;
 import cn.hutool.core.util.StrUtil;
 import cn.hutool.http.HttpRequest;
 import cn.hutool.http.HttpResponse;
 import cn.hutool.json.JSONObject;
 import cn.hutool.json.JSONUtil;
 import com.jfinal.kit.PropKit;
+import com.jfinal.plugin.activerecord.Record;
 
 import java.io.File;
+import java.util.List;
+
+public class MaxKbImpl {
+    public static MaxKbModel mm = new MaxKbModel();
 
-public class MaxKbCallUtil {
     //MaxKB的url
     static final String baseUrl = PropKit.get("baseUrl");
     //获取身份识别码接口
@@ -95,8 +100,9 @@ public class MaxKbCallUtil {
 
     /**
      * 删除知识库文档
-     * @param authCode 身份识别码
+     *
-     * @param zskId 知识库id
+     * @param authCode   身份识别码
+     * @param zskId      知识库id
      * @param documentId 文档id
      * @return
      */

src/main/java/Tools/MaxKb/Service/MaxKbService.java

@@ -0,0 +1,54 @@
+package Tools.MaxKb.Service;
+
+import Tools.MaxKb.Impl.MaxKbImpl;
+import Tools.MaxKb.Model.MaxKbModel;
+import com.jfinal.kit.PropKit;
+import com.jfinal.plugin.activerecord.Record;
+
+import java.io.File;
+import java.util.List;
+
+public class MaxKbService {
+    public static MaxKbModel mm = new MaxKbModel();
+
+    /**
+     * 上传文件到知识库
+     *
+     * @param uploadFile 文件路径
+     */
+    public static void uploadQA(String uploadFile) {
+        //获取数据集名称
+        String dataSetName = PropKit.get("dataSetName");
+        List<com.jfinal.plugin.activerecord.Record> list = mm.getDataSetList(dataSetName);
+        if (list.size() > 1) {
+            System.out.println("数据集:【" + dataSetName + "】名称重复,请检查后重新运行！");
+            return;
+        }
+        if (list.isEmpty()) {
+            System.out.println("数据集:【" + dataSetName + "】不存在,请检查后重新运行！");
+            return;
+        }
+        //数据集ID
+        String zskId = list.getFirst().getStr("id");
+
+        //在上传文档前，需要删除掉旧的同名文档
+        File fi = new File(uploadFile);
+        String fileName = fi.getName();
+        //获取身份识别
+        String authCode = MaxKbImpl.getAuthorization();
+        //上传文档
+        List<com.jfinal.plugin.activerecord.Record> listDocuments = mm.getDocumentListByDataSetId(zskId);
+        for (Record record : listDocuments) {
+            String documentName = record.getStr("name");
+            if (documentName.equals(fileName)) {
+                String documentId = record.getStr("id");
+                //准备删除掉旧的文档
+                MaxKbImpl.delDocument(authCode, zskId, documentId);
+                System.out.println("删除旧的文档成功：" + documentName);
+            }
+        }
+        //上传新的文档
+        MaxKbImpl.uploadQA(authCode, zskId, uploadFile);
+        System.out.println("恭喜，所有操作成功完成！");
+    }
+}

src/main/java/Tools/MaxKb/UploadQA.java

@@ -1,61 +1,20 @@
 package Tools.MaxKb;
 
-import Tools.MaxKb.Model.MaxKbModel;
+import Tools.MaxKb.Service.MaxKbService;
-import Tools.MaxKb.Util.MaxKbCallUtil;
 import Tools.MaxKb.Util.MaxKbDb;
 import com.jfinal.kit.PropKit;
-import com.jfinal.plugin.activerecord.Record;
-
-import java.io.File;
-import java.util.List;
 
 public class UploadQA {
 
-    public static MaxKbModel mm = new MaxKbModel();
-
-
     //要上传的QA知识内容
     static String uploadFile = "D:\\dsWork\\QingLong\\Doc\\MaxKB\\各学校人员和班级统计【结果】.xlsx";
 
     public static void main(String[] args) throws Exception {
         //加载配置文件
         PropKit.use("MaxKb.properties");
-
         //初始化数据库
         MaxKbDb.Init();
-
+        //上传文件
-        //获取数据集名称
+        MaxKbService.uploadQA(uploadFile);
-        String dataSetName = PropKit.get("dataSetName");
-        List<Record> list = mm.getDataSetList(dataSetName);
-        if (list.size() > 1) {
-            System.out.println("数据集:【" + dataSetName + "】名称重复,请检查后重新运行！");
-            return;
-        }
-        if (list.isEmpty()) {
-            System.out.println("数据集:【" + dataSetName + "】不存在,请检查后重新运行！");
-            return;
-        }
-        //数据集ID
-        String zskId = list.getFirst().getStr("id");
-
-        //在上传文档前，需要删除掉旧的同名文档
-        File fi = new File(uploadFile);
-        String fileName = fi.getName();
-        //获取身份识别
-        String authCode = MaxKbCallUtil.getAuthorization();
-        //上传文档
-        List<Record> listDocuments = mm.getDocumentListByDataSetId(zskId);
-        for (Record record : listDocuments) {
-            String documentName = record.getStr("name");
-            if (documentName.equals(fileName)) {
-                String documentId = record.getStr("id");
-                //准备删除掉旧的文档
-                MaxKbCallUtil.delDocument(authCode, zskId, documentId);
-                System.out.println("删除旧的文档成功：" + documentName);
-            }
-        }
-        //上传新的文档
-        MaxKbCallUtil.uploadQA(authCode, zskId, uploadFile);
-        System.out.println("恭喜，所有操作成功完成！");
     }
 }