Создание ViewModel с зависимостями (Views) — часть Android Jetpack .

Концепции и реализация Jetpack Compose

Следуя лучшим практикам внедрения зависимостей , ViewModels могут принимать зависимости в качестве параметров в своем конструкторе. В основном это типы из предметной области или уровня данных . Поскольку фреймворк предоставляет ViewModels, для создания их экземпляров требуется специальный механизм. Этот механизм — интерфейс ViewModelProvider.Factory . Только реализации этого интерфейса могут создавать экземпляры ViewModels в нужной области видимости .

ViewModels with CreationExtras

Если класс ViewModel получает зависимости в своем конструкторе, предоставьте фабрику, реализующую интерфейс ViewModelProvider.Factory . Переопределите функцию create(Class<T>, CreationExtras) , чтобы предоставить новый экземпляр ViewModel.

CreationExtras с APPLICATION_KEY

Ниже приведён пример того, как предоставить экземпляр ViewModel , который принимает в качестве зависимостей репозиторий , ограниченный областью видимости класса Application , и SavedStateHandle :

import static androidx.lifecycle.SavedStateHandleSupport.createSavedStateHandle;
import static androidx.lifecycle.ViewModelProvider.AndroidViewModelFactory.APPLICATION_KEY;

import androidx.lifecycle.SavedStateHandle;
import androidx.lifecycle.ViewModel;
import androidx.lifecycle.viewmodel.ViewModelInitializer;

public class MyViewModel extends ViewModel {

    public MyViewModel(
        MyRepository myRepository,
        SavedStateHandle savedStateHandle
    ) { /* Init ViewModel here */ }

    static final ViewModelInitializer<MyViewModel> initializer = new ViewModelInitializer<>(
        MyViewModel.class,
        creationExtras -> {
            MyApplication app = (MyApplication) creationExtras.get(APPLICATION_KEY);
            assert app != null;
            SavedStateHandle savedStateHandle = createSavedStateHandle(creationExtras);

            return new MyViewModel(app.getMyRepository(), savedStateHandle);
        }
    );
}

Затем вы можете использовать эту фабрику при получении экземпляра ViewModel:

Котлин 

import androidx.activity.viewModels

class MyActivity : AppCompatActivity() {

    private val viewModel: MyViewModel by viewModels { MyViewModel.Factory }

    // Rest of Activity code
}

Java 

import androidx.appcompat.app.AppCompatActivity;
import androidx.lifecycle.ViewModelProvider;

public class MyActivity extends AppCompatActivity {

MyViewModel myViewModel = new ViewModelProvider(
    this,
    ViewModelProvider.Factory.from(MyViewModel.initializer)
).get(MyViewModel.class);

// Rest of Activity code
}

Передайте пользовательские параметры в качестве CreationExtras.

Вы можете передавать зависимости в вашу ViewModel через CreationExtras , создав пользовательский ключ. Это может быть полезно, если ваша ViewModel зависит от объектов, недоступных через класс Application и APPLICATION_KEY . Например, когда ваша ViewModel создается внутри модуля Kotlin Multiplatform и, следовательно, не имеет доступа к зависимостям Android.

В этом примере ViewModel определяет пользовательский ключ и использует его в ViewModelProvider.Factory .

import androidx.lifecycle.ViewModel
import androidx.lifecycle.ViewModelProvider
import androidx.lifecycle.viewModelScope
import androidx.lifecycle.viewmodel.CreationExtras
import androidx.lifecycle.viewmodel.initializer
import androidx.lifecycle.viewmodel.viewModelFactory

class MyViewModel(
    private val myRepository: MyRepository,
) : ViewModel() {
    // ViewModel logic

    // Define ViewModel factory in a companion object
    companion object {

        // Define a custom key for your dependency
        val MY_REPOSITORY_KEY = object : CreationExtras.Key<MyRepository> {}

        val Factory: ViewModelProvider.Factory = viewModelFactory {
            initializer {
                // Get the dependency in your factory
                val myRepository = this[MY_REPOSITORY_KEY] as MyRepository
                MyViewModel(
                    myRepository = myRepository,
                )
            }
        }
    }
}

Вы можете создать экземпляр ViewModel , используя CreationExtras.Key из ViewModelStoreOwner такого как ComponentActivity , Fragment или NavBackStackEntry .

import androidx.lifecycle.ViewModelProvider
import androidx.lifecycle.ViewModelStoreOwner
import androidx.lifecycle.viewmodel.CreationExtras
import androidx.lifecycle.viewmodel.MutableCreationExtras
// ...
    // Use from ComponentActivity, Fragment, NavBackStackEntry,
    // or another ViewModelStoreOwner.
    val viewModelStoreOwner: ViewModelStoreOwner = this
    val myViewModel: MyViewModel = ViewModelProvider.create(
        viewModelStoreOwner,
        factory = MyViewModel.Factory,
        extras = MutableCreationExtras().apply {
            set(MyViewModel.MY_REPOSITORY_KEY, myRepository)
        },
    )[MyViewModel::class]

Фабрики для версий ViewModel старше 2.5.0

Если вы используете версию ViewModel старше 2.5.0, вам необходимо предоставить фабрики из подмножества классов, которые наследуют ViewModelProvider.Factory и реализуют функцию create(Class<T>) . В зависимости от того, какие зависимости требуются ViewModel , необходимо наследовать другой класс от:

Если Application или SavedStateHandle не требуются, просто наследуйте класс от ViewModelProvider.Factory .

В следующем примере используется AbstractSavedStateViewModelFactory для ViewModel, которая принимает в качестве зависимости репозиторий и тип SavedStateHandle :

Котлин 

class MyViewModel(
private val myRepository: MyRepository,
private val savedStateHandle: SavedStateHandle
) : ViewModel() {

// ViewModel logic ...

// Define ViewModel factory in a companion object
companion object {
    fun provideFactory(
        myRepository: MyRepository,
        owner: SavedStateRegistryOwner,
        defaultArgs: Bundle? = null,
    ): AbstractSavedStateViewModelFactory =
        object : AbstractSavedStateViewModelFactory(owner, defaultArgs) {
            @Suppress("UNCHECKED_CAST")
            override fun <T : ViewModel> create(
                key: String,
                modelClass: Class<T>,
                handle: SavedStateHandle
            ): T {
                return MyViewModel(myRepository, handle) as T
            }
        }
    }
}

Java 

import androidx.annotation.NonNull;
import androidx.lifecycle.AbstractSavedStateViewModelFactory;
import androidx.lifecycle.SavedStateHandle;
import androidx.lifecycle.ViewModel;

public class MyViewModel extends ViewModel {
    public MyViewModel(
        MyRepository myRepository,
        SavedStateHandle savedStateHandle
    ) { /* Init ViewModel here */ }
}

public class MyViewModelFactory extends AbstractSavedStateViewModelFactory {

    private final MyRepository myRepository;

    public MyViewModelFactory(
        MyRepository myRepository
    ) {
        this.myRepository = myRepository;
    }

    @SuppressWarnings("unchecked")
    @NonNull
    @Override
    protected <T extends ViewModel> T create(
        @NonNull String key, @NonNull Class<T> modelClass, @NonNull SavedStateHandle handle
    ) {
        return (T) new MyViewModel(myRepository, handle);
    }
}

Затем вы можете использовать фабрику для получения вашей ViewModel:

Котлин 

import androidx.activity.viewModels

class MyActivity : AppCompatActivity() {

    private val viewModel: MyViewModel by viewModels {
        MyViewModel.provideFactory((application as MyApplication).myRepository, this)
    }

    // Rest of Activity code
}

Java 

import androidx.appcompat.app.AppCompatActivity;
import androidx.lifecycle.ViewModelProvider;

public class MyActivity extends AppCompatActivity {

    MyViewModel myViewModel = new ViewModelProvider(
        this,
        ViewModelProvider.Factory.from(MyViewModel.initializer)
    ).get(MyViewModel.class);

    // Rest of Activity code
}
{% verbatim %} {% endverbatim %} {% verbatim %} {% endverbatim %}