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

小鱼儿 2024-03-22 22:50 88阅读 0赞

## 在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方法
        @Override
        public void afterPropertiesSet() throws Exception {
        
            //本方法里可以做一些初始化的操作，如设置类静态变量值 / 将类注册到策略工厂/ 执行一些其他方法或动作/...
    
            //设置类静态变量值
            str = "qwe123!";
    
            //将k1策略 (本类) 注册到 策略工厂里
            MyStrategyFactory.register("k1",this);
            log.info("注册了策略:{}到策略工厂里",this);
        }
    
        @Override
        public 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();
            }
        }
    }