be83555761b1207b86d69033bb74335e11102fb4
[ccsdk/cds.git] /
1 /*
2  * Copyright © 2017-2019 AT&T, Bell Canada
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *     http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 package org.onap.ccsdk.apps.blueprintsprocessor.functions.netconf.executor.core
18
19 import org.onap.ccsdk.apps.blueprintsprocessor.functions.netconf.executor.api.DeviceInfo
20 import org.onap.ccsdk.apps.blueprintsprocessor.functions.netconf.executor.api.DeviceResponse
21 import org.onap.ccsdk.apps.blueprintsprocessor.functions.netconf.executor.api.NetconfException
22 import org.onap.ccsdk.apps.blueprintsprocessor.functions.netconf.executor.api.NetconfRpcService
23 import org.onap.ccsdk.apps.blueprintsprocessor.functions.netconf.executor.api.NetconfSession
24 import org.onap.ccsdk.apps.blueprintsprocessor.functions.netconf.executor.utils.NetconfMessageUtils
25 import org.onap.ccsdk.apps.blueprintsprocessor.functions.netconf.executor.utils.RpcStatus
26 import org.slf4j.LoggerFactory
27 import java.util.concurrent.TimeUnit
28 import java.util.concurrent.atomic.AtomicInteger
29
30 class NetconfRpcServiceImpl(private var deviceInfo: DeviceInfo) : NetconfRpcService {
31
32     private val log = LoggerFactory.getLogger(NetconfRpcService::class.java)
33
34     private var responseTimeout: Int = deviceInfo.replyTimeout
35
36     private lateinit var netconfSession: NetconfSession
37
38     private val messageIdInteger = AtomicInteger(1)
39
40     fun setNetconfSession(netconfSession: NetconfSession) {
41         this.netconfSession = netconfSession
42     }
43
44     override fun getConfig(filter: String, configTarget: String): DeviceResponse {
45         var output = DeviceResponse()
46         val messageId = messageIdInteger.getAndIncrement().toString()
47         log.info("$deviceInfo: getConfig: messageId($messageId)")
48         try {
49             val message = NetconfMessageUtils.getConfig(messageId, configTarget, filter)
50             output = asyncRpc(message, messageId)
51         } catch (e: Exception) {
52             output.status = RpcStatus.FAILURE
53             output.errorMessage = "$deviceInfo: failed in get-config command $e.message"
54         }
55         return output
56     }
57
58     override fun deleteConfig(configTarget: String): DeviceResponse {
59         var output = DeviceResponse()
60         val messageId = messageIdInteger.getAndIncrement().toString()
61         log.info("$deviceInfo: deleteConfig: messageId($messageId)")
62         try {
63             val deleteConfigMessage = NetconfMessageUtils.deleteConfig(messageId, configTarget)
64             output.requestMessage = deleteConfigMessage
65             output = asyncRpc(deleteConfigMessage, messageId)
66         } catch (e: Exception) {
67             output.status = RpcStatus.FAILURE
68             output.errorMessage = "$deviceInfo: failed in delete config command $e.message"
69         }
70         return output
71     }
72
73     override fun lock(configTarget: String): DeviceResponse {
74         var output = DeviceResponse()
75         val messageId = messageIdInteger.getAndIncrement().toString()
76         log.info("$deviceInfo: lock: messageId($messageId)")
77         try {
78             val lockMessage = NetconfMessageUtils.lock(messageId, configTarget)
79             output.requestMessage = lockMessage
80             output = asyncRpc(lockMessage, messageId)
81         } catch (e: Exception) {
82             output.status = RpcStatus.FAILURE
83             output.errorMessage = "$deviceInfo: failed in lock command $e.message"
84         }
85
86         return output
87     }
88
89     override fun unLock(configTarget: String): DeviceResponse {
90         var output = DeviceResponse()
91         val messageId = messageIdInteger.getAndIncrement().toString()
92         log.info("$deviceInfo: unLock: messageId($messageId)")
93         try {
94             val unlockMessage = NetconfMessageUtils.unlock(messageId, configTarget)
95             output.requestMessage = unlockMessage
96             output = asyncRpc(unlockMessage, messageId)
97         } catch (e: Exception) {
98             output.status = RpcStatus.FAILURE
99             output.errorMessage = "$deviceInfo: failed in lock command $e.message"
100         }
101         return output
102     }
103
104     override fun commit(): DeviceResponse {
105         var output = DeviceResponse()
106         val messageId = messageIdInteger.getAndIncrement().toString()
107         log.info("$deviceInfo: commit: messageId($messageId)")
108         try {
109             val messageContent = NetconfMessageUtils.commit(messageId)
110             output = asyncRpc(messageContent, messageId)
111         } catch (e: Exception) {
112             output.status = RpcStatus.FAILURE
113             output.errorMessage = "$deviceInfo: failed in commit command $e.message"
114         }
115         return output
116     }
117
118     override fun discardConfig(): DeviceResponse {
119         var output = DeviceResponse()
120         val messageId = messageIdInteger.getAndIncrement().toString()
121         log.info("$deviceInfo: discard: messageId($messageId)")
122         try {
123             val discardChangesMessage = NetconfMessageUtils.discardChanges(messageId)
124             output.requestMessage = discardChangesMessage
125             output = asyncRpc(discardChangesMessage, messageId)
126         } catch (e: Exception) {
127             output.status = RpcStatus.FAILURE
128             output.errorMessage = "$deviceInfo: failed in discard changes command " + e.message
129         }
130         return output
131     }
132
133     override fun editConfig(messageContent: String, configTarget: String,
134                             editDefaultOperation: String): DeviceResponse {
135         var response = DeviceResponse()
136         val messageId = messageIdInteger.getAndIncrement().toString()
137         log.info("$deviceInfo: editConfig: messageId($messageId)")
138         try {
139             val editMessage =
140                 NetconfMessageUtils.editConfig(messageId, configTarget, editDefaultOperation, messageContent)
141             response.requestMessage = editMessage
142             response = asyncRpc(editMessage, messageId)
143         } catch (e: Exception) {
144             response.status = RpcStatus.FAILURE
145             response.errorMessage = e.message
146         }
147         return response
148     }
149
150     override fun validate(configTarget: String): DeviceResponse {
151         var output = DeviceResponse()
152         val messageId = messageIdInteger.getAndIncrement().toString()
153         try {
154             val validateMessage = NetconfMessageUtils.validate(messageId, configTarget)
155             output.requestMessage = validateMessage
156             output = asyncRpc(validateMessage, messageId)
157         } catch (e: Exception) {
158             output.status = RpcStatus.FAILURE
159             output.errorMessage = "$deviceInfo: failed in validate command " + e.message
160         }
161         return output
162     }
163
164     override fun closeSession(force: Boolean): DeviceResponse {
165         var output = DeviceResponse()
166         val messageId = messageIdInteger.getAndIncrement().toString()
167         log.info("$deviceInfo: closeSession: messageId($messageId)")
168         try {
169             val messageContent = NetconfMessageUtils.closeSession(messageId, force)
170             output = asyncRpc(messageContent, messageId)
171         } catch (e: Exception) {
172             output.status = RpcStatus.FAILURE
173             output.errorMessage = "$deviceInfo: failed in closeSession command " + e.message
174         }
175         return output
176     }
177
178     @Throws(NetconfException::class)
179     override fun asyncRpc(request: String, messageId: String): DeviceResponse {
180         val response = DeviceResponse()
181         log.info("$deviceInfo: send asyncRpc with messageId($messageId)")
182         response.requestMessage = request
183
184         val rpcResponse = netconfSession.asyncRpc(request, messageId).get(responseTimeout.toLong(), TimeUnit.SECONDS)
185         if (!NetconfMessageUtils.checkReply(rpcResponse)) {
186             throw NetconfException(rpcResponse)
187         }
188         response.responseMessage = rpcResponse
189         response.status = RpcStatus.SUCCESS
190         response.errorMessage = null
191         return response
192     }
193 }