带你读《2022技术人的百宝黑皮书》——一个搞定责任链的注解（7）

/**
* 消息限流器
*/ @AutoPipeline
public interface MessageThrottler {
/**
*节流单个消息
*
*@param messageThrottlerToken 消息限流令牌
*@return 是否被节流
*/
boolean throttle(MessageThrottlerToken messageThrottlerToken);

  /**
  * 消息分类节流器
  *
  * <p>
  * 目前仅针对营销消息进行节流7  */
  public class ClassificationThrottlerHandler implements MessageThrottlerHandler { 9
@Override
public boolean throttle(MessageThrottlerToken messageThrottlerToken, MessageThrottlerHandler- Context context) {
if (!ClassificationConstant.MARKETING.equals(messageThrottlerToken.getClassification())) { return false;
}
return context.throttle(messageThrottlerToken);
}
@Override
  public boolean anyThrottle(List<MessageThrottlerToken> messageThrottlerTokens, MessageThrot- tlerHandlerContext context) {
if (CollectionUtils.isEmpty(messageThrottlerTokens)) {
return false;
  }
// 获取营销消息
  List<MessageThrottlerToken> marketingMessageThrottlerTokens = messageThrottlerTo- kens.stream().filter(messageThrottlerToken -> {
return ClassificationConstant.MARKETING.equals(messageThrottlerToken.getClassification());
}).collect(Collectors.toList()); 
// 如果营销消息为空，说明消息均不需要被限流，直接返回false
if (CollectionUtils.isEmpty(marketingMessageThrottlerTokens)) {
return false;
  }
    return context.anyThrottle(marketingMessageThrottlerTokens); 
}
@Override
  public boolean allThrottle(List<MessageThrottlerToken> messageThrottlerTokens, MessageThrot- tlerHandlerContext context) {
if (CollectionUtils.isEmpty(messageThrottlerTokens)) {
return false;
  }
// 获取营销消息
  List<MessageThrottlerToken> marketingMessageThrottlerTokens = messageThrottlerTo- kens.stream().filter(messageThrottlerToken -> {
return ClassificationConstant.MARKETING.equals(messageThrottlerToken.getClassification());
}).collect(Collectors.toList()); 
// 存在非营销消息，非营销消息不会被限流
if (marketingMessageThrottlerTokens.size() < messageThrottlerTokens.size()) {
return false;
  }
  return context.allThrottle(marketingMessageThrottlerTokens);
}
}