转转基于MQ的分布式重试框架设计方案

/**
 * 需要保证最终一致性的函数
 */
public void doSomething(Object args) {
    try {
      // 执行事务的操作
      executeTransaction();
      // 提交事务
      commitTransaction();
    } catch (Exception e) {
        // 回滚事务
        rollbackTransaction();
        // 记录日志
        log.error(e);
        // 序列化参数
        byte[] body = serialize(args);
        // 构建消息, 指定Topic、Body
        Message msg = new Message("doSomethingTopic", body);
        // 发送失败重试消息
        mq.send(msg);
    }
}

/**
 * 消费者，用于失败重试处理
 */
@Consumer(topic = "doSomethingTopic")
public void consume(Message msg) {
    // 反序列化
    Object args = msg.deserialize();
    // 重试
    doSomething(args);
}

@Service
class Service {
 
    @MQRetry
    public void doSomething(String params1, String params2, List<String> params3) {
        //throw new RuntimeException(); 抛异常将重试
        //RetryContext.markRetryLater(); 标记为需要下次重试
 
        //int retryCount = RetryContext.getRetryCount(); 获取重试次数
    }
 
}
 
@Controller
class Controller {
     
    @Autowired
    private Service service;
 
    service.doSomething("1", "2", Arrays.asList("3", "4"));
}

/**
 * 基于MQ的分布式重试组件
 */
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface MQRetry {
    /**
     * 最大重试次数，默认与上限为16次
     */
    int maxAttempts() default 16;
    
    /**
     * 忽略的异常类列表，默认所有异常都重试
     */
    Class<? extends Throwable>[] exclude() default {};
    
    /**
     * 需要重试的异常类列表，默认所有异常都重试
     */
    Class<? extends Throwable>[] include() default {};
    
    /**
     * 出现异常时的处理函数, 格式: Bean名.方法名. 如: smsService.onError
     * 也可以只设置函数名, 不设置Bean名将执行本类的函数. 如: onError
     * 要求函数参数必须与重试函数的参数完全一致
     */
    String errorHandler() default "";
    
    /**
     * true: 第一次调用时, 同步执行@MQRetry函数, 如果失败再使用MQ
     * false: 调用@MQRetry函数时, 只会发送MQ
     */
    boolean firstSyncCall() default true;
    
    /**
     * 消费线程数，默认为20个
     */
    int consumeThread() default 20;
    
}