• Java使用Function包&策略模式，优化大量if...else语句

    目录

    场景模拟

    Consumer与ToIntBiFunction简介,u>

     场景Demo业务代码改造

    最终结果

    业务代码中，若存在大量无法避免的if...else代码，可以尝试使用JDK8提供的函数式编程包。

    场景模拟

    设计一个简单的招聘业务：

    招聘条件是：男18~60岁，女18~50岁。再根据男女不同年龄段，分配不同的事情任务做。

    模拟代码如下

    1. package cn.daydayup.designmode.factorymode.demov4;
    2.  
    3.  
    4. /**
    5.  * @author ShangHai
    6.  * @desc
    7.  */
    8. public class Demo {
    9.  
    10.     /**
    11.      * 模拟招聘业务
    12.      * @param sex 性别
    13.      * @param age 年龄
    14.      * @param name 名字
    15.      */
    16.     public void recruit(String sex, int age, String name){
    17.         if(CommonConstants.MAN.equals(sex)){
    18.             // 男性
    19.             if(age > CommonConstants.SIXTY){
    20.                 throw new RuntimeException("年龄不满足");
    21.             } else if(age > CommonConstants.FORTY_FIVE){
    22.                 this.doSomeThings001(name);
    23.             } else if(age >= CommonConstants.THIRTY){
    24.                 this.doSomeThings002(name);
    25.             } else if(age >= CommonConstants.EIGHTEEN){
    26.                 this.doSomeThings003(name);
    27.             } else {
    28.                 throw new RuntimeException("年龄不满足");
    29.             }
    30.         } else if(CommonConstants.WOMAN.equals(sex)) {
    31.             // 女性
    32.             if(age > CommonConstants.FIFTY){
    33.                 throw new RuntimeException("年龄不满足");
    34.             } else if(age >= CommonConstants.THIRTY){
    35.                 this.doSomeThings004(name);
    36.             } else if(age >= CommonConstants.EIGHTEEN){
    37.                 this.doSomeThings005(name);
    38.             } else {
    39.                 throw new RuntimeException("年龄不满足");
    40.             }
    41.         } else {
    42.             throw new RuntimeException("性别不满足");
    43.         }
    44.     }
    45.  
    46.     private void doSomeThings001(String name){
    47.         System.out.println(name + "[男]年龄45-60");
    48.     }
    49.  
    50.     private void doSomeThings002(String name){
    51.         System.out.println(name + "[男]年龄30-45");
    52.     }
    53.  
    54.     private void doSomeThings003(String name){
    55.         System.out.println(name + "[男]年龄18-30");
    56.     }
    57.  
    58.     private void doSomeThings004(String name){
    59.         System.out.println(name + "[女]年龄30-60");
    60.     }
    61.  
    62.     private void doSomeThings005(String name){
    63.         System.out.println(name + "[女]年龄18-30");
    64.     }
    65.  
    66. }

    常量代码 

    1. package cn.daydayup.designmode.factorymode.demov4;
    2.  
    3. /**
    4.  * @author ShangHai
    5.  * @desc
    6.  */
    7. public class CommonConstants {
    8.  
    9.     public final static int ZERO = 0;
    10.  
    11.     public final static int ONE = 1;
    12.  
    13.     public final static int TWO = 2;
    14.  
    15.     public final static int THRRE = 3;
    16.  
    17.     public final static int FOUR = 4;
    18.  
    19.     public final static int FIVE = 5;
    20.  
    21.     public final static int EIGHTEEN = 18;
    22.  
    23.     public final static int THIRTY = 30;
    24.  
    25.     public final static int FORTY_FIVE = 45;
    26.  
    27.     public final static int FIFTY = 50;
    28.  
    29.     public final static int SIXTY = 60;
    30.  
    31.     public final static String MAN = "男";
    32.  
    33.     public final static String WOMAN = "女";
    34.  
    35. }

     main方法执行【控制台：GeGeDa[男]年龄45-60】

    1. public static void main(String[] args) {
    2.         // 控制台输出：​ 【GeGeDa[男]年龄45-60】
    3.         new Demo().recruit(CommonConstants.MAN, 49, "GeGeDa");
    4.     }

    上述的业务方法中，存在较多的if...else条件语句。

    若条件维度根据业务需求增多，if...else语句将会不断叠加，业务代码将变得不再优雅。

    分析以上业务代码，可采用Function包中的Consumer与ToIntBiFunction配合进行改造。

    Consumer与ToIntBiFunction简介

    Consumer

    Consumer接口是java.util.function包的其中一个接口，

    包含一个核心方法accept(T t)，其意义即传入一个参数进行消费，无返回值。

    代码测试如下：

    1. package cn.daydayup.designmode.factorymode.demov4.test;
    2.  
    3. import java.util.function.Consumer;
    4.  
    5. /**
    6.  * @author ShangHai
    7.  * @desc
    8.  */
    9. public class TestConsumer {
    10.  
    11.     public static void main(String[] args) {
    12.         Consumer consumer = a -> {
    13.             System.out.println("输出参数值：" + a);
    14.         };
    15.  
    16.         // 传入字符串参数
    17.         consumer.accept("aaaa");
    18.     }
    19.  
    20. }

    以上consumer对象声明的方式，等同于

    1. Consumer consumer = new Consumer() {
    2.             @Override
    3.             public void accept(String a) {
    4.                 System.out.println("输出参数值：" + a);
    5.             }
    6.         };

    main方法执行【控制台：输出参数值：aaaa】

     ToIntBiFunction

    ToIntBiFunction接口是java.util.function包的其中一个接口，

    包含一个核心方法applyAsInt(T t,U u)，其意义即传入两个参数进行消费，返回一个int值。

    代码测试如下：

    1. package cn.daydayup.designmode.factorymode.demov4.test;
    2.  
    3. import cn.daydayup.designmode.CommonConstant;
    4. import cn.daydayup.designmode.factorymode.demov4.CommonConstants;
    5.  
    6. import java.util.function.ToIntBiFunction;
    7.  
    8. /**
    9.  * @author ShangHai
    10.  * @desc
    11.  */
    12. public class TestToIntBiFunction {
    13.  
    14.     public static void main(String[] args) {
    15.         // 传入两个String参数,得到一个int值
    16.         ToIntBiFunction toIntBiFunction = (a, b) -> {
    17.             if(a.equals(b)){
    18.                 return CommonConstants.ONE;
    19.             } else {
    20.                 return CommonConstants.TWO;
    21.             }
    22.         };
    23.  
    24.         // 传入两个字符串参数
    25.         int result = toIntBiFunction.applyAsInt("参数1", "参数2");
    26.         System.out.println("得到的int值" + result);
    27.     }
    28.  
    29. }

    以上toIntBiFunction对象声明的方式，等同于

    1. ToIntBiFunction toIntBiFunction = new ToIntBiFunction() {
    2.             @Override
    3.             public int applyAsInt(String a, String b) {
    4.                 if(a.equals(b)){
    5.                     return CommonConstants.ONE;
    6.                 } else {
    7.                     return CommonConstants.TWO;
    8.                 }
    9.             }
    10.         };

    main方法执行【控制台：得到的int值2】

     场景Demo业务代码改造

    一、if条件&逻辑代码抽离

    创建一个策略列表业务执行类RecruitConsumer，

    用来存储if标签体内实际的执行方法。

    1. package cn.daydayup.designmode.factorymode.demov4.function;
    2.  
    3. import cn.daydayup.designmode.factorymode.demov4.CommonConstants;
    4.  
    5. import java.util.HashMap;
    6. import java.util.Map;
    7. import java.util.function.Consumer;
    8.  
    9. /**
    10.  * @author ShangHai
    11.  * @desc
    12.  */
    13. public class RecruitConsumer {
    14.  
    15.     // 策略列表
    16.     public static final Map> CONSUMER_MAP = new HashMap<>();
    17.  
    18.     static {
    19.         // 根据传入不同的int值,将执行不同方法
    20.         CONSUMER_MAP.put(CommonConstants.ONE, RecruitConsumer::doSomeThings001);
    21.         CONSUMER_MAP.put(CommonConstants.TWO, RecruitConsumer::doSomeThings002);
    22.         CONSUMER_MAP.put(CommonConstants.THRRE, RecruitConsumer::doSomeThings003);
    23.         CONSUMER_MAP.put(CommonConstants.FOUR, RecruitConsumer::doSomeThings004);
    24.         CONSUMER_MAP.put(CommonConstants.FIVE, RecruitConsumer::doSomeThings005);
    25.     }
    26.  
    27.     private static void doSomeThings001(String name){
    28.         System.out.println(name + "[男]年龄45-60");
    29.     }
    30.  
    31.     private static void doSomeThings002(String name){
    32.         System.out.println(name + "[男]年龄30-45");
    33.     }
    34.  
    35.     private static void doSomeThings003(String name){
    36.         System.out.println(name + "[男]年龄18-30");
    37.     }
    38.  
    39.     private static void doSomeThings004(String name){
    40.         System.out.println(name + "[女]年龄30-60");
    41.     }
    42.  
    43.     private static void doSomeThings005(String name){
    44.         System.out.println(name + "[女]年龄18-30");
    45.     }
    46.  
    47. }

    创建一个"条件收集&策略数值"返回类RecruitToIntBiFunction，

    用来收集需求下所有的if分支，且根据if满足的条件返回一个策略列表需要的策略int数值。

    1. package cn.daydayup.designmode.factorymode.demov4.function;
    2.  
    3. import cn.daydayup.designmode.CommonConstant;
    4. import cn.daydayup.designmode.factorymode.demov4.CommonConstants;
    5.  
    6. import java.util.function.ToIntBiFunction;
    7.  
    8. /**
    9.  * @author ShangHai
    10.  * @desc
    11.  */
    12. public class RecruitToIntBiFunction implements ToIntBiFunction{
    13.  
    14.     @Override
    15.     public int applyAsInt(Object o, Object o2) {
    16.         if(o==null || o2==null){
    17.             // 返回0,RecruitConsumer策略列表不具备0策略,不执行任何逻辑
    18.             return CommonConstants.ZERO;
    19.         }
    20.  
    21.         String sex = o.toString();
    22.         int age = Integer.parseInt(o2.toString());
    23.         if(CommonConstants.MAN.equals(sex)){
    24.             // 男性
    25.             if(age > CommonConstants.SIXTY){
    26.                 // 返回0,RecruitConsumer策略列表不具备0策略,不执行任何逻辑
    27.                 return CommonConstants.ZERO;
    28.             } else if(age > CommonConstants.FORTY_FIVE){
    29.                 return CommonConstants.ONE;
    30.             } else if(age >= CommonConstants.THIRTY){
    31.                 return CommonConstants.TWO;
    32.             } else if(age >= CommonConstants.EIGHTEEN){
    33.                 return CommonConstants.THRRE;
    34.             } else {
    35.                 // 返回0,RecruitConsumer策略列表不具备0策略,不执行任何逻辑
    36.                 return CommonConstants.ZERO;
    37.             }
    38.         } else if(CommonConstants.WOMAN.equals(sex)) {
    39.             // 女性
    40.             if(age > CommonConstants.FIFTY){
    41.                 // 返回0,RecruitConsumer策略列表不具备0策略,不执行任何逻辑
    42.                 return CommonConstants.ZERO;
    43.             } else if(age >= CommonConstants.THIRTY){
    44.                 return CommonConstants.FOUR;
    45.             } else if(age >= CommonConstants.EIGHTEEN){
    46.                 return CommonConstants.FIVE;
    47.             } else {
    48.                 // 返回0,RecruitConsumer策略列表不具备0策略,不执行任何逻辑
    49.                 return CommonConstants.ZERO;
    50.             }
    51.         } else {
    52.             // 返回0,RecruitConsumer策略列表不具备0策略,不执行任何逻辑
    53.             return CommonConstants.ZERO;
    54.         }
    55.     }
    56.  
    57. }

    二、业务逻辑代码改造（最终仅需两行）

    1. package cn.daydayup.designmode.factorymode.demov4;
    2.  
    3.  
    4. import cn.daydayup.designmode.factorymode.demov4.function.RecruitConsumer;
    5. import cn.daydayup.designmode.factorymode.demov4.function.RecruitToIntBiFunction;
    6.  
    7. /**
    8.  * @author ShangHai
    9.  * @desc
    10.  */
    11. public class Demo {
    12.  
    13.     /**
    14.      * 模拟招聘业务
    15.      * @param sex 性别
    16.      * @param age 年龄
    17.      * @param name 名字
    18.      */
    19.     public void recruit(String sex, int age, String name){
    20.         // 将if条件所需参数传入,得到策略int数值
    21.         int strategyInt = new RecruitToIntBiFunction().applyAsInt(sex, age);
    22.         // 将策略int数值传入策略列表,consumer会自动执行对应的策略方法
    23.         RecruitConsumer.CONSUMER_MAP.get(strategyInt).accept(name);
    24.     }
    25.  
    26.     public static void main(String[] args) {
    27.         // 控制台输出 ShangHai[男]年龄18-30
    28.         new Demo().recruit(CommonConstants.MAN, 18, "ShangHai");
    29.     }
    30. }
    31.  
    32.  
    33.  
    34.

     main方法执行【控制台：ShangHai[男]年龄18-30】

    最终结果

    业务方法经过"Funtion包与策略模式"改造后，

    代码变得简洁；

    业务方法不再关注if如何实现，交由定义好的策略消费即可；

    if...else分支语句及业务实际执行方法分类被收集，代码整体层次变得规范。

  • 相关阅读:
    电脑重装系统后Word表格自动换行的方法
    计算机毕业设计源码丨基于java的企业人力资源（人事）管理系统
    基于Java的城市桥梁道路管理系统(Vue.js+SpringBoot)
    【教3妹学编程-java实战4】Map遍历删除元素的几种方法
    王道数据结构C语言循环链表基本操作实现
    【Spring Boot项目】根据用户的角色控制数据库访问权限
    关于二阶低通滤波的C代码及入门测试
    Springboot+网上眼镜商场 毕业设计-附源码241659
    Java:实现动态数组类算法（附完整源码）
    Android 12添加系统服务
  • 原文地址：https://blog.csdn.net/ShangHai0123/article/details/127672277