面试官:使用 RocketMQ 怎么进行灰度发布?

今天来聊一聊 RocketMQ 的灰度方案。

灰度发布是指在黑与白之间,平滑过渡的一种发布方式。在大流量的系统中,如果一次升级改造范围比较大,或者影响内容不太确定,一般会采用切量的方式进行升级,这样可以减少生产变更带来的影响。

编辑切换为居中

添加图片注释,不超过 140 字(可选)

如上图,对 ServiceA 这个服务进行升级,采用灰度发布,先升级 Server5,一周后如果没有问题,升级 Server4 和 Server 3,再运行一周没有问题,把剩下两个节点都升级。

上面的案例是一个 RPC 的调用。但如果使用消息队列该怎么做呢?使用消息队列,并不能使用网关来进行流量转发。这里需要分不同场景进行分析。

1 只升级消费者

这是最简单的情况,比如只有消费者修改了消费逻辑,就是 RPC 调用的情况类似,我们只要把消费者进行灰度发布就可以。如下图:

编辑切换为居中

添加图片注释,不超过 140 字(可选)

2 生产者也升级

下面是一个订单的实体类,我们新加了一个属性,订单生成时间:

 
 

public class Order { private Long id; private Long userId; private Long productId; private Integer count; private BigDecimal payAmount; /**订单状态:0:创建中;1:已完结*/ private Integer status; /**新加属性,订单生成时间*/ private String createTime; }

消费端的改造是需要对 createTime 这个属性进行处理。

2.1 消费端过滤

在生产者的 Order 类中增加 createTime 属性,如果我们直接使用 createTime 属性来过滤,消费者并不能实现灰度,因为所有的消费者都可能会拉取到带有 createTime 属性的消息。

RocketMQ 中 Message 的定义如下:

 
 

public class Message implements Serializable { private String topic; private int flag; private Map<String, String> properties; private byte[] body; private String transactionId; }

可以在 properties 属性中增加一个灰度标识,比如生产者发送消息的时候封装如下:

 
 

Message msg = buildMessage(topic); msg.putUserProperty("gray", "true");

注意:也可以在 SendMessageHook 这个钩子函数中定义。通过这种方式可以在消费端新增加一个灰度 Consumer Group,用来对灰度消息则进行消费。如下图:

编辑切换为居中

添加图片注释,不超过 140 字(可选)

对于灰度 Consumer Group 判断到 gray 属性是 true 时进行消费,而对于普通 Consumer Group,判断到 gray 属性不等于 true 时再进行消费。这里可以借助 RocketMQ 客户端的 FilterMessageHook,代码如下:

 
 

defaultMQPushConsumerImpl.registerFilterMessageHook(new FilterMessageHook() { @Override public String hookName() { return "filterHook"; } @Override public void filterMessage(FilterMessageContext context) { List<MessageExt> messages = context.getMsgList(); context.setMsgList(messages.stream().filter(m -> StringUtils.equals(m.getProperty("gray"),"true")) .collect(Collectors.toList())); } });

不过这样会有两个问题,灰度和正常的两个 Consumer Group 相当于是广播组:

  1. 两个组都要对所有的消息进行拉取,比如本来使用灰度发布计划切 10% 的流量,但实际上全部流量都切过去了,只是根据属性做了判断。这让消费端整体承担了两倍的压力;

  2. 因为两个消费者组都要去 Broker 拉取消息,Broker 的压力也增加了一倍。

2.2 Broker 过滤

2.2.1 使用 tag 过滤

如果一个 Consumer 不订阅一个 Topic 中的全部消息,可以通过 Tag 来过滤。比如一个 Consumer 订阅了 TopicA 这个 Topic 中的 Tag1 和 Tag2 这两个 tag,那这个 Consumer 的订阅关系如下图:

SubscriptionData 这个对象封装了 Topic、tag 以及所订阅 tag 的 hashcode 集合。

Consumer 发送拉取消息请求时,会把订阅关系传给 Broker(Broker 解析成 SubscriptionData 对象),Broker 使用 consumequeue 获取消息时,首先判断判断最后 8 个字节的 tag hashcode 是否在 SubscriptionData 的 codeSet 中,如果不在就跳过,如果存在把消息返回给 Consumer。如下图:

编辑切换为居中

添加图片注释,不超过 140 字(可选)

这样可以在灰度 Producer 发送消息时加上 Tag,如下代码:

 
 

Message msg = new Message(); msg.setBody("Test"); msg.setTopic("Topic"); msg.setTags("Gray");

而在灰度消费者订阅 Gray 这个 tag。这样就避免了 2.1 节中消息全量拉取的问题。

2.2.2 使用 SQL92 过滤

使用 SQL92 过滤,可以应对更加复杂的场景,不仅可以过滤 Tag,还可以过滤 UserProperty。

比如下面是一个生产者的代码:

 
 

Message msg = new Message(); msg.setTopic("testTopic"); msg.setTags("tag1"); msg.putUserProperty("gray","true");

这样消费者初始化的时候,可以定义使用 SQL92 过滤,代码如下:

 
 

consumer.subscribe("testTopic", MessageSelector.bySql("(TAGS is not null and TAGS in TAGS='''''tag1''''')" + "and (gray is not null gray='true')"));

下面是 bySql 的源代码:

 
 

public static MessageSelector bySql(String sql) { return new MessageSelector(ExpressionType.SQL92, sql); }

3 总结

本文介绍了 RocketMQ 灰度消息的使用方法,场景比较简单。对于全链路的复杂灰度场景,可以参考使用阿里的微服引擎 MSE。

                       资源获取:
大家 点赞、收藏、关注、评论啦 、 查看👇🏻👇🏻👇🏻 微信公众号获取联系方式👇🏻👇🏻👇🏻
精彩专栏推荐订阅:下方专栏👇🏻👇🏻👇🏻👇🏻
每天学四小时:Java+Spring+JVM+分布式高并发,架构师指日可待

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值