Migrate netconf users of submit() to commit()
[netconf.git] / netconf / sal-netconf-connector / src / main / java / org / opendaylight / netconf / sal / connect / util / NetconfTopologyRPCProvider.java
1 /*
2  * Copyright (c) 2017 Brocade Communication Systems and others.  All rights reserved.
3  *
4  * This program and the accompanying materials are made available under the
5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6  * and is available at http://www.eclipse.org/legal/epl-v10.html
7  */
8 package org.opendaylight.netconf.sal.connect.util;
9
10 import com.google.common.annotations.VisibleForTesting;
11 import com.google.common.base.Preconditions;
12 import com.google.common.util.concurrent.FutureCallback;
13 import com.google.common.util.concurrent.ListenableFuture;
14 import com.google.common.util.concurrent.MoreExecutors;
15 import com.google.common.util.concurrent.SettableFuture;
16 import org.opendaylight.aaa.encrypt.AAAEncryptionService;
17 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
18 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
19 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
20 import org.opendaylight.mdsal.common.api.CommitInfo;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.CreateDeviceInput;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.CreateDeviceOutput;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.CreateDeviceOutputBuilder;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.DeleteDeviceInput;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.DeleteDeviceOutput;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.DeleteDeviceOutputBuilder;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.NetconfNode;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.NetconfNodeBuilder;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.NetconfNodeTopologyService;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.netconf.node.credentials.Credentials;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.netconf.node.credentials.credentials.LoginPw;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.netconf.node.credentials.credentials.LoginPwBuilder;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.netconf.node.credentials.credentials.login.pw.LoginPassword;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.netconf.node.credentials.credentials.login.pw.LoginPasswordBuilder;
35 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NetworkTopology;
36 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
37 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.TopologyId;
38 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
39 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyKey;
40 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
41 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeKey;
42 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
43 import org.opendaylight.yangtools.yang.common.RpcResult;
44 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
45 import org.slf4j.Logger;
46 import org.slf4j.LoggerFactory;
47
48 public class NetconfTopologyRPCProvider implements NetconfNodeTopologyService {
49     private static final Logger LOG = LoggerFactory.getLogger(NetconfTopologyRPCProvider.class);
50
51     private final InstanceIdentifier<Topology> topologyPath;
52     private final AAAEncryptionService encryptionService;
53     private final DataBroker dataBroker;
54
55     public NetconfTopologyRPCProvider(final DataBroker dataBroker,
56                                       final AAAEncryptionService encryptionService,
57                                       final String topologyId) {
58         this.dataBroker = dataBroker;
59         this.encryptionService = Preconditions.checkNotNull(encryptionService);
60         this.topologyPath = InstanceIdentifier.builder(NetworkTopology.class)
61                 .child(Topology.class, new TopologyKey(new TopologyId(Preconditions.checkNotNull(topologyId)))).build();
62     }
63
64     @Override
65     public ListenableFuture<RpcResult<CreateDeviceOutput>> createDevice(final CreateDeviceInput input) {
66         final NetconfNode node = this.encryptPassword(input);
67         final SettableFuture<RpcResult<CreateDeviceOutput>> futureResult = SettableFuture.create();
68         final NodeId nodeId = new NodeId(input.getNodeId());
69         writeToConfigDS(node, nodeId, futureResult);
70         return futureResult;
71     }
72
73     @VisibleForTesting
74     public NetconfNode encryptPassword(final CreateDeviceInput input) {
75         final NetconfNodeBuilder builder = new NetconfNodeBuilder();
76         builder.fieldsFrom(input);
77
78         final Credentials credentials = handleEncryption(input.getCredentials());
79         builder.setCredentials(credentials);
80
81         return builder.build();
82     }
83
84     private Credentials handleEncryption(final Credentials credentials) {
85         if (credentials instanceof LoginPw) {
86             final LoginPassword loginPassword = ((LoginPw) credentials).getLoginPassword();
87             final String encryptedPassword =
88                     encryptionService.encrypt(loginPassword.getPassword());
89
90             return new LoginPwBuilder().setLoginPassword(new LoginPasswordBuilder()
91                     .setPassword(encryptedPassword)
92                     .setUsername(loginPassword.getUsername()).build()).build();
93         }
94
95         // nothing else needs to be encrypted
96         return credentials;
97     }
98
99     private void writeToConfigDS(final NetconfNode node, final NodeId nodeId,
100             final SettableFuture<RpcResult<CreateDeviceOutput>> futureResult) {
101
102         final WriteTransaction writeTransaction = dataBroker.newWriteOnlyTransaction();
103         final InstanceIdentifier<NetconfNode> niid = topologyPath.child(Node.class,
104                 new NodeKey(nodeId)).augmentation(NetconfNode.class);
105         writeTransaction.merge(LogicalDatastoreType.CONFIGURATION, niid, node, true);
106         writeTransaction.commit().addCallback(new FutureCallback<CommitInfo>() {
107
108             @Override
109             public void onSuccess(final CommitInfo result) {
110                 LOG.info("add-netconf-node RPC: Added netconf node successfully.");
111                 futureResult.set(RpcResultBuilder.success(new CreateDeviceOutputBuilder().build()).build());
112             }
113
114             @Override
115             public void onFailure(final Throwable exception) {
116                 LOG.error("add-netconf-node RPC: Unable to add netconf node.", exception);
117                 futureResult.setException(exception);
118             }
119         }, MoreExecutors.directExecutor());
120     }
121
122
123     @Override
124     public ListenableFuture<RpcResult<DeleteDeviceOutput>> deleteDevice(final DeleteDeviceInput input) {
125         final NodeId nodeId = new NodeId(input.getNodeId());
126
127         final InstanceIdentifier<Node> niid = topologyPath.child(Node.class, new NodeKey(nodeId));
128
129         final WriteTransaction wtx = dataBroker.newWriteOnlyTransaction();
130         wtx.delete(LogicalDatastoreType.CONFIGURATION, niid);
131
132         final SettableFuture<RpcResult<DeleteDeviceOutput>> rpcFuture = SettableFuture.create();
133
134         wtx.commit().addCallback(new FutureCallback<CommitInfo>() {
135
136             @Override
137             public void onSuccess(final CommitInfo result) {
138                 LOG.info("delete-device RPC: Removed netconf node successfully.");
139                 rpcFuture.set(RpcResultBuilder.success(new DeleteDeviceOutputBuilder().build()).build());
140             }
141
142             @Override
143             public void onFailure(final Throwable exception) {
144                 LOG.error("delete-device RPC: Unable to remove netconf node.", exception);
145                 rpcFuture.setException(exception);
146             }
147         }, MoreExecutors.directExecutor());
148
149         return rpcFuture;
150     }
151 }