Skip to content

Commit

Permalink
rls: double import rls protos (#5003)
Browse files Browse the repository at this point in the history
Use `rlspb` for messages and `rlsgrpc` for services
  • Loading branch information
easwars committed Nov 29, 2021
1 parent d542bfc commit 6f8796b
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions balancer/rls/internal/client.go
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@ import (
"time"

"google.golang.org/grpc"
rlsgrpc "google.golang.org/grpc/internal/proto/grpc_lookup_v1"
rlspb "google.golang.org/grpc/internal/proto/grpc_lookup_v1"
)

Expand All @@ -43,7 +44,7 @@ const grpcTargetType = "grpc"
// throttling and asks this client to make an RPC call only after checking with
// the throttler.
type rlsClient struct {
stub rlspb.RouteLookupServiceClient
stub rlsgrpc.RouteLookupServiceClient
// origDialTarget is the original dial target of the user and sent in each
// RouteLookup RPC made to the RLS server.
origDialTarget string
Expand All @@ -54,7 +55,7 @@ type rlsClient struct {

func newRLSClient(cc *grpc.ClientConn, dialTarget string, rpcTimeout time.Duration) *rlsClient {
return &rlsClient{
stub: rlspb.NewRouteLookupServiceClient(cc),
stub: rlsgrpc.NewRouteLookupServiceClient(cc),
origDialTarget: dialTarget,
rpcTimeout: rpcTimeout,
}
Expand Down

0 comments on commit 6f8796b

Please sign in to comment.