Try something

Author: zasweq

examples/features/customloadbalancer/client/main.go

@@ -28,7 +28,7 @@ import (
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/credentials/insecure"
 	_ "google.golang.org/grpc/examples/features/customloadbalancer/client/customroundrobin" // To register custom_round_robin.
-	"google.golang.org/grpc/examples/features/proto/echo"
+	pb "google.golang.org/grpc/examples/features/proto/echo"
 	"google.golang.org/grpc/internal"
 	"google.golang.org/grpc/peer"
 	"google.golang.org/grpc/resolver"
@@ -64,7 +64,7 @@ func main() {
 	}
 	ctx, cancel := context.WithTimeout(context.Background(), time.Second*10)
 	defer cancel()
-	ec := echo.NewEchoClient(cc)
+	ec := pb.NewEchoClient(cc)
 	if err := waitForDistribution(ctx, ec); err != nil {
 		log.Fatalf(err.Error())
 	}
@@ -74,7 +74,7 @@ func main() {
 // waitForDistribution makes RPC's on the echo client until 3 RPC's follow the
 // same 1:2 address ratio for the peer. Returns an error if fails to do so
 // before context timeout.
-func waitForDistribution(ctx context.Context, ec echo.EchoClient) error {
+func waitForDistribution(ctx context.Context, ec pb.EchoClient) error {
 	for {
 		results := make(map[string]uint32)
 	InnerLoop:
@@ -87,7 +87,7 @@ func waitForDistribution(ctx context.Context, ec echo.EchoClient) error {
 				res := make(map[string]uint32)
 				for j := 0; j < 3; j++ {
 					var peer peer.Peer
-					r, err := ec.UnaryEcho(ctx, &echo.EchoRequest{Message: "this is examples/customloadbalancing"}, grpc.Peer(&peer))
+					r, err := ec.UnaryEcho(ctx, &pb.EchoRequest{Message: "this is examples/customloadbalancing"}, grpc.Peer(&peer))
 					if err != nil {
 						return fmt.Errorf("UnaryEcho failed: %v", err)
 					}

examples/features/customloadbalancer/server/main.go

@@ -26,20 +26,20 @@ import (
 	"sync"
 
 	"google.golang.org/grpc"
-	"google.golang.org/grpc/examples/features/proto/echo"
+	pb "google.golang.org/grpc/examples/features/proto/echo"
 )
 
 var (
 	addrs = []string{"localhost:20000", "localhost:20001"}
 )
 
 type echoServer struct {
-	echo.UnimplementedEchoServer
+	pb.UnimplementedEchoServer
 	addr string
 }
 
-func (s *echoServer) UnaryEcho(ctx context.Context, req *echo.EchoRequest) (*echo.EchoResponse, error) {
-	return &echo.EchoResponse{Message: fmt.Sprintf("%s (from %s)", req.Message, s.addr)}, nil
+func (s *echoServer) UnaryEcho(ctx context.Context, req *pb.EchoRequest) (*pb.EchoResponse, error) {
+	return &pb.EchoResponse{Message: fmt.Sprintf("%s (from %s)", req.Message, s.addr)}, nil
 }
 
 func main() {
@@ -50,7 +50,7 @@ func main() {
 			log.Fatalf("failed to listen: %v", err)
 		}
 		s := grpc.NewServer()
-		echo.RegisterEchoServer(s, &echoServer{
+		pb.RegisterEchoServer(s, &echoServer{
 			addr: addr,
 		})
 		log.Printf("serving on %s\n", addr)