diff --git a/src/main/java/com/github/hcsp/polymorphism/User.java b/src/main/java/com/github/hcsp/polymorphism/User.java index 51c95dc..2431f63 100644 --- a/src/main/java/com/github/hcsp/polymorphism/User.java +++ b/src/main/java/com/github/hcsp/polymorphism/User.java @@ -5,10 +5,14 @@ import java.util.function.Predicate; public class User { - /** 用户ID,数据库主键,全局唯一 */ + /** + * 用户ID,数据库主键,全局唯一 + */ private final Integer id; - /** 用户名 */ + /** + * 用户名 + */ private final String name; public User(Integer id, String name) { @@ -26,38 +30,31 @@ public String getName() { // 过滤ID为偶数的用户 public static List filterUsersWithEvenId(List users) { - List results = new ArrayList<>(); - for (User user : users) { - if (user.id % 2 == 0) { - results.add(user); - } - } - return results; + return filter(users, user -> user.id % 2 == 0); } - // 过滤姓张的用户 + + // 过滤姓张的用户 public static List filterZhangUsers(List users) { - List results = new ArrayList<>(); - for (User user : users) { - if (user.name.startsWith("张")) { - results.add(user); - } - } - return results; + return filter(users, user -> user.name.startsWith("张")); } - // 过滤姓王的用户 - public static List filterWangUsers(List users) { - List results = new ArrayList<>(); - for (User user : users) { - if (user.name.startsWith("王")) { - results.add(user); + // 过滤姓王的用户 + public static List filterWangUsers(List users) { + return filter(users, user -> user.name.startsWith("王")); + } + // 你可以发现,在上面三个函数中包含大量的重复代码。 + // 请尝试通过Predicate接口将上述代码抽取成一个公用的过滤器函数 + // 并简化上面三个函数 + public static List filter(List users, Predicate predicate) { + List results = new ArrayList<>(); + for (User user : users) { + if (predicate.test(user)) { + results.add(user); + // System.out.println(user.id); + } } + return results; } - return results; - } - // 你可以发现,在上面三个函数中包含大量的重复代码。 - // 请尝试通过Predicate接口将上述代码抽取成一个公用的过滤器函数 - // 并简化上面三个函数 - public static List filter(List users, Predicate predicate) {} } +