使用协程在 api 请求之前从房间数据库加载数据

Question

我需要实现以下逻辑。 当应用程序启动时，从共享首选项加载 userId，然后从 Room 数据库加载用户对象，最终能够使用来自用户对象的访问令牌发出 Retrofit 请求。

这是我目前的情况。

在应用程序启动时使用 Dagger 初始化用户存储库

class App : Application() {
    companion object {
        lateinit var userRepository: UserRepository
    }
    override fun onCreate() {
        super.onCreate()
        appComponent = DaggerAppComponent.factory().create(applicationContext)
        userRepository = appComponent.userRepository()
    }
}

从 sharedPrefs 加载userId ，然后在存储库初始化时从 db 加载用户

@Singleton
class UserRepository @Inject constructor(
    private val dao: UserDao,
    private val sharedPrefs: SharedPreferences
) {
    var user: User
    private var userId: Int

    init {
        userId = sharedPrefs.getInt(PREFS_KEY_UID, 0)
        GlobalScope.launch(Dispatchers.IO) {
            user = dao.geUser(userId)
        }
    }
}

拦截 Retrofit 请求以将令牌插入标头

interface Webservice {
    companion object Factory {
        private val headerInterceptor = Interceptor() {
            val newRequest = it.request().newBuilder()
                .addHeader("Authorization", "Bearer ${App.userRepository.user.token}")
                .build()
            it.proceed(newRequest)
        }
        fun create(): Webservice {
            val client = OkHttpClient.Builder()
                .addInterceptor(headerInterceptor)
                .build();
            val retrofit = retrofit2.Retrofit.Builder()
                .client(client)
                .baseUrl(BASE_URL)
                .build()
            return retrofit.create(Webservice::class.java)
        }
    }
}

创建 MainActivity 的 ViewModel 时发出第一个 Web 请求

class MainViewModel @Inject constructor(
    private val repository: DataRepository
): ViewModel() {
    val ticket: LiveData<Data> = repository.getData(viewModelScope)
}

@Singleton
class DataRepository @Inject constructor(
    private val webservice: Webservice,
    private val dao: DataDao
) {
    fun getData(scope: CoroutineScope): LiveData<Data> {
        refreshData(scope)
        return dao.getData()
    }
    suspend fun refreshData(scope: CoroutineScope) {
        scope.launch(Dispatchers.IO) {
            ...
            val response = webservice.getData().awaitResponse()
            ...
            dao.insert(response.body()!!)
            ...
        }
    }
}

您可能已经注意到我正在使用协程。

如何确保在发送 Web 请求之前从数据库加载user ？

Answer 1

为确保用户是从数据库加载的，您可以使用带有async协程构建器的Deferred并在访问用户变量时调用await函数。 例如：

val user: Deferred<User>

init {

   userId = sharedPrefs.getInt(PREFS_KEY_UID, 0)
   user = GlobalScope.aync {
             dao.geUser(userId)
          }
}

....

val user = repository.user.await()

Answer 2

这是我现在的解决方案。 感谢 Marko Topolnik 的评论。

只需像这样通过runBlocking替换GlobalScope.launch

init {
    userId = sharedPrefs.getInt(PREFS_KEY_UID, 0)
    runBlocking(Dispatchers.IO) {
        user = dao.geUser(userId)
    }
}

就我而言，该操作是轻量级的，因此不需要启动画面。