server_better.go

package main

import (
	"fmt"
	"net"

	"golang.org/x/net/context"
	"google.golang.org/grpc"
	"google.golang.org/grpc/grpclog"

	pb "github.com/ppg/grpc-intro/proto"
)

// START SERVICE IMPLEMENTATION OMIT
type testBetterService struct {
}

func (s *testBetterService) Add(ctx context.Context, req *pb.BetterNumericRequest) (*pb.BetterNumericResponse, error) {
	var sum int32
	for _, v := range req.Values {
		grpclog.Printf(" + %v", v)
		sum += v
	}
	prefix := fmt.Sprintf("This is type %d", req.Type)
	if req.Type == pb.TestType_TYPE_1 {
		prefix = "Everything is awesome"
	}
	return &pb.BetterNumericResponse{Prefix: prefix, R: sum}, nil
}

// END SERVICE IMPLEMENTATION OMIT

func main() {
	// Create a TCP listener like you normally would
	listener, err := net.Listen("tcp", ":1234")
	if err != nil {
		grpclog.Fatalf("failed to listen: %v", err)
	}

	// Create a gRPC server and attach our service to it
	server := grpc.NewServer()
	pb.RegisterBetterTestServer(server, &testBetterService{})

	// Start handling requests
	grpclog.Printf("Starting to listen on %s", listener.Addr())
	err = server.Serve(listener)
	if err != nil {
		grpclog.Fatalf("failed to serve: %v", err)
	}
}