Bug 2364: Migrated Binding MD-SAL to not use composites nodes

[controller.git] / opendaylight / md-sal / sal-binding-broker / src / main / java / org / opendaylight / controller / md / sal / binding / compat / CompositeRoutedRpcRegistration.java

diff --git a/opendaylight/md-sal/sal-binding-broker/src/main/java/org/opendaylight/controller/md/sal/binding/compat/CompositeRoutedRpcRegistration.java b/opendaylight/md-sal/sal-binding-broker/src/main/java/org/opendaylight/controller/md/sal/binding/compat/CompositeRoutedRpcRegistration.java
new file mode 100644 (file)
index 0000000..bb6c2e7
--- /dev/null
+++ b/opendaylight/md-sal/sal-binding-broker/src/main/java/org/opendaylight/controller/md/sal/binding/compat/CompositeRoutedRpcRegistration.java
@@ -0,0 +1,88 @@
+/*
+ * Copyright (c) 2015 Cisco Systems, Inc. and others.  All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * 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.controller.md.sal.binding.compat;
+
+import com.google.common.base.Throwables;
+import com.google.common.collect.ImmutableSet;
+import java.util.HashMap;
+import java.util.Map;
+import org.opendaylight.controller.md.sal.binding.impl.BindingDOMRpcProviderServiceAdapter;
+import org.opendaylight.controller.sal.binding.api.BindingAwareBroker.RoutedRpcRegistration;
+import org.opendaylight.yangtools.concepts.ObjectRegistration;
+import org.opendaylight.yangtools.yang.binding.BaseIdentity;
+import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
+import org.opendaylight.yangtools.yang.binding.RpcService;
+
+final class CompositeRoutedRpcRegistration<T extends RpcService> implements RoutedRpcRegistration<T> {
+
+    private final Class<T> type;
+    private final T instance;
+    private final BindingDOMRpcProviderServiceAdapter adapter;
+    private final Map<InstanceIdentifier<?>, ObjectRegistration<T>> registrations = new HashMap<>(2);
+
+    CompositeRoutedRpcRegistration(final Class<T> type, final T impl, final BindingDOMRpcProviderServiceAdapter providerAdapter) {
+        this.type = type;
+        this.instance = impl;
+        this.adapter = providerAdapter;
+    }
+
+    @Override
+    public Class<T> getServiceType() {
+        return type;
+    }
+
+    @Override
+    public T getInstance() {
+        return instance;
+    }
+
+    @Deprecated
+    @Override
+    public void registerInstance(final Class<? extends BaseIdentity> context, final InstanceIdentifier<?> path) {
+        registerPath(context, path);
+    }
+
+    @Override
+    public synchronized void registerPath(final Class<? extends BaseIdentity> context, final InstanceIdentifier<?> path) {
+        if(!registrations.containsKey(path)) {
+            registrations.put(path, adapter.registerRpcImplementation(type, instance, ImmutableSet.<InstanceIdentifier<?>>of(path)));
+        }
+    }
+
+
+    @Override
+    @Deprecated
+    public void unregisterInstance(final Class<? extends BaseIdentity> context, final InstanceIdentifier<?> path) {
+        unregisterPath(context, path);
+    }
+
+    @Override
+    public synchronized  void unregisterPath(final Class<? extends BaseIdentity> context, final InstanceIdentifier<?> path) {
+        final ObjectRegistration<T> reg = registrations.remove(path);
+        if(reg != null) {
+            try {
+                reg.close();
+            } catch (final Exception e) {
+                // FIXME: Once we have proper subclass of ObjectRegistrationo
+                throw Throwables.propagate(e);
+            }
+        }
+    }
+
+    @Override
+    public synchronized void close() {
+        try {
+            for(final ObjectRegistration<T> reg : registrations.values()) {
+                    reg.close();
+            }
+        } catch (final Exception e) {
+            throw Throwables.propagate(e);
+        }
+    }
+
+}