From 67b54608dae5739e91f0bfabd1fbfe41b580e604 Mon Sep 17 00:00:00 2001
From: Chengyuan Zhang <chengyuanzhang@google.com>
Date: Wed, 14 Oct 2020 13:37:16 -0700
Subject: [PATCH] alts: migrate java proto map getter from get<field> to
 get<field>Map (#7522)

Migrate java proto map getter from get to getMap.

This is part of a set of changes to java proto map API described here: go/java-proto-maplike

More information: go/java-proto-maplike-getFooMap
---
 .../main/java/io/grpc/alts/internal/AltsAuthContext.java    | 2 +-
 .../grpc/xds/internal/rbac/engine/AuthorizationEngine.java  | 6 +++---
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/alts/src/main/java/io/grpc/alts/internal/AltsAuthContext.java b/alts/src/main/java/io/grpc/alts/internal/AltsAuthContext.java
index b35d4faf98..6ec7410003 100644
--- a/alts/src/main/java/io/grpc/alts/internal/AltsAuthContext.java
+++ b/alts/src/main/java/io/grpc/alts/internal/AltsAuthContext.java
@@ -34,7 +34,7 @@ public final class AltsAuthContext {
             .setPeerServiceAccount(result.getPeerIdentity().getServiceAccount())
             .setLocalServiceAccount(result.getLocalIdentity().getServiceAccount())
             .setPeerRpcVersions(result.getPeerRpcVersions())
-            .putAllPeerAttributes(result.getPeerIdentity().getAttributes())
+            .putAllPeerAttributes(result.getPeerIdentity().getAttributesMap())
             .build();
   }
 
diff --git a/xds/src/main/java/io/grpc/xds/internal/rbac/engine/AuthorizationEngine.java b/xds/src/main/java/io/grpc/xds/internal/rbac/engine/AuthorizationEngine.java
index a7e09b0f3f..9ed439953a 100644
--- a/xds/src/main/java/io/grpc/xds/internal/rbac/engine/AuthorizationEngine.java
+++ b/xds/src/main/java/io/grpc/xds/internal/rbac/engine/AuthorizationEngine.java
@@ -85,7 +85,7 @@ public class AuthorizationEngine {
    */
   public AuthorizationEngine(RBAC rbacPolicy) {
     Map<String, Expr> conditions = new LinkedHashMap<>();
-    for (Map.Entry<String, Policy> policy: rbacPolicy.getPolicies().entrySet()) {
+    for (Map.Entry<String, Policy> policy: rbacPolicy.getPoliciesMap().entrySet()) {
       conditions.put(policy.getKey(), policy.getValue().getCondition());
     }
     allowEngine = (rbacPolicy.getAction() == Action.ALLOW) 
@@ -108,12 +108,12 @@ public class AuthorizationEngine {
         "Invalid RBAC list, " 
         + "must provide a RBAC with DENY action followed by a RBAC with ALLOW action. ");
     Map<String, Expr> denyConditions = new LinkedHashMap<>();
-    for (Map.Entry<String, Policy> policy: denyPolicy.getPolicies().entrySet()) {
+    for (Map.Entry<String, Policy> policy: denyPolicy.getPoliciesMap().entrySet()) {
       denyConditions.put(policy.getKey(), policy.getValue().getCondition());
     }
     denyEngine = new RbacEngine(Action.DENY, ImmutableMap.copyOf(denyConditions));
     Map<String, Expr> allowConditions = new LinkedHashMap<>();
-    for (Map.Entry<String, Policy> policy: allowPolicy.getPolicies().entrySet()) {
+    for (Map.Entry<String, Policy> policy: allowPolicy.getPoliciesMap().entrySet()) {
       allowConditions.put(policy.getKey(), policy.getValue().getCondition());
     }
     allowEngine = new RbacEngine(Action.ALLOW, ImmutableMap.copyOf(allowConditions));