如何在Room中将user_roles的role_name映射为User的Set<String>属性？

现有SQLite表结构：

CREATE TABLE users (id INT, name VARCHAR(255))
CREATE TABLE user_roles (user_id INT REFERENCES users(id), role_name VARCHAR(255))

尝试编写的Room实体定义：

@Entity(tableName = "users")
data class User(
  @PrimaryKey
  val id: Int,
  val roles: Set<String>  // 期望从user_roles表的role_name字段获取数据
)

可选的第二个实体：

@Entity(tableName = "user_roles")
data class UserRole(
  val user_id: Int,
  val role_name: String
)

DAO定义：

@Dao
interface UserDao {
  @Query("SELECT * FROM users")
  fun getAllUsers(): Flow<List<User>>
}

由于SQLite不支持集合类型，上述代码无法编译。需要解决：如何将user_roles表的role_name映射到User实体的roles属性？Room能否自动完成该映射？

Room本身不会自动跨表映射集合类型，但可以通过两种官方推荐的方式实现需求：

方法一：使用@Relation实现一对多关联映射 #

这是处理实体间一对多关系的标准方案，Room会自动关联两张表并组装数据。

1. 调整User实体，通过@Relation注解关联UserRole实体：

@Entity(tableName = "users")
data class User(
  @PrimaryKey
  val id: Int,
  val name: String,
  // 关联user_roles表，parentColumn是当前表的关联字段，entityColumn是关联表的外键字段
  @Relation(
    parentColumn = "id",
    entityColumn = "user_id",
    entity = UserRole::class
  )
  private val roleEntities: List<UserRole>
) {
  // 转换为期望的Set<String>类型
  val roles: Set<String> get() = roleEntities.map { it.role_name }.toSet()
}

2. 修改DAO，添加@Transaction保证关联查询的原子性：

@Dao
interface UserDao {
  @Transaction
  @Query("SELECT * FROM users")
  fun getAllUsers(): Flow<List<User>>
}

Room会自动执行两步操作：先查询所有用户，再批量查询每个用户对应的角色，最后将结果组装到User的roles属性中。

方法二：自定义SQL聚合查询+手动映射 #

如果不需要单独操作user_roles表，可以通过SQL聚合函数将角色名拼接，再转换为集合。

1. 编写聚合查询，用GROUP_CONCAT将同一用户的角色名拼接成字符串：

@Dao
interface UserDao {
  @Query("""
    SELECT u.id, u.name, GROUP_CONCAT(ur.role_name, '|') as roles_str
    FROM users u
    LEFT JOIN user_roles ur ON u.id = ur.user_id
    GROUP BY u.id, u.name
  """)
  fun getAllUsersWithRoles(): Flow<List<UserWithRolesString>>

  // 中间数据类，用于接收SQL查询的原始结果
  data class UserWithRolesString(
    val id: Int,
    val name: String,
    val roles_str: String?
  )
}

2. 手动将拼接字符串转换为Set<String>：

// 在Repository或ViewModel中处理映射逻辑
fun Flow<List<UserDao.UserWithRolesString>>.convertToUsers(): Flow<List<User>> {
  return this.map { userList ->
    userList.map { item ->
      User(
        id = item.id,
        name = item.name,
        roles = item.roles_str?.split("|")?.toSet() ?: emptySet()
      )
    }
  }
}

注意：选择的分隔符要避免和角色名冲突，SQLite 3.33.0及以上版本也可以用JSON_GROUP_ARRAY直接生成JSON数组，再通过TypeConverter转换为集合。

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