1 files changed, 124 insertions, 0 deletions
diff --git a/vnfs/TestVNF/netconfserver/src/main/java/com/ericsson/testvnf/server/helper/ActionHelper.java b/vnfs/TestVNF/netconfserver/src/main/java/com/ericsson/testvnf/server/helper/ActionHelper.java
new file mode 100644
index 00000000..d9714ff7
--- /dev/null
+++ b/vnfs/TestVNF/netconfserver/src/main/java/com/ericsson/testvnf/server/helper/ActionHelper.java
@@ -0,0 +1,124 @@
+/*
+ * ============LICENSE_START=======================================================
+ *  Copyright (C) 2016-2018 Ericsson. All rights reserved.
+ * ================================================================================
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ * 
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ *
+ * SPDX-License-Identifier: Apache-2.0
+ * ============LICENSE_END=========================================================
+ */
+
+package com.ericsson.testvnf.server.helper;
+
+import java.io.OutputStream;
+import java.util.Map;
+
+import org.apache.commons.logging.Log;
+import org.apache.commons.logging.LogFactory;
+
+import com.ericsson.testvnf.server.Server;
+import com.ericsson.testvnf.server.models.NetconfMessage;
+import com.ericsson.testvnf.server.models.RpcData;
+
+/*
+ * The helper class which performs actions based on netconf requests
+ */
+public class ActionHelper {
+	
+	private static final Log log = LogFactory.getLog(ActionHelper.class);
+	private OutputStream					out;
+	private Thread groovyCallerThread;
+	
+	public ActionHelper(OutputStream out){
+		this.out = out;
+	}
+	
+	public boolean doActionForRPCRequest(NetconfMessage netconfMessage, Map<String, Boolean> connectionResetMap, boolean sessionClosedStatus){
+		// if the client is sending an rpc request for any protocol operation
+		RpcData rpcData = (RpcData) netconfMessage;
+		if (null != rpcData.getTargetName()) {
+			// the connectionResetMap is a shared between multiple client connections coming in for a single netconf server instance
+			// true value for a target indicates that sending events to that target needs to be stopped
+			// edit-config operation with a target determines the termination of sending events to that target.
+			if (rpcData.getOperation().equals("edit-config")) {
+				connectionResetMap.put(rpcData.getTargetName(), true);
+			} else {
+				connectionResetMap.put(rpcData.getTargetName(), false);
+			}
+		}
+		if (rpcData.getOperation().equals("\"close-session\"")) {
+			sessionClosedStatus = true;
+		}
+		String operation = rpcData.getOperation();
+		log.info("Received query");
+
+		// on getting a get-schema request from client, the server sends back the yang
+		// model of the particular functionality/capability asked for.
+		if (operation.equals("get-schema")) {
+			doActionForGetSchemaOperation(rpcData);
+		} else {
+			doActionForOperations(rpcData, operation, connectionResetMap);
+		}
+		return sessionClosedStatus;
+	}
+	
+	public void doActionForOperations(RpcData rpcData, String operation, Map<String, Boolean> connectionResetMap) {
+		groovyCallerThread = new Thread("Groovy caller") {
+			@Override
+			public void run() {
+				rpcData.setTimeDelayForSendingEvents(Server.getTimeDelayForSendingEvents());
+				String result ="";
+				// pick up and execute the correct groovy file based on the operation and configuration datastore identified from the rpc request
+				result = HelperUtils.executeGroovy(Server.getNetconfTemplatesDirectory()
+						+ "/netconftemplates/"+operation+"/"+rpcData.getConfigurationDatastore()+"/response.groovy", rpcData, connectionResetMap);
+				if (!result.equals("ignore")) {
+					result = result.replaceAll("<MID>", rpcData.getMessageId());
+					HelperUtils.sendAsBytesToOutput(result.getBytes(), out);
+					log.info("Finished writing " + result);
+				}
+			}
+		};
+		groovyCallerThread.start();
+		log.info("groovyCallerThread started");
+	}
+	
+	public void doActionForGetSchemaOperation(RpcData rpcData){
+		log.info("get-schema received.");
+		if (null != rpcData.getSchemaDetails()
+				&& null != rpcData.getSchemaDetails().getIdentifier()) {
+			log.info("Sending schema for capability...");
+			String schemaString = HelperUtils.readFile(Server.getNetconfTemplatesDirectory()
+					+ "/netconftemplates/"+rpcData.getSchemaDetails().getIdentifier()+"-schema.yang"); // the yang model has to be in the netconftemplates directory with the proper naming format : '<operation>-schema.yang'
+			String schemaStringValue = schemaString.replaceAll("<MID>", rpcData.getMessageId()); // Put the correct message id in the response as it is in the request.
+			HelperUtils.sendAsBytesToOutput(schemaStringValue.getBytes(), out);
+			log.info("Finished writing the schema information");
+		}
+	}
+	
+	// the method sends a hello message to the client, hello.xml in the netconftemplates directory is used to send the hello message
+	public void sendHelloMessage(){
+		log.info("Send hello message.");
+		String helloMessage = HelperUtils.readFile(Server.getNetconfTemplatesDirectory()+"/netconftemplates/hello.xml");
+    	HelperUtils.sendAsBytesToOutput(helloMessage.getBytes(), out);
+    	log.info("Hello message sent.");
+	}
+
+	public void interruptGroovyCallerThread() {
+		// kill thread if it don't finish naturally
+		if (groovyCallerThread!=null && groovyCallerThread.isAlive()) {
+			log.info("Killing groovy caller thread");
+			groovyCallerThread.interrupt();
+		}
+		
+	}
+}