新增情感词删除接口

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();
+    }
+
 }

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;
+    }
 }

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);
     }
 }