Code re-organization

commands/grpc.go

@@ -0,0 +1,53 @@
+package commands
+
+import (
+	"fmt"
+	"github.com/siteworxpro/img-proxy-url-generator/config"
+	proto "github.com/siteworxpro/img-proxy-url-generator/grpc"
+	"github.com/urfave/cli/v2"
+	"google.golang.org/grpc"
+	"log"
+	"net"
+)
+
+func GrpcCommand() *cli.Command {
+	return &cli.Command{
+		Name:  "grpc",
+		Usage: "Start a grpc service",
+		Flags: []cli.Flag{
+			&cli.IntFlag{
+				Name:     "port",
+				Aliases:  []string{"p"},
+				Usage:    "Port to listen on",
+				Required: false,
+				Value:    9000,
+			},
+		},
+		Action: func(c *cli.Context) error {
+			cfg, err := config.NewConfig(c.String("config"))
+			if err != nil {
+				return err
+			}
+
+			s := grpc.NewServer()
+			addr := fmt.Sprintf(":%d", c.Int("port"))
+			println("listening on", addr)
+			lis, err := net.Listen("tcp", addr)
+			if err != nil {
+				log.Fatalf("failed to listen: %v", err)
+			}
+			svc, err := proto.NewService(cfg)
+			if err != nil {
+				log.Fatalf("failed to serve: %v", err)
+			}
+
+			proto.RegisterGeneratorServer(s, svc)
+			err = s.Serve(lis)
+			if err != nil {
+				log.Fatalf("failed to serve: %v", err)
+			}
+
+			return nil
+		},
+	}
+}