From fe66b6da9b0588008d952cd9f6402203ee504a20 Mon Sep 17 00:00:00 2001
From: hujunbo <9094908@qq.com>
Date: Wed, 30 Nov 2022 23:32:01 +0800
Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9E=E6=83=85=E6=84=9F=E8=AF=8D?=
 =?UTF-8?q?=E5=88=A0=E9=99=A4=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../business/controller/ReportController.java | 13 ++++++
 .../biemo/nlp/SentimentAnalysisUtils.java     | 42 +++++++++++++++++++
 .../java/com/hankcs/demo/DemoKeyword.java     |  2 +-
 3 files changed, 56 insertions(+), 1 deletion(-)

diff --git a/ruoyi-biemo/src/main/java/com/ruoyi/biemo/business/controller/ReportController.java b/ruoyi-biemo/src/main/java/com/ruoyi/biemo/business/controller/ReportController.java
index df5e5b5..19e978e 100644
--- a/ruoyi-biemo/src/main/java/com/ruoyi/biemo/business/controller/ReportController.java
+++ b/ruoyi-biemo/src/main/java/com/ruoyi/biemo/business/controller/ReportController.java
@@ -140,9 +140,22 @@ public class ReportController
         return AjaxResult.success("模型训练成功",SentimentAnalysisUtils.analysisImport(files,sentimentName));
     }
 
+    /**
+     * 获取当前的情感词
+     * @return
+     */
     @GetMapping("/sentiment/getCurrSentiments")
     public AjaxResult getCurrSentiments(){
         return AjaxResult.success("查询成功",SentimentAnalysisUtils.getCurrSentiments());
     }
 
+    /**
+     * 删除情感词目录
+     */
+    @DeleteMapping("/sentiment/removeSentiments/{sentiments}")
+    public AjaxResult removeSentiments(@PathVariable("sentiments") String[] sentimentsArray){
+        SentimentAnalysisUtils.removeSentiments(sentimentsArray);
+        return AjaxResult.success();
+    }
+
 }
diff --git a/ruoyi-biemo/src/main/java/com/ruoyi/biemo/nlp/SentimentAnalysisUtils.java b/ruoyi-biemo/src/main/java/com/ruoyi/biemo/nlp/SentimentAnalysisUtils.java
index 3777624..41970b4 100644
--- a/ruoyi-biemo/src/main/java/com/ruoyi/biemo/nlp/SentimentAnalysisUtils.java
+++ b/ruoyi-biemo/src/main/java/com/ruoyi/biemo/nlp/SentimentAnalysisUtils.java
@@ -93,4 +93,46 @@ public class SentimentAnalysisUtils {
         }
         return result;
     }
+
+    public static void removeSentiments(String[] sentimentsArray) {
+        if(sentimentsArray!=null&&sentimentsArray.length>0){
+            File file = new File(CORPUS_FOLDER);
+            for(int i=0;i<sentimentsArray.length;i++){
+                File f = new File(file.getAbsolutePath()+"/"+sentimentsArray[i]);
+                deleteFile(f);
+            }
+        }
+    }
+
+    public static void main(String[] args) {
+        String[] arr = new String[]{"测试","中性"};
+        removeSentiments(arr);
+    }
+
+    public static Boolean deleteFile(File file) {
+        //判断文件不为null或文件目录存在
+        if (file == null || !file.exists()) {
+            System.out.println("文件删除失败,请检查文件是否存在以及文件路径是否正确");
+            return false;
+        }
+        //获取目录下子文件
+        File[] files = file.listFiles();
+        //遍历该目录下的文件对象
+        for (File f : files) {
+            //判断子目录是否存在子目录,如果是文件则删除
+            if (f.isDirectory()) {
+                //递归删除目录下的文件
+                deleteFile(f);
+            } else {
+                //文件删除
+                f.delete();
+                //打印文件名
+                System.out.println("文件名：" + f.getName());
+            }
+        }
+        //文件夹删除
+        file.delete();
+        System.out.println("目录名：" + file.getName());
+        return true;
+    }
 }
diff --git a/ruoyi-hanlp/src/test/java/com/hankcs/demo/DemoKeyword.java b/ruoyi-hanlp/src/test/java/com/hankcs/demo/DemoKeyword.java
index 5342c73..d82a2cb 100644
--- a/ruoyi-hanlp/src/test/java/com/hankcs/demo/DemoKeyword.java
+++ b/ruoyi-hanlp/src/test/java/com/hankcs/demo/DemoKeyword.java
@@ -29,7 +29,7 @@ public class DemoKeyword
                 "但两者的界限并不非常清楚，特别是在中国。" +
                 "软件从业人员分为初级程序员、高级程序员、系统" +
                 "分析员和项目经理四大类。";
-        List<String> keywordList = HanLP.extractKeyword(content,3);
+        List<String> keywordList = HanLP.extractKeyword(content,5);
         System.out.println(keywordList);
     }
 }