将 Broadcast 接收器包装成 Flow(协程)

Wrap Broadcast receiver into Flow (coroutine)

我有一个用于 wifi 扫描结果的广播接收器作为数据源,我想以协程方式实现它。我在这里找到了暂停功能的答案:

suspend fun getCurrentScanResult(): List<ScanResult> =
    suspendCancellableCoroutine { cont ->
        //define broadcast reciever
        val wifiScanReceiver = object : BroadcastReceiver() {
            override fun onReceive(c: Context, intent: Intent) {
                if (intent.action?.equals(WifiManager.SCAN_RESULTS_AVAILABLE_ACTION) == true) {
                    context.unregisterReceiver(this)
                    cont.resume(wifiManager.scanResults)
                }
            }
        }
        //setup cancellation action on the continuation
        cont.invokeOnCancellation {
            context.unregisterReceiver(wifiScanReceiver)
        }
        //register broadcast reciever
        context.registerReceiver(wifiScanReceiver, IntentFilter(WifiManager.SCAN_RESULTS_AVAILABLE_ACTION))
        //kick off scanning to eventually receive the broadcast
        wifiManager.startScan()
    }

这对于 signle emit 来说很好,但是如果我想在扫描过程中获得结果,那么我会崩溃,因为 cont.resume() 只能被调用一次。然后我决定尝试Flow。这是我的代码:

suspend fun getCurrentScanResult(): Flow<List<ScanResult>> =
    flow{
        val wifiScanReceiver = object : BroadcastReceiver() {
            override fun onReceive(c: Context, intent: Intent) {
                if (intent.action?.equals(WifiManager.SCAN_RESULTS_AVAILABLE_ACTION) == true) {
                    //context.unregisterReceiver(this)
                    emit(wifiManager.scanResults)
                }
            }
        }
        //setup cancellation action on the continuation
        //register broadcast reciever
        context.registerReceiver(wifiScanReceiver, IntentFilter(WifiManager.SCAN_RESULTS_AVAILABLE_ACTION))
        //kick off scanning to eventually receive the broadcast
        wifiManager.startScan()
    }

但是现在 Android Stuidio 说 Suspension functions can be called only within coroutine body 函数 emit(wifiManager.scanResults) 有没有办法在这里使用 Flow?

请查看专门为此用例设计的 callback flow。这样的事情就可以完成工作:

callbackFlow {
  val receiver = object : BroadcastReceiver() {
    override fun onReceive(context: Context?, intent: Intent) {
      if (intent.action == WifiManager.SCAN_RESULTS_AVAILABLE_ACTION) {
        sendBlocking(wifiManager.scanResults) // or non-blocking offer()
      }
    }
  } 
  context.registerReceiver(receiver, intentFilter)

  awaitClose {
      context.unregisterReceiver(receiver)
  }
}

您可能还想与其他人分享此流程,例如shareIn 运营商避免为每个流量订阅者注册一个新的接收器。