JPA中父级User如何获取所有多级OneToMany关联子用户

当前代码仅能获取直接子用户（如A只能拿到B，无法获取B的子用户C），核心问题是实体关联映射不完整+缺少层级查询逻辑，以下是具体修复方案：

1. 修正User实体的关联映射 #

现有@OneToMany未正确维护父子关系的外键，且缺少反向父用户关联，导致数据库无法追溯完整层级结构。修改User实体：

@XmlRootElement(name = "users")
@Table(name = "table_users")
@Entity
public class User extends StorageInformation {
    // 保留原有字段...

    // 添加父用户关联，维护外键
    @ManyToOne(fetch = FetchType.LAZY)
    @JoinColumn(name = "parent_id") // 数据库表需新增parent_id字段，关联自身id
    private User parent;

    // 修正OneToMany关联，指定mappedBy指向parent字段，维护双向关联
    @OneToMany(mappedBy = "parent", cascade = CascadeType.ALL, orphanRemoval = true)
    private List<User> childs = new ArrayList<>(); // 初始化集合避免空指针

    // 保留原有getter/setter...

    // 添加parent的getter/setter
    public User getParent() {
        return parent;
    }

    public void setParent(User parent) {
        this.parent = parent;
    }

    // 辅助方法：维护双向关联一致性
    public void addChild(User child) {
        childs.add(child);
        child.setParent(this);
    }
}

2. 修正用户创建逻辑 #

创建子用户时，需同时维护父用户的子列表和子用户的父引用，确保关联关系完整：

public void save() {
    FacesContext facesContext = FacesContext.getCurrentInstance();
    boolean res = false;
    if (getUsername() != null) {
        try {
            Application.logger.info("User Domain : " + this.getScope() + " For : " + this.getUsername());
            User masterUser = service.loginUser(facesContext);
            MasterRepository repo = new MasterRepository();
            User childUser = new User();
            childUser.setStatus(getStatus());
            childUser.setScope(getScope());
            childUser.setTel(getTel());
            childUser.setPassword(getPassword());
            childUser.setDomain(getDomain());
            childUser.setName(getName());
            childUser.setUsername(getUsername());
            
            // 用辅助方法添加子用户，自动维护双向关联
            masterUser.addChild(childUser);
            
            if (getId() > 0) {
                childUser.setId(getId());
            }
            res = repo.save(masterUser);
        } catch (Exception e) {
            Application.logger.fatal(e.getMessage());
        }
    }
}

3. 实现全层级子用户查询 #

方案一：代码递归加载（兼容所有数据库） #

在User类中添加递归方法，获取所有层级的子用户：

// 在User类中添加方法
public List<User> getAllDescendants() {
    List<User> allDescendants = new ArrayList<>(childs);
    for (User child : childs) {
        allDescendants.addAll(child.getAllDescendants());
    }
    return allDescendants;
}

注意：懒加载模式下需确保在EntityManager会话内调用该方法，避免LazyInitializationException，可在Repository查询时提前fetch子用户，或在Service层开启事务。

方案二：JPQL递归查询（适用于支持WITH RECURSIVE的数据库，如PostgreSQL、MySQL 8+） #

在MasterRepository中添加递归查询方法，一次性获取所有层级子用户：

public List<User> getAllDescendantsOfUser(Long userId) {
    EntityManager entityManager = hibernateService.getEntityManagerFactory().createEntityManager();
    String jpql = """
        WITH RECURSIVE user_hierarchy AS (
            SELECT u FROM User u WHERE u.id = :userId
            UNION ALL
            SELECT child FROM User child JOIN user_hierarchy parent ON child.parent.id = parent.id
        )
        SELECT u FROM user_hierarchy u WHERE u.id != :userId
    """;
    TypedQuery<User> query = entityManager.createQuery(jpql, User.class);
    query.setParameter("userId", userId);
    return query.getResultList();
}

4. 优化Repository查询逻辑 #

若需在获取用户时直接加载直接子用户，可修改查询方法使用JOIN FETCH：

// 修改findByUsername方法，加载直接子用户
public User findByUsernameWithChilds(String username) {
    username = username.toLowerCase();
    EntityManager entityManager = hibernateService.getEntityManagerFactory().createEntityManager();
    String jpql = "SELECT u FROM User u LEFT JOIN FETCH u.childs WHERE u.username = :username";
    TypedQuery<User> query = entityManager.createQuery(jpql, User.class);
    query.setParameter("username", username);
    List<User> result = query.getResultList();
    return result.isEmpty() ? null : result.get(0);
}

内容的提问来源于stack exchange，提问作者FarshiD Naqizadeh