java - 使用 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 帖子并尝试复制它，即:

Pre-populating Room database with Hilt without creating an extra instance of the database

在上述之后，应用程序再次崩溃，提供了一个巨大的错误，该错误引用了自动生成的类并且没有任何意义可以使它更有意义

我尝试的另一种方法是将 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)
}
}

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