Merge "InstallApp Grpc proto and server for rsync"

1 files changed, 58 insertions, 2 deletions

diff --git a/src/rsync/cmd/main.go b/src/rsync/cmd/main.go
index 6e30b60b..ce98edd2 100644
--- a/src/rsync/cmd/main.go
+++ b/src/rsync/cmd/main.go
@@ -17,11 +17,63 @@ import (
 	"fmt"
 	"log"
 	"math/rand"
+	"net"
+	register "rsync/pkg/grpc"
+	installpb "rsync/pkg/grpc/installapp"
+	"rsync/pkg/grpc/installappserver"
+	"strings"
 	"time"
+
+	"github.com/onap/multicloud-k8s/src/orchestrator/pkg/infra/config"
 	contextDb "github.com/onap/multicloud-k8s/src/orchestrator/pkg/infra/contextdb"
 	"github.com/onap/multicloud-k8s/src/orchestrator/pkg/infra/db"
+	"google.golang.org/grpc"
+	"google.golang.org/grpc/credentials"
+	"google.golang.org/grpc/testdata"
 )
 
+func startGrpcServer() error {
+	var tls bool
+
+	if strings.Contains(config.GetConfiguration().GrpcEnableTLS, "enable") {
+		tls = true
+	} else {
+		tls = false
+	}
+	certFile := config.GetConfiguration().GrpcServerCert
+	keyFile := config.GetConfiguration().GrpcServerKey
+
+	host, port := register.GetServerHostPort()
+
+	lis, err := net.Listen("tcp", fmt.Sprintf("%s:%d", host, port))
+	if err != nil {
+		log.Fatalf("Could not listen to port: %v", err)
+	}
+	var opts []grpc.ServerOption
+	if tls {
+		if certFile == "" {
+			certFile = testdata.Path("server.pem")
+		}
+		if keyFile == "" {
+			keyFile = testdata.Path("server.key")
+		}
+		creds, err := credentials.NewServerTLSFromFile(certFile, keyFile)
+		if err != nil {
+			log.Fatalf("Could not generate credentials %v", err)
+		}
+		opts = []grpc.ServerOption{grpc.Creds(creds)}
+	}
+	grpcServer := grpc.NewServer(opts...)
+	installpb.RegisterInstallappServer(grpcServer, installappserver.NewInstallAppServer())
+
+	log.Println("Starting rsync gRPC Server")
+	err = grpcServer.Serve(lis)
+	if err != nil {
+		log.Fatalf("rsync grpc server is not serving %v", err)
+	}
+	return err
+}
+
 func main() {
 
 	rand.Seed(time.Now().UnixNano())
@@ -44,6 +96,10 @@ func main() {
 		log.Fatalln("Exiting...")
 	}
 
-	// Initialize grpc
-
+	// Start grpc
+	fmt.Println("starting rsync GRPC server..")
+	err = startGrpcServer()
+	if err != nil {
+		log.Fatalf("GRPC server failed to start")
+	}
 }