Add grpc reflection

HOU-SZ · Jul 10, 2022 · e98c2e2 · e98c2e2
1 parent 1a1f853
commit e98c2e2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/cmd/server/main.go b/cmd/server/main.go
@@ -9,6 +9,7 @@ import (
 	"github.com/HOU-SZ/pcbook/pb"
 	"github.com/HOU-SZ/pcbook/service"
 	"google.golang.org/grpc"
+	"google.golang.org/grpc/reflection"
 )
 
 func main() {
@@ -22,6 +23,8 @@ func main() {
 	laptopServer := service.NewLaptopServer(laptopStore, imageStore, ratingStore)
 	grpcServer := grpc.NewServer()
 	pb.RegisterLaptopServiceServer(grpcServer, laptopServer)
+	// Register reflection service on gRPC server.
+	reflection.Register(grpcServer)
 
 	address := fmt.Sprintf("0.0.0.0:%d", *port)
 	listener, err := net.Listen("tcp", address)