什么是通过继承获得正确返回类的好习惯？

Question

我有不同类型的用户，他们都继承自父类 User：

class User(string username, string password){...}

class UserAdmin(string username, string password) extends User{
   ...
   // new methods
}

class User(string username, string password) extends User{
   ...
   // new methods
}

我有一种登录方法，但问题是，如果我使用 UserAdmin 的凭据登录，我想返回一个 UserAdmin，对于 User 的所有其他子类也是如此。

public User login(String userName, String password) {
    for (User user : usersList) {
        if (user.isCorrectLogin(userName, password)) {
            return user;
        }
    }
    throw new RuntimeException("Login failed !");
}

我该怎么做？我应该为每种类型的用户设置多个 login() 方法吗？或者是否有不同的方法来设计用户一开始就没有这个问题？

Answer 1

单一登录方法应该适合您的用例，但您应该确保User 的正确类型已经在usersList 中。例如：

List<User> usersList = new ArrayList<>();
usersList.add(new User("steve", "some_secret_password"));
usersList.add(new UserAdmin("bob", "some_even_more_secret_password"));

这样，您现有的login 方法将自动返回正确的类型。

如果要检查返回值是否属于特定类型，可以使用instanceof 运算符：

User user = login(username, password);
if (user instanceof UserAdmin) {
  // Do privileged action
}

不过，还有其他几种方法可以解决这个问题。您可以简单地让每个特定的用户类型返回它拥有的一组角色，这样您就可以完全避免instanceof：

// User
public Set<String> getRoles() {
  return Set.of("user");
}

// UserAdmin
public Set<String> getRoles() {
  return Set.of("user", "admin");
}

那么你的逻辑可能变成：

User user = login(username, password);
if (user.getRoles().contains("admin")) {
  // Privileged action
}