VTP: grpc server impl

Issue-ID: VNFSDK-304

Change-Id: I27c03f931d25615d6de0bb85a4c506c842b069f0
Signed-off-by: Kanagaraj Manickam k00365106 <kanagaraj.manickam@huawei.com>

3 files changed, 203 insertions, 0 deletions

diff --git a/grpc/grpc-server/src/main/java/org/open/infc/grpc/server/OpenInterfaceGrpcServer.java b/grpc/grpc-server/src/main/java/org/open/infc/grpc/server/OpenInterfaceGrpcServer.java
new file mode 100644
index 00000000..0f389f31
--- /dev/null
+++ b/grpc/grpc-server/src/main/java/org/open/infc/grpc/server/OpenInterfaceGrpcServer.java
@@ -0,0 +1,156 @@
+/*
+ * Copyright 2018 Huawei Technologies Co., Ltd.
+ *
+ * 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.
+ */
+
+package org.open.infc.grpc.server;
+
+import java.io.IOException;
+import java.util.Map.Entry;
+import java.util.logging.Logger;
+
+import org.onap.cli.fw.cmd.OnapCommand;
+import org.onap.cli.fw.conf.OnapCommandConfig;
+import org.onap.cli.fw.conf.OnapCommandConstants;
+import org.onap.cli.fw.error.OnapCommandException;
+import org.onap.cli.fw.output.OnapCommandResultType;
+import org.onap.cli.fw.registrar.OnapCommandRegistrar;
+import org.onap.cli.main.OnapCli;
+import org.open.infc.grpc.Args;
+import org.open.infc.grpc.Input;
+import org.open.infc.grpc.OpenInterfaceGrpc;
+import org.open.infc.grpc.Output;
+import org.open.infc.grpc.Output.Builder;
+import org.open.infc.grpc.Result;
+
+import io.grpc.Server;
+import io.grpc.ServerBuilder;
+import io.grpc.stub.StreamObserver;
+
+public class OpenInterfaceGrpcServer {
+
+      private static final Logger logger = Logger.getLogger(OpenInterfaceGrpcServer.class.getName());
+
+      private static final String CONF_FILE = "oclip-grpc-server.properties";
+      private static final String CONF_SERVER_PORT = "oclip.grpc_server_port";
+
+      static {
+          OnapCommandConfig.addProperties(CONF_FILE);
+      }
+      private Server server;
+
+      private void start() throws IOException {
+        /* The port on which the server should run */
+        int port = Integer.parseInt(OnapCommandConfig.getPropertyValue(CONF_SERVER_PORT));
+        server = ServerBuilder.forPort(port)
+            .addService(new OpenInterfaceGrpcImpl())
+            .build()
+            .start();
+        logger.info("Server started, listening on " + port);
+        Runtime.getRuntime().addShutdownHook(new Thread() {
+          @Override
+          public void run() {
+            // Use stderr here since the logger may have been reset by its JVM shutdown hook.
+            System.err.println("*** shutting down gRPC server since JVM is shutting down");
+            OpenInterfaceGrpcServer.this.stop();
+            System.err.println("*** server shut down");
+          }
+        });
+      }
+
+      private void stop() {
+        if (server != null) {
+          server.shutdown();
+        }
+      }
+
+      /**
+       * Await termination on the main thread since the grpc library uses daemon threads.
+       */
+      private void blockUntilShutdown() throws InterruptedException {
+        if (server != null) {
+          server.awaitTermination();
+        }
+      }
+
+      /**
+       * Main launches the server from the command line.
+       */
+      public static void main(String[] args) throws IOException, InterruptedException {
+        final OpenInterfaceGrpcServer server = new OpenInterfaceGrpcServer();
+        server.start();
+        server.blockUntilShutdown();
+      }
+
+      static class OpenRemoteCli extends OnapCli {
+          private String outputs = "";
+          public OpenRemoteCli(String product, String[] args) {
+            super(product, args);
+          }
+
+          public void print(String msg) {
+              outputs += msg + "\n";
+          }
+
+          public String getResult() {
+              return outputs;
+          }
+      }
+
+      static class OpenInterfaceGrpcImpl extends OpenInterfaceGrpc.OpenInterfaceImplBase {
+
+        @Override
+        public void invoke(Input req, StreamObserver<Output> responseObserver) {
+            Builder reply = Output.newBuilder();
+            logger.info(req.toString());
+
+            String product = req.getOptionsMap().get(OnapCommandConstants.OPEN_CLI_PRODUCT_NAME);
+            String format =  req.getOptionsMap().getOrDefault(OnapCommandConstants.DEFAULT_PARAMETER_OUTPUT_FORMAT, OnapCommandResultType.JSON.name().toLowerCase());
+            String command = req.getAction();
+
+            try {
+                OnapCommand cmd = OnapCommandRegistrar.getRegistrar().get(command, product);
+                cmd.getParametersMap().get(OnapCommandConstants.DEFAULT_PARAMETER_OUTPUT_FORMAT).setValue(format);
+                for (Entry<String, String> arg: req.getParams().entrySet()) {
+                    cmd.getParametersMap().get(arg.getKey()).setValue(arg.getValue());
+                }
+                cmd.execute();
+
+                reply.putAttrs(OnapCommandConstants.RESULTS, cmd.getResult().print());
+                reply.setSuccess(true);
+                reply.putAttrs(OnapCommandConstants.ERROR, "{}");
+            } catch (OnapCommandException e) {
+                logger.info(e.getMessage());
+                reply.putAttrs(OnapCommandConstants.RESULTS, "{}");
+                reply.setSuccess(false);
+                reply.putAttrs(OnapCommandConstants.ERROR, e.toJsonString());
+            }
+
+            responseObserver.onNext(reply.build());
+            responseObserver.onCompleted();
+        }
+
+        @Override
+        public void remoteCli(Args req, StreamObserver<Result> responseObserver) {
+            logger.info(req.toString());
+
+            OpenRemoteCli cli = new OpenRemoteCli(req.getProduct(), req.getArgsList().toArray(new String [] {}));
+            cli.handle();
+
+            Result reply = Result.newBuilder().setExitCode(cli.getExitCode()).setOutput(cli.getResult()).build();
+            responseObserver.onNext(reply);
+            responseObserver.onCompleted();
+        }
+      }
+}
diff --git a/grpc/grpc-server/src/main/resources/log4j.properties b/grpc/grpc-server/src/main/resources/log4j.properties
new file mode 100644
index 00000000..29a491cd
--- /dev/null
+++ b/grpc/grpc-server/src/main/resources/log4j.properties
@@ -0,0 +1,32 @@
+# Copyright 2018 Huawei Technologies Co., Ltd.
+#
+# 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.
+
+log4j.rootLogger=ERROR, file
+
+log4j.logger.org.onap.cli=DEBUG, file, stdout
+log4j.logger.org.open.infc=DEBUG, file, stdout
+
+# Direct log messages to stdout
+log4j.appender.stdout=org.apache.log4j.ConsoleAppender
+log4j.appender.stdout.Target=System.out
+log4j.appender.stdout.layout=org.apache.log4j.PatternLayout
+log4j.appender.stdout.layout.ConversionPattern=%d{yyyy-MM-dd HH:mm:ss} %-5p %c{1}:%L - %m%n
+
+# Redirect log messages to a log file, support file rolling.
+log4j.appender.file=org.apache.log4j.RollingFileAppender
+log4j.appender.file.File=${OPEN_CLI_HOME}/logs/open-cli.log
+log4j.appender.file.MaxFileSize=5MB
+log4j.appender.file.MaxBackupIndex=10
+log4j.appender.file.layout=org.apache.log4j.PatternLayout
+log4j.appender.file.layout.ConversionPattern=%d{yyyy-MM-dd HH:mm:ss} %-5p %c{1}:%L - %m%n
diff --git a/grpc/grpc-server/src/main/resources/oclip-grpc-server.properties b/grpc/grpc-server/src/main/resources/oclip-grpc-server.properties
new file mode 100644
index 00000000..972fdebb
--- /dev/null
+++ b/grpc/grpc-server/src/main/resources/oclip-grpc-server.properties
@@ -0,0 +1,15 @@
+# Copyright 2018 Huawei Technologies Co., Ltd.
+#
+# 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.
+
+oclip.grpc_server_port=50051
\ No newline at end of file