Android Arch-ViewModelProviderHelper

ViewModelProviderHelper.kt

import android.arch.lifecycle.ViewModelProvider
import android.arch.lifecycle.ViewModelStores
import android.support.v4.app.Fragment

/**
 * Created by Shen on 2/4/2018.
 */
object ViewModelProviderHelper {

    fun forFragmentInActivity(fragment: Fragment): ViewModelProvider {
        val factory = ViewModelProvider.AndroidViewModelFactory.getInstance(fragment.activity.application)
        val store = ViewModelStores.of(fragment.activity)
        return ViewModelProvider(store, factory)
    }

    fun forFragment(fragment: Fragment): ViewModelProvider {
        val factory = ViewModelProvider.AndroidViewModelFactory.getInstance(fragment.activity.application)
        val store = ViewModelStores.of(fragment)
        return ViewModelProvider(store, factory)
    }
}