Save changes.

Author: kannanjgithub

rls/src/main/java/io/grpc/rls/RlsProtoConverters.java

@@ -64,6 +64,7 @@ static final class RouteLookupRequestConverter
     @Override
     protected RlsProtoData.RouteLookupRequest doForward(RouteLookupRequest routeLookupRequest) {
       return RlsProtoData.RouteLookupRequest.create(
+          RlsProtoData.RouteLookupRequest.Reason.valueOf(routeLookupRequest.getReason().name()),
           ImmutableMap.copyOf(routeLookupRequest.getKeyMapMap()));
     }
 
@@ -72,6 +73,7 @@ protected RouteLookupRequest doBackward(RlsProtoData.RouteLookupRequest routeLoo
       return
           RouteLookupRequest.newBuilder()
               .setTargetType("grpc")
+              .setReason(RouteLookupRequest.Reason.valueOf(routeLookupRequest.reason().name()))
               .putAllKeyMap(routeLookupRequest.keyMap())
               .build();
     }

rls/src/main/java/io/grpc/rls/RlsProtoData.java

@@ -32,11 +32,19 @@ private RlsProtoData() {}
   @Immutable
   abstract static class RouteLookupRequest {
 
+    // Names should match those in grpc.lookup.v1.RouteLookupRequest.Reason
+    enum Reason {
+      REASON_UNKNOWN,  // Unused
+      REASON_MISS,     // No data available in local cache
+      REASON_STALE;    // Data in local cache is stale
+    }
+    // Reason for making this request.
+    abstract Reason reason();
     /** Returns a map of key values extracted via key builders for the gRPC or HTTP request. */
     abstract ImmutableMap<String, String> keyMap();
 
-    static RouteLookupRequest create(ImmutableMap<String, String> keyMap) {
-      return new AutoValue_RlsProtoData_RouteLookupRequest(keyMap);
+    static RouteLookupRequest create(Reason reason, ImmutableMap<String, String> keyMap) {
+      return new AutoValue_RlsProtoData_RouteLookupRequest(reason, keyMap);
     }
   }