SpringBoot结合策略模式

结合SpringBoot的策略模式套路，代码少些if-else

1、编写接口

/**
 * 计算策略接口
 * @author xuchen
 * @Date: 2021/3/8 19:07
 */
public interface CalculateStrategy {
    int doOperation(int num1,int num2);
}

 2、实现类

/**
 * @author xuchen
 * @Date: 2021/3/8 19:08
 */
@Component
public class AddOperation implements CalculateStrategy {
    @Override
    public int doOperation(int num1, int num2) {
        return num1 + num2;
    }
}

/**
 * @author xuchen
 * @Date: 2021/3/8 19:09
 */
@Component
public class MultiplyOperation implements CalculateStrategy {
    @Override
    public int doOperation(int num1, int num2) {
        return num1 * num2;
    }

}

/**
 * @author xuchen
 * @Date: 2021/3/8 19:09
 */
@Component
public class SubstractOperation implements CalculateStrategy {
    @Override
    public int doOperation(int num1, int num2) {
        return num1 - num2;
    }

}

 3、上下文编写

创建上下文管理，用于提取策略

import com.google.common.base.Preconditions;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
 * 创建上下文管理，用于提取策略。
 * @autowired注解放的位置和对应的参数列表
 * 实际上它还可以注入到Map和List，Map的key就是它注入时的名，List则是存放全部实例对象
 * @author xuchen
 * @Date: 2021/3/8 19:10
 */
@Component
public class CalculatelOperationContext {
    private final Map<String, CalculateStrategy> strategyMap = new ConcurrentHashMap<>();

    @Autowired
    public void stragegyInteface(Map<String, CalculateStrategy> strategyMap) {
        this.strategyMap.clear();
        strategyMap.forEach(this.strategyMap::put);
        System.out.println(this.strategyMap);
    }


    @Autowired
    public void stragegyInteface2(List<CalculateStrategy> strategyMap) {
        strategyMap.forEach(System.out::println);
    }

    public CalculateStrategy strategySelect(String mode) {
        Preconditions.checkArgument(!StringUtils.isEmpty(mode), "不允许输入空字符串");
        return this.strategyMap.get(mode);
    }
}

4、测试controller （此处使用了swagger，可以去除此块）

import com.xc.luckysheet.utils.strategy.CalculatelOperationContext;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

/**
 * @author xuchen
 * @Date: 2021/3/8 20:01
 */
@Slf4j
@RestController
@Api(description = "测试接口")
@RequestMapping("test/Strategy")
public class TestStrategyController {

    @Autowired
    private CalculatelOperationContext calculatelOperationContext;

    @ApiOperation(value = "Strategy查询",notes = "Strategy查询")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "mode", value = "multiplyOperation", paramType = "query", required = true, dataType = "String")
    })
    @GetMapping(value = "/operation")
    public String strategySelect(@RequestParam("mode") String mode) {
        return String.valueOf(calculatelOperationContext.strategySelect(mode).doOperation(20, 5));
    }
}

5、测试结果

 

完