在springboot项目中使用策略工厂模式

在springboot项目中使用策略工厂模式

策略接口类

package cn.test.ext;public interface ITestStrategy {void execTestMethod();
}

策略实现类

package cn.test.ext.beanlife;import cn.test.ext.ITestStrategy;
import cn.test.ext.MyStrategyFactory;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.stereotype.Component;/*** InitializingBean* 实现该接口，你的bean可以在 spring容器初始化当前bean时 执行自定义逻辑** 必须在afterPropertiesSet方法里写你自定义的执行逻辑*/
@Slf4j
@Component
public class TestInitializingBean implements ITestStrategy, InitializingBean {private static String str ;//实现InitializingBean接口，必须重写afterPropertiesSet方法//这样当springboot启动加载TestInitializingBean时，会自动执行里边的afterPropertiesSet方法@Overridepublic void afterPropertiesSet() throws Exception {//本方法里可以做一些初始化的操作，如设置类静态变量值 / 将类注册到策略工厂/ 执行一些其他方法或动作/...//设置类静态变量值str = "qwe123!";//将k1策略 (本类) 注册到 策略工厂里MyStrategyFactory.register("k1",this);log.info("注册了策略:{}到策略工厂里",this);}@Overridepublic void execTestMethod() {log.info("start----执行k1策略...");System.err.println(str);}
}

策略工厂类

package cn.test.ext;import cn.hutool.core.util.StrUtil;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;/*** 策略工厂*/
@Component
@Slf4j
public class MyStrategyFactory {private static Map<String, ITestStrategy> strategyMap = new ConcurrentHashMap<>();/*** 将每个策略注册到策略工厂** 利用策略类里的 InitializingBean 的afterPropertiesSet方法，* 来把策略类初始化到StrategyFactory 策略工厂里** @param strategyCode String* @param itemStrategy IPersonAddrSimilarComputeHandleStrategy*/public static void register(String strategyCode, ITestStrategy itemStrategy) {strategyMap.put(strategyCode, itemStrategy);}/*** 从策略工厂取出需要的策略* @param key* @return*/public static ITestStrategy getReceiptHandleStrategy(String key) {for (String strate : strategyMap.keySet()) {if (StrUtil.equals(strate, key)) {//按key取出对应的策略类，来执行处理策略ITestStrategy strategy = strategyMap.get(key);log.info("拿到了策略:{}", strategy.getClass().getName());return strategy;}}log.info("没拿到对应的策略类,不存在 策略:{}", key);return null;}
}

策略上下文类

package cn.test.ext;import org.springframework.stereotype.Component;@Component
public class MyStrategyContext {private ITestStrategy iTestStrategy;//1.按key get某个策略类public void getStrategyInstance(String key) {ITestStrategy strategyInstance = MyStrategyFactory.getReceiptHandleStrategy(key);this.iTestStrategy = strategyInstance;}//2.通过context执行该策略类里的方法public void execMethod(){if (iTestStrategy!=null){iTestStrategy.execTestMethod();}}
}