设计模式 ​(策略模式+注解)

标签： 设计模式  

0.前记
之前写过一篇文章【springboot通过xml方式自动注入的策略模式】，这篇文章是通过xml的方式来注入具体的策略实现类的，以下即将写的文章是通过把具体的策略实现类交由spring容器管理（利用@Component注解）来实现的。

1.策略类的接口
目录结构:

demo:

package com.hcr.strategy;



public interface IAnimalStrategy {
    
    int getType();

    
    String dealBusiness(String name);
}

2.策略类的实现类
demo:

package com.hcr.strategy.impl;

import com.hcr.strategy.IAnimalStrategy;
import org.springframework.stereotype.Component;



@Component
public class CatStrategy implements IAnimalStrategy {
    @Override
    public int getType() {
        // 设置1代表猫类
        return 1;
    }

    @Override
    public String dealBusiness(String name) {
        return "猫类的实现类" + name;
    }
}

package com.hcr.strategy.impl;

import com.hcr.strategy.IAnimalStrategy;
import org.springframework.stereotype.Component;



@Component
public class DogStrategy implements IAnimalStrategy {
    @Override
    public int getType() {
        // 设置1代表猫类狗类
        return 2;
    }

    @Override
    public String dealBusiness(String name) {
        return "狗类的实现类" + name;
    }
}

3.获取具体策略实现类的工厂类
demo:

package com.hcr.strategy;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;



@Component
public class AnimalStrategyFactory {

    @Autowired
    private List animalStrategyList;

    private static Map animalStrategyMap;

    
    @PostConstruct
    public void init() {
        animalStrategyMap = animalStrategyList.stream().collect(Collectors.toMap(IAnimalStrategy::getType, a -> a));
    }

    
    public IAnimalStrategy getStrategy(int type) {
        return animalStrategyMap.get(type);
    }
}

4.测试接口
demo:

package com.hcr.controller;

import com.hcr.strategy.AnimalStrategyFactory;
import com.hcr.strategy.IAnimalStrategy;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

@RestController
public class StrategyController {
    @Autowired
    private AnimalStrategyFactory animalStrategyFactory;

    @PostMapping(value = "test/strategy")
    public Object testStrategy(@RequestParam(value = "type", defaultValue = "1", required = true) Integer type
            , @RequestParam(value = "name", required = true) String name) {

        IAnimalStrategy strategy = animalStrategyFactory.getStrategy(type);
        String resultName = strategy.dealBusiness(name);

        return resultName;
    }
}

5.测试结果