5. Jetpack源码解析---ViewModel基本使用及源码解析

Hankkin

发布于 2019-08-14 14:56:44

8420

发布于 2019-08-14 14:56:44

文章被收录于专栏：Android干货园

截止到目前为止，JetpackNote源码分析的文章已经有四篇文章了，这一系列的文章我的初衷是想仔细研究一下Jetpack，最终使用Jetpack组件写一个Demo，上一篇已经分析了LiveData,本篇文章将分析ViewModel.

1.背景

Jetpack源码解析系列文章：

1. Jetpack源码解析—看完你就知道Navigation是什么了？

2. Jetpack源码解析—Navigation为什么切换Fragment会重绘？

3. Jetpack源码解析—用Lifecycles管理生命周期

4. Jetpack源码解析—LiveData的使用及工作原理

上篇我们对LiveData进行了分析，已清楚了它的主要作用，我们再来温习一下：

LiveData是一个可以感知Activity、Fragment生命周期的数据容器。其本身是基于观察者模式设计的，当LiveData所持有的数据发生改变时，它会通知对应的界面所持有该数据的UI进行更新，并且LiveData中持有Lifecycle的引用，所以只会在LifecycleOwner处于Active的状态下通知数据改变，果数据改变发生在非 active 状态，数据会变化，但是不发送通知，等 owner 回到 active 的状态下，再发送通知. 而LiveData配合今天所讲的ViewModel使用起来真的是非常方便，接下来我们开始分析：

2.简介

2.1 是什么？

简单来说：ViewModel是以关联生命周期的方式来存储和管理UI相关数据的类，即使configuration发生改变(例如屏幕旋转),数据仍然可以存在不会销毁.

举个例子来说：我们在开发中经常会遇到这种场景，当我们的Activity和Fragment被销毁重建之后，它们中的数据将会丢失，而我们一般的解决方案就是通过onSaveInstanceState()中保存数据，然后在onCreate()中恢复过来，但是当这些数据较大时或者数据可能又需要重新请求接口，这时候ViewModel就派上用场了，也就是说当我们把数据保存在ViewModel中，不管Activity和Fragment被销毁了还是屏幕旋转导致configuration发生了变化，保存在其中的数据依然存在。

不仅如此，ViewModel还可以帮助我们轻易的实现Fragment及Activity之间的数据共享和通信，下面会详细介绍。

2.2 ViewModel生命周期

先来看一下官方给出的图：

图中展示了当一个Activity经过屏幕旋转后的生命周期状态改变，右侧则是ViewModel的生命周期状态。我们可以看到ViewModel的生命周期并不会跟随着Activity的生命周期变化而变化，虽然ViewModel是由该Activity创建的。我们会在源码分析部分再去看ViewModel的生命周期具体是怎样的。下面我们看下ViewModel该怎样使用？

3. 基本使用

3.1 数据存储

我们参考官方Demo实现一个计时器的功能，并且演示当屏幕发生旋转时，计时器会不会重新启动：

DemoViewModel

class DemoViewModel : ViewModel() {

    var time: Long? = null

    var seekbarValue = MutableLiveData<Int>()
}

ViewModelFragment

class ViewModelFragment : Fragment() {

    override fun onCreateView(
        inflater: LayoutInflater, container: ViewGroup?,
        savedInstanceState: Bundle?
    ): View? {
        return inflater.inflate(R.layout.fragment_view_model, container, false)
    }


    override fun onActivityCreated(savedInstanceState: Bundle?) {
        super.onActivityCreated(savedInstanceState)
        val vm = ViewModelProviders.of(this).get(DemoViewModel::class.java)
        if (vm.time == null) {
            vm.time = SystemClock.elapsedRealtime()
        }
        chronometer.base = vm.time!!
        chronometer.start()

        btn_fragment_share.setOnClickListener {
            findNavController().navigate(R.id.viewModelShareActivity)
        }
    }

}

代码很简单，只是在ViewModel中存储了一个time值，在fragment中启动计时器，当我们旋转屏幕的时候你会发现，计时器的值并没有变化，仍然按照旋转之前的数值进行计数。

3.2 Fragment数据共享

在ViewModelShareActivity中展示了ViewModel中的数据进行Fragment数据共享的功能。

在之前的DemoViewModel中我们存储了seekbar的值，然后我们看Fragment中是怎么实现的？

class DataShareFragment : Fragment() {

    private lateinit var mViewModel: DemoViewModel

    override fun onCreateView(
        inflater: LayoutInflater, container: ViewGroup?,
        savedInstanceState: Bundle?
    ): View? {
        mViewModel = ViewModelProviders.of(activity!!).get(DemoViewModel::class.java)
        return inflater.inflate(R.layout.fragment_data_share, container, false)
    }


    override fun onActivityCreated(savedInstanceState: Bundle?) {
        super.onActivityCreated(savedInstanceState)
        subscribeSeekBar()
    }

