Android源码解析-ViewModel

前言

最近一个月主要实现了IM功能，其中UI框架使用了ViewModel和LiveData的存储和通知机制，实现后代码简洁易于维护。
感慨于Android arch components控件强大同时，需要顺带分析一波其中源码实现。今天先来分析一下ViewModel的源码实现。

ViewModel简介

ViewModel是通过关联生命周期的方式来存储和管理跟UI相关的数据。即使configuration发生变化，在ViewModel中存放的数据是不会被销毁的。
没使用ViewModel的时候，如果系统Configuration发生变化，我们的Activity会被销毁重建，导致Activity中的UI数据丢失。为了规避这个问题，我们只能在onSaveInstanceState()将UI数据进行保存，在onCreate方法中判断savedInstanceState中是否有我们存储的数据。
有了ViewModel之后，我们只需要将数据存储到ViewModel即可，ViewModel中的数据不会随着Activity的销毁重建而消失。同时，如果不同的Fragment使用相同的Activity对象来获取ViewModel，可以轻易的实现数据共享和通信。

使用ViewModel的例子:

// 自定义一个ViewModel，存储一个字符串
public class TestViewModel extends ViewModel {
    public String content;

    @Override
    protected void onCleared() {
        // 数据清理工作
        content = null;
    }
}

// 在Activity中获取并使用ViewModel
TestViewModel viewModel = ViewModelProviders.of(activity).get(TestViewModel.class);
Log.d("tag", viewModel.content);

三个问题

  1. ViewModel是如何创建出来的？
  2. 为什么不同的Fragment使用相同的Activity对象来获取ViewModel，可以轻易的实现ViewModel共享?
  3. ViewModel为什么在Activity销毁重建时不会被销毁回收？

我个人习惯，看源码时候总是要带着问题去分析理解，这样才会有所收获。

源码分析

ViewModel是如何创建出来的？

通过示例代码，去掉链式调用后，我们可以看到ViewModel是通过如下两步创建出来的：

// 1\. 创建ViewModelProvider
ViewModelProvider viewModelProvider = ViewModelProviders.of(activity);

// 2\. 通过反射获取ViewModel
TestViewModel viewModel = viewModelProvider.get(TestViewModel.class);

创建ViewModelProvider

先看下第一步的源码实现：

public static ViewModelProvider of(@NonNull FragmentActivity activity, @Nullable Factory factory) {
    if (factory == null) {
        // 如果传入的对象创建工厂类为null，则使用默认的AndroidViewModelFactory来创建对象
        factory = AndroidViewModelFactory.getInstance(application);
    }

    // 创建一个ViewModelProvider
    return new ViewModelProvider(ViewModelStores.of(activity), (Factory)factory);
}

源码中，我们发现创建一个ViewModelProvider需要传入两个参数：ViewModelStore和Factory。我们先看下Factory的实现。

Factory

Factory顾名思义，定义了创建ViewModel的行为接口。里面只有一个create方法，用于子类自行决定如何实现一个ViewModel对象的创建。

public interface Factory {
    <T extends ViewModel> T create(@NonNull Class<T> modelClass);
}

同时，ViewModelProvider源码内部也提供了两个默认Factory实现：NewInstanceFactory和AndroidViewModelFactory。

// 直接反射Class对象的无参构造函数来创建ViewModel
public static class NewInstanceFactory implements Factory {
    @Override
    public <T extends ViewModel> T create(@NonNull Class<T> modelClass) {
        try {
            return modelClass.newInstance();
        } catch (InstantiationException e) {
            throw new RuntimeException("Cannot create an instance of " + modelClass, e);
        } catch (IllegalAccessException e) {
            throw new RuntimeException("Cannot create an instance of " + modelClass, e);
        }
    }
}

// 1\. 如果对象继承自AndroidViewModel，发射调用带Application参数的构造函数创建ViewModel对象；
// 2\. 如果对象不继承自AndroidViewModel，则直接调用父类，即调用Class的无参构造函数创建ViewModel对象。
public static class AndroidViewModelFactory extends ViewModelProvider.NewInstanceFactory {
    private static AndroidViewModelFactory sInstance;
    @NonNull
    public static AndroidViewModelFactory getInstance(@NonNull Application application) {
        if (sInstance == null) {
            sInstance = new AndroidViewModelFactory(application);
        }
        return sInstance;
    }

    private Application mApplication;
    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);
            } catch (NoSuchMethodException e) {
                throw new RuntimeException("Cannot create an instance of " + modelClass, e);
            } catch (IllegalAccessException e) {
                throw new RuntimeException("Cannot create an instance of " + modelClass, e);
            } catch (InstantiationException e) {
                throw new RuntimeException("Cannot create an instance of " + modelClass, e);
            } catch (InvocationTargetException e) {
                throw new RuntimeException("Cannot create an instance of " + modelClass, e);
            }
        }
        return super.create(modelClass);
    }
}

启示：我们自定义的ViewModel对象如果构造函数中需要其他各种参数，我们只需要创建一个自定义的Factory类，然后调用该class的有参构造函数进行创建即可。

ViewModelStore

ViewModelStore就是个HashMap，通过key来获取ViewModel对象。

public class ViewModelStore {

    private final HashMap<String, ViewModel> mMap = new HashMap<>();

    final void put(String key, ViewModel viewModel) {
        ViewModel oldViewModel = mMap.get(key);
        if (oldViewModel != null) {
            oldViewModel.onCleared();
        }
        mMap.put(key, viewModel);
    }

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

ViewModelProvider

了解了Factory实现和ViewModelStore实现后，我们来看一下ViewModelProvider的get方法是如何创建ViewModel对象的。

public class ViewModelProvider {
    private final Factory mFactory;
    private final ViewModelStore mViewModelStore;

    public ViewModelProvider(@NonNull ViewModelStore store, @NonNull Factory factory) {
        mFactory = factory;
        this.mViewModelStore = store;
    }

    public <T extends ViewModel> T get(@NonNull Class<T> modelClass) {
        String canonicalName = modelClass.getCanonicalName();
        return get(DEFAULT_KEY + ":" + canonicalName, modelClass);
    }    

    @NonNull
    @MainThread
    public <T extends ViewModel> T get(@NonNull String key, @NonNull Class<T> modelClass) {
        ViewModel viewModel = mViewModelStore.get(key);

        if (modelClass.isInstance(viewModel)) {
            return (T) viewModel;
        }
        viewModel = mFactory.create(modelClass);
        mViewModelStore.put(key, viewModel);
        //noinspection unchecked
        return (T) viewModel;
    }
}

ViewModleProvider的get方法实现也比较简单，总结一下就是：

  1. 使用ViewModel Class的canonicalName作为ViewModel在ViewModelStore中的唯一标识。
  2. 通过唯一标识，先查询一下ViewModelStore中是否有该ViewModel对象，如果有则直接返回。
  3. 如果ViewModelStore中没有该ViewModel对象，则通过Factory工厂类反射创建出ViewModel对象，存入ViewModelStore中，并返回给调用者。

小结

至此，第一个问题ViewModel是如何创建出来的已经分析完毕。接下来，我们看第二个问题。

为什么不同的Fragment使用相同的Activity对象来获取ViewModel，可以轻易的实现ViewModel共享？

讲道理，如果同学们仔细看了ViewModel的创建流程，这个问题自然迎刃而解。
因为不同的Fragment使用相同的Activity对象来获取ViewModel，在创建ViewModel之前都会先从Activity提供的ViewModelStore中先查询一遍是否已经存在该ViewModel对象。所以我们只需要先在Activity中同样调用一遍ViewModel的获取代码，即可让ViewModel存在于ViewModelStore中，从而不同的Fragment可以共享一份ViewModel了。

ViewModel为什么在Activity销毁重建时不会被销毁回收？

在看这个问题之前，我们回到ViewModelProviders.of()源码中：

public static ViewModelProvider of(@NonNull FragmentActivity activity, @Nullable Factory factory) {
    if (factory == null) {
        // 如果传入的对象创建工厂类为null，则使用默认的AndroidViewModelFactory来创建对象
        factory = AndroidViewModelFactory.getInstance(application);
    }

    // 创建一个ViewModelProvider
    return new ViewModelProvider(ViewModelStores.of(activity), (Factory)factory);
}

ViewModelProvider获取ViewModelStore是通过ViewModelStores.of(activity)实现的。我们先看一下这个ViewModelStores做了什么操作。

ViewModelStores

看名字又像是一个封装好的工厂类。看下源码：

public static ViewModelStore of(@NonNull FragmentActivity activity) {
    return activity instanceof ViewModelStoreOwner ? ((ViewModelStoreOwner)activity).getViewModelStore() : HolderFragment.holderFragmentFor(activity).getViewModelStore();
}

通过源码可以发现，如果Activity是ViewModelStoreOwner的实现类，则直接通过activity获取ViewModelStore，如果不是，则通过HolderFragment.holderFragmentFor(activity).getViewModelStore()来获取。

通过查看FragmentActivity源码，发现其已经是实现了ViewModelStoreOwner接口。

public class FragmentActivity extends BaseFragmentActivityApi16 implements
        ViewModelStoreOwner,
        ActivityCompat.OnRequestPermissionsResultCallback,
        ActivityCompat.RequestPermissionsRequestCodeValidator {
}

所以这里我们已经不需要理会HolderFragment了，很多分析ViewModel源码的文章都会花很多篇幅分析HolderFragment，我实在是搞不懂为什么要这种操作。（初步怀疑是互相抄袭。。.）

虽然FragmentActivity实现了ViewModelStoreOwner接口，能够提供ViewModelStore，但是ViewModelStore是如何跟Activity生命周期关联起来的呢?

FragmentActivity中ViewModelStore的生命周期处理

搜索了一下FragmentActivity中关于ViewModelStore的调用，发现这里的实现应该跟生命周期处理有关。


image.png

google搜索了一下Activity的onRetainNonConfigurationInstance的作用：大部分同学知道Activity因为configuration变化销毁和重建时会调用onSaveInstanceState和onRestoreInstanceState。与此同时，Activity其实还会回调onRetainNonConfigurationInstance和getLastNonConfigurationInstance方法。

onRetainNonConfigurationInstance和onSaveInstanceState作用相同，用来保存UI相关变量，当Activity意外销毁时，Activity的ViewModelStore对象就是在这里进行了保存。

那什么时机进行的恢复呢？
当Activity的onCreate调用时，会调用getLastNonConfigurationInstance，获取之前保存的ViewModelStore，如果ViewModelStore不为空，就进行赋值。这里进行了ViewModelStore的恢复。

image.png

小结

这里我们又学到了Activity的两个跟生命周期相关的函数调用：onRetainNonConfigurationInstance和getLastNonConfigurationInstance。

