Java开发聊天机器人

Shiro是使用JAVA开发的基于 OneBo-V11 协议的 QQ机器人快速开发框架

文档：快速开始 | Shiro SDK Document

示例Demo：https://gitee.com/hidewnd/shiro-demo.git

SpringBoot3.0.0+
jdk17+

机器人框架基于springboot开发，可嵌入现有springboot框架项目中或作为独立子服务运行，目前框架支持go-cphttp协议的绝大部分oneBot事件或请求，支持自行扩展事件及请求

其他语言框架：

Python：NoneBot
TypeScript：Koishi

客户端配置反向连接：ws://127.0.0.1:5555/ws/shiro

非机器人端连接需配置header：x-self-id={机器人账号}

实现效果：

配置文件

1
server:
2
  port: 5555
3
shiro:
4
  interceptor: com.hidewnd.shiro.plugins.InterceptorExample
5
  # 注解方式编写的插件无需在插件列表（plugin-list）定义
6
  # 插件列表为顺序执行，如果前一个插件返回了 MESSAGE_BLOCK 将不会执行后续插件
7
  plugin-list:
8
    - com.hidewnd.shiro.plugins.OverridePlugin
9
  ws:
10
    # 访问密钥，强烈推荐在公网的服务器设置
11
    access-token: ""
12
    # 超时回收，默认10秒
13
    timeout: 10
14
    # 最大文本消息缓冲区
15
    max-text-message-buffer-size: 512000
16
    # 二进制消息的最大长度
17
    max-binary-message-buffer-size: 512000
18
    # 反向 Websocket 连接地址
19
    server:
20
      enable: true
21
      # 默认值 "/ws/shiro"
22
      url: "/ws/shiro"
23
      # 最大空闲时间，超过这个时间将关闭会话
24
      max-session-idle-timeout: 900000
25
  # 限速器（令牌桶算法）
26
  limiter:
27
    # 是否启用限速器
28
    enable: false
29
    # 补充速率（每秒补充的令牌数量）
30
    rate: 1
31
    # 令牌桶容量
32
    capacity: 1
33
    # 如果该值为 false 时，当令牌获取失败则会直接丢次本次请求
34
    # 如果该值为 true 时，当令牌获取失败则会阻塞当前线程，后续任务将被添加到等待队列
35
    awaitTask: true
36
    # 等待超时
37
    timeout: 10
38
  # 线程池配置
39
  task-pool:
40
    # 核心线程数（默认线程数）
41
    core-pool-size: 10
42
    # 缓冲队列大小
43
    queue-capacity: 200
44
    # 允许线程空闲时间（单位：默认为秒）
45
    keep-alive-time: 10
46
    # 最大线程数
47
    max-pool-size: 30
48
    # 线程池名前缀
49
    thread-name-prefix: "ShiroTaskPool-"

具体实现

在机器人项目框架中，通常以插件的概念作为机器人的功能体现，一个机器人可拥有多个插件，进而获得多种功能；

插件概念类似于MVC架构中的Controller层，作为对外提供的接口，插件中声明的不同命令即代表可用的接口；

通常一个插件下的命令代表同一主题功能的不同命令，可通过项目构造香一个大型功能拆分为多个子插件开发，便于项目维护；

编写插件-重写父类方式

继承父类BotPlugin
重写方法：onGroupMessage、onPrivateMessage、onAnyMessage…
配置文件shiro.plugin-list中声明重写类

需注意的点：

重写方法实现的插件根据配置文件上下声明顺序依次执行
重写方法，如果前一个插件返回了 MESSAGE_BLOCK 将不会执行后续插件

代码示例：

1
package com.hidewnd.shiro.plugins;
2

3
import ch.qos.logback.core.util.StringUtil;
4
import com.mikuac.shiro.annotation.common.Order;
5
import com.mikuac.shiro.common.utils.MsgUtils;
6
import com.mikuac.shiro.core.Bot;
7
import com.mikuac.shiro.core.BotPlugin;
8
import com.mikuac.shiro.dto.action.common.ActionData;
9
import com.mikuac.shiro.dto.action.common.MsgId;
10
import com.mikuac.shiro.dto.event.message.GroupMessageEvent;
11
import com.mikuac.shiro.dto.event.message.PrivateMessageEvent;
12
import org.springframework.stereotype.Component;
13

14
/**
15
 * 目前发现晚于注解方式编写的插件
16
 */
