Java留言分类方法详解与实现

public class MessageClassifier {

    private Map<String, List<String>> keywordMap = new HashMap<>();

    public MessageClassifier() {
        // 初始化关键词
        keywordMap.put("投诉", Arrays.asList("不满", "投诉", "差劲", "糟糕", "气愤"));
        keywordMap.put("咨询", Arrays.asList("请问", "怎么", "如何", "哪里", "什么时候"));
        keywordMap.put("建议", Arrays.asList("建议", "可以改进", "希望", "提议"));
    }

    public String classify(String message) {
        for (Map.Entry<String, List<String>> entry : keywordMap.entrySet()) {
            for (String keyword : entry.getValue()) {
                if (message.contains(keyword)) {
                    return entry.getKey();
                }
            }
        }
        return "其他"; // 默认分类
    }
}

MessageClassifier classifier = new MessageClassifier();
String category = classifier.classify("请问这个功能怎么用？");
System.out.println(category); // 输出：咨询

<dependency>
    <groupId>com.huaban</groupId>
    <artifactId>jieba-analysis</artifactId>
    <version>1.0.2</version>
</dependency>

rule "classify_complaint"
when
    $m : Message( content matches "(?i).*.(投诉|不满|差劲).*" )
then
    $m.setCategory("投诉");
end

@RestController
public class MessageController {

    private final MessageClassifier classifier = new MessageClassifier();

    @PostMapping("/classify")
    public Map<String, String> classify(@RequestBody Map<String, String> request) {
        String msg = request.get("message");
        String category = classifier.classify(msg);
        return Map.of("message", msg, "category", category);
    }
}