* terms of the Eclipse Public License v1.0 which accompanies this distribution,
* and is available at http://www.eclipse.org/legal/epl-v10.html
*/
-
package org.opendaylight.netconf.mdsal.connector;
-import com.google.common.base.Preconditions;
-import com.google.common.collect.Sets;
+import static com.google.common.base.Preconditions.checkState;
+
import java.util.Collections;
+import java.util.HashSet;
import java.util.Set;
import java.util.concurrent.atomic.AtomicReference;
-import org.opendaylight.controller.config.util.capability.Capability;
-import org.opendaylight.netconf.api.monitoring.CapabilityListener;
-import org.opendaylight.controller.sal.core.api.model.SchemaService;
-import org.opendaylight.yangtools.concepts.ListenerRegistration;
-import org.opendaylight.yangtools.yang.model.api.SchemaContext;
-import org.opendaylight.yangtools.yang.model.api.SchemaContextListener;
+import org.eclipse.jdt.annotation.NonNull;
+import org.opendaylight.mdsal.dom.api.DOMSchemaService;
+import org.opendaylight.netconf.api.capability.Capability;
+import org.opendaylight.netconf.server.api.monitoring.CapabilityListener;
+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.repo.api.YangTextSchemaSource;
import org.opendaylight.yangtools.yang.model.repo.spi.SchemaSourceProvider;
-public class CurrentSchemaContext implements SchemaContextListener, AutoCloseable {
- private final AtomicReference<SchemaContext> currentContext = new AtomicReference();
- private final ListenerRegistration<SchemaContextListener> schemaContextListenerListenerRegistration;
- private final Set<CapabilityListener> listeners1 = Collections.synchronizedSet(Sets.newHashSet());
+// Non-final for mocking
+@SuppressWarnings("checkstyle:FinalClass")
+public class CurrentSchemaContext implements EffectiveModelContextListener, AutoCloseable {
+ private final AtomicReference<EffectiveModelContext> currentContext = new AtomicReference<>();
+ private final Set<CapabilityListener> listeners1 = Collections.synchronizedSet(new HashSet<>());
private final SchemaSourceProvider<YangTextSchemaSource> rootSchemaSourceProvider;
- public SchemaContext getCurrentContext() {
- Preconditions.checkState(currentContext.get() != null, "Current context not received");
- return currentContext.get();
- }
+ private Registration schemaContextListenerListenerRegistration;
- public CurrentSchemaContext(final SchemaService schemaService, final SchemaSourceProvider<YangTextSchemaSource> rootSchemaSourceProvider) {
+ private CurrentSchemaContext(final SchemaSourceProvider<YangTextSchemaSource> rootSchemaSourceProvider) {
this.rootSchemaSourceProvider = rootSchemaSourceProvider;
- schemaContextListenerListenerRegistration = schemaService.registerSchemaContextListener(this);
+ }
+
+ // keep spotbugs from complaining about overridable method in constructor
+ public static CurrentSchemaContext create(final DOMSchemaService schemaService,
+ final SchemaSourceProvider<YangTextSchemaSource> rootSchemaSourceProvider) {
+ var context = new CurrentSchemaContext(rootSchemaSourceProvider);
+ final Registration registration = schemaService.registerSchemaContextListener(context);
+ context.setRegistration(registration);
+ return context;
+ }
+
+ private void setRegistration(final Registration registration) {
+ schemaContextListenerListenerRegistration = registration;
+ }
+
+ public @NonNull EffectiveModelContext getCurrentContext() {
+ final var ret = currentContext.get();
+ checkState(ret != null, "Current context not received");
+ return ret;
}
@Override
- public void onGlobalContextUpdated(final SchemaContext schemaContext) {
+ public void onModelContextUpdated(final EffectiveModelContext schemaContext) {
currentContext.set(schemaContext);
// FIXME is notifying all the listeners from this callback wise ?
- final Set<Capability> addedCaps = MdsalNetconfOperationServiceFactory.transformCapabilities(currentContext.get(), rootSchemaSourceProvider);
+ final Set<Capability> addedCaps = MdsalNetconfOperationServiceFactory.transformCapabilities(
+ currentContext.get(), rootSchemaSourceProvider);
for (final CapabilityListener listener : listeners1) {
- listener.onCapabilitiesChanged(addedCaps, Collections.emptySet());
+ listener.onCapabilitiesChanged(addedCaps, Set.of());
}
}
@Override
- public void close() throws Exception {
+ public void close() {
listeners1.clear();
schemaContextListenerListenerRegistration.close();
currentContext.set(null);
}
- public AutoCloseable registerCapabilityListener(final CapabilityListener listener) {
- listener.onCapabilitiesChanged(MdsalNetconfOperationServiceFactory.transformCapabilities(currentContext.get(), rootSchemaSourceProvider), Collections.<Capability>emptySet());
+ public Registration registerCapabilityListener(final CapabilityListener listener) {
+ listener.onCapabilitiesChanged(MdsalNetconfOperationServiceFactory.transformCapabilities(currentContext.get(),
+ rootSchemaSourceProvider), Set.of());
listeners1.add(listener);
return () -> listeners1.remove(listener);
}
-}
\ No newline at end of file
+}