自定义注解

 适应于策略模式

/**
 * 自定义注解
 */
@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
@Component
public @interface Action{

     Type value();
}

/**
 * 自定义类型
 */
@Getter
@AllArgsConstructor
public enum Type {

    BALANCE("B","余额"),
    POINTS("P","积分")
    ;

    private String code;

    private String message;

    public static Type of(String code) {
        String message = String.format(ErrorCode.ENUM_CODE_NOT_SUPPORT.message(), "ActionType", code);
        return Stream.of(ActionType.values()).filter(actionType -> actionType.getCode().equals(code)).findFirst().orElseThrow(() -> new BizException(ErrorCode.ENUM_CODE_NOT_SUPPORT.code(), message));
    }
}
/**
 * 选择上下文,把含有注解的类 注入到spring
 */
@Component
public class ActionContext implements BeanPostProcessor {

    public final static Map<ActionType, IActionStrategy> strategyMap = Maps.newHashMap();

    private final static TransmittableThreadLocal<TriggerActionDP> context = new TransmittableThreadLocal<>();

    public static void clear() {
        ActionContext.context.remove();
    }


    @Override
    public Object postProcessBeforeInitialization(Object bean, String beanName) throws BeansException {
        return bean;
    }

    @Override
    public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
        ActionKey actionKey = AnnotationUtils.findAnnotation(bean.getClass(), ActionKey.class);
        if (actionKey != null && IActionStrategy.class.isAssignableFrom(bean.getClass())) {
            strategyMap.put(actionKey.value(), (IActionStrategy) bean);
        }
        return bean;
    }

    public static void set(TriggerActionDP triggerAction) {
        context.set(triggerAction);
    }

    public static TriggerActionDP get() {
        return context.get();
    }

}
/**
 * 策略类,做一些业务处理
 */ 
@Slf4j
@ActionKey(ActionType.BALANCE)
public class ActionBalanceStrategy extends AbstractFundStrategy {


    /**
     *  
     */
    @Override
    public void doSomething() {
       

    }
}
     
  //选择处理具体的子类执行任务
   Type type =  Type.of(actionType);
  ActionContext.strategyMap.get(type).doSomething();

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值