parent
22f682b13b
commit
d87f85f621
@ -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("恭喜,所有操作成功完成!");
|
||||
}
|
||||
}
|
@ -1,61 +1,20 @@
|
||||
package Tools.MaxKb;
|
||||
|
||||
import Tools.MaxKb.Model.MaxKbModel;
|
||||
import Tools.MaxKb.Util.MaxKbCallUtil;
|
||||
import Tools.MaxKb.Service.MaxKbService;
|
||||
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();
|
||||
|
||||
//获取数据集名称
|
||||
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("恭喜,所有操作成功完成!");
|
||||
//上传文件
|
||||
MaxKbService.uploadQA(uploadFile);
|
||||
}
|
||||
}
|
Loading…
Reference in new issue