策略模式优化if-else

前⾔：

　　当if else的条件少的话，代码可阅读性及逻辑不影响阅读和扩展。⼀旦if else过多的话会导致逻辑⽐较混乱，不易扩展并且很容易出错。实现⽅案：

　　1、定义⼀个@HandlerType注解，⼀个value属性，其值对应 if 条件的值

　　2、定义⼀个接⼝或者抽象类，抽AbstractHandler，抽象逻辑处理⽅法handler⽅法交给具体的业务实现　　3、根据⾃⼰的业务，定义多个类实现AbstractHandler接⼝。每个实现类都注册到Spring容器中　　4、对每个AbstractHandler的实现类都标注上@HandlerType注解，值为其要处理 if 条件的值

　　5、定义⼀个HandlerContext，⽤来扫描具有 @HandlerType注解的类，并将注解中的value作为key，对应的类作为value，初始化其属性handlerMap

　　6、通过条件从Spring容器中获取⼀个对应的AbstractHandler实例　　7、执⾏实例对应的handler⽅法

举个例⼦：

　　有这样⼀个需求，需要监听⼀个Kafka topic下的不同类型的消息，类型从监听到的message中去区分，不同的类型的消息，处理逻辑也不同。代码：

　　1、定义⾃定义注解

/**

* Kafka监听消息的topic动作编号 *

* @author yangyongjie * @date 2019/11/7 * @desc */

@Target({ElementType.TYPE})

@Retention(RetentionPolicy.RUNTIME)@Documented

public @interface ListenerAction { /**

* topic动作编号 *

* @return */

int value();}

　　2、定义抽象处理器

/**

* 抽象的Kafka消息处理器 *

* @author yangyongjie * @date 2019/11/7 * @desc */

public abstract class AbstractListenerHandler { /**

* 消息处理 *

* @param data */

public abstract void handler(String data);}

　　3、具体实现处理器，多个，这⾥只列⼀个

　　4、在具体的处理器上标注@HandlerType注解，表⽰其要处理的逻辑分⽀，这⾥表⽰只处理action为2001的消息

/**

* PartnerId 变更消息监听处理器 * @author yangyongjie * @date 2019/11/7 * @desc */

@ListenerAction(2001)@Component

public class PartnerIdListenerHandler extends AbstractListenerHandler { @Override

public void handler(String data) { // dosomething }}

　　5、扫描具有 @HandlerType注解的类，并将注解中的value作为key，对应的类作为value，初始化其属性handlerMap

**

* 消息处理器的处理器

* 1.扫描包中标有@ListenerAction注解的类

* 2.将注解中的动作编号值作为key，对应的类作为value，初始化handlerMap *

* @author yangyongjie * @date 2019/11/7 * @desc */

@Component

public class ListenerHandlerContext implements ApplicationContextAware { private ApplicationContext applicationContext;

/**

* 保存处理器的map，key为action，value为处理器Class */

private static final Map HANDLER_MAP = new HashMap<>(4);

public AbstractListenerHandler getListenerHandlerInstance(Integer action) throws BssException { Class clazz = HANDLER_MAP.get(action); if (clazz == null) {

throw new BssException(ResponseEnum.ERROR_HANDLER_NOT_FOUND.getCode(), ResponseEnum.ERROR_HANDLER_NOT_FOUND.getMsg()); }

return (AbstractListenerHandler) applicationContext.getBean(clazz); }

/**

* 扫描@ListenerAction，初始化ListenerHandlerContext，将其注册到Spring容器中 *

* @param applicationContext * @throws BeansException */

@Override

public void setApplicationContext(ApplicationContext applicationContext) throws BeansException { this.applicationContext = applicationContext; // 遍历所有带有@ListenerAction注解的类

Map listenerHandlerBeans = applicationContext.getBeansWithAnnotation(ListenerAction.class); if (MapUtils.isNotEmpty(listenerHandlerBeans)) {

for (Object listenerHandlerBean : listenerHandlerBeans.values()) {

Integer action = listenerHandlerBean.getClass().getAnnotation(ListenerAction.class).value(); HANDLER_MAP.put(action, listenerHandlerBean.getClass()); } } }}

　　6、获取处理器，执⾏逻辑分⽀

@Autowired

private ListenerHandlerContext listenerHandlerContext;...

// 获取消息监听处理器

AbstractListenerHandler listenerHandler = listenerHandlerContext.getListenerHandlerInstance(action);// kafka消息逻辑处理

listenerHandler.handler(data);

　　END.