    private fun subscribeSeekBar() {
        // 监听SeekBar改变ViewModel的中的值
        seekBar.setOnSeekBarChangeListener(object : SeekBar.OnSeekBarChangeListener {
            override fun onProgressChanged(seekBar: SeekBar, progress: Int, fromUser: Boolean) {
                if (fromUser) {
                    mViewModel.seekbarValue.value = progress
                }
            }

            override fun onStartTrackingTouch(seekBar: SeekBar) {}

            override fun onStopTrackingTouch(seekBar: SeekBar) {}
        })
        // 当ViewModel中的值发生变化时，更新SeekBar
        activity?.let {
            mViewModel.seekbarValue.observe(it, Observer<Int> { value ->
                if (value != null) {
                    seekBar.progress = value
                }
            })
        }
    }
}

看代码其实挺简单的，只做了两个操作：

监听SeekBar改变ViewModel的中的值
当ViewModel中的值发生变化时，更新SeekBar

同样，当旋转屏幕之后，SeekBar的值也不会改变。到这里ViewModel的基本使用方式我们已经了解了，接下来我们来分析一下它具体是怎么实现的？

没错，又到了源码分析的部分了，相信很多童鞋也都比较不喜欢源码的部分，包括我也是，之前很不愿意去看源码，但是当你尝试看了一些源码之后，你会发现很有意思的，因为有些东西你是有必要去分析源码实现的，这是作为一个开发者必备的基本的素质，而且当你使用一个组件的时候，一步一步的跟着代码走，慢慢的分析了整个的组件设计方式，最后站在开发这个组件的角度，去看他的设计思想和一些模式的时候，对自己本身也是一个很大的提高，所以我真的建议有兴趣的可以跟着自己的思路一步一步的看下源码。好了废话不多说。

4. 源码分析

ViewModelProviders

在使用VM的时候一般都通过val vm = ViewModelProviders.of(this).get(DemoViewModel::class.java)去创建，具体来看一下of方法：

@NonNull
    @MainThread
    public static ViewModelProvider of(@NonNull Fragment fragment, @Nullable Factory factory) {
        Application application = checkApplication(checkActivity(fragment));
        if (factory == null) {
            factory = ViewModelProvider.AndroidViewModelFactory.getInstance(application);
        }
        return new ViewModelProvider(fragment.getViewModelStore(), factory);
    }

of方法通过ViewModelProvider中的一个单例AndroidViewModelFactory创建了Factory帮助我们去创建VM，并且返回了一个ViewModelProvider。大家注意一下第一个参数fragment.getViewModelStore()，这个ViewModelStore是什么呢？接着看

public class ViewModelStore {

    private final HashMap<String, ViewModel> mMap = new HashMap<>();
		//存储VM
    final void put(String key, ViewModel viewModel) {
        ViewModel oldViewModel = mMap.put(key, viewModel);
        if (oldViewModel != null) {
            oldViewModel.onCleared();
        }
    }

    final ViewModel get(String key) {
        return mMap.get(key);
    }

    Set<String> keys() {
        return new HashSet<>(mMap.keySet());
    }

    /**
     *  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();
    }
}

其实就是一个存储VM的容器，里面通过HashMap进行存和取。

下面我们看创建VM时的get()方法

@NonNull
    @MainThread
    public <T extends ViewModel> T get(@NonNull String key, @NonNull Class<T> modelClass) {
    		//在ViewModelStore中拿到VM实例
        ViewModel viewModel = mViewModelStore.get(key);

        if (modelClass.isInstance(viewModel)) {
            //noinspection unchecked
            return (T) viewModel;
        } else {
            //noinspection StatementWithEmptyBody
            if (viewModel != null) {
                // TODO: log a warning.
            }
        }
        //工厂创建VM，并保存VM
        if (mFactory instanceof KeyedFactory) {
            viewModel = ((KeyedFactory) (mFactory)).create(key, modelClass);
        } else {
            viewModel = (mFactory).create(modelClass);
        }
        mViewModelStore.put(key, viewModel);
        //noinspection unchecked
        return (T) viewModel;
    }

get()方法就是获取VM实例的过程，如果VM不存在，则通过工厂去创建实例。具体工厂创建实现：

public static class AndroidViewModelFactory extends ViewModelProvider.NewInstanceFactory {

        private static AndroidViewModelFactory sInstance;

        /**
         * Retrieve a singleton instance of AndroidViewModelFactory.
         *
         * @param application an application to pass in {@link AndroidViewModel}
         * @return A valid {@link AndroidViewModelFactory}
         */
        @NonNull
        public static AndroidViewModelFactory getInstance(@NonNull Application application) {
            if (sInstance == null) {
                sInstance = new AndroidViewModelFactory(application);
            }
            return sInstance;
        }

