Kotlin教程 Flow(数据流)-中级讲解

一.代码使用

1.ViewModel操作数据

class UserInfoViewModel : ViewModel() { /** * 保存用户信息 * */ fun saveUserInfo(userInfoDao: UserInfoDao, userInfo: UserInfo) { viewModelScope.launch(Dispatchers.IO) { val dbUserInfo = fromJson2Bean(fromObject2Json(userInfo), DBUserInfo::class.java) dbUserInfo.id = 0 userInfoDao.insert(dbUserInfo) Log.d("数据流", "生产数据 json:" + fromObject2Json(dbUserInfo)) } } /** * 获取用户信息 * */ fun getUserInfo(userInfoDao: UserInfoDao, userId: String, callback: ICallback) { viewModelScope.launch { getUserInfoData(userInfoDao, userId) .flowOn(Dispatchers.IO)//指定操作运行在IO线程 .catch { exception ->//异常处理 callback.onComplete(UserInfo(), exception.message.toString()) } .collect { userInfo ->//collect方法自动切换到主线程接收数据 callback.onComplete(userInfo, "获取成功") } } } /** * 获取用户信息 * */ private fun getUserInfoData(userInfoDao: UserInfoDao, userId: String) = flow { try { val dbUserInfo = userInfoDao.queryByUserId(userId) val userInfo = fromJson2Bean(fromObject2Json(dbUserInfo), UserInfo::class.java) emit(userInfo) Log.d("数据流", "生产数据 所处线程:" + Thread.currentThread().name) } catch (e: Exception) { throw e } } /** * 回调 * */ interface ICallback { fun onComplete(userInfo: UserInfo, errorMsg: String) } }

2.Activity使用

class KotlinFlowActivity : CoroutineBaseActivity() { private val dao = RoomManager.impl.userInfoDao() private val viewModel by viewModels<UserInfoViewModel>() private lateinit var addTextView: TextView private lateinit var queryTextView: TextView private lateinit var textView: TextView override fun onCreate(savedInstanceState: Bundle?) { super.onCreate(savedInstanceState) setContentView(R.layout.activity_flow1) addTextView = findViewById(R.id.flow_tv1) addTextView.setOnClickListener { add() } queryTextView = findViewById(R.id.flow_tv2) queryTextView.setOnClickListener { query() } textView = findViewById(R.id.flow_content) } /** * 新增 * */ private fun add() { val userInfo = UserInfo() userInfo.userId = "abcd123456" userInfo.userName = "张三" userInfo.remark = "备注11111111111" viewModel.saveUserInfo(dao, userInfo) } /** * 查询 * */ private fun query() { viewModel.getUserInfo(dao, "abcd123456", object : UserInfoViewModel.ICallback { override fun onComplete(userInfo: UserInfo, errorMsg: String) { textView.text = fromObject2Json(userInfo) Log.d("数据流", "获取数据 所处线程:" + Thread.currentThread().name) } }) } }

3.细节说明

3.1.因为最终收集collect方法，是挂起方法，所以调用的时候需要在协程中操作，因为本身在ViewModel中，所以可以直接使用ViewModel的协程方法。

//collect方法 public suspend fun collect(collector: FlowCollector<T>) //ViewModel协程方法调用 viewModelScope.launch { XXX }

3.2.因Flow流一般用在获取Room数据库(本例就是)或者网络请求中，所以需要指定操作的线程为IO线程，不可直接在UI主线程中做耗时操作

.flowOn(Dispatchers.IO)//指定操作运行在IO线程

3.3.使用数据流时，切记不要忘记catch