grpc test

Author: irr

grpc/book.proto

@@ -0,0 +1,11 @@
+syntax = "proto3";
+package book;
+
+message Book {
+    string name = 1;
+    int32 isbn = 2;
+}
+
+service BookService {
+    rpc GetBook(Book) returns (Book) {}
+}

grpc/book/book.go

@@ -0,0 +1,22 @@
+package book
+
+import (
+	log "github.com/sirupsen/logrus"
+
+	"golang.org/x/net/context"
+)
+
+// Server interface for our service methods
+type Server struct {
+}
+
+// GetBook logs Book from client and returns new Book
+func (s *Server) GetBook(ctx context.Context, input *Book) (*Book, error) {
+
+	log.WithFields(log.Fields{
+		"Name": input.Name,
+		"Isbn": input.Isbn,
+	}).Info("Book data received from client")
+
+	return &Book{Name: "The Great Gatsby", Isbn: 90393}, nil
+}

grpc/book/book.pb.go

@@ -0,0 +1,15 @@
+module grpc_using_go
+
+go 1.15
+
+require (
+	github.com/fullstorydev/grpcurl v1.7.0 // indirect
+	github.com/golang/protobuf v1.4.3
+	github.com/sirupsen/logrus v1.7.0
+	golang.org/x/net v0.0.0-20201202161906-c7110b5ffcbb
+	golang.org/x/sys v0.0.0-20201204225414-ed752295db88 // indirect
+	golang.org/x/text v0.3.4 // indirect
+	google.golang.org/genproto v0.0.0-20201204160425-06b3db808446 // indirect
+	google.golang.org/grpc v1.34.0
+	google.golang.org/protobuf v1.25.0
+)

grpc/go.mod

@@ -0,0 +1,69 @@
+package main
+
+import (
+	"net"
+	"os"
+
+	"grpc_using_go/book"
+
+	log "github.com/sirupsen/logrus"
+	"google.golang.org/grpc"
+	"google.golang.org/grpc/reflection"
+)
+
+/*
+sudo apt install protobuf-compiler
+go get -u github.com/golang/protobuf/protoc-gen-go
+
+go install github.com/fullstorydev/grpcurl/cmd/grpcurl
+*/
+
+func main() {
+	log.SetFormatter(&log.TextFormatter{
+		FullTimestamp: true,
+	})
+
+	var port string
+	var ok bool
+	port, ok = os.LookupEnv("PORT")
+	if ok {
+		log.WithFields(log.Fields{
+			"PORT": port,
+		}).Info("PORT env var defined")
+
+	} else {
+		port = "9000"
+		log.WithFields(log.Fields{
+			"PORT": port,
+		}).Warn("PORT env var not defined. Going with default")
+
+	}
+
+	l, err := net.Listen("tcp", ":"+port)
+	if err != nil {
+		log.WithFields(log.Fields{
+			"Error": err.Error(),
+		}).Fatal("Failed to listen")
+	}
+
+	s := book.Server{}
+
+	grpcServer := grpc.NewServer()
+
+	/*
+		grpcurl -plaintext localhost:9000 list
+		grpcurl -plaintext localhost:9000 describe
+		grpcurl -d '{"name": "To Kill a Mockingbird", "isbn": 12345}' -plaintext localhost:9000 book.BookService/GetBook
+	*/
+	reflection.Register(grpcServer)
+
+	book.RegisterBookServiceServer(grpcServer, &s)
+
+	log.Info("gRPC server started at ", port)
+	if err := grpcServer.Serve(l); err != nil {
+		log.WithFields(log.Fields{
+			"Error": err.Error(),
+		}).Fatal("Failed to serve")
+	}
+
+}