Xác định và truy vấn mối quan hệ lồng nhau

Đôi khi, bạn cần truy vấn một tập hợp gồm ba bảng trở lên và tất cả đều liên quan đến nhau. Trong trường hợp đó, bạn sẽ xác định mối quan hệ lồng ghép giữa các bảng.

Giả sử trong ví dụ về ứng dụng phát nhạc trực tuyến, bạn muốn truy vấn tất cả người dùng, danh sách phát của mỗi người dùng và tất cả bài hát trong từng danh sách phát của từng người dùng. Người dùng có mối quan hệ một với nhiều với danh sách phát và danh sách phát có mối quan hệ nhiều với nhiều với bài hát. Ví dụ về mã sau đây cho thấy các lớp đại diện cho các thực thể này, cũng như bảng tham chiếu chéo cho mối quan hệ nhiều với nhiều giữa danh sách phát và bài hát:

@Entity
data class User(
    @PrimaryKey val userId: Long,
    val name: String,
    val age: Int
)

@Entity
data class Playlist(
    @PrimaryKey val playlistId: Long,
    val userCreatorId: Long,
    val playlistName: String
)

@Entity
data class Song(
    @PrimaryKey val songId: Long,
    val songName: String,
    val artist: String
)

@Entity(primaryKeys = ["playlistId", "songId"], indices = [Index("playlistId", "songId")])
data class PlaylistSongCrossRef(
    val playlistId: Long,
    val songId: Long
)

Trước tiên, hãy mô hình hoá mối quan hệ giữa hai bảng trong tập hợp của bạn như bình thường, sử dụng lớp dữ liệu và chú giải @Relation. Ví dụ sau đây hiển thị một lớp PlaylistWithSongs mà mô hình hoá mối quan hệ nhiều với nhiều giữa lớp thực thể Playlist và lớp thực thể Song:

data class PlaylistWithSongs(
    @Embedded val playlist: Playlist,
    @Relation(
        parentColumns = ["playlistId"],
        entityColumns = ["songId"],
        associateBy = Junction(PlaylistSongCrossRef::class)
    )
    val songs: List<Song>
)

Sau khi xác định một lớp dữ liệu đại diện cho mối quan hệ này, hãy tạo một lớp dữ liệu khác mà mô hình hoá mối quan hệ giữa một bảng khác từ tập hợp của bạn và lớp mối quan hệ đầu tiên, "lồng" mối quan hệ hiện có vào trong mối quan hệ mới. Ví dụ sau đây hiển thị một lớp UserWithPlaylistsAndSongs mà mô hình hoá mối quan hệ một với nhiều giữa lớp thực thể User và lớp mối quan hệ PlaylistWithSongs:

data class UserWithPlaylistsAndSongs(
    @Embedded val user: User,
    @Relation(
        entity = Playlist::class,
        parentColumns = ["userId"],
        entityColumns = ["userCreatorId"]
    )
    val playlists: List<PlaylistWithSongs>
)

Lớp UserWithPlaylistsAndSongs gián tiếp mô hình hoá mối quan hệ giữa cả ba lớp thực thể: User, PlaylistSong. Điều này được minh hoạ trong hình 1.

UserWithPlaylistsAndSongs mô hình hoá mối quan hệ giữa Người dùng và PlaylistWithSongs. Tác vụ này trả về mối quan hệ giữa Danh sách phát và Bài hát.
Hình 1. Sơ đồ các lớp mối quan hệ trong ví dụ về ứng dụng phát nhạc trực tuyến.

Nếu tập hợp của bạn có nhiều bảng hơn, hãy tạo một lớp để mô hình hoá mối quan hệ giữa từng bảng còn lại và lớp mối quan hệ trước đó. Quy trình này tạo ra một chuỗi các mối quan hệ lồng ghép giữa tất cả các bảng bạn muốn truy vấn.

Cuối cùng, hãy thêm một hàm vào lớp đối tượng truy cập dữ liệu (DAO) để hiển thị hàm truy vấn mà ứng dụng của bạn cần. Hàm này đòi hỏi Room chạy nhiều truy vấn. Vì vậy, hãy thêm chú thích @Transaction để toàn bộ thao tác chạy một cách tỉ mỉ:

@Transaction
@Query("SELECT * FROM User")
suspend fun getUsersWithPlaylistsAndSongs(): List<UserWithPlaylistsAndSongs>