jetpack-ViewModel 源码解析

//通过key 获取value
T getTag(String key) {
if (mBagOfTags == null) {
return null;
}
synchronized (mBagOfTags) {
return (T) mBagOfTags.get(key);
}
}
// 关闭操作
private static void closeWithRuntimeException(Object obj) {
if (obj instanceof Closeable) {
try {
((Closeable) obj).close();
} catch (IOException e) {
throw new RuntimeException(e);
}
}
}
}

---

二、ViewModelStore

ViewModelStore是用来存储 各种ViewModel的，内部是一个HashMap 来保存数据

public class ViewModelStore {
//存放 ViewModel
private final HashMap<String, ViewModel> mMap = new HashMap<>();
//保存ViewModel 如果已经存在，则 触发一次清空逻辑
final void put(String key, ViewModel viewModel) {
ViewModel oldViewModel = mMap.put(key, viewModel);
if (oldViewModel != null) {
oldViewModel.onCleared();
}
}
//通过key 获取 ViewModel
final ViewModel get(String key) {
return mMap.get(key);
}
//获取所有存储的key
Set keys() {
return new HashSet<>(mMap.keySet());
}
//清除所有ViewModel 并触发 clear 操作
public final void clear() {
for (ViewModel vm : mMap.values()) {
vm.clear();
}
mMap.clear();
}
}

ViewModelStore 的获取方式是怎的呢 ？

---

三、 ViewModelStoreOwner

public interface ViewModelStoreOwner {
/**

• Returns owned {@link ViewModelStore}
• @return a {@code ViewModelStore}
  */
  @NonNull
  ViewModelStore getViewModelStore();
  }

ViewModelStoreOwner 是ViewModelStore 的持有者，是个接口类，那就是谁实现了这个接口，谁就是提供ViewModelStore的生产者。

public class ComponentActivity extends androidx.core.app.ComponentActivity implements
ViewModelStoreOwner,
HasDefaultViewModelProviderFactory {

private ViewModelStore mViewModelStore;
private ViewModelProvider.Factory mDefaultFactory;
//实现 ViewModelStoreOwner 接口，可以获取 ViewModelStore
@NonNull
@Override
public ViewModelStore getViewModelStore() {
if (getApplication() == null) {
throw new IllegalStateException("Your activity is not yet attached to the "

• “Application instance. You can’t request ViewModel before onCreate call.”);
  }
  if (mViewModelStore == null) {
  NonConfigurationInstances nc =
  (NonConfigurationInstances) getLastNonConfigurationInstance();
  if (nc != null) {
  // Restore the ViewModelStore from NonConfigurationInstances
  mViewModelStore = nc.viewModelStore;
  }
  if (mViewModelStore == null) {
  mViewModelStore = new ViewModelStore();
  }
  }
  return mViewModelStore;
  }

//实现 HasDefaultViewModelProviderFactory 接口，对外提供 ViewModelProvider.Factory 生成ViewModel
@NonNull
@Override
public ViewModelProvider.Factory getDefaultViewModelProviderFactory() {
if (getApplication() == null) {
throw new IllegalStateException("Your activity is not yet attached to the "

• “Application instance. You can’t request ViewModel before onCreate call.”);
  }
  if (mDefaultFactory == null) {
  mDefaultFactory = new SavedStateViewModelFactory(
  getApplication(),
  this,
  getIntent() != null ? getIntent().getExtras() : null);
  }
  return mDefaultFactory;
  }

}

ComponentActivity 实现了 ViewStoreOwner 和 HasDefaultViewModelProviderFactory 接口 用来返回 ViewModelStore 和 ViewModelProvider.Factory

​ 我们看到 都判断了 getApplication() 为null 的情况下会抛出异常，getApplication 返回的 mApplication 是在 Activity 的 attach中赋值的 ，也就是在 attach 之前会一直为空，这之前使用会抛出异常。

Activity.Java

@UnsupportedAppUsage
final void attach(Context context, ActivityThread aThread,
Instrumentation instr, IBinder token, int ident,
Application application, Intent intent, ActivityInfo info,
CharSequence title, Activity parent, String id,
NonConfigurationInstances lastNonConfigurationInstances,
Configuration config, String referrer, IVoiceInteractor voiceInteractor,
Window window, ActivityConfigCallback activityConfigCallback, IBinder assistToken) {
/省略其他代码*/
mApplication = application;
/省略其他代码/

}

四、ViewModelProvider.Factory

我们看到 ViewModelProvider中定义了接口 Factory,实现该接口可以返回一个 Factory.

public interface Factory {
@NonNull
T create(@NonNull Class modelClass);
}

ComponentActivity 默认返回的是 SavedStateViewModelFactory，该类实现了 Factory，并禁用了一个参数的 create 接口，实现了两个参数的接口,需要带一个key：

@NonNull
public abstract T create(@NonNull String key,
@NonNull Class modelClass);

@NonNull
@Override
public T create(@NonNull Class modelClass) {
throw new UnsupportedOperationException("create(String, Class<?>) must be called on "

• “implementaions of KeyedFactory”);
  }

