@Override
  public void dataSnapshot(@NotNull DataSink sink) {
    sink.lazy(Location.DATA_KEY, () -> {
      PsiElement e = myElement.getElement();
      return e != null && e.isValid() ? new PsiLocation<>(e) : null;
    });
  }
@Override
  public void dataSnapshot(@NotNull DataSink sink)