Make ViewModel survive Activity tear down

BaseFragment.java

public class BaseFragment extends Fragment {
    
    protected long dbId = -1;
    protected ViewModelFactory viewModelFactory;
    
    @Override
    public void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);
        if (savedInstanceState != null) {
            dbId = savedInstanceState.getLong("dbId", -1);
        }
        viewModelFactory = new ViewModelFactory(dbId);
    }
  
    @Override
    public void onSaveInstanceState(Bundle outState) {
        super.onSaveInstanceState(outState);
        dbId = viewModelFactory.saveToDb();
        outState.putLong("dbId", dbId);
    }
  
}

MyFragment.java

public class MyFragment extends BaseFragment {
    
    private MyViewModel state;

    @Override
    public void onCreate(Bundle savedInstanceState) {
      super.onCreate(savedInstanceState);
      state = ViewModelProviders.of(getActivity(), viewModelFactory)
                .get(MyViewModel.class);
    }

}

ViewModelFactory.java

public class ViewModelFactory implements ViewModelProvider.Factory {
    private Object data;
    private final long dbId;

    public ViewModelFactory(long dbId) {
        this.dbId = dbId;
    }

    @Override
    public <T extends ViewModel> T create(Class<T> modelClass) {
        try {
            if (dbId > -1) {
                data = YourTable.get(dbId, modelClass); // this is your implementation of getting data from local db
            }

            if (data == null) {
                data = modelClass.newInstance();
            }

            return (T) data;
        } catch (InstantiationException e) {
            Crashlytics.logException(e);
        } catch (IllegalAccessException e) {
            Crashlytics.logException(e);
        }
        return null;
    }

    public long saveToDb() {
        long id = -1;
        if (data != null) {
            id = YourTable.put(data); // this is your implementation of saving to local db
        }
        return id;
    }
}