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 的協(xié)程作用域了。那..這也只是表述了我們 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<>();

    /**
     *  Clears internal storage and notifies ViewModels that they are no longer used.
     */
    public final void clear() {
        for (ViewModel vm : mMap.values()) {
            vm.clear();
        }
        mMap.clear();
    }
}

可以看到這個 ViewModelStore 內部實現(xiàn) 用 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();
    }

可以發(fā)現(xiàn)我們最初存放 viewmodelScope 的 mBagOfTags

這里面的邏輯就是對 mBagOfTags 存儲的數(shù)據挨個提取出來并且調用 closeWithRuntimeException

跟進 closeWithRuntimeException：

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

該方法內會逐個判斷對象是否實現(xiàn) Closeable 如果實現(xiàn)就會調用這個接口的 close 方法，

再回到最初我們 viewModel 的擴展方法那邊，看看我們 viewModelScope 的真正面目

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

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

可以明確的看到我們的 ViewModelScope 實現(xiàn)了 Closeable 并且充寫了 close 方法，

close 方法內的實現(xiàn) 會對協(xié)程上下文進行 cancel。

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

viewModelScope 是 ViewModel 的擴展成員，該對象是 CloseableCoroutineScope，并且實現(xiàn)了 Closeable 接口
ViewModelScope 存儲在 ViewModel 的名叫 mBagOfTags 的HashMap中啊
ViewModel 存儲在 Activity 的 ViewModelStore 中，并且會監(jiān)聽 Activity 的 Lifecycle 的狀態(tài)變更，在ON_DESTROY 且非配置變更引起的事件中對 viewModelStore 進行清空
ViewModelStore 清空會對 ViewModelStore 內的所有 ViewModel 逐個調用 clear 方法。
ViewModel的clear方法會對 ViewModel的 mBagOfTags 內存儲的對象進行調用 close 方法(該對象需實現(xiàn)Closeable 接口)
最終會會調用我們 ViewModelScope 的實現(xiàn)類 CloseableCoroutineScope 的 close 方法中。close 方法會對協(xié)程進行 cancel。

到此這篇關于一文了解Android ViewModelScope 如何自動取消協(xié)程的文章就介紹到這了,更多相關Android ViewModel Scope 取消協(xié)程內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: