Comment rendre cette ViewModelFactory plus flexible et accepter différents types de classes ViewModel ?

Question

J'ai copié un exemple de MVVM avec les composants d'architecture Android, Retrofit, Dagger et la liaison de données. J'utilise ce code comme point de départ pour mon application afin de commencer à utiliser de meilleures architectures dans le développement d'applications Android. Cependant, prenez ces codes :

interface ViewModelInjector {
    /**
     * Injects required dependencies into the specified PostListViewModel.
     * @param postListViewModel PostListViewModel in which to inject the dependencies
     */
    fun inject(postListViewModel: PostListViewModel)

    @Component.Builder
    interface Builder {
        fun build(): ViewModelInjector
        fun networkModule(networkModule: NetworkModule): Builder
    }
}

Et

class ViewModelFactory(private val activity: AppCompatActivity) : ViewModelProvider.Factory {
    override fun <T : ViewModel?> create(modelClass: Class<T>): T {
        if (modelClass.isAssignableFrom(PostListViewModel::class.java)) {
            val db = Room.databaseBuilder(
                activity.applicationContext,
                AppDatabase::class.java,
                "posts"
            ).build()

            @Suppress("UNCHECKED_CAST")
            return PostListViewModel(db.postDao()) as T
        }

        throw IllegalArgumentException("Unknown ViewModel class")
    }
}

Et

abstract class BaseViewModel : ViewModel() {
    private val injector: ViewModelInjector = DaggerViewModelInjector
        .builder()
        .networkModule(NetworkModule)
        .build()

    init {
        inject()
    }

    private fun inject() {
        when (this) {
            is PostListViewModel -> injector.inject(this)
        }
    }
}

Le principal problème est qu'il est bloqué avec PostListViewModel . J'aimerais le faire d'une manière dynamique, en acceptant tout type de [Name]ViewModel classe. J'ai essayé quelques méthodes en utilisant Class<T> mais je n'ai plus le code. J'ai également essayé de faire une recherche, mais je n'ai pas trouvé de résultat satisfaisant. Peut-être que je n'ai pas cherché les bons termes. Je vous remercie de votre aide.

Answer 1

J'ai récemment rencontré le même problème et j'ai trouvé une solution. generic ViewModelFactory c'est java mais

@Singleton
public class ViewModelFactory implements ViewModelProvider.Factory {
private final Map<Class<? extends ViewModel>, Provider<ViewModel>> creators;

@Inject
public ViewModelFactory(Map<Class<? extends ViewModel>, Provider<ViewModel>> creators) {
    this.creators = creators;
}

@NonNull
@SuppressWarnings("unchecked")
@Override
public <T extends ViewModel> T create(@NonNull Class<T> modelClass) {
    Provider<? extends ViewModel> creator = creators.get(modelClass);
    if (creator == null) {
        for (Map.Entry<Class<? extends ViewModel>, Provider<ViewModel>> 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);
    }
}
}

Je mettrais bien le lien de référence, mais je ne l'ai pas encore trouvé

Edita:

@Singleton
@Component(modules={ActivityModule.class, FragmentModule.class, AppModule.class})
public interface AppComponents {

@Component.Builder
interface Builder {
    @BindsInstance
    Builder application(Application application);
    AppComponents build();
}

void inject(Weather weatherApp);
}

Maintenant, ce qui est dans votre intérêt dans la classe ci-dessus devrait être AppModule.class

@Module(includes = {UserModelModule.class /* other model modules */})
public class AppModule {

// --- DATABASE INJECTION ---

@Provides
@Singleton
YourDatabase provideDatabase(Application application) {
    return Room.databaseBuilder(application,
            YourDatabase.class, "YourDatabase.db")
            //.allowMainThreadQueries() // do NOT DO THIS IN REAL APPLICATIONs
            .fallbackToDestructiveMigration()
            .build();
}

@Provides
@Singleton
UserDao provideUserDao(YourDatabase database) { return database.userDao(); }
.
.
// other Daos that you will have (the above is an example from [https://developer.android.com/jetpack/docs/guide]

Enfin, je pense que vous voudrez voir à quoi ressemble la classe ModelModule

@Module
public abstract class UserModelModule {

@Binds
@IntoMap
@ViewModelKey(UserViewModel.class)
abstract ViewModel bindUserProfileViewModel(UserViewModel repoViewModel);

@Binds
abstract ViewModelProvider.Factory bindViewModelFactory(ViewModelFactory factory);
}