Split up MessageTransformer
[netconf.git] / netconf / sal-netconf-connector / src / main / java / org / opendaylight / netconf / sal / connect / netconf / sal / SchemalessNetconfDeviceRpc.java
index 84b824000eacc78103d9874e9bf7a36f97fb0ec1..1d66547a4d72408022278d88d21d99527c58eb9f 100644 (file)
@@ -7,97 +7,91 @@
  */
 package org.opendaylight.netconf.sal.connect.netconf.sal;
 
-import com.google.common.base.Function;
-import com.google.common.util.concurrent.CheckedFuture;
+import com.google.common.util.concurrent.FutureCallback;
 import com.google.common.util.concurrent.Futures;
 import com.google.common.util.concurrent.ListenableFuture;
-import javax.annotation.Nonnull;
-import javax.annotation.Nullable;
-import org.opendaylight.controller.md.sal.dom.api.DOMRpcAvailabilityListener;
-import org.opendaylight.controller.md.sal.dom.api.DOMRpcException;
-import org.opendaylight.controller.md.sal.dom.api.DOMRpcImplementationNotAvailableException;
-import org.opendaylight.controller.md.sal.dom.api.DOMRpcResult;
-import org.opendaylight.controller.md.sal.dom.api.DOMRpcService;
-import org.opendaylight.controller.md.sal.dom.spi.DefaultDOMRpcResult;
+import com.google.common.util.concurrent.MoreExecutors;
+import com.google.common.util.concurrent.SettableFuture;
+import org.eclipse.jdt.annotation.NonNull;
+import org.opendaylight.mdsal.dom.api.DOMRpcImplementationNotAvailableException;
+import org.opendaylight.mdsal.dom.api.DOMRpcResult;
+import org.opendaylight.mdsal.dom.spi.DefaultDOMRpcResult;
 import org.opendaylight.netconf.api.NetconfMessage;
-import org.opendaylight.netconf.sal.connect.api.MessageTransformer;
 import org.opendaylight.netconf.sal.connect.api.RemoteDeviceCommunicator;
+import org.opendaylight.netconf.sal.connect.api.RemoteDeviceServices.Rpcs;
+import org.opendaylight.netconf.sal.connect.api.RpcTransformer;
 import org.opendaylight.netconf.sal.connect.netconf.schema.mapping.BaseRpcSchemalessTransformer;
 import org.opendaylight.netconf.sal.connect.netconf.schema.mapping.SchemalessMessageTransformer;
 import org.opendaylight.netconf.sal.connect.netconf.util.NetconfMessageTransformUtil;
 import org.opendaylight.netconf.sal.connect.util.RemoteDeviceId;
-import org.opendaylight.yangtools.concepts.ListenerRegistration;
+import org.opendaylight.yangtools.yang.common.QName;
 import org.opendaylight.yangtools.yang.common.RpcResult;
-import org.opendaylight.yangtools.yang.data.api.schema.AnyXmlNode;
+import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
+import org.opendaylight.yangtools.yang.data.api.schema.DOMSourceAnyxmlNode;
 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
-import org.opendaylight.yangtools.yang.model.api.SchemaPath;
 
 /**
  * Invokes RPC by sending netconf message via listener. Also transforms result from NetconfMessage to CompositeNode.
  */