17
@Component
18
public class OverridePlugin extends BotPlugin {
19

20
    @Order(1)
21
    @Override
22
    public int onPrivateMessage(Bot bot, PrivateMessageEvent event) {
23
        String message = StringUtil.nullStringToEmpty(event.getMessage());
24
        bot.sendPrivateMsg(event.getUserId(), message, false);
25
        // 返回 MESSAGE_IGNORE 执行 plugin-list 下一个插件，返回 MESSAGE_BLOCK 则不执行下一个插件
26
        return MESSAGE_IGNORE;
27
    }
28

29
    @Order(2)
30
    @Override
31
    public int onGroupMessage(Bot bot, GroupMessageEvent event) {
32
        String message = StringUtil.nullStringToEmpty(event.getMessage());
33
        bot.sendGroupMsg(event.getGroupId(), message, false);
34
        // 返回 MESSAGE_IGNORE 执行 plugin-list 下一个插件，返回 MESSAGE_BLOCK 则不执行下一个插件
35
        return MESSAGE_IGNORE;
36
    }
37
}

编写插件-注解声明方式

一个类代表一个插件，在类上添加注解@Shiro、@Component的方式由spring接管实例，并识别成一个插件类
插件类中的方法通过方法注解@PrivateMessageHandler、@GroupMessageHandler、AnyMessageHandler声明对不同的消息事件进行处理，常用入参包括：Bot、PrivateMessageEvent、GroupMessageEvent、Matcher…，对于不同类型的事件，可用的入参有所不同。
命令支持全匹配、前后匹配、正则匹配，正则匹配可通过Matcher获取匹配后的参数信息

代码示例：

1
package com.hidewnd.shiro.plugins;
2

3
import cn.hutool.core.util.StrUtil;
4
import com.mikuac.shiro.annotation.AnyMessageHandler;
5
import com.mikuac.shiro.annotation.GroupMessageHandler;
6
import com.mikuac.shiro.annotation.MessageHandlerFilter;
7
import com.mikuac.shiro.annotation.PrivateMessageHandler;
8
import com.mikuac.shiro.annotation.common.Shiro;
9
import com.mikuac.shiro.common.utils.MsgUtils;
10
import com.mikuac.shiro.core.Bot;
11
import com.mikuac.shiro.dto.event.message.AnyMessageEvent;
12
import com.mikuac.shiro.dto.event.message.GroupMessageEvent;
13
import com.mikuac.shiro.dto.event.message.PrivateMessageEvent;
14
import com.mikuac.shiro.enums.AtEnum;
15
import lombok.extern.slf4j.Slf4j;
16
import org.springframework.stereotype.Component;
17

18
import java.util.regex.Matcher;
19

20

21
@Slf4j
22
@Shiro
23
@Component
24
public class ExamplePlugin {
25

26
    // 当机器人收到的私聊消息消息符合 cmd 值 "hi" 时，这个方法会被调用。
27
    @PrivateMessageHandler
28
    @MessageHandlerFilter(cmd = "hi")
29
    public void fun1(Bot bot, PrivateMessageEvent event, Matcher matcher) {
30
        // 构建消息
31
        String sendMsg = MsgUtils.builder().face(66).text("this is privateMessage callBack").build();
32
        // 发送私聊消息
33
        bot.sendPrivateMsg(event.getUserId(), sendMsg, false);
34
    }
35

36
    // 如果 at 参数设定为 AtEnum.NEED 则只有 at 了机器人的消息会被响应
37
    @GroupMessageHandler
38
    @MessageHandlerFilter(at = AtEnum.NEED)
39
    public void fun2(GroupMessageEvent event) {
40
        // 以注解方式调用可以根据自己的需要来为方法设定参数
41
        // 例如群组消息可以传递 GroupMessageEvent, Bot, Matcher 多余的参数会被设定为 null
42
        log.info("群组消息: {}", event.getMessage());
43
    }
44

45
    @GroupMessageHandler
46
    @MessageHandlerFilter(startWith = "测试")
47
    public void fun3(Bot bot, GroupMessageEvent event, Matcher matcher) {
48
        String message = event.getMessage();
49
        String group = matcher.group();
50
        String text = StrUtil.format("收到群组消息: {}，匹配到的内容: {}", message, group);
51
        bot.sendGroupMsg(event.getGroupId(), text, false);
52
    }
53

54
    // 同时监听群组及私聊消息 并根据消息类型（私聊，群聊）回复
55
    @AnyMessageHandler
56
    @MessageHandlerFilter(cmd = "hi")
57
    public void fun3(Bot bot, AnyMessageEvent event) {
58
        bot.sendMsg(event, "this is GroupMessage callBack", false);
59
    }
60

61
    @GroupMessageHandler
62
    @MessageHandlerFilter(cmd = "^匹配 (.*)?$")
63
    public void fun4(Bot bot, GroupMessageEvent event, Matcher matcher) {
64
        int index = 1;
65
        String message = matcher.group(0);
66
        String group = matcher.group(index);
67
        while (StrUtil.isNotEmpty(group)) {
68
            String text = StrUtil.format("收到群组消息: {}，匹配到的内容: {}", message , group);
69
            bot.sendGroupMsg(event.getGroupId(), text, false);
70
            group = matcher.groupCount() < index ? matcher.group(++index) : null;
71
        }
72

73
    }
74
}

机器人上下线事件处理

继承父类CoreEvent，并重写相关方法，可用于对ws上下线事件的处理及客户端黑名单处理

代码示例：

1
import ch.qos.logback.core.util.StringUtil;
2
import com.mikuac.shiro.core.Bot;
3
import com.mikuac.shiro.core.CoreEvent;
4
import lombok.extern.slf4j.Slf4j;
5
import org.springframework.context.annotation.Primary;
6
import org.springframework.stereotype.Component;
7
import org.springframework.web.socket.WebSocketSession;
8

9
@Slf4j
10
@Primary
11
@Component
12
public class BotRuntimeEvent extends CoreEvent {
13

14
    @Override
15
    public void online(Bot bot) {
16
        // 客户端上线事件
17
        // 例如上线后发送消息给指定的群或好友
18
        // 如需获取上线的机器人账号可以调用 bot.getSelfId()
19
        log.info("机器人({}) 上线了", bot.getSelfId());
20
    }
21

22
    @Override
23
    public void offline(long account) {
24
        // 客户端离线事件
25
        log.info("机器人({}) 下线了", account);
26
    }
27

28
    @Override
29
    public boolean session(WebSocketSession session) {
30
        // 可以通过 session.getHandshakeHeaders().getFirst("x-self-id") 获取上线的机器人账号
31
        // 例如当服务端为开放服务时，并且只有白名单内的账号才允许连接，此时可以检查账号是否存在于白名内
32
        // 不存在的话返回 false 即可禁止连接
33
        String botId = session.getHandshakeHeaders().getFirst("x-self-id");
34
        if (StringUtil.isNullOrEmpty(botId)) {
35
            log.warn("未知的连接请求");
36
        }
37
        log.info("机器人({})请求连接", botId);
38
        return true;
39
    }
40
}

WS拦截器处理

在事件开始响应前后进行预处理操作，可用于对敏感、违规参数预处理、功能命令权限限制等操作

具体步骤：

实现接口BotMessageEventInterceptor，并实现方法：preHandle()、afterCompletion()
配置文件声明类：shiro.interceptor= com.hidewnd.shiro.plugins.InterceptorExample

代码示例：

1
import com.mikuac.shiro.common.utils.MsgUtils;
2
import com.mikuac.shiro.core.Bot;
3
import com.mikuac.shiro.core.BotMessageEventInterceptor;
4
import com.mikuac.shiro.dto.event.message.GroupMessageEvent;
5
import com.mikuac.shiro.dto.event.message.MessageEvent;
6
import com.mikuac.shiro.exception.ShiroException;
7
import org.springframework.stereotype.Component;
8

9
/**
10
 * 拦截器
11
 */
12
@Component
13
public class InterceptorExample implements BotMessageEventInterceptor {
14
    @Override
15
    public boolean preHandle(Bot bot, MessageEvent event) throws ShiroException {
16
        // 预处理方法，可以在触发事件前做一些处理，返回值为 false 时本次事件将会被拦截
17
        // 使用场景可以是黑名单检查或检查权限等等，具体的使用场景可以发挥自己的想象力
18
        // MessageEvent 可能为右边这三种类型 PrivateMessageEvent、GroupMessageEvent、GuildMessageEvent
19
        String message = event.getMessage();
20
        String messageType = event.getMessageType();
21
        if (message.contains("敏感")) {
22
            String sendMsg = MsgUtils.builder()
23
                    .face(37)
24
                    .text("检测敏感数据，中止执行")
25
                    .build();
26
            if ("group".equals(messageType)) {
27
                bot.sendGroupMsg(((GroupMessageEvent) event).getGroupId(), sendMsg, false);
28
            }
29
            return false;
30
        }
31
        return true;
32
    }
33

34
    @Override
35
    public void afterCompletion(Bot bot, MessageEvent event) throws ShiroException {
36
        // 后置处理方法
37
        // 当 preHandle 返回值为 true 时则会执行 plugin-list，执行完毕后进入到 afterCompletion
38
    }
39
}

Java开发聊天机器人

相关依赖

配置文件

具体实现

编写插件-重写父类方式

编写插件-注解声明方式

机器人上下线事件处理

WS拦截器处理