Tixeon
Tixeon

Reputation: 930

Create custom params android viewmodel class

This will create MyViewModel object with default or zero parameters.

MyActivity

ViewModelProviders.of(this)
                .get(MyViewModel::class.java)

How can i create if MyViewModel class have custom parameters constructors ?

MyViewModel

class MyViewModel(context: Application,
                        private val repository: MistakesRepository) : AndroidViewModel(context) {


    val showLoading = ObservableBoolean(false)
    val liveItems = MutableLiveData<List<Mistake>>()

    fun loadAllMistakes(forceLoad: Boolean) {

    }

}

After searching for few blog posts, it could be done by having custom factory object. But why there is no documentation about this developer page ? Or I miss out something here ?

Upvotes: 0

Views: 3925

Answers (2)

Guru Karthi R
Guru Karthi R

Reputation: 320

ViewModelProviders.of(this, object : ViewModelProvider.Factory {
            override fun <T : ViewModel?> create(modelClass: Class<T>): T {
                @Suppress("UNCHECKED_CAST")
                return MyViewModel(applicationContext, mistakesRepository) as T
            }
        })[MyViewModel::class.java]

Upvotes: 5

Udit
Udit

Reputation: 1037

I can help you in JAVA. Hopefully you can convert to kotlin better than I can.

class MyViewModelFactory implements ViewModelProvider.Factory {

    private final Context context;
    private final MistakesRepository repository;

    @Inject
    public MyViewModelFactory(Context context, MistakesRepository repository) {
       this.context = context;
       this.repository = repository;
    }

    @Override
    public MyViewModel create(Class modelClass) {
        return new MyViewModel(context, repository);
    }
}

Then finally, to use:

@Inject MyViewModelFactory factory;
...
MyViewModel createViewModel() {
    return ViewModelProviders.of(this,factory)
                             .get(MyViewModel.class);
}

An attempt in kotlin

class MyViewModelFactory @Inject constructor(val context: Context, val repository: MistakesRepository) : ViewModelProvider.Factory {

        override fun create(modelClass: Class): MyViewModel {
            return MyViewModel(context, repository);
        }
    }

@Inject lateinit var factory : MyViewModelFactory
...
fun createViewModel() : MyViewModel {
    return ViewModelProviders.of(this,factory)
                             .get(MyViewModel::java.class);
}

Upvotes: 1

Related Questions