Activity与Fragment通过ViewModel通讯的原理

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 get(@NonNull String key, @NonNull Class 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通讯的基础。

你可能感兴趣的:(Activity与Fragment通过ViewModel通讯的原理)