android - 如何从 Worker 类中获取 ViewModel？

Question

1. 我的应用程序中有一个Activivity名为 MainActivity。

2. MainActivity 与ViewModelOnCreate 方法中的a 相关联

   ...
   val someViewModel = ViewModelProviders.of(this).get(SomeViewModel::class.java)
   ...
   

3. 我做了一些背景工作Workers。

   val someWork = OneTimeWorkRequestBuilder<LoginWorker>().build() WorkManager.getInstance().beginUniqueWork("SOMEWORK", ExistingWorkPolicy.REPLACE, captivePortalWork).enqueue()

4. 问题来了...worker例如循环10次，每次只睡一秒钟...所以我们有10秒的后台工作。我想更新ViewModel以显示后台工作的进度并每次更新 UI。

   A）我可以从 MainActivity 访问 View 模型并观察worker但是我只能更新ViewModel进度成功或失败的时间......这意味着我可以在工作完成 0 或 100% 时显示进度。

   B) 如果我可以从 Worker 类访问 ViewModel，我可以随时更新 ViewModel。但我不知道如何......这种方法在 Worker 类中不起作用。

   val someWork = ViewModelProviders.of(this).get(SomeViewModel::class.java)

   我怎样才能ViewModel从Worker课堂上得到一个？

Answer 1

您必须将值从Worker输出分配Data给ViewModel，并且在您的 中观察到此数据Activity，如下所示：

工人代码：

public class MyWorker extends Worker {

    public static final String MY_KEY_DATA_FROM_WORKER = "MY_KEY_DATA_FROM_WORKER";

    public MyWorker(@NonNull Context appContext, @NonNull WorkerParameters workerParams) {
        super(appContext, workerParams);
    }

    @NonNull
    @Override
    public Worker.Result doWork() {
        //here do http or database requests
        String result = "my needed result";

        if (result != null) {
            setOutputData(new Data.Builder()
                    .putString(MY_KEY_DATA_FROM_WORKER, result)
                    .build());

            return Worker.Result.SUCCESS;
        }

        return Result.FAILURE;
    }
}

ViewModel 的代码：

 public class MyActivityModel extends ViewModel {

    private final MutableLiveData<String> myLiveData = new MutableLiveData<String>();

    public void loadDataFromWorker(LifecycleOwner lifecycleOwner) {
        OneTimeWorkRequest myWorkerReq = new OneTimeWorkRequest.Builder(MyWorker.class)
                .build();

        WorkManager mWorkManager = WorkManager.getInstance();
        mWorkManager
                .beginWith(myWorkerReq)
                .enqueue();

        mWorkManager.getStatusByIdLiveData(myWorkerReq.getId()).observe(lifecycleOwner, new Observer<WorkStatus>() {
            @Override
            public void onChanged(WorkStatus workStatus) {
                if (workStatus.getState().isFinished()) {

                    // here you asign data to ViewModel
                    Data workOutputData = workStatus.getOutputData();
                    myLiveData.setValue(workOutputData.getString(MyWorker .MY_KEY_DATA_FROM_WORKER));
                }
            }
        });
    }

    public MutableLiveData<String> getData() {
        return myLiveData;
    }
  }

活动代码：

//...

  @Override
protected void onCreate(Bundle savedInstanceState) {
    super.onCreate(savedInstanceState);
    setContentView(R.layout.my_activity);

    //declare model and start loading data in Worker
    MyActivityModel model = ViewModelProviders.of(this).get(MyActivityModel.class);
    model.loadDataFromWorker(this);

    //observe data from model
    model.getData().observe(this, new Observer<String>() {
        @Override
        public void onChanged(String str) {

        //here you get new String value or null if Work failed 

        }
    });
}

//...

Answer 2

简而言之，您不想将您的ViewModeltoWorkManager或Workerclass 公开，而是将百分比公开给您的ViewModel.

WorkManager 2.3.1+ 添加了对设置和观察中间进度的支持。

对于使用 ListenableWorker 或 Worker 的 Java 开发人员，setProgressAsync() API 返回一个 ListenableFuture；更新进度是异步的，因为更新过程涉及将进度信息存储在数据库中。在 Kotlin 中，您可以使用 CoroutineWorker 对象的 setProgress() 扩展函数来更新进度信息。

资源

对于 Kotlin，使用CoroutineWorker以下内容：

companion object {
    const val Progress = "Progress"
    private const val delayDuration = 1L
}

override suspend fun doWork(): Result {
    val firstUpdate = workDataOf(Progress to 0)
    val lastUpdate = workDataOf(Progress to 100)
    setProgress(firstUpdate)
    delay(delayDuration)
    setProgress(lastUpdate)
    return Result.success()
}

在您的ViewModel中，您可以像这样观察您的进度：

WorkManager.getInstance(applicationContext)
// requestId is the WorkRequest id
.getWorkInfoByIdLiveData(requestId)
.observe(observer, Observer { workInfo: WorkInfo? ->
        if (workInfo != null) {
            val progress = workInfo.progress
            val value = progress.getInt(Progress, 0)
            // Do something with progress information
        }
})

如果您正在使用WorkContinuation，并且想要整体百分比，我发现最简单的方法是这样使用workInfosLiveData：

WorkContinuation.combine(workForSyncingFlights).also {
it.enqueue()
it.workInfosLiveData.asFlow().conflate().collect { listOfWorkInfo ->
    if (listOfWorkInfo.isNullOrEmpty()) return@collect
    listOfWorkInfo.forEach { workInfo ->
        if (workInfo.state.isFinished) {
            val finished = listOfWorkInfo.filter { info -> info.state.isFinished }.size
            val totalPercent = finished.toDouble().div(listOfWorkInfo.size.toDouble()) * 100
        }
    }
}

Answer 3

来自非常好的 Google 文档的两个附加信息。

a) 如果您来到这里是因为没有 viewmodelscope.launch 就无法运行您的任务，请查看 Android“ Threading in WorkManager ”文档。

b）这里的问题可能可以通过“观察中间工人进度”来解决。

Answer 4

您不能ViewModel为此使用 a 。您需要使用自己的房间数据库LiveData并向您的 UI 公开一个以显示进度。