You've already forked img-proxy-url-generator
54 lines
1.1 KiB
Go
54 lines
1.1 KiB
Go
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
|
|
},
|
|
}
|
|
}
|