github.com/erda-project/erda-infra@v1.0.9/examples/service/protocol/client/client.go (about)

     1  // Code generated by protoc-gen-go-client. DO NOT EDIT.
     2  // Sources: greeter.proto, user.proto
     3  
     4  package client
     5  
     6  import (
     7  	context "context"
     8  
     9  	pb "github.com/erda-project/erda-infra/examples/service/protocol/pb"
    10  	grpc "github.com/erda-project/erda-infra/pkg/transport/grpc"
    11  	grpc1 "google.golang.org/grpc"
    12  )
    13  
    14  // Client provide all service clients.
    15  type Client interface {
    16  	// GreeterService greeter.proto
    17  	GreeterService() pb.GreeterServiceClient
    18  	// UserService user.proto
    19  	UserService() pb.UserServiceClient
    20  }
    21  
    22  // New create client
    23  func New(cc grpc.ClientConnInterface) Client {
    24  	return &serviceClients{
    25  		greeterService: pb.NewGreeterServiceClient(cc),
    26  		userService:    pb.NewUserServiceClient(cc),
    27  	}
    28  }
    29  
    30  type serviceClients struct {
    31  	greeterService pb.GreeterServiceClient
    32  	userService    pb.UserServiceClient
    33  }
    34  
    35  func (c *serviceClients) GreeterService() pb.GreeterServiceClient {
    36  	return c.greeterService
    37  }
    38  
    39  func (c *serviceClients) UserService() pb.UserServiceClient {
    40  	return c.userService
    41  }
    42  
    43  type greeterServiceWrapper struct {
    44  	client pb.GreeterServiceClient
    45  	opts   []grpc1.CallOption
    46  }
    47  
    48  func (s *greeterServiceWrapper) SayHello(ctx context.Context, req *pb.HelloRequest) (*pb.HelloResponse, error) {
    49  	return s.client.SayHello(ctx, req, append(grpc.CallOptionFromContext(ctx), s.opts...)...)
    50  }
    51  
    52  type userServiceWrapper struct {
    53  	client pb.UserServiceClient
    54  	opts   []grpc1.CallOption
    55  }
    56  
    57  func (s *userServiceWrapper) GetUser(ctx context.Context, req *pb.GetUserRequest) (*pb.GetUserResponse, error) {
    58  	return s.client.GetUser(ctx, req, append(grpc.CallOptionFromContext(ctx), s.opts...)...)
    59  }
    60  
    61  func (s *userServiceWrapper) UpdateUser(ctx context.Context, req *pb.UpdateUserRequest) (*pb.UpdateUserResponse, error) {
    62  	return s.client.UpdateUser(ctx, req, append(grpc.CallOptionFromContext(ctx), s.opts...)...)
    63  }