【高质量代码系列1】过多的if else 代码重构技巧总结

一、if else 的滥用

在实际业务开发中，常常面临多个条件判断的情况，如果不加思索就写代码，很容易出现一堆if else 的代码，导致代码逻辑不清晰，可阅读性很差，笔者现在改造中的一个历史遗留项目，大片大片的if else，给后来人带来很大的理解成本。过多的if else ，这是很明显的坏代码的味道，这时就应该考虑自己的设计是否合理，有没有更好的表达方式了。

二、if else 重构技巧

1.适当提前return

这是判断条件取反的做法，代码在逻辑表达上会更清晰，先看看下面这段demo示例代码:

1/**

2    * 判断用户余额是否足够取现

3    * @param userId

4    * @param withdrawAmount 取现金额

5    * @return

6    */

7    public Boolean checkBanlance(Long userId,BigDecimal withdrawAmount) {

8

9        User user = userDao.getUser(userId);

10        if (user != null) {

11            if (withdrawAmount.compareTo(user.getBalance()) <= 0) {

12                return Boolean.TRUE;

13            } else {

14                return false;

15            }

16        }

17        return false;

18    }

这段就可以通过if条件判断提前return来避免if嵌套的问题，如果user对象为空直接return false，如下：

1public Boolean checkBanlance2(Long userId,BigDecimal withdrawAmount) {

2        User user = userDao.getUser(userId);

3        if (user == null) {

4            return Boolean.FALSE;

5        }

6        if (withdrawAmount.compareTo(user.getBalance()) <= 0) {

7            return Boolean.TRUE;

8        } else {

9            return false;

10        }

11        return false;

12    }

2.用布尔表达式来代替

上面的函数返回的结果是一个Boolean，那我们可以用一个布尔表达式来表示，如下：

1public Boolean checkBanlance3(Long userId,BigDecimal withdrawAmount) {

2

3        User user = userDao.getUser(userId);

4        if (user == null) {

5            return Boolean.FALSE;

6        }

7        return withdrawAmount.compareTo(user.getBalance()) <= 0;

8    }

checkBanlance3 相比checkBanlance2的代码又可以简化了许多！

3.Map结合反射来代替if else

以一个获取手机号的demo代码来说明这种用法，代码如下：

1public class UserMobile {

2

3    public String getMobile1() {

4

5        return "114774440";

6    }

7

8    public String getMobile2() {

9

10        return "1448720444";

11    }

12

13    public String getMobile3() {

14

15        return "1447191222";

16    }

17

18    public String getMobiel(String name) {

19

20        if (StringUtils.equals(name, "xiaoming")) {

21            return this.getMobile1();

22        } else if (StringUtils.equals(name, "xiaohua")) {

23            return this.getMobile2();

24        } else if (StringUtils.equals(name, "xiaozhang")) {

25            return this.getMobile3();

26        }

27        return "";

28    }

29}

getMobiel方法根据姓名来获取手机号，不同的人返回不同的手机号，按照现在的写法，if else 很容易被滥用，尤其是要新增其他用户的时候。这时候我们可以考虑使用map来代替if else，

思路是：将name作为key，将getMobile1/getMobile2/getMobile3这三个方法名作为value，放在map中，然后根据参数name，获取对应的方法名，然后使用反射来调用相关的方法。示例代码如下：

1public class UserMobile {

2

3    static Map map = new HashMap();

4    static {

5        map.put("xiaoming", "getMobile1");

6        map.put("xiaohua", "getMobile2");

7        map.put("xiaozhang", "getMobile3");

8    }

9

10    public String getMobile1() {

11

12        return "114774440";

13    }

14

15    public String getMobile2() {

16

17        return "1448720444";

18    }

19

20    public String getMobile3() {

21

22        return "1447191222";

23    }

24

25

26    public String getMobiel(String name) throws Exception {

27

28        String methodName = map.get(name);

29        if (StringUtils.isEmpty(methodName)) {

30            return "";

31        }

32        Method method = this.getClass().getMethod(methodName,new Class[]{});

33        String mobile = (String) method.invoke(this);

34        return mobile;

35    }

36

37    public static void main(String[] args) throws Exception {

38

39        UserMobile userMobile = new  UserMobile();

40        String mobile = userMobile.getMobiel("xiaoming");

41        System.out.println("mobile:"+mobile);

42    }

43}

这种用法一般多用于需要选择不同的执行方法时可以尝试使用! 除了这种处理方式外，还可以使用枚举。

4.用枚举来代替if else

1public enum UserInfo {

2

3    XIAOMING("xiaoming"){

4        @Override

5        public String getUserMobile() {

6            return "13652222";

7        }

8    },

9    XIAOHUA("xiaohua"){

10        @Override

11        public String getUserMobile() {

12            return "12555555896";

13        }

14    },

15    XIAOZHANG("xiaozhang") {

16        @Override

17        public String getUserMobile() {

18            return "12555555896";

19        }

20    };

21

22    private String userName;

23

24    private String mobile;

25

26    UserInfo(String userName) {

27        this.userName = userName;

28    }

29

30    public String getUserName() {

31        return userName;

32    }

33

34    public void setUserName(String userName) {

35        this.userName = userName;

36    }

37

38    public void setMobile(String mobile) {

39        this.mobile = mobile;

40    }

41

42    public String getMobile(String name) {

43

44        for (UserInfo userInfo : UserInfo.values()) {

45            if (StringUtils.equals(name,userInfo.getUserName())) {

46                userInfo.getUserMobile();

47            }

48        }

49        return "";

50    }

51

52    public abstract String getUserMobile();

53}

5.应用多态来代替if else

1public class RestaurantService {

2

3    public void provideService(String worker) {

4        if (StringUtils.equals(worker,"waiter")) {

5            System.out.println("拿瓶红酒");

6        } else if (StringUtils.equals(worker,"cook")){

7            System.out.println("做一盘番茄鸡蛋");

8        } else if (StringUtils.equals(worker,"cleaner")) {

9            System.out.println("打扫一下卫生");

10        } else {

11            System.out.println("sorry,没有此项服务");

12        }

13    }

14}

这是以餐厅的服务场景为例一个demo，这段代码写法，很容易引发if else的滥用，如果再增加一些服务，继续这种用法，只能不断的if else，那代码的可阅读性极差，且逻辑繁杂。

我们可以考虑使用多态，将不同的服务定义为一个子类，将每种服务封装到子类中。

1public interface RestService {

2

3    void provideService();

4}

5

6

7public class Cleaner implements RestService {

8

9    @Override

10    public void provideService() {

11        System.out.println("打扫一下卫生");

12    }

13}

14

15public class Cook implements RestService {

16

17    @Override

18    public void provideService() {

19        System.out.println("做一盘番茄鸡蛋");

20    }

21}

22

23public class Waiter implements RestService {

24

25    @Override

26    public void provideService() {

27        System.out.println("拿瓶红酒");

28    }

29}

30

31public class RestaurantService {

32

33    static Map registerService = new HashMap<>();

34    static {

35        registerService.put("waiter", new Waiter());

36        registerService.put("cook", new Cook());

37        registerService.put("cleaner", new Cleaner());

38    }

39

40    public void getService(String name) {

41

42        RestService restService = registerService.get(name);

43

44        if (restService==null) {

45            System.out.println("sorry,没有此项服务");

46            return;

47        }

48        restService.provideService();

49    }

50

51

52    public static void main(String[] args) {

53

54        RestaurantService factory = new RestaurantService();

55        factory.getService("cleaner");

56    }

57}

将waiter、cook、cleaner这三个角色分别定义三个子类，通过接口来调用需要用到的服务。

RestaurantService类中，将三个服务注册到map中，根据key来获取对应的服务，这也是用map来代替if else的一种用法。

注意，本文探讨的是过多的if else的重构技巧，不要为了重构而重构。

更多内容欢迎关注个人微信公众号，一起成长！

————————————————

版权声明：本文为CSDN博主「灿若繁星」的原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接及本声明。

原文链接：https://blog.csdn.net/song19890528/article/details/79574032