五、ViewModelProvider

ViewModelProvider 有两个成员变量： ViewModel 工厂类，和ViewModel 存储的地方

private final Factory mFactory;
private final ViewModelStore mViewModelStore;

ViewModelProvider 有三个构造方法：

//传入 ViewModelStoreOwner owner 实现接口返回 ViewModelStore 和 Factoty,如果不是 HasDefaultViewModelProviderFactory,则 NewInstanceFactory 生成一个（采用单例）
public ViewModelProvider(@NonNull ViewModelStoreOwner owner) {
this(owner.getViewModelStore(), owner instanceof HasDefaultViewModelProviderFactory
? ((HasDefaultViewModelProviderFactory) owner).getDefaultViewModelProviderFactory()
: NewInstanceFactory.getInstance());
}
//传入 ViewModelStoreOwner owner 实现接口返回 ViewModelStore ，Factory 自定义传入
public ViewModelProvider(@NonNull ViewModelStoreOwner owner, @NonNull Factory factory) {
this(owner.getViewModelStore(), factory);
}
//传入 ViewModelStoreOwner ViewModelStore 和 Factory 自定义传入
public ViewModelProvider(@NonNull ViewModelStore store, @NonNull Factory factory) {
mFactory = factory;
mViewModelStore = store;
}

ViewModelProvider 获取ViewModel有两种方法：

//不传入key，默认key 是 androidx.lifecycle.ViewModelProvider.DefaultKey + modelClass.getCanonicalName()
@NonNull
@MainThread
public T get(@NonNull Class modelClass) {
String canonicalName = modelClass.getCanonicalName();
if (canonicalName == null) {
throw new IllegalArgumentException(“Local and anonymous classes can not be ViewModels”);
}
return get(DEFAULT_KEY + “:” + canonicalName, modelClass);
}

//传入key，如果有已经存在 则返回已经存在的否则生成一个 放入 ViewModelStore 中
@SuppressWarnings(“unchecked”)
@NonNull
@MainThread
public T get(@NonNull String key, @NonNull Class modelClass) {
ViewModel viewModel = mViewModelStore.get(key);
//这里判断了 ViewModel 是都已经存在
if (modelClass.isInstance(viewModel)) {
if (mFactory instanceof OnRequeryFactory) {
((OnRequeryFactory) mFactory).onRequery(viewModel);
}
return (T) viewModel;
} else {
//noinspection StatementWithEmptyBody
if (viewModel != null) {
// TODO: log a warning.
}
}
//新生成ViewModel 并放入 ViewModelStore中
if (mFactory instanceof KeyedFactory) {
viewModel = ((KeyedFactory) mFactory).create(key, modelClass);
} else {
viewModel = mFactory.create(modelClass);
}
mViewModelStore.put(key, viewModel);
return (T) viewModel;
}

可以看到 viewModel 是通过 ViewModelStore 按key和 value 存储的，那么就为数据可恢复提供了机制。

六、数据恢复

我们先看下要恢复数据，就要看如何恢复 ViewModelStore

@NonNull
@Override
public ViewModelStore getViewModelStore() {
if (getApplication() == null) {
throw new IllegalStateException("Your activity is not yet attached to the "

• “Application instance. You can’t request ViewModel before onCreate call.”);
  }
  if (mViewModelStore == null) {
  NonConfigurationInstances nc =
  (NonConfigurationInstances) getLastNonConfigurationInstance();
  if (nc != null) {
  // Restore the ViewModelStore from NonConfigurationInstances
  mViewModelStore = nc.viewModelStore;
  }
  if (mViewModelStore == null) {
  mViewModelStore = new ViewModelStore();
  }
  }
  return mViewModelStore;
  }

可以看到 获取ViewModelStore会先从 NonConfigurationInstances 获取

接下来我们看下 NonConfigurationInstances 这个是从哪里得来的？，可以看到 mLastNonConfigurationInstances 是在activity attach 时参数传入和赋值的

最后

自我介绍一下，小编13年上海交大毕业，曾经在小公司待过，也去过华为、OPPO等大厂，18年进入阿里一直到现在。

深知大多数初中级Android工程师，想要提升技能，往往是自己摸索成长，自己不成体系的自学效果低效漫长且无助。

因此我收集整理了一份《2024年Android移动开发全套学习资料》，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友，同时减轻大家的负担。

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，基本涵盖了95%以上Android开发知识点！不论你是刚入门Android开发的新手，还是希望在技术上不断提升的资深开发者，这些资料都将为你打开新的学习之门

如果你觉得这些内容对你有帮助，需要这份全套学习资料的朋友可以戳我获取！！

由于文件比较大，这里只是将部分目录截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频，并且会持续更新！
-1715757231244)]

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，基本涵盖了95%以上Android开发知识点！不论你是刚入门Android开发的新手，还是希望在技术上不断提升的资深开发者，这些资料都将为你打开新的学习之门

如果你觉得这些内容对你有帮助，需要这份全套学习资料的朋友可以戳我获取！！

由于文件比较大，这里只是将部分目录截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频，并且会持续更新！