Spring中实现策略模式的几种方式
一.背景
在写业务代码的时候,难免会遇到很多if-else,这个时候如果if-else不是很多可以用if-else。如果此时场景过多,太多的if-else会导致代码比较臃肿,所以这个时候就需要抽象化,将每个场景独立开来,定义一个顶层接口,不同场景有不同实现,这个时候策略模式就出现了。本文主要阐述工作中常用的实现策略模式的几种方式。
二.实现一
2.1 定义接口
java">package com.ljm.service;import com.ljm.constant.StrategyEnum;/*** @Author: ljm* @Date: 2024/4/29 15:09* @Version: v1.0.0* @Description: TODO**/
public interface IHandler {void handler();StrategyEnum getHandleStrategy();
}
2.2 定义枚举
java">package com.ljm.constant;/*** @Author: ljm* @Date: 2024/4/29 15:10* @Version: v1.0.0* @Description: TODO**/
public enum StrategyEnum {FIRST, SECOND, THIRD, FOUR
}
2.3 定义实现类
java">import com.ljm.constant.StrategyEnum;
import com.ljm.service.IHandler;
import org.springframework.stereotype.Service;/*** @Author: ljm* @Date: 2024/4/29 15:12* @Version: v1.0.0* @Description: TODO**/
@Service
public class FirstHandler implements IHandler {@Overridepublic void handler() {System.out.println("Execute First Handler");}@Overridepublic StrategyEnum getHandleStrategy() {return StrategyEnum.FIRST;}
}
java">package com.ljm.service.impl;import com.ljm.constant.StrategyEnum;
import com.ljm.service.IHandler;
import org.springframework.stereotype.Service;/*** @Author: ljm* @Date: 2024/4/29 15:12* @Version: v1.0.0* @Description: TODO**/
@Service
public class SecondHandler implements IHandler {@Overridepublic void handler() {System.out.println("Execute Second Handler");}@Overridepublic StrategyEnum getHandleStrategy() {return StrategyEnum.SECOND;}
}
java">package com.ljm.service.impl;import com.ljm.constant.StrategyEnum;
import com.ljm.service.IHandler;
import org.springframework.stereotype.Service;/*** @Author: ljm* @Date: 2024/4/29 15:12* @Version: v1.0.0* @Description: TODO**/
@Service
public class ThirdHandler implements IHandler {@Overridepublic void handler() {System.out.println("Execute Third Handler");}@Overridepublic StrategyEnum getHandleStrategy() {return StrategyEnum.THIRD;}
}
java">package com.ljm.service.impl;import com.ljm.constant.StrategyEnum;
import com.ljm.service.IHandler;
import org.springframework.stereotype.Service;/*** @Author: ljm* @Date: 2024/4/29 15:12* @Version: v1.0.0* @Description: TODO**/
@Service
public class FourHandler implements IHandler {@Overridepublic void handler() {System.out.println("Execute Four Handler");}@Overridepublic StrategyEnum getHandleStrategy() {return StrategyEnum.FOUR;}
}
2.4 定义策略工厂类
java">package com.ljm.service;import com.ljm.constant.StrategyEnum;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;import java.util.List;
import java.util.concurrent.ConcurrentHashMap;/*** @Author: ljm* @Date: 2024/4/29 15:16* @Version: v1.0.0* @Description: TODO**/
@Service
public class HandlerStrategyFactory {public final ConcurrentHashMap<StrategyEnum,IHandler> handlerStrategyMap = new ConcurrentHashMap<>();@Autowiredpublic HandlerStrategyFactory(List<IHandler> iHandlers){iHandlers.forEach(x -> handlerStrategyMap.put(x.getHandleStrategy(),x));}public IHandler getHandleStrategy(StrategyEnum strategyEnum){return handlerStrategyMap.get(strategyEnum);}
}
2.5 测试
java">package com.ljm;import com.MyApplication;
import com.ljm.constant.StrategyEnum;
import com.ljm.service.HandlerStrategyFactory;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;/*** @Author: ljm* @Date: 2024/2/26 10:58* @Version: v1.0.0* @Description: TODO**/
@RunWith(SpringJUnit4ClassRunner.class)
@SpringBootTest(classes = MyApplication.class)
public class MyTest {@Autowiredprivate HandlerStrategyFactory handlerStrategyFactory;@Testpublic void testStrategy() {handlerStrategyFactory.getHandleStrategy(StrategyEnum.FIRST).handler();handlerStrategyFactory.getHandleStrategy(StrategyEnum.SECOND).handler();handlerStrategyFactory.getHandleStrategy(StrategyEnum.THIRD).handler();handlerStrategyFactory.getHandleStrategy(StrategyEnum.FOUR).handler();/* 测试结果Execute First HandlerExecute Second HandlerExecute Third HandlerExecute Four Handler*/}}
实现一主要是为IHandler接口新增一个用于区分使用何种策略的方法,再构建一个策略工厂类,利用构造器注入的方式,将Spring中的实现类与相应的枚举类绑定,在使用的时候只需要传入StrategyEnum相应的值就可以调用想调用的策略.
三.实现二
3.1 定义抽象类
java">package com.ljm.service;import com.ljm.constant.StrategyEnum;
import org.springframework.beans.factory.InitializingBean;/*** @Author: ljm* @Date: 2024/4/29 15:09* @Version: v1.0.0**/
public abstract class AbstractHandler implements InitializingBean {public abstract void handler();public abstract StrategyEnum getHandleStrategy();@Overridepublic void afterPropertiesSet() throws Exception {HandlerStrategyFactory.registerHandlerStrategy(this);}
}
3.2 定义枚举
java">package com.ljm.constant;/*** @Author: ljm* @Date: 2024/4/29 15:10* @Version: v1.0.0* @Description: TODO**/
public enum StrategyEnum {FIRST, SECOND, THIRD, FOUR
}
3.3 定义实现类
java">package com.ljm.service.impl;import com.ljm.constant.StrategyEnum;
import com.ljm.service.AbstractHandler;
import org.springframework.stereotype.Service;/*** @Author: ljm* @Date: 2024/4/29 15:12* @Version: v1.0.0* @Description: TODO**/
@Service
public class FirstHandler extends AbstractHandler {@Overridepublic void handler() {System.out.println("Execute First Handler");}@Overridepublic StrategyEnum getHandleStrategy() {return StrategyEnum.FIRST;}
}
java">package com.ljm.service.impl;import com.ljm.constant.StrategyEnum;
import com.ljm.service.AbstractHandler;
import org.springframework.stereotype.Service;/*** @Author: ljm* @Date: 2024/4/29 15:12* @Version: v1.0.0* @Description: TODO**/
@Service
public class SecondHandler extends AbstractHandler {@Overridepublic void handler() {System.out.println("Execute Second Handler");}@Overridepublic StrategyEnum getHandleStrategy() {return StrategyEnum.SECOND;}
}
java">package com.ljm.service.impl;import com.ljm.constant.StrategyEnum;
import com.ljm.service.AbstractHandler;
import org.springframework.stereotype.Service;/*** @Author: ljm* @Date: 2024/4/29 15:12* @Version: v1.0.0* @Description: TODO**/
@Service
public class ThirdHandler extends AbstractHandler {@Overridepublic void handler() {System.out.println("Execute Third Handler");}@Overridepublic StrategyEnum getHandleStrategy() {return StrategyEnum.THIRD;}
}
java">package com.ljm.service.impl;import com.ljm.constant.StrategyEnum;
import com.ljm.service.AbstractHandler;
import org.springframework.stereotype.Service;/*** @Author: ljm* @Date: 2024/4/29 15:12* @Version: v1.0.0* @Description: TODO**/
@Service
public class FourHandler extends AbstractHandler {@Overridepublic void handler() {System.out.println("Execute Four Handler");}@Overridepublic StrategyEnum getHandleStrategy() {return StrategyEnum.FOUR;}
}
3.4 定义策略工厂类
java">package com.ljm.service;import com.ljm.constant.StrategyEnum;import java.util.concurrent.ConcurrentHashMap;/*** @Author: ljm* @Date: 2024/4/29 15:16* @Version: v1.0.0* @Description: TODO**/
public class HandlerStrategyFactory {public static final ConcurrentHashMap<StrategyEnum, AbstractHandler> handlerStrategyMap = new ConcurrentHashMap<>();public static void registerHandlerStrategy(AbstractHandler handler) {handlerStrategyMap.put(handler.getHandleStrategy(), handler);}public static AbstractHandler getHandleStrategy(StrategyEnum strategyEnum) {return handlerStrategyMap.get(strategyEnum);}
}
3.5 测试
java">package com.ljm;import com.MyApplication;
import com.ljm.constant.StrategyEnum;
import com.ljm.service.HandlerStrategyFactory;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;/*** @Author: ljm* @Date: 2024/2/26 10:58* @Version: v1.0.0* @Description: TODO**/
@RunWith(SpringJUnit4ClassRunner.class)
@SpringBootTest(classes = MyApplication.class)
public class MyTest {@Testpublic void testStrategy() {HandlerStrategyFactory.getHandleStrategy(StrategyEnum.FIRST).handler();HandlerStrategyFactory.getHandleStrategy(StrategyEnum.SECOND).handler();HandlerStrategyFactory.getHandleStrategy(StrategyEnum.THIRD).handler();HandlerStrategyFactory.getHandleStrategy(StrategyEnum.FOUR).handler();/* 测试结果Execute First HandlerExecute Second HandlerExecute Third HandlerExecute Four Handler*/}}
实现二主要是为AbstractHandler抽象类实现InitializingBean接口,再对象初始化的时候将其注册到我们自己构建的策略工厂类中,此时的对象由Spring生成并与相应的枚举类绑定,在使用的时候只需要传入StrategyEnum相应的值就可以调用想调用的策略.