import static java.util.Objects.requireNonNull;
import com.google.common.collect.Collections2;
-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 com.google.common.util.concurrent.MoreExecutors;
-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.DOMRpcIdentifier;
-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.SettableFuture;
+import org.opendaylight.mdsal.dom.api.DOMRpcAvailabilityListener;
+import org.opendaylight.mdsal.dom.api.DOMRpcIdentifier;
+import org.opendaylight.mdsal.dom.api.DOMRpcImplementationNotAvailableException;
+import org.opendaylight.mdsal.dom.api.DOMRpcResult;
+import org.opendaylight.mdsal.dom.api.DefaultDOMRpcException;
+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.yangtools.concepts.ListenerRegistration;
import org.opendaylight.yangtools.concepts.NoOpListenerRegistration;
+import org.opendaylight.yangtools.yang.common.QName;
import org.opendaylight.yangtools.yang.common.RpcResult;
import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
-import org.opendaylight.yangtools.yang.model.api.SchemaContext;
-import org.opendaylight.yangtools.yang.model.api.SchemaPath;
+import org.opendaylight.yangtools.yang.model.api.EffectiveModelContext;
/**
- * Invokes RPC by sending netconf message via listener. Also transforms result from NetconfMessage to CompositeNode.
+ * Invokes RPC by sending netconf message via listener. Also transforms result from NetconfMessage to
+ * {@link NormalizedNode}.
*/
-public final class NetconfDeviceRpc implements DOMRpcService {
+public final class NetconfDeviceRpc implements Rpcs.Normalized {
+ private final RemoteDeviceCommunicator communicator;
+ private final RpcTransformer transformer;
+ private final EffectiveModelContext modelContext;
- private final RemoteDeviceCommunicator<NetconfMessage> communicator;
- private final MessageTransformer<NetconfMessage> transformer;
- private final SchemaContext schemaContext;
-
- public NetconfDeviceRpc(final SchemaContext schemaContext,
- final RemoteDeviceCommunicator<NetconfMessage> communicator,
- final MessageTransformer<NetconfMessage> transformer) {
+ public NetconfDeviceRpc(final EffectiveModelContext modelContext, final RemoteDeviceCommunicator communicator,
+ final RpcTransformer transformer) {
+ this.modelContext = requireNonNull(modelContext);
this.communicator = communicator;
this.transformer = transformer;
- this.schemaContext = requireNonNull(schemaContext);
}
- @Nonnull
@Override
- public CheckedFuture<DOMRpcResult, DOMRpcException> invokeRpc(@Nonnull final SchemaPath type,
- @Nullable final NormalizedNode<?, ?> input) {
- final NetconfMessage message = transformer.toRpcRequest(type, input);
- final ListenableFuture<RpcResult<NetconfMessage>> delegateFutureWithPureResult =
- communicator.sendRequest(message, type.getLastComponent());
+ @SuppressWarnings("checkstyle:IllegalCatch")
+ public ListenableFuture<DOMRpcResult> invokeRpc(final QName type, final NormalizedNode input) {
+ final ListenableFuture<RpcResult<NetconfMessage>> delegateFuture = communicator.sendRequest(
+ transformer.toRpcRequest(type, input), type);
- final ListenableFuture<DOMRpcResult> transformed =
- Futures.transform(delegateFutureWithPureResult, input1 -> {
- if (input1.isSuccessful()) {
- return transformer.toRpcResult(input1.getResult(), type);
- } else {
- return new DefaultDOMRpcResult(input1.getErrors());
+ final SettableFuture<DOMRpcResult> ret = SettableFuture.create();
+ Futures.addCallback(delegateFuture, new FutureCallback<RpcResult<NetconfMessage>>() {
+ @Override
+ public void onSuccess(final RpcResult<NetconfMessage> result) {
+ try {
+ ret.set(result.isSuccessful() ? transformer.toRpcResult(result.getResult(), type)
+ : new DefaultDOMRpcResult(result.getErrors()));
+ } catch (Exception cause) {
+ ret.setException(new DefaultDOMRpcException(
+ "Unable to parse rpc reply. type: " + type + " input: " + input, cause));
}
- }, MoreExecutors.directExecutor());
+ }
+
+ @Override
+ public void onFailure(final Throwable cause) {
+ ret.setException(new DOMRpcImplementationNotAvailableException(cause, "Unable to invoke rpc %s", type));
+ }
- return Futures.makeChecked(transformed, exception ->
- new DOMRpcImplementationNotAvailableException(exception, "Unable to invoke rpc %s", type));
+ }, MoreExecutors.directExecutor());
+ return ret;
}
- @Nonnull
@Override
- public <T extends DOMRpcAvailabilityListener> ListenerRegistration<T> registerRpcListener(
- @Nonnull final T listener) {
- listener.onRpcAvailable(Collections2.transform(schemaContext.getOperations(),
- input -> DOMRpcIdentifier.create(input.getPath())));
+ public <T extends DOMRpcAvailabilityListener> ListenerRegistration<T> registerRpcListener(final T listener) {
+ listener.onRpcAvailable(Collections2.transform(modelContext.getOperations(),
+ input -> DOMRpcIdentifier.create(input.getQName())));
// NOOP, no rpcs appear and disappear in this implementation
return NoOpListenerRegistration.of(listener);
Code Review / netconf.git / blobdiff