  1. Activity实现了ViewModelStoreOwner接口，创建了ViewModelStore对象。
  2. 当Activity意外销毁时，onRetainNonConfigurationInstance函数被回调，在此函数中对ViewModelStore对象进行了保存。
  3. 当Activity重建时，onCreate方法中会先获取getLastNonConfigurationInstance，如果其中的ViewModelStore对象不为空，就直接引用，不再重新创建ViewModelStore对象了。

总结

三个问题分析完毕，相信大家已经对ViewModel的实现原理比较熟悉了。建议大家以后学习源码时，也带着问题去分析思考，事半功倍。

©著作权归作者所有,转载或内容合作请联系作者
  • 人面猴
    序言：七十年代末，一起剥皮案震惊了整个滨河市，随后出现的几起案子，更是在滨河造成了极大的恐慌，老刑警刘岩，带你破解...
    沈念sama阅读 203,937评论 6赞 478
  • 死咒
    序言：滨河连续发生了三起死亡事件，死亡现场离奇诡异，居然都是意外死亡，警方通过查阅死者的电脑和手机，发现死者居然都...
    沈念sama阅读 85,503评论 2赞 381
  • 救了他两次的神仙让他今天三更去死
    文/潘晓璐 我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
    开封第一讲书人阅读 150,712评论 0赞 337
  • 道士缉凶录：失踪的卖姜人
    文/不坏的土叔 我叫张陵，是天一观的道长。 经常有香客问我，道长，这世上最难降的妖魔是什么？ 我笑而不...
    开封第一讲书人阅读 54,668评论 1赞 276
  • 港岛之恋（遗憾婚礼）
    正文 为了忘掉前任，我火速办了婚礼，结果婚礼上，老公的妹妹穿的比我还像新娘。我一直安慰自己，他们只是感情好，可当我...
    茶点故事阅读 63,677评论 5赞 366
  • 恶毒庶女顶嫁案：这布局不是一般人想出来的
    文/花漫 我一把揭开白布。 她就那样静静地躺着，像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上，一...
    开封第一讲书人阅读 48,601评论 1赞 281
  • 城市分裂传说
    那天，我揣着相机与录音，去河边找鬼。 笑死，一个胖子当着我的面吹牛，可吹牛的内容都是我干的。 我是一名探鬼主播，决...
    沈念sama阅读 37,975评论 3赞 396
  • 双鸳鸯连环套：你想象不到人心有多黑
    文/苍兰香墨 我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
    开封第一讲书人阅读 36,637评论 0赞 258
  • 万荣杀人案实录
    序言：老挝万荣一对情侣失踪，失踪者是张志新（化名）和其女友刘颖，没想到半个月后，有当地人在树林里发现了一具尸体，经...
    沈念sama阅读 40,881评论 1赞 298
  • 护林员之死
    正文 独居荒郊野岭守林人离奇死亡，尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,621评论 2赞 321
  • 白月光启示录
    正文 我和宋清朗相恋三年，在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,710评论 1赞 329
  • 活死人
    序言：一个原本活蹦乱跳的男人离奇死亡，死状恐怖，灵堂内的尸体忽然破棺而出，到底是诈尸还是另有隐情，我是刑警宁泽，带...
    沈念sama阅读 33,387评论 4赞 319
  • 日本核电站爆炸内幕
    正文 年R本政府宣布，位于F岛的核电站，受9级特大地震影响，放射性物质发生泄漏。R本人自食恶果不足惜，却给世界环境...
    茶点故事阅读 38,971评论 3赞 307
  • 男人毒药：我在死后第九天来索命
    文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹，春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,947评论 0赞 19
  • 一桩弑父案，背后竟有这般阴谋
    文/苍兰香墨 我抬头看了看天上的太阳。三九已至，却和暖如春，着一层夹袄步出监牢的瞬间，已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,189评论 1赞 260
  • 情欲美人皮
    我被黑心中介骗来泰国打工， 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留，地道东北人。 一个月前我还...
    沈念sama阅读 44,805评论 2赞 349
  • 代替公主和亲
    正文 我出身青楼，却偏偏与公主长得像，于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子，可洞房花烛夜当晚...
    茶点故事阅读 42,449评论 2赞 342

推荐阅读更多精彩内容