ViewModel 是 Google 在 Android Jetpack 项目中提供的关键组件,可以方便开发者建立UI模型,保存UI状态变量,优雅的编写基于MVVM架构的代码。
Dagger2 是 Google 提供的依赖注入框架,使用依赖注入框架可以优化项目架构,写出更整洁的代码。
但是当这两者遇到一起时,会对使用者产生些小小的麻烦。
问题描述
ViewModel 在 Activity 中初始化,依赖一个Provider方法 ViewModelProviders:
private ViewModel viewModel;
@Override
protected void onCreate(Bundle savedInstanceState) {
super.onCreate(savedInstanceState);
setContentView(R.layout.activity_main);
/**
* 使用 ViewModelProviders 方法获取viewmodel
*/
viewModel = ViewModelProviders.of(this).get(MainViewModel.class);
}
这样,ViewModelProviders 要求在ViewModel中必须提供一个无参构造函数,否则运行时会报错:
Caused by: java.lang.RuntimeException: Cannot create an instance of class MainViewModel
Caused by: java.lang.InstantiationException: java.lang.Class has no zero argument constructor
于是,要想为 ViewModel 类注入依赖,使用 @inject 标注构造函数的方式变得不可取。
@Inject
public MainViewModel(AppExecutors executors, QueensModel queensModel, SharedPreferences sharedPreferences) {
this.executors = executors;
this.sharedPreferences = sharedPreferences;
setQueensModel(queensModel);
}
这时需要怎么办?
解决方案
首先想到的不太合理的方案为:
1 放弃构造函数注入,使用成员变量注入方案
- ViewModel 中需要注入的成员变量用@Inject标注
@Inject
private AppExecutors executors;
@Inject
private SharedPreferences sharedPreferences;
@Inject
private QueensModel queensModel;
2 AppComponent 添加inject方法
public interface AppComponent extends AndroidInjector {
void inject(MainViewModel mainViewModel);
……
}
3 ViewModel的无参构造函数中,获取DaggerAppComponent,调用inject方法
public MainViewModel() {
……
mContext.getApplication().getAppComponent().inject(this);
}
这个方案的问题在于:
MainViewModel 的构造函数引入了对DaggerAppComponent的依赖,被依赖注入的对象需要知道依赖注入的方法,破坏了依赖注入的原则.
参考 Google 官方实现 android-architecture-components, 为了解决上述问题,采用了如下方式为 ViewModel 提供依赖注入。
2 Google 提供的 ViewModel 依赖注入方案
- ViewModel 依然采用构造函数注入的方式,@Inject 注解构造函数而不是成员变量。
@Inject
public QueensViewModel(AppExecutors executors, QueensModel queensModel, SharedPreferences sharedPreferences) {
this.executors = executors;
this.sharedPreferences = sharedPreferences;
setQueensModel(queensModel);
}
- 提供一个工厂类,根据 class name 生产 ViewModel 对象。这个类构造函数需要注入一个Map,这个 Map 就是 ViewModel 的 class name 和 对应的 provider。
@Singleton
public class QueensViewModelFactory implements ViewModelProvider.Factory{
private final Map, Provider> creators;
@Inject
public QueensViewModelFactory(Map, Provider> creators) {
this.creators = creators;
}
@NonNull
@Override
public T create(@NonNull Class modelClass) {
Provider extends ViewModel> creator = creators.get(modelClass);
if (creator == null) {
for (Map.Entry, Provider> entry : creators.entrySet()) {
if (modelClass.isAssignableFrom(entry.getKey())) {
creator = entry.getValue();
break;
}
}
}
if (creator == null) {
throw new IllegalArgumentException("unknown model class " + modelClass);
}
try {
return (T) creator.get();
} catch (Exception e) {
throw new RuntimeException(e);
}
}
}
- 通过Dagger2 的 MultiBindings 方法提供方法2 中需要的Map。并通过@Binds 绑定 ViewModelProvider.Factory 接口和 2中提供的实现类。关于MultiBindings的用法可以参考multibindings
@Module
abstract public class ViewModelModule {
@Binds
@IntoMap
@ViewModelKey(QueensViewModel.class)
abstract ViewModel bindQueensViewModel(QueensViewModel queensViewmodel);
@Binds
abstract ViewModelProvider.Factory bindViewModelFactory(QueensViewModelFactory factory);
}
- ViewModelModule 加入AppModule
@Module(includes = ViewModelModule.class)
public class AppModule {
}
这样 ViewModel 就已经添加进入Dagger2 的依赖图中了,Activity中获取viewModel的方法为:
@Inject
ViewModelProvider.Factory viewModelFactory;
QueensViewModel viewModel;
@Override
protected void onCreate(Bundle savedInstanceState) {
super.onCreate(savedInstanceState);
setContentView(R.layout.activity_main);
/**
* 使用 ViewModelProviders 方法获取viewmodel
*/
viewModel = ViewModelProviders.of(this,viewModelFactory).get(QueensViewModel .class);
}
对比方案一,Google 的方案对 ViewModel 的修改极小,后续添加 ViewModel 只需要 在 ViewModelModule 中,把新增的 ViewModel 绑定进 Map 即可。
参考
- 方案二完整代码可以参考我的GitHub项目:https://github.com/lixiangthinker/NQueens
- 感谢 Google 爸爸提供的参考设计 android-architecture-components