spring中策略模式使用

策略模式

工作中經常使用到策略模式+工廠模式，實現一個接口多種實現的靈活調用與后續代碼的擴展性。在spring中使用策略模式更為簡單，所有的bean均為spring容器管理，只需獲取該接口的所有實現類即可。

下面以事件處理功能為例，接收到事件之后，根據事件類型調用不同的實現接口去處理。如需新增事件，只需擴展實現類即可，無需改動之前的代碼。這樣即做到了功能的隔離，又可防止改動原代碼導致的bug。

類圖

代碼示例

定義接口

public interface IBaseEventService {

    /**
     * 處理事件
     * @param eventObject
     * @return
     * @throws Exception
     */
    public boolean dealEvent(String eventObject);

    /**
     * 獲取事件類型
     * @return
     */
    public String getType();

}

接口實現

@Service
public class AddUserEventServiceImpl implements IBaseEventService {
    @Override
    public boolean dealEvent(String eventObject) {
        // TODO 業務處理邏輯
        return false;
    }

    @Override
    public String getType() {
        return EventTypeEnum.ADD_USER_AFTER.getKey();
    }
}

常量定義

public enum EventTypeEnum {
    ADD_USER_AFTER("ADD_USER_AFTER"),
    PLACE_ORDER_AFTER("PLACE_ORDER_AFTER");

    private String key;

    EventTypeEnum(String key) {
        this.key = key;
    }

    public String getKey() {
        return key;
    }

    public void setKey(String key) {
        this.key = key;
    }
}

策略類

@Service
public class EventStrategyService {

    Map<String, IBaseEventService> eventServiceMap = new HashMap<>();

    /**
     * 構造函數
     * @param eventServices spring容器中所有IBaseEventService的實現類
     */
    public EventStrategyService(List<IBaseEventService> eventServices) {
        for (IBaseEventService eventService : eventServices) {
            eventServiceMap.put(eventService.getType(), eventService);
        }
    }
    
    /**
     * 根據事件類型調用不同的實現類處理
     */
    public boolean dealEvent(String eventType, String eventObject) {
        IBaseEventService eventService = eventServiceMap.get(eventType);
        if (eventService == null){
            throw new BizException("未找到事件處理實現類，eventType：" + eventType);
        }
        return eventService.dealEvent(eventObject);
    }

}

接口調用

    @Autowired
    private EventStrategyService eventStrategyService;
    
    //處理事件
    eventStrategyService.dealEvent(eventType, userObject);