巧用策略模式实现if-else的判断

巧用策略模式实现if-else的判断

1、Strategy接口

/**
 * @ClassName StrategyService
 * @Description TODO
 * @Author hl
 * @Date 2021/3/17 11:24
 * @Version 1.0
 */
public interface StrategyService {

    /**
     * 策略模式的处理方法
     * @return String
     */
    String doOperation();
}

2、StrategyImpl实现类

import com.example.strategystudy.strategy.service.StrategyService;
import org.springframework.stereotype.Component;

/**
 * @ClassName StrategyOne
 * @Description TODO
 * @Author hl
 * @Date 2021/3/17 11:29
 * @Version 1.0
 */
@Component("Tencent")
public class StrategyTencentImpl implements StrategyService {

    @Override
    public String doOperation() {
        return "print Tencent";
    }
}


import com.example.strategystudy.strategy.service.StrategyService;
import org.springframework.stereotype.Component;

/**
 * @ClassName StrategyTwoImpl
 * @Description TODO
 * @Author hl
 * @Date 2021/3/17 11:32
 * @Version 1.0
 */
@Component("Wechat")
public class StrategyWechatImpl implements StrategyService {

    @Override
    public String doOperation() {
        return "print Wechat";
    }
}

3、将实现Strategy的实现类都自动注入到strategyMap类

import com.example.strategystudy.strategy.service.StrategyService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
 * @ClassName FactoryForStrategy
 * @Description TODO
 * @Author hl
 * @Date 2021/3/17 11:18
 * @Version 1.0
 */
@Service
public class FactoryForStrategy {

    /**
     * 初始化的时候将所有的StrategyService自动加载到Map中
     */
    @Autowired
    private final Map<String, StrategyService> strategies = new ConcurrentHashMap<>();

    public StrategyService getStrategy(String component) throws Exception{
        StrategyService strategy = strategies.get(component);
        if(strategy == null) {
            throw new RuntimeException("no strategy defined");
        }
        return strategy;
    }
}

4、测试结果

  @Test
    public void pattern(){
        try {
            String str = factoryForStrategy.getStrategy("Wechat").doOperation();
            log.info("测试结果{}", str);
        } catch (Exception e) {
            log.error("error:{}", e.getMessage());
        }
    }