首页 > 解决方案 > 使用 Hilt 预填充房间数据库

问题描述

我正在尝试使用 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 传递给回调类并使其工作。我要求提出一些方法来实现这一点,或者任何替代方案也将不胜感激。

标签: javaandroidkotlinandroid-room

解决方案


根据您在问题中添加的链接,您找到解决方案只需编辑第一个方法提供 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)
}
}

我尝试这段代码对我很好


推荐阅读