Activity中的viewmodel初始化分为两步
mRecordViewModel = ViewModelProviders.of(this).get(RecordViewModel.class);
第一步:构建ViewModelProvider
而ViewModelProvider的构造函数,需要两个参数:
//创建一个ViewModelProvider对象
public static ViewModelProvider of(@NonNull FragmentActivity activity) {
ViewModelProvider.AndroidViewModelFactory factory =
ViewModelProvider.AndroidViewModelFactory.getInstance(
checkApplication(activity));
return new ViewModelProvider(ViewModelStores.of(activity), factory);
}
//创建ViewModelProvider时,需要一个ViewModelStore的参数和factory参数,
//factory参数用的是单例,基本上本app中所有的factory都是一样的。
//下面函数是先创建ViewModelStore对象
public static ViewModelStore of(@NonNull FragmentActivity activity) {
if (activity instanceof ViewModelStoreOwner) {
//我代码中用的是FragmentActivity,这个是实现了ViewModelStoreOwner接口的,所以走了这个分支
//这里会走到FragmentActivty中的getViewModelStore函数
return ((ViewModelStoreOwner) activity).getViewModelStore();
}
return holderFragmentFor(activity).getViewModelStore();
}
//FragmentActivity
//可以看到对于新创建的Activity来说,这里也是new了一个ViewModelStore对象。
public ViewModelStore getViewModelStore() {
...
if (this.mViewModelStore == null) {
this.mViewModelStore = new ViewModelStore();
}
}
return this.mViewModelStore;
}
}
总结:第一步构建ViewModelProvider到此结束,这里总共new了两个东西,一个是ViewModelStore,一个是ViewModelProvider,后者把前者给封装到进了自己对象中,第二步会用到。
第二步 生成ViewModel对象
第一步中使用of函数,得到了ViewModelProvider,然后紧接着调用了这个对象的get函数
public <T extends ViewModel> T get(@NonNull String key, @NonNull Class<T> modelClass) {
ViewModel viewModel = mViewModelStore.get(key);
...
viewModel = mFactory.create(modelClass);
mViewModelStore.put(key, viewModel);
//noinspection unchecked
return (T) viewModel;
}
这个函数结构也很简单,先获取,获取不到就新建,新建完成就加入mViewModelStore中,mViewModelStore本质上就是对Hashmap做了简单的封装。
mFactory.create这个函数也比较简单,就是通过class的方式new了一个对象出来。
第三步 Activity与Fragment通讯
Fragment创建ViewModel对象的方式与Activity基本一致,也是用了activity作为参数传入的,如果Fragment是基于当前Activity显示的,则使用的activity也是同一个,所以后面流程完全一致。
mRecordViewModel = ViewModelProviders.of(getActivity()).get(RecordViewModel.class);
Activity和Fragment创建ViewModel时的区别在于:
分别会创建一个ViewModelProvider对象,这个不同的ViewModelProvider对象中又封装了相同的ViewModelStore对象和factory对象,通过相同的ViewModelStore,又会获取到相同的ViewModel对象,这也是Activity和Fragment通过ViewModel通讯的基础。