ECS Service Connect で gRPC のロードバランシングを試してみた

package main

import (
    "context"
    "log"
    "net"

    "google.golang.org/grpc"
    "google.golang.org/grpc/reflection"

    pb "google.golang.org/grpc/examples/helloworld/helloworld"
)

type server struct{
    pb.UnimplementedGreeterServer
}

func (s *server) SayHello(ctx context.Context, in *pb.HelloRequest) (*pb.HelloReply, error) {
	log.Printf("Received: %v", in.Name)
	return &pb.HelloReply{Message: "Hello " + in.Name}, nil
}
func main() {
    lis, err := net.Listen("tcp", ":50051")
    if err != nil {
        log.Fatalf("failed to listen: %v", err)
    }
    s := grpc.NewServer()
    pb.RegisterGreeterServer(s, &server{})
    reflection.Register(s)
    log.Printf("serving on :50051\n")
    if err := s.Serve(lis); err != nil {
        log.Fatalf("failed to serve: %v", err)
    }
}

package main

import (
    "context"
    "log"
    "os"
    "time"
    "strconv"

    "google.golang.org/grpc"
    "google.golang.org/grpc/credentials/insecure"
    pb "google.golang.org/grpc/examples/helloworld/helloworld"
)

func main() {
    // gRPCサーバーに接続
    grpcTarget := os.Getenv("GRPC_TARGET")
    if grpcTarget == "" {
        log.Fatal("GRPC_TARGET environment variable is required")
    }
    conn, err := grpc.Dial(
        grpcTarget,
        grpc.WithTransportCredentials(insecure.NewCredentials()),
    )
    if err != nil {
        log.Fatalf("did not connect: %v", err)
    }
    defer conn.Close()

    // クライアントを作成
    c := pb.NewGreeterClient(conn)
    ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
    defer cancel()

    // リクエストを送信
    name := os.Getenv("NAME")
    if name == "" {
        log.Fatal("NAME environment variable is required")
    }
    iterationStr := os.Getenv("ITERATION")
    iteration, err := strconv.Atoi(iterationStr)
    if err != nil {
        log.Fatalf("Failed to parse Iteration environment variable: %v", err)
    }
    for i := 0; i < iteration; i++ {
        reply, err := c.SayHello(ctx, &pb.HelloRequest{Name: name + "(" +strconv.Itoa(i) + ")"})
      if err != nil {
      log.Fatalf("リクエスト失敗: %v", err)
      }
        log.Printf("サーバーからの応答: %s", reply.GetMessage())
    }
}

    conn, err := grpc.Dial(
        grpcTarget,
        grpc.WithTransportCredentials(insecure.NewCredentials()),
    )