        private Application mApplication;

        /**
         * Creates a {@code AndroidViewModelFactory}
         *
         * @param application an application to pass in {@link AndroidViewModel}
         */
        public AndroidViewModelFactory(@NonNull Application application) {
            mApplication = application;
        }

        @NonNull
        @Override
        public <T extends ViewModel> T create(@NonNull Class<T> modelClass) {
            if (AndroidViewModel.class.isAssignableFrom(modelClass)) {
                //noinspection TryWithIdenticalCatches
                try {
                    return modelClass.getConstructor(Application.class).newInstance(mApplication);
                } 
              .......
            }
            return super.create(modelClass);
        }
    }

这个工厂就很简单了，就是通过反射来创建VM实例。

到这里VM的创建过程就差不多了，而我们发现他并没有和生命周期有什么相关的东西，或者说VM是怎样保证的的数据不被销毁的呢？看了网上的一些文章，很多都说在of方法的时候传入了Fragment,并且通过HolderFragment持有ViewModelStore对象等等……比如：

但是在我这里发现跟他们的都不一样，我搜了一下ViewModelStores，发现它已经‘退役’了。

并且它的注释也告诉了我们它的继承者：

也就是我们在of()方法中的:

也就是说谷歌已经将ViewModelStore集成到了Fragment中，一起去看一下吧：

Fragment.java

 		@NonNull
    @Override
    public ViewModelStore getViewModelStore() {
        if (mFragmentManager == null) {
            throw new IllegalStateException("Can't access ViewModels from detached fragment");
        }
        return mFragmentManager.getViewModelStore(this);
    }
FragmentManagerImpl.java

		private FragmentManagerViewModel mNonConfig;
		......
    @NonNull
    ViewModelStore getViewModelStore(@NonNull Fragment f) {
        return mNonConfig.getViewModelStore(f);
    }

我们可以看到代码到了我们熟悉的FragmentManagerImpl，它的里面持有了一个FragmentManagerViewModel实例，进去看看这个是个什么东西：

FragmentManagerViewModel.java

class FragmentManagerViewModel extends ViewModel {

    //创建FragmentVM的工厂
    private static final ViewModelProvider.Factory FACTORY = new ViewModelProvider.Factory() {
        @NonNull
        @Override
        @SuppressWarnings("unchecked")
        public <T extends ViewModel> T create(@NonNull Class<T> modelClass) {
            FragmentManagerViewModel viewModel = new FragmentManagerViewModel(true);
            return (T) viewModel;
        }
    };

    //从viewModelProvider中获取FragmentVM实例
    @NonNull
    static FragmentManagerViewModel getInstance(ViewModelStore viewModelStore) {
        ViewModelProvider viewModelProvider = new ViewModelProvider(viewModelStore,
                FACTORY);
        return viewModelProvider.get(FragmentManagerViewModel.class);
    }
    //非中断的Fragment集合
    private final HashSet<Fragment> mRetainedFragments = new HashSet<>();
    private final HashMap<String, FragmentManagerViewModel> mChildNonConfigs = new HashMap<>();
    private final HashMap<String, ViewModelStore> mViewModelStores = new HashMap<>();

    private final boolean mStateAutomaticallySaved;
    // Only used when mStateAutomaticallySaved is true
    private boolean mHasBeenCleared = false;
    // Only used when mStateAutomaticallySaved is false
    private boolean mHasSavedSnapshot = false;

 
    FragmentManagerViewModel(boolean stateAutomaticallySaved) {
        mStateAutomaticallySaved = stateAutomaticallySaved;
    }

    
    //添加非中断Fragment
    boolean addRetainedFragment(@NonNull Fragment fragment) {
        return mRetainedFragments.add(fragment);
    }

    @NonNull
    Collection<Fragment> getRetainedFragments() {
        return mRetainedFragments;
    }
    //是否改销毁
    boolean shouldDestroy(@NonNull Fragment fragment) {
        if (!mRetainedFragments.contains(fragment)) {
            // Always destroy non-retained Fragments
            return true;
        }
        if (mStateAutomaticallySaved) {
            // If we automatically save our state, then only
            // destroy a retained Fragment when we've been cleared
            return mHasBeenCleared;
        } else {
            // Else, only destroy retained Fragments if they've
            // been reaped before the state has been saved
            return !mHasSavedSnapshot;
        }
    }

    boolean removeRetainedFragment(@NonNull Fragment fragment) {
        return mRetainedFragments.remove(fragment);
    }

