Spring中策略模式实现方法

一、定义

在策略模式（Strategy Pattern）中，一个类的行为或其算法可以在运行时更改。这种类型的设计模式属于行为型模式。在策略模式中，我们创建表示各种策略的对象和一个行为随着策略对象改变而改变的 context 对象。策略对象改变 context 对象的执行算法。由于spring中我们需要注册对象，不能通过直接通过 **new** 来创建策略实现类，需要使用工厂模式来获取bean对象

二、实现方法

1.策略模式接口

public interface StrangeService {

     int executeStrange(int id, double credit);

}

2.定义策略模式的实现类

StrangeActivityAddServiceImpl和、StrangeActivitySubServiceImpl

StrangeActivityAddServiceImpl

@Service
@Slf4j
public class StrangeActivityAddServiceImpl implements StrangeService{

    @Autowired
    private ActivityMapper activityMapper;

    @Override
    public int executeStrange(int id, double credit) {
        log.info("传入参数:id=[{}],credit=[{}]",id,credit);
        Activity activity=activityMapper.selectByPrimaryKey(id);
        activity.setCredit(activity.getCredit()+credit);
        int res=activityMapper.updateByPrimaryKeySelective(activity);
        System.out.println("增加积分策略结果:"+res);
        return res;
    }
}

StrangeActivitySubServiceImpl

@Service
@Slf4j
public class StrangeActivitySubServiceImpl implements StrangeService{
    @Autowired
    private ActivityMapper activityMapper;

    @Override
    public int executeStrange(int id, double credit) {
        log.info("传入参数:id=[{}],credit=[{}]",id,credit);
        Activity activity=activityMapper.selectByPrimaryKey(id);
        activity.setCredit(activity.getCredit()-credit);
        int res=activityMapper.updateByPrimaryKeySelective(activity);
        System.out.println("增加积分策略结果:"+res);
        return res;
    }
}

3.策略模式上下文

用于获取注入的策略实现类的bean对象

（1）通过ConcurrentHashMap的形式获取注入的对象

@Service
public class StrangeContext {

    @Autowired
    private final Map<String, StrangeService> map = new ConcurrentHashMap<>();

    /**
     * 根据服务实现类名称获取对应的服务
     * @param serviceName
     * @return
     */
    public StrangeService strangeService(String serviceName) {
        StrangeService strangeService = map.get(serviceName);
        if (strangeService == null) {
            throw new RuntimeException("未定义此策略实现类!");
        }
        return strangeService;
    }

}

（2）通过SpringContextHolder重写ApplicationContextAware获取上下文对象获取指定bean

StrangeService strangeService=SpringContextHolder.getBean(StrangeActivityConstant.STRANGE_CREDIT_ADD_SERVICE,StrangeActivityAddServiceImpl.class);
strangeService.executeStrange(1,1);

4.策略模式实现类常量名

public interface StrangeActivityConstant {

    String STRANGE_CREDIT_ADD_SERVICE = "strangeActivityAddServiceImpl";

    String STRANGE_CREDIT_MULTIPLY_SERVICE = "strangeActivityMultiServiceImpl";

    String STRANGE_CREDIT_SUBTRACT_SERVICE = "strangeActivitySubServiceImpl";
}

5.使用方式

在spring中注入上下文StrangeContext对象，然后执行方法获取currentHashMap中策略实现类Impl的bean对象，再调用策略实现类Impl的具体实现方法

@Autowired
private StrangeContext strangeContext;

@RequestMapping("/test15")
public void testStrangePattern(){
    //为活动id=1的活动增加积分
strangeContext.strangeService(StrangeActivityConstant.STRANGE_CREDIT_ADD_SERVICE).executeStrange(1,9);

    //为活动id=3的活动减去积分
strangeContext.strangeService(StrangeActivityConstant.STRANGE_CREDIT_SUBTRACT_SERVICE).executeStrange(3,1);
}