一文了解Android ViewModelScope 如何自动取消协程

2024-04-02 19:04:59 248人浏览薄情痞子

摘要

先看一下 ViewModel 中的 ViewModelScope 是何方神圣 val ViewModel.viewModelScope: Co

先看一下 ViewModel 中的 ViewModelScope 是何方神圣

val ViewModel.viewModelScope: CoroutineScope
        get() {
            val scope: CoroutineScope? = this.getTag(JOB_KEY)
            if (scope != null) {
                return scope
            }
            return setTagIfAbsent(JOB_KEY,
                CloseableCoroutineScope(SupervisorJob() + Dispatchers.Main.immediate))
        }

可以看到这个是一个扩展方法，

再点击 setTagIfAbsent 方法进去

 <T> T setTagIfAbsent(String key, T newValue) {
        T previous;
        synchronized (mBaGofTags) {
            previous = (T) mBagOfTags.get(key);//第一次肯定为null
            if (previous == null) {
                mBagOfTags.put(key, newValue);//null 存储
            }
        }
        T result = previous == null ? newValue : previous;
        if (mCleared) {//判断是否已经clear了
            // It is possible that we'll call close() multiple times on the same object, but
            // Closeable interface requires close method to be idempotent:
            // "if the stream is already closed then invoking this method has no effect." (c)
            closeWithRuntimeException(result);
        }
        return result;
    }

可以看到这边会把我们的 ViewModel 存储到 ViewModel 内的 mBagOfTags 中

这个 mBagOfTags 是

    private final Map<String, Object> mBagOfTags = new HashMap<>();

这个时候我们 viewModel 就会持有我们 viewModelScope 的协程作用域了。那..这也只是表述了我们 viewModelScope 存在哪里而已，什么时候清除呢？

先看一下 ViewModel 的生命周期：

可以看到 ViewModel 的生命周期会在 Activity onDestory 之后会被调用。那...具体哪里调的？

翻看源码可以追溯到 ComponentActivity 的默认构造器内

 public ComponentActivity() {
     
        getLifecycle().addObserver(new LifecycleEventObserver() {
            @Override
            public void onStateChanged(@NonNull LifecycleOwner source,
                    @NonNull Lifecycle.Event event) {
                if (event == Lifecycle.Event.ON_DESTROY) {
                    if (!isChanginGConfigurations()) {
                        getViewModelStore().clear();
                    }
                }
            }
        });
  }

可以看到内部会通对 Lifecycle 添加一个观察者，观察当前 Activity 的生命周期变更事件，如果走到了 Destory ，并且本次 Destory 并非由于配置变更引起的，才会真正调用 ViewModelStore 的 clear 方法。

跟进 clear 方法看看：

public class ViewModelStore {
    private final HashMap<String, ViewModel> mMap = new HashMap<>();

    
    public final void clear() {
        for (ViewModel vm : mMap.values()) {
            vm.clear();
        }
        mMap.clear();
    }
}

可以看到这个 ViewModelStore 内部实现用 HashMap 存储 ViewModel

于是在 clear 的时候，会逐个遍历调用 clear方法，再次跟进 ViewModel 的 clear 方法

 @MainThread
    final void clear() {
        mCleared = true;
        // Since clear() is final, this method is still called on mock objects
        // and in those cases, mBagOfTags is null. It'll always be empty though
        // because setTagIfAbsent and getTag are not final so we can skip
        // clearing it
        if (mBagOfTags != null) {
            synchronized (mBagOfTags) {
                for (Object value : mBagOfTags.values()) {
                    // see comment for the similar call in setTagIfAbsent
                    closeWithRuntimeException(value);
                }
            }
        }
        onCleared();
    }

可以发现我们最初存放 viewmodelScope 的 mBagOfTags

这里面的逻辑就是对 mBagOfTags 存储的数据挨个提取出来并且调用 closeWithRuntimeException

跟进 closeWithRuntimeException：

 private static void closeWithRuntimeException(Object obj) {
        if (obj instanceof Closeable) {
            try {
                ((Closeable) obj).close();
            } catch (IOException e) {
                throw new RuntimeException(e);
            }
        }
    }

该方法内会逐个判断对象是否实现 Closeable 如果实现就会调用这个接口的 close 方法，

再回到最初我们 viewModel 的扩展方法那边，看看我们 viewModelScope 的真正面目

internal class CloseableCoroutineScope(context: CoroutineContext) 
    : Closeable, CoroutineScope {
    override val coroutineContext: CoroutineContext = context

    override fun close() {
        coroutineContext.cancel()
    }
}

可以明确的看到我们的 ViewModelScope 实现了 Closeable 并且充写了 close 方法，

close 方法内的实现会对协程上下文进行 cancel。

至此我们可以大致整理一下：

viewModelScope 是 ViewModel 的扩展成员，该对象是 CloseableCoroutineScope，并且实现了 Closeable 接口
ViewModelScope 存储在 ViewModel 的名叫 mBagOfTags 的HashMap中啊
ViewModel 存储在 Activity 的 ViewModelStore 中，并且会监听 Activity 的 Lifecycle 的状态变更，在ON_DESTROY 且非配置变更引起的事件中对 viewModelStore 进行清空
ViewModelStore 清空会对 ViewModelStore 内的所有 ViewModel 逐个调用 clear 方法。
ViewModel的clear方法会对 ViewModel的 mBagOfTags 内存储的对象进行调用 close 方法(该对象需实现Closeable 接口)
最终会会调用我们 ViewModelScope 的实现类 CloseableCoroutineScope 的 close 方法中。close 方法会对协程进行 cancel。

到此这篇关于一文了解Android ViewModelScope 如何自动取消协程的文章就介绍到这了,更多相关Android ViewModel Scope 取消协程内容请搜索编程网以前的文章或继续浏览下面的相关文章希望大家以后多多支持编程网！

--结束END--

本文标题: 一文了解Android ViewModelScope 如何自动取消协程

本文链接: https://lsjlt.com/news/153281.html(转载时请注明来源链接)

有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341