Java使用函数式接口处理if-else的策略模式

最新推荐文章于 2024-07-03 21:31:53 发布

boonya

最新推荐文章于 2024-07-03 21:31:53 发布

阅读量1.2k

点赞数

分类专栏： Java # 设计模式文章标签： java 策略模式开发语言

本文链接：https://blog.csdn.net/boonya/article/details/122719460

版权

Java 同时被 2 个专栏收录

104 篇文章 8 订阅

订阅专栏

设计模式

7 篇文章 0 订阅

订阅专栏

常规策略模式示例

public abstract class Strategy {

    abstract void doSomething();
}

// 策略A
public class AStrategy extends Strategy{

    @Override
    void doSomething() {

    }
}

// 策略B

public class BStrategy extends Strategy{

    @Override
    void doSomething() {

    }
}

函数式接口策略

定义函数方法

@Slf4j
public class Functions {

    public String strategyA(String resourceId){
        log.info(resourceId);
        return "A";
    }

    public String strategyB(String resourceId){
        log.info(resourceId);
        return "B";
    }

    public String strategyC(String resourceId){
        log.info(resourceId);
        return "C";
    }
}

函数策略调用

@Slf4j
public class FunctionStrategy {

    private Functions functions = new Functions();

    /**
     * 函数式表
     */
    private Map<String, Function<String,String>> strategyMap = new HashMap<>();

    /**
     * 初始化函数匹配策略
     */
    public void init(){
        strategyMap.put("A",resourceId->functions.strategyA(resourceId));
        strategyMap.put("B",resourceId->functions.strategyB(resourceId));
        strategyMap.put("C",resourceId->functions.strategyC(resourceId));
    }

    /**
     * 获取结果
     * @param resourceId
     * @return
     */
    public String getResult(String resourceId){
        Function<String,String> result = strategyMap.get(resourceId);
        if(null != result){
            return result.apply(resourceId);
        }else{
            new RuntimeException("错误调用");
        }
        return null;
    }

    public static void main(String[] args) {
        FunctionStrategy functionStrategy = new FunctionStrategy();
        functionStrategy.init();
        log.debug("->{}",functionStrategy.getResult("A"));
    }
}