使用策略模式 + Function优化if-else

由于Function接口的入参只支持一个参数,所以自定义一个MyFunction,支持多个参数,接口如下:

@FunctionalInterface
public interface MyFunction<T, R> {
    R apply(T... t);
}

下面就是如何使用了:

@RunWith(SpringRunner.class)
@SpringBootTest(classes = Application.class)
public class MyFunctionTest {

    private static Map<String, MyFunction<Object, String>> map = new HashMap<>();

    @PostConstruct
    public void init(){
        map.put("A", MyFunctionTest::fun01);
        map.put("B", MyFunctionTest::fun02);
        map.put("C", MyFunctionTest::fun03);
    }

    private static String fun03(Object... o) {
        System.out.println("fun03=====>");
        for (Object o1 : o) {
            System.out.println(o1);
        }
        return "CCC";
    }

    private static String fun02(Object... o) {
        System.out.println("fun02=====>");
        for (Object o1 : o) {
            System.out.println(o1);
        }
        return "BBB";
    }

    private static String fun01(Object... o) {
        System.out.println("fun01=====>");
        for (Object o1 : o) {
            System.out.println(o1);
        }
        return "AAA";
    }

    @Test
    public void test(){
        String type = "A";
        // 以前的if-else
//        if(type.equals("A")){
//            fun01("a", 1);
//        }else if(type.equals("B")){
//            fun02("b", 2);
//        }else if(type.equals("C")){
//            fun03("c", 3);
//        }
        // 优化的if-else
        map.get(type).apply("a", 1);
    }
}
  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值