使用 Hilt 预填充房间数据库

Question

我正在尝试使用 RoomDatabase.Callback() 方法用数据预填充 Room 数据库，并且也成功做到了这一点。后来我尝试对 Hilt 进行同样的操作，但无法弄清楚如何在 hilt 模块中提供数据库时添加回调。

这是数据库模块：

@Module
@InstallIn(ApplicationComponent::class)
object DatabaseModule {

    @Singleton
    @Provides
    fun provideDatabase(
        @ApplicationContext context: Context
    ): PersonDatabase {
        return Room.databaseBuilder(
            context,
            PersonDatabase::class.java,
            "person_database"
        ).build()

    }

    @Singleton
    @Provides
    fun provideDao(database: PersonDatabase) = database.personDao()

}

这是回调类：

class PersonCallback @Inject constructor(
    private val dao: PersonDao
) : RoomDatabase.Callback() {

    private val applicationScope = CoroutineScope(SupervisorJob())

    override fun onCreate(db: SupportSQLiteDatabase) {
        super.onCreate(db)
        applicationScope.launch(Dispatchers.IO) {
            populateDatabase()
        }
    }

    private suspend fun populateDatabase() {
        val person = Person("FirstName", "LastName", 20)
        dao.insertData(person)
    }
}

我尝试过使用数据库并提供这样的 dao，但它陷入了循环并且应用程序崩溃了。简而言之，错误表明以下方式是递归的，因此是不允许的

    @Singleton
    @Provides
    fun provideDatabase(
        @ApplicationContext context: Context
    ): PersonDatabase {
        return Room.databaseBuilder(
            context,
            PersonDatabase::class.java,
            "person_database"
        ).addCallback(
            PersonCallback(provideDatabase(context).personDao())
        ).build()

    }

然后，我按照 SO 帖子并尝试复制它，即：

使用 Hilt 预填充 Room 数据库，无需创建额外的数据库实例

执行上述操作后，应用程序再次崩溃，并提供了一个引用自动生成的类的巨大错误，并且没有任何内容可以使其更有意义。

我尝试的另一种方法是将 dao 作为参数传递，但它再次崩溃并出现错误，指出它会导致依赖循环，因为我也将 dao 作为构造函数参数传递给存储库。

    @Singleton
    @Provides
    fun provideDatabase(
        @ApplicationContext context: Context,
        personDao: PersonDao
    ): PersonDatabase {
        return Room.databaseBuilder(
            context,
            PersonDatabase::class.java,
            "person_database"
        ).addCallback(
            PersonCallback(personDao)
        ).build()

    }

经过所有这些尝试，我无法弄清楚应该如何将 dao 传递给回调类并使其工作。我请求建议一些方法来实现这一点，或者任何替代方案也将受到赞赏。

Answer 1

根据您在问题中添加的链接，您找到解决方案只需编辑第一个方法提供 PersonDatabase

@Singleton
@Provides
fun provideDatabase(
    @ApplicationContext context: Context,
    provider: Provider<PersonDao>
): PersonDatabase {
    return Room.databaseBuilder(
        context,
        PersonDatabase::class.java,
        "person_database"
    ).addCallback(
        PersonCallback(provider)
    ).build()

}

然后你的回调类采用 PersonDao 的提供程序，只需为你的类进行编辑

class PersonCallback (
private val provider: Provider<PersonDao>
  ) : RoomDatabase.Callback() {

private val applicationScope = CoroutineScope(SupervisorJob())

override fun onCreate(db: SupportSQLiteDatabase) {
    super.onCreate(db)
    applicationScope.launch(Dispatchers.IO) {
        populateDatabase()
    }
}

private suspend fun populateDatabase() {
    val person = Person("FirstName", "LastName", 20)
    provider.get().insertData(person)
}
}

我尝试这段代码对我来说工作得很好