		//获取VMStore
    @NonNull
    ViewModelStore getViewModelStore(@NonNull Fragment f) {
        ViewModelStore viewModelStore = mViewModelStores.get(f.mWho);
        if (viewModelStore == null) {
            viewModelStore = new ViewModelStore();
            mViewModelStores.put(f.mWho, viewModelStore);
        }
        return viewModelStore;
    }
    //销毁并清空VM
    void clearNonConfigState(@NonNull Fragment f) {
        if (FragmentManagerImpl.DEBUG) {
            Log.d(FragmentManagerImpl.TAG, "Clearing non-config state for " + f);
        }
        // Clear and remove the Fragment's child non config state
        FragmentManagerViewModel childNonConfig = mChildNonConfigs.get(f.mWho);
        if (childNonConfig != null) {
            childNonConfig.onCleared();
            mChildNonConfigs.remove(f.mWho);
        }
        // Clear and remove the Fragment's ViewModelStore
        ViewModelStore viewModelStore = mViewModelStores.get(f.mWho);
        if (viewModelStore != null) {
            viewModelStore.clear();
            mViewModelStores.remove(f.mWho);
        }
    }

看代码我们发现它继承了VM，并且里面保存了VMStore,也就是说保存了VM，同时清空的操作也在这里面：clearNonConfigState()

ViewModelStore.java

/**
     *  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();
    }

那么到底是什么时候来清空VM的呢？也就是说clear()是什么时候调用的呢？查看源码我们发现有两处：

也就是上面提到的FragmentViewModel.java里面的clearNonConfigState()方法，而这个方法只在一个地方被调用了：

if (DEBUG) Log.v(TAG, "movefrom CREATED: " + f);
boolean beingRemoved = f.mRemoving && !f.isInBackStack();
if (beingRemoved || mNonConfig.shouldDestroy(f)) {
    boolean shouldClear;
    if (mHost instanceof ViewModelStoreOwner) {
        shouldClear = mNonConfig.isCleared();
    } else if (mHost.getContext() instanceof Activity) {
        Activity activity = (Activity) mHost.getContext();
        shouldClear = !activity.isChangingConfigurations();
    } else {
        shouldClear = true;
    }
  	//Fragment正在被移除或者应该清空的状态下
    if (beingRemoved || shouldClear) {
        mNonConfig.clearNonConfigState(f);
    }
    f.performDestroy();
    dispatchOnFragmentDestroyed(f, false);
} else {
    f.mState = Fragment.INITIALIZING;
}

这个方法是在FragmentManagerImpl.java中的moveToState方法里面的，这个方法是跟随着Fragment的生命周期的，当这个方法被调用时，判断两个状态beingRemoved和shoudClear然后调用clear()方法。关于moveToState()可以查看这篇文章：Fragment之底层关键操作函数moveToState

第二个调用的地方就是ComponentActivity.java

getLifecycle().addObserver(new GenericLifecycleObserver() {
    @Override
    public void onStateChanged(@NonNull LifecycleOwner source,
            @NonNull Lifecycle.Event event) {
            if (event == Lifecycle.Event.ON_DESTROY) {
                if (!isChangingConfigurations()) {//不是改变状态配置的
                    getViewModelStore().clear();
                }
            }
        }
});

关于ComponentActivity，如果有看过之前我分析过的关于Lifecycles的应该对它有所了解3. Jetpack源码解析—用Lifecycles管理生命周期。FragmentActivity继承了它，而ComponentActivity里面通过Lifecycles观察生命周期，当接受到ON_DESTROY的事件时，清空VM。

5. 总结

分析完源码之后，我们来总结一下整个流程：

通过ViewModelProviders.of(this).get(DemoViewModel::class.java)创建VM
of()方法中传入 fragment.getViewModelStore()并且返回VMProviders
查看FragmentManagerImpl中getViewModelStore()，持有FragmentManagerViewModel对象
在FragmentManagerViewModel中清空VM操作 clearNonConfigState()，同时在ViewModelStore中clear()了ViewModel的value值
最后我们发现只有在ComponentActivity中观察到接收到ON_DESTROY的事件时同时并不是由于configuration发生变化时才会执行clear()操作；另外一处是在moveToState()方法中，满足beingRemoved和shouldClear状态也会清空VM

好了整个流程就是这样了，并没有特别深入的去分析，但是基本的原理我们已经清楚了，Demo中也只是简单的使用了VM。

JetPack源码分析系列文章到这里应该就结束了，后面还有Paging、WorkManager、Room，以及Camera2和ViewPager2，这些目前暂时不分析了，但是也会出基本的使用和简单的分析，一共5篇源码分析，文章中的Demo我写了一个APP—JetpackNote，里面有基本的Demo例子，和文章的分析；一直也没有提到是因为功能还不完善，我会尽快完善它的，也希望有什么意见的小伙伴可以和我沟通交流。

Github源码地址在：

JetpackNote

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2019年07月16日，如有侵权请联系 cloudcommunity@tencent.com 删除

android

本文分享自作者个人站点/博客前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与腾讯云自媒体同步曝光计划，欢迎热爱写作的你一起参与！

android

登录后参与评论

0 条评论

热度