怎么用redis发布订阅方式实现简易的消息系统

WBOY转载: 2023-05-26 20:52:31753浏览

I. 基本使用

1. 配置

我们使用 SpringBoot 2.2.1.RELEASE来搭建项目环境，直接在pom.xml中添加 redis 依赖

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-data-redis</artifactId>
</dependency>

如果我们的 redis 是默认配置，则可以不额外添加任何配置；也可以直接在application.yml配置中，如下

spring:
  redis:
    host: 127.0.0.1
    port: 6379
    password:

2. 使用姿势

redis 的发布/订阅，主要就是利用两个命令publish/subscribe; 在 SpringBoot 中使用发布订阅模式比较简单，借助 RedisTemplate 可以很方便的实现

a. 消息发布

@Service
public class PubSubBean {
    @Autowired
    private StringRedisTemplate redisTemplate;

    public void publish(String key, String value) {
        redisTemplate.execute(new RedisCallback<Object>() {
            @Override
            public Object doInRedis(RedisConnection redisConnection) throws DataAccessException {
                redisConnection.publish(key.getBytes(), value.getBytes());
                return null;
            }
        });
    }
}

b. 订阅消息

消息订阅这里，需要注意我们借助org.springframework.data.redis.connection.MessageListener来实现消费逻辑

public void subscribe(MessageListener messageListener, String key) {
    redisTemplate.execute(new RedisCallback<Object>() {
        @Override
        public Object doInRedis(RedisConnection redisConnection) throws DataAccessException {
            redisConnection.subscribe(messageListener, key.getBytes());
            return null;
        }
    });
}

c. 测试 case

写一个简单的测试 case，来验证一下上面的发布订阅，顺带理解一下这个MessageListener的使用姿势；我们创建一个简单的 WEB 工程，提供两个 rest 接口

@RestController
@RequestMapping(path = "rest")
public class DemoRest {
    @Autowired
    private PubSubBean pubSubBean;

    // 发布消息
    @GetMapping(path = "pub")
    public String pubTest(String key, String value) {
        pubSubBean.publish(key, value);
        return "over";
    }

    // 新增消费者
    @GetMapping(path = "sub")
    public String subscribe(String key, String uuid) {
        pubSubBean.subscribe(new MessageListener() {
            @Override
            public void onMessage(Message message, byte[] bytes) {
                System.out.println(uuid + " ==> msg:" + message);
            }
        }, key);
        return "over";
    }
}

我们先创建了两个消费者，然后发送消息时，两个都收到；再新增一个消费者，发送消息时，三个都能收到

3. 使用说明与应用场景

redis 的发布订阅，只适用于比较简单的场景，从上面的使用说明也能看出，它就是一个简单的发布订阅模型，支持 1 对 N，而且发送的消息，只有在线的消费者才能 get 到（至于不在线的，那就只能说遗憾了）而且对于 redis 而言，消息推出去之后就完事了，至于消费者能不能正常消费，那就不 care 了

划重点：