java - Spring Boot如何在发送响应之前修改实体中的列表

Question

如何在发送之前修改实体中的列表？

说明:

我有一个建筑实体，其中包含公寓列表:

@Entity
@Table(name = "building")
public class Building {

...

@OneToMany(mappedBy = "building", cascade = CascadeType.ALL)
private Set<Flat> flats;
}

我还有用户对象，其角色:经理或商业。经理可以看到所有单位，但商业只能看到他创建的单位。

现在我需要一个端点来查找建筑物:

@RestController
@RequestMapping("/building")
public class BuildingController {
    @GetMapping("/{id}")
    public Building getBuildingById(@PathVariable(value = "id") Long id) {
        return buildingService.findById(id);
    }
}

但在这种情况下，即使用户是商业用户，该建筑拥有的所有公寓都会发送。 如何修改公寓列表以仅发送好的公寓？

Answer 1

我发现的解决方案是使用数据传输对象(DTO)。当我从数据库获取实体时，我会创建一个 DTO 并在发送之前仅填充所需的信息。这是 BuildingDto.java 的示例:

public class BuildingDto implements Serializable {
    ...
    private List<FlatDto> flats;
    private List<FlatUnauthorizedDto> flatsUnauthorized;

    // Constructors + getters + setters
}