代码重构之Spring+工厂+策略模式

策略模式定义

将一组算法封装到实现共同接口的独立类中，使得它们可以相互替换。

策略举例

项目中执行需求：有关类型1的需求，由A做；类型2的需求，由B做；类型3的需求，由C做。

代码实现

    public void doItem(int type) {
        if (type == 1) {
            // do action
            workA();
            System.out.println("A do work.");
        } else if (type == 2) {
            // do action
            workB();
            System.out.println("B do work.");
        } else if (type == 3) {
            // do action
            System.out.println("C do work.");
        } else {
            // do action
            workC();
            System.out.println("do it yourself.");
        }
    }

我们最快想到的，就是if…else…或者case。大多数情况下，这样是合理的。
但是，如果type越来越多，action越来越复杂，doItem方法就会变得越来越庞大，代码的可读性也就会越来越差。
下面，介绍一下前段时间在项目中的一次重构经历，主要用到策略模式。

策略模式简单示例

    // 定义策略接口
    public interface Strategy {
        public void work();
    }
    // 策略A
    public class StrategyA implements Strategy {
        @Override  
        public void work() {
            System.out.println("A do work.");
        }
    }
    // 策略B
    public class strategyB implements Strategy {
        @Override  
        public void work() {
            System.out.println("B do work.");
        }
    }
    // 策略C
    public class strategyC implements Strategy {
        @Override  
        public void work() {
            System.out.println("C do work.");
        }
    }
    // 环境类
    public class Context {
        private Strategy strategy;

        public Context(Strategy strategy) {
             this.strategy = strategy;
        }
        public void setStrategy(Strategy strategy) {
             this.strategy = strategy;
        }
        public void work() {
             this.strategy.work();
        }
    }
    public void doItem() {
        Context ct = new Context(new StrategyA());
        ct.work();
        ct.setStrategy(new StrategyB());
        ct.work();
    }

项目重构实战

原部分代码

    public void pushData2Sap(String type, List<Long> ids) {
        Object[] dtos = null;
        if (type.equals("customer")) {
            List<CustomerDTO> dtoList = customerService.query(ids);
            dtos = new Object[dtoList.size()];
            dtos = dtoList.toArray();
        } else if (type.equals("product")) {
            List<ProductDTO> dtoList = new ArrayList<ProductDTO>();
            for (Long id : ids) {
                ProductDTO d = productService.queryById(id);
                dtoList.add(d);
            }
            dtos = new Object[dtoList.size()];
            dtos = dtoList.toArray();
        } else if (type.equals("provider")) {
            List<ProviderDTO> dtoList = providerService.query(ids);
            dtos = new Object[dtoList.size()];
            dtos = dtoList.toArray();
        } else if (type.equals("contact")) {
            List<ContactDTO> dtoList = contactService.query(ids);
            dtos = new Object[dtoList.size()];
            dtos = dtoList.toArray();
        } 
        RedisHelper.push(dtos);
    }

Spring + 工厂 + 策略模式重构后

    // applicationContext-bean.xml
    <!-- 推送sap策略工厂 -->
    <bean id="contextSpringFactory" class="core.util.ContextSpringFactory">
        <property name="stgMap">   
            <map>
                <entry key="customer" value-ref="customerServiceImpl"/>
                <entry key="product" value-ref="productServiceImpl"/>
                <entry key="provider" value-ref="providerServiceImpl"/>
                <entry key="contact" value-ref="contactServiceImpl"/>
            </map>   
        </property> 
    </bean>
    // 策略工厂
    @Component
    public class ContextSpringFactory {
        private Map<String, IPushSapStrategy<?>> stgMap = new HashMap<String, IPushSapStrategy<?>>();
        public Map<String, IPushSapStrategy<?>> getStgMap() {
            return stgMap;
        }
        public void setStgMap(Map<String, IPushSapStrategy<?>> stgMap) {
            this.stgMap = stgMap;
        }
        public List<?> doAction(String strType, List<Long> ids) {
            return this.stgMap.get(strType).querySapObjs(ids);
        }
    }
    // RedisHelper
    // 引入策略工厂
    @Resource
    private ContextSpringFactory contectSpringFactory;

    public void pushData2Sap(String type, List<Long> ids) {
        List<?> sapDtos = contectSpringFactory.doAction(typeId, ids);
        Object[] dtos = dtoList.toArray();
        RedisHelper.push(dtos);
    }

这样，利用Spring+工厂模式+策略模式完成了代码重构