From 737146b3c460dab0ea9e14376c3599d7bfb83adf Mon Sep 17 00:00:00 2001
From: yz <3614508250@qq.com>
Date: Wed, 31 May 2023 16:09:36 +0800
Subject: [PATCH] =?UTF-8?q?=E8=AF=8D=E6=80=A7=E6=A0=87=E6=B3=A8=E6=96=B0?=
=?UTF-8?q?=E5=A2=9E=E6=9F=B1=E7=8A=B6=E5=9B=BE=E5=8F=82=E6=95=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit
---
.../controller/DocInfoController.java | 29 +++++++++++++++++++
.../business/service/CategoryService.java | 3 +-
2 files changed, 31 insertions(+), 1 deletion(-)
diff --git a/ruoyi-biemo/src/main/java/com/ruoyi/biemo/business/controller/DocInfoController.java b/ruoyi-biemo/src/main/java/com/ruoyi/biemo/business/controller/DocInfoController.java
index cb2da75..f9be45c 100644
--- a/ruoyi-biemo/src/main/java/com/ruoyi/biemo/business/controller/DocInfoController.java
+++ b/ruoyi-biemo/src/main/java/com/ruoyi/biemo/business/controller/DocInfoController.java
@@ -103,6 +103,13 @@ public class DocInfoController extends BaseController {
result.put("labelCategories",MyObjects.labelCategories);
Sentence sentence = DependencyParserUtils.nerAnalysis(content);
List<IWord> wordList = sentence.wordList;
+ int noun=0;
+ int personalName=0;
+ int placeName=0;
+ int InstitutionalGroup=0;
+ int verb=0;
+ int adjective=0;
+ int adverb=0;
if(wordList!=null&&wordList.size()>0){
int offset = 0;
int id = 0;
@@ -115,6 +122,21 @@ public class DocInfoController extends BaseController {
Integer categoryId = MyObjects.wordsMapping.get(label);
JSONObject jsonObject = new JSONObject();
jsonObject.put("id",id);
+ if(categoryId==0){
+ noun++;
+ }else if(categoryId==1){
+ personalName++;
+ }else if(categoryId==2){
+ placeName++;
+ }else if(categoryId==3){
+ InstitutionalGroup++;
+ }else if(categoryId==4){
+ verb++;
+ }else if(categoryId==5){
+ adjective++;
+ }else if(categoryId==6){
+ adverb++;
+ }
jsonObject.put("categoryId",categoryId);
jsonObject.put("startIndex",startIndex);
jsonObject.put("endIndex",endIndex);
@@ -125,6 +147,13 @@ public class DocInfoController extends BaseController {
}
result.put("labels",labels);
}
+ result.put("noun",noun);
+ result.put("personalName",personalName);
+ result.put("placeName",placeName);
+ result.put("institutionalGroup",InstitutionalGroup);
+ result.put("verb",verb);
+ result.put("adjective",adjective);
+ result.put("adverb",adverb);
return AjaxResult.success("查询成功",result);
}
diff --git a/ruoyi-biemo/src/main/java/com/ruoyi/biemo/business/service/CategoryService.java b/ruoyi-biemo/src/main/java/com/ruoyi/biemo/business/service/CategoryService.java
index b84890c..01c82ea 100644
--- a/ruoyi-biemo/src/main/java/com/ruoyi/biemo/business/service/CategoryService.java
+++ b/ruoyi-biemo/src/main/java/com/ruoyi/biemo/business/service/CategoryService.java
@@ -63,7 +63,8 @@ public class CategoryService extends EsService<Category> {
if(id==null||id.trim().length()==0){
return null;
}
- List<Category> categories = esLambdaQuery().eq(Category::getId,id).query();
+// List<Category> categories = esLambdaQuery().eq(Category::getId,id).query();
+ List<Category> categories = esLambdaQuery().getDoc_InfoBy_Id(Category::getId,id).query();
return categories!=null&&categories.size()>0?categories.get(0):null;
}