-public final class SchemalessNetconfDeviceRpc implements DOMRpcService {
-
-    private final RemoteDeviceCommunicator<NetconfMessage> listener;
+public final class SchemalessNetconfDeviceRpc implements Rpcs.Schemaless {
+    private final RemoteDeviceCommunicator listener;
     private final BaseRpcSchemalessTransformer baseRpcTransformer;
     private final SchemalessMessageTransformer schemalessTransformer;
     private final RemoteDeviceId deviceId;
 
-    public SchemalessNetconfDeviceRpc(final RemoteDeviceId deviceId,
-                                      final RemoteDeviceCommunicator<NetconfMessage> listener,
-                                      final BaseRpcSchemalessTransformer baseRpcTransformer,
-                                      final SchemalessMessageTransformer messageTransformer) {
+    public SchemalessNetconfDeviceRpc(final RemoteDeviceId deviceId, final RemoteDeviceCommunicator listener,
+            final BaseRpcSchemalessTransformer baseRpcTransformer,
+            final SchemalessMessageTransformer messageTransformer) {
         this.deviceId = deviceId;
         this.listener = listener;
         this.baseRpcTransformer = baseRpcTransformer;
-        this.schemalessTransformer = messageTransformer;
+        schemalessTransformer = messageTransformer;
+    }
+
+    @Override
+    public ListenableFuture<? extends DOMRpcResult> invokeNetconf(final QName type, final ContainerNode input) {
+        if (!isBaseRpc(type)) {
+            throw new IllegalArgumentException("Cannot handle " + type);
+        }
+        return handleRpc(type, input, baseRpcTransformer);
     }
 
-    @Nonnull
     @Override
-    public CheckedFuture<DOMRpcResult, DOMRpcException> invokeRpc(@Nonnull final SchemaPath type,
-                                                                  @Nullable final NormalizedNode<?, ?> input) {
-        final MessageTransformer<NetconfMessage> transformer;
-        if (input instanceof AnyXmlNode) {
+    public ListenableFuture<? extends DOMRpcResult> invokeRpc(final QName type, final NormalizedNode input) {
+        final RpcTransformer transformer;
+        if (input instanceof DOMSourceAnyxmlNode) {
             transformer = schemalessTransformer;
         } else if (isBaseRpc(type)) {
             transformer = baseRpcTransformer;
         } else {
-            return Futures.immediateFailedCheckedFuture(
-                    new DOMRpcImplementationNotAvailableException("Unable to invoke rpc %s", type));
+            return Futures.immediateFailedFuture(new DOMRpcImplementationNotAvailableException(
+                "Unable to invoke rpc %s", type));
         }
         return handleRpc(type, input, transformer);
     }
 
-    private CheckedFuture<DOMRpcResult, DOMRpcException> handleRpc(
-            @Nonnull final SchemaPath type, @Nullable final NormalizedNode<?, ?> input,
-            final MessageTransformer<NetconfMessage> transformer) {
-        final NetconfMessage netconfMessage = transformer.toRpcRequest(type, input);
-        final ListenableFuture<RpcResult<NetconfMessage>> rpcResultListenableFuture =
-                listener.sendRequest(netconfMessage, type.getLastComponent());
+    private @NonNull ListenableFuture<DOMRpcResult> handleRpc(final @NonNull QName type,
+            final @NonNull NormalizedNode input, final RpcTransformer transformer) {
+        final var delegateFuture = listener.sendRequest(transformer.toRpcRequest(type, input), type);
+        final var ret = SettableFuture.<DOMRpcResult>create();
+        Futures.addCallback(delegateFuture, new FutureCallback<>() {
+            @Override
+            public void onSuccess(final RpcResult<NetconfMessage> result) {
+                ret.set(result.isSuccessful() ? transformer.toRpcResult(result.getResult(), type)
+                        : new DefaultDOMRpcResult(result.getErrors()));
+            }
 
-        final ListenableFuture<DOMRpcResult> transformed =
-            Futures.transform(rpcResultListenableFuture, (Function<RpcResult<NetconfMessage>, DOMRpcResult>) input1 -> {
-                if (input1.isSuccessful()) {
-                    return transformer.toRpcResult(input1.getResult(), type);
-                } else {
-                    return new DefaultDOMRpcResult(input1.getErrors());
-                }
-            });
+            @Override
+            public void onFailure(final Throwable cause) {
+                ret.setException(new DOMRpcImplementationNotAvailableException(cause,
+                    "Unable to invoke rpc %s on device %s", type, deviceId));
+            }
 
-        return Futures.makeChecked(transformed,
-            e -> new DOMRpcImplementationNotAvailableException(e,
-                "Unable to invoke rpc %s on device %s", type, deviceId));
+        }, MoreExecutors.directExecutor());
+        return ret;
     }
 
-    private static boolean isBaseRpc(final SchemaPath type) {
-        return NetconfMessageTransformUtil.NETCONF_URI.equals(type.getLastComponent().getNamespace());
+    private static boolean isBaseRpc(final QName type) {
+        return NetconfMessageTransformUtil.NETCONF_URI.equals(type.getNamespace());
     }
-
-    @Nonnull
-    @Override
-    public <T extends DOMRpcAvailabilityListener> ListenerRegistration<T> registerRpcListener(
-            @Nonnull final T listener) {
-        throw new UnsupportedOperationException("Not available for netconf 1.0");
-    }
-
 }