【Java】一次代码优化

有两个类型相同的类，分了三个方法分别比较、获取这两个类的三个值。
看不懂这句话直接看代码↓

@Data
@Accessors(chain = true) //链式调用
public class Student {
    private Integer valueOne;
    private Integer valueTwo;
    private Integer valueThree;
}

public class Service {
	private Integer getValueOne(Student student1, Student student2) {
	    Integer value = student1.getValueOne();
	    if (value != null && value > 0) {
	        return value;
	    }
	    value = student2.getValueOne();
	    if (value != null && value > 0) {
	        return value;
	    }
	    return 0;
	}

	private Integer getValueTwo(Student student1, Student student2) {
	    Integer value = student1.getValueTwo();
	    if (value != null && value > 0) {
	        return value;
	    }
	    value = student2.getValueTwo();
	    if (value != null && value > 0) {
	        return value;
	    }
	    return 0;
	}

	private Integer getValueThree(Student student1, Student student2) {
	    Integer value = student1.getValueThree();
	    if (value != null && value > 0) {
	        return value;
	    }
	    value = student2.getValueThree();
	    if (value != null && value > 0) {
	        return value;
	    }
	    return 0;
	}
}

public static void main(String[] args) {
    Student student1 = new Student().setValueOne(100).setValueTwo(200).setValueThree(0);
    Student student2 = new Student().setValueOne(0).setValueTwo(0).setValueThree(500);

    Service service = new Service();
    System.out.println(">>>>>>>> valueOne：" + service.getValueOne(student1, student2));
    System.out.println(">>>>>>>> valueTwo：" + service.getValueTwo(student1, student2));
    System.out.println(">>>>>>>> valueThree：" + service.getValueThree(student1, student2));
}

打印结果：
>>>>>>>> valueOne：100
>>>>>>>> valueTwo：200
>>>>>>>> valueThree：500

只要student1的属性值有值就返回student1的，否则student2的有值就返回，不然返回0。可以发现，这3个方法极其相似，只是取的属性值不同。

改进：
使用Function函数式接口(需要入参和返参)改进，不懂的自个学lambda表达式(涉及函数式接口和方法引用)去。

private Integer getAttribute(Student student1, Student student2, Function<Student, Integer> attributeGetter) {
    Integer value1 = this.getValue(student1, attributeGetter);
    Integer value2 = this.getValue(student2, attributeGetter);
    return value1 > 0 ? value1 : value2;
}

private Integer getValue(Student student, Function<Student, Integer> attributeGetter) {
    Integer value;
    if (null != student) {
        value = attributeGetter.apply(student);
        if (null != value && value > 0) {
            return value;
        }
    }
    return 0;
}

public static void main(String[] args) {
    Student student1 = new Student().setValueOne(100).setValueTwo(200).setValueThree(0);
    Student student2 = new Student().setValueOne(0).setValueTwo(0).setValueThree(500);
    
    Service service = new Service();
    /*
       Student::getValueOne为方法引用：表示将Student的getValueOne方法体作为Function函数式接口的抽象方法   R apply(T t)   的实现
       Student为引用对象，与Function<Student, Integer>泛型中的第一个对应，是apply方法的入参T
       getValueOne的返回值与泛型中的第二个对应，是apply方法的返参R
    */
    System.out.println(">>>>>>>> valueOne：" + service.getAttribute(student1, student2, Student::getValueOne));
    System.out.println(">>>>>>>> valueTwo：" + service.getAttribute(student1, student2, Student::getValueTwo));
    System.out.println(">>>>>>>> valueThree：" + service.getAttribute(student1, student2, Student::getValueThree));
}

打印结果：
>>>>>>>> valueOne：100
>>>>>>>> valueTwo：200
>>>>>>>> valueThree：500