Refactor DOMYangTextSourceProvider

[mdsal.git] / dom / mdsal-dom-spi / src / main / java / org / opendaylight / mdsal / dom / spi / FixedDOMSchemaService.java

diff --git a/dom/mdsal-dom-spi/src/main/java/org/opendaylight/mdsal/dom/spi/FixedDOMSchemaService.java b/dom/mdsal-dom-spi/src/main/java/org/opendaylight/mdsal/dom/spi/FixedDOMSchemaService.java
index 42b364665a6864a7602dea3786d44b221ee2e513..48cbd4b996ecbdf1e600b0956241fc8f87cefb4a 100644 (file)
--- a/dom/mdsal-dom-spi/src/main/java/org/opendaylight/mdsal/dom/spi/FixedDOMSchemaService.java
+++ b/dom/mdsal-dom-spi/src/main/java/org/opendaylight/mdsal/dom/spi/FixedDOMSchemaService.java
@@ -10,76 +10,54 @@ package org.opendaylight.mdsal.dom.spi;
 import static java.util.Objects.requireNonNull;
 
 import com.google.common.annotations.Beta;
-import com.google.common.util.concurrent.ListenableFuture;
 import java.util.List;
-import org.eclipse.jdt.annotation.NonNull;
+import java.util.function.Consumer;
+import java.util.function.Supplier;
+import org.eclipse.jdt.annotation.NonNullByDefault;
+import org.eclipse.jdt.annotation.Nullable;
 import org.opendaylight.mdsal.dom.api.DOMSchemaService;
-import org.opendaylight.mdsal.dom.api.DOMYangTextSourceProvider;
 import org.opendaylight.yangtools.concepts.Registration;
 import org.opendaylight.yangtools.yang.model.api.EffectiveModelContext;
-import org.opendaylight.yangtools.yang.model.api.EffectiveModelContextListener;
-import org.opendaylight.yangtools.yang.model.api.EffectiveModelContextProvider;
-import org.opendaylight.yangtools.yang.model.repo.api.SourceIdentifier;
-import org.opendaylight.yangtools.yang.model.repo.api.YangTextSchemaSource;
-import org.opendaylight.yangtools.yang.model.repo.spi.SchemaSourceProvider;
 
 /**
- * {@link DOMSchemaService} (and {@link DOMYangTextSourceProvider}) implementations backed by a
- * {@link EffectiveModelContextProvider} (and {@link SchemaSourceProvider}) which are known to be fixed and never change
- * schemas.
+ * {@link DOMSchemaService} backed by a {@code Supplier<EffectiveModelContext>}
+ * (and potentially a {@link YangTextSourceExtension}) which are known to be fixed
+ * and never change schemas.
  *
  * @author Michael Vorburger.ch
  */
 @Beta
-public class FixedDOMSchemaService extends AbstractDOMSchemaService {
-    private static final class WithYangTextSources extends FixedDOMSchemaService implements DOMYangTextSourceProvider {
-        private final @NonNull SchemaSourceProvider<YangTextSchemaSource> schemaSourceProvider;
-
-        WithYangTextSources(final EffectiveModelContextProvider schemaContextProvider,
-                final SchemaSourceProvider<YangTextSchemaSource> schemaSourceProvider) {
-            super(schemaContextProvider);
-            this.schemaSourceProvider = requireNonNull(schemaSourceProvider);
-        }
-
-        @Override
-        public List<Extension> supportedExtensions() {
-            return List.of(this);
-        }
-
-        @Override
-        public ListenableFuture<? extends YangTextSchemaSource> getSource(final SourceIdentifier sourceIdentifier) {
-            return schemaSourceProvider.getSource(sourceIdentifier);
-        }
+@NonNullByDefault
+public record FixedDOMSchemaService(
+        Supplier<EffectiveModelContext> modelContextSupplier,
+        @Nullable YangTextSourceExtension extension) implements DOMSchemaService {
+    public FixedDOMSchemaService {
+        requireNonNull(modelContextSupplier);
     }
 
-    private final @NonNull EffectiveModelContextProvider schemaContextProvider;
-
-    private FixedDOMSchemaService(final EffectiveModelContextProvider schemaContextProvider) {
-        this.schemaContextProvider = requireNonNull(schemaContextProvider);
+    public FixedDOMSchemaService(final Supplier<EffectiveModelContext> modelContextSupplier) {
+        this(modelContextSupplier, null);
     }
 
-    public static @NonNull DOMSchemaService of(final EffectiveModelContext effectiveModel) {
-        final var checked = requireNonNull(effectiveModel);
-        return of(() -> checked);
+    public FixedDOMSchemaService(final EffectiveModelContext effectiveModel) {
+        this(() -> effectiveModel, null);
+        requireNonNull(effectiveModel);
     }
 
-    public static @NonNull DOMSchemaService of(final EffectiveModelContextProvider schemaContextProvider) {
-        return new FixedDOMSchemaService(schemaContextProvider);
-    }
-
-    public static @NonNull DOMSchemaService of(final EffectiveModelContextProvider schemaContextProvider,
-            final SchemaSourceProvider<YangTextSchemaSource> yangTextSourceProvider) {
-        return new WithYangTextSources(schemaContextProvider, requireNonNull(yangTextSourceProvider));
+    @Override
+    public List<Extension> supportedExtensions() {
+        final var local = extension;
+        return local == null ? List.of() : List.of(local);
     }
 
     @Override
-    public final @NonNull EffectiveModelContext getGlobalContext() {
-        return schemaContextProvider.getEffectiveModelContext();
+    public final EffectiveModelContext getGlobalContext() {
+        return modelContextSupplier.get();
     }
 
     @Override
-    public final Registration registerSchemaContextListener(final EffectiveModelContextListener listener) {
-        listener.onModelContextUpdated(getGlobalContext());
+    public final Registration registerSchemaContextListener(final Consumer<EffectiveModelContext> listener) {
+        listener.accept(getGlobalContext());
         return () -> { };
     }
 }