rls: double import rls protos (#5003)

Use `rlspb` for messages and `rlsgrpc` for services
This commit is contained in:
Easwar Swaminathan 2021-11-29 14:27:43 -08:00 committed by GitHub
parent d542bfcee4
commit 6f8796bc00
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions

View File

@ -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"
)
@ -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
@ -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,
}