add --empty-package-service

Makefile

@@ -3,3 +3,4 @@ SHELL := /bin/bash
 .PHONY: proto
 proto:
 	protoc --proto_path=$(GOPATH)/src --proto_path api --go_out=plugins=grpc:api api/api.proto
+	protoc --proto_path=$(GOPATH)/src --proto_path api --go_out=plugins=grpc:api/emptypackage api/emptypackage.proto

api/emptypackage.proto

@@ -0,0 +1,13 @@
+syntax = "proto3";
+
+service EmptyPackageService {
+  rpc Unary (SimpleRequest) returns (SimpleResponse) {}
+}
+
+message SimpleRequest {
+  string name = 1;
+}
+
+message SimpleResponse {
+  string message = 1;
+}

main.go

@@ -17,6 +17,7 @@ func main() {
 	cert := pflag.String("cert", "", "cert")
 	certKey := pflag.String("cert-key", "", "cert key")
 	rootCACert := pflag.String("root-ca-cert", "", "cert key")
+	emptyPackageService := pflag.Bool("empty-package-service", false, "register empty package service")
 	pflag.Parse()
 
 	var opts []server.Option
@@ -41,6 +42,9 @@ func main() {
 	if *rootCACert != "" {
 		opts = append(opts, server.WithRootCACert(*rootCACert))
 	}
+	if *emptyPackageService {
+		opts = append(opts, server.WithEmptyPackageService())
+	}
 
 	defer server.New(opts...).Serve().Stop()
 

server/emptypackage.go

@@ -0,0 +1,19 @@
+package server
+
+import (
+	"context"
+	"fmt"
+	"log"
+
+	"github.com/ktr0731/grpc-test/api/emptypackage"
+)
+
+type EmptyPackageService struct {
+	logger *log.Logger
+}
+
+func (s *EmptyPackageService) Unary(ctx context.Context, req *emptypackage.SimpleRequest) (*emptypackage.SimpleResponse, error) {
+	return &emptypackage.SimpleResponse{
+		Message: fmt.Sprintf("hello, %s", req.GetName()),
+	}, nil
+}

server/option.go

@@ -8,18 +8,19 @@ type opt struct {
 	protocol Protocol
 	addr     string
 
-	tls           bool
-	cert, certKey string
-	rootCACert    string
-	reflection    bool
-	verbose       bool
-	compressor    string
+	tls                 bool
+	cert, certKey       string
+	rootCACert          string
+	reflection          bool
+	verbose             bool
+	compressor          string
+	emptyPackageService bool
 }
 
 type Option func(*opt)
 
 var defaultOption = opt{
-	addr: ":50051",
+	addr: "0.0.0.0:50051",
 }
 
 type Protocol int
@@ -81,3 +82,9 @@ func WithCompressor(c string) Option {
 		o.compressor = c
 	}
 }
+
+func WithEmptyPackageService() Option {
+	return func(o *opt) {
+		o.emptyPackageService = true
+	}
+}

server/server.go

@@ -14,6 +14,7 @@ import (
 	"github.com/improbable-eng/grpc-web/go/grpcweb"
 	"github.com/ktr0731/dept/logger"
 	"github.com/ktr0731/grpc-test/api"
+	"github.com/ktr0731/grpc-test/api/emptypackage"
 	"github.com/rakyll/statik/fs"
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/credentials"
@@ -80,6 +81,9 @@ func New(opts ...Option) *Server {
 
 	s := grpc.NewServer(grpcOpts...)
 	api.RegisterExampleServer(s, &ExampleService{logger: logger})
+	if opt.emptyPackageService {
+		emptypackage.RegisterEmptyPackageServiceServer(s, &EmptyPackageService{logger: logger})
+	}
 
 	if opt.reflection {
 		reflection.Register(s)