Spring Boot - 搜索类别树

Question

我正在编写基本上是广告服务的 Spring Boot 应用程序。在数据库中（我使用 PostgreSQL 方言）我存储类别树。每个类别都包含对其直接父级的引用。每个广告都有自己的类别，为了解决问题，您可以仅将广告分配给叶类别。

下面的图像捕获了结构。

Category tree

现在我需要从数据库中检索特定类别的广告。但是假设我要检索类别 E 的所有广告。这意味着我要检索类别 E 的子类别的广告，但不一定是直接子类别的广告。在这种情况下，我想检索类别 G、H、I 和 J 的广告，因为它们都是叶子并且都是类别 E 的子节点。

如何通过 Spring 存储库实现它？以下是此示例的简化实体类。

类别...

@Entity
@Table(name = "categories")
public class Category {

    @Id
    @Column(name = "cat_id")
    private int id;

    @Column(name = "cat_name", length = 50, nullable = false, unique = true)
    private String name;

    @ManyToOne(cascade = CascadeType.ALL)
    @JoinColumn(name = "cat_parent", referencedColumnName = "cat_id", nullable = false)
    private Category parent;

    ...
}

...和广告。

@Entity
@Table(name = "advertisements")
public class Advertisement {

    @Id
    @Column(name = "adv_id")
    private long id;

    @Column(name = "adv_title", length = 100, nullable = false)
    private String title;

    @Column(name = "adv_description", length = 5000, nullable = false)
    private String description;

    @ManyToOne(cascade = CascadeType.ALL)
    @JoinColumn(name = "adv_category", referencedColumnName = "cat_id", nullable = false)
    private Category category;

    @ManyToOne(cascade = CascadeType.ALL)
    @JoinColumn(name = "adv_owner", referencedColumnName = "user_id", nullable = false)
    private User owner;

    ...
}

Answer 1

这可能不是最优雅的方法，但您可以先收集类别包，然后使用 in 子句 查询广告存储库，将结果限制为仅这些类别中的广告。

首先将父/子关联的另一端添加到 Category 对象（如果它不存在），然后创建一个方法，该方法返回一个包含该类别和所有子（不仅仅是直接子）的集合，例如：

@JoinColumn(name = "cat_parent")
@OneToMany
Set<Category> children;
... 
public Set<Category> collectLeafChildren() { 
   Set<Category> results = new HashSet<>(); 
   if (children.isEmpty()) { 
      results.add(this); 
   } else { 
      children.forEach(child -> { 
        results.addAll(child.collectLeafChildren()); 
      }); 
   } 
   return results; 
}

然后为Advertisement 存储库创建一个新方法，该方法返回给定类别集的所有广告，例如：

List<Advertisement> findByCategoryIn(Set<Category> categories);

并使用那些：

Set<Category> categories = category.collectLeafChildren();
List<Advertisement> advertisementRepository.findByCategoryIn(categories);