Java Stream 流式编程实战演练

Java 8 新引入的流式编程 (Stream) 功能很强大,用的好可以大大简化代码,提高效率。同时,大数据编程框架的源码以及业务代码大量使用了流式编程的思想。所以,这一块必须熟练掌握。

但是,流式编程不是 "the silver bullet" ,有些场景使用可能 overkill ,让代码不那么易读,而且打 log 或者 debug 的时候费劲。下面就举 2 例。

例1

业务逻辑:写一个函数,判断 Student 的性别是男生还是女生。

  • 实现1:是比较老派的写法,中规中矩。
  • 实现2:是流式编程的写法。使用了 Optional + map + filter

废话不多说,上代码:

实体类

public class Student {
    private String name;
    private Gender gender;

    public Student(String name, Gender gender) {
        this.name = name;
        this.gender = gender;
    }
    public String getName() {
        return name;
    }
    public void setName(String name) {
        this.name = name;
    }
    public Gender getGender() {
        return gender;
    }
    public void setGender(Gender gender) {
        this.gender = gender;
    }
}

enum Gender {
    Boy, Girl
}

实现1

public static boolean isBoy(Student student) {
	if (student == null || student.getGender() == null) return false;

	if (student.getGender() == Gender.Boy){
		return true;
	} else {
		return false;
	}
}

实现2

public static boolean isGirl(Student student) {
	return Optional.ofNullable(student)
			.map(Student::getGender)
			.filter(gender -> gender == Gender.Girl).isPresent();
}

主函数

public static void main(String[] args) {

	Student s1 = new Student("Alice", Gender.Girl);
	Student s2 = new Student("Bob", Gender.Boy);

	System.out.println(s1.getName() + " is a girl: " + isGirl(s1));
	System.out.println(s2.getName() + " is a boy: " + isBoy(s2));

}

输出

Alice is a girl: true
Bob is a boy: true

个人认为,这里使用流式编程有一点点 overkill ,因为老派的实现方法已经足够简洁。

例2

业务逻辑:有一个 list ,里面放了一堆 map ,而 map 里面放的是键值对:【姓名-成绩】。要求,把所有学生的名字收集起来,做成一个字符串输出。

  • 实现1:使用 for 循环的常规写法。
  • 实现2:使用了 flatmap + collect
public static void main(String[] args) {

	// prep data
	List<Map<String, String>> scoreList = new ArrayList<>();
	Map<String, String> scoreGroup1 = new HashMap();
	scoreGroup1.put("Alice", "70");
	scoreGroup1.put("Bob", "90");
	Map<String, String> scoreGroup2 = new HashMap();
	scoreGroup2.put("Charles", "80");
	scoreList.add(scoreGroup1);
	scoreList.add(scoreGroup2);

	// process 1
	StringBuilder names1 = new StringBuilder();
	for (Map<String, String> map: scoreList) {
		Set<String> set = map.keySet();
		for(String key: set) {
			if (!names1.toString().isEmpty()){
				names1.append(", " + key);
			} else {
				names1.append(key);
			}
		}
	}
	System.out.println("Process 1 - Student names: " + names1.toString());

	// process 2
	String names2 = scoreList
			.stream()
			.flatMap(e -> e.keySet().stream())
			.collect(Collectors.joining(", "));
	System.out.println("Process 2 - Student names: " + names2);
}

输出

Process 1 - Student names: Bob, Alice, Charles
Process 2 - Student names: Bob, Alice, Charles

这个例子中,使用 flatpmap 还是大大简化了代码的。

总结

从个人角度出发,打铁还需自身硬。有些流式编程的代码一开始看确实晦涩难懂,那我们就平时多用 Stream 练练手,写的多了,自然就熟了。

  • 如果是一个 legacy 的项目,就尽量跟原来的 coding style 保持一致,这样减少 bug 率。
  • 如果是一个新的或者大数据的项目,那就多用流式编程。

参考

上一篇:JavaScript原型对象


下一篇:MongoDB的增删改查