From 295b64b5ff03cf7ec87ec86408436040fa26446d Mon Sep 17 00:00:00 2001
From: Chengyuan Zhang <chengyuanzhang@google.com>
Date: Fri, 31 Jan 2020 17:50:04 -0800
Subject: [PATCH] grpclb: expose balancer address related attributes in
 internal accessor (#6669)

---
 .../InternalGrpclbConstantsAccessor.java       | 18 +++++-------------
 1 file changed, 5 insertions(+), 13 deletions(-)

diff --git a/grpclb/src/main/java/io/grpc/grpclb/InternalGrpclbConstantsAccessor.java b/grpclb/src/main/java/io/grpc/grpclb/InternalGrpclbConstantsAccessor.java
index 298add4b73..3360a701b0 100644
--- a/grpclb/src/main/java/io/grpc/grpclb/InternalGrpclbConstantsAccessor.java
+++ b/grpclb/src/main/java/io/grpc/grpclb/InternalGrpclbConstantsAccessor.java
@@ -32,18 +32,10 @@ public class InternalGrpclbConstantsAccessor {
   private InternalGrpclbConstantsAccessor() {
   }
 
-  /**
-   * Sets attribute for gRPC LB address authority.
-   */
-  public static Attributes setLbAddrAuthorityAttr(
-      @EquivalentAddressGroup.Attr Attributes attrs, String authority) {
-    return attrs.toBuilder().set(GrpclbConstants.ATTR_LB_ADDR_AUTHORITY, authority).build();
-  }
+  public static Attributes.Key<List<EquivalentAddressGroup>> ATTR_LB_ADDRS =
+      GrpclbConstants.ATTR_LB_ADDRS;
 
-  /**
-   * Sets attribute for gRPC LB addresses.
-   */
-  public static Attributes setLbAddrAttr(Attributes attrs, List<EquivalentAddressGroup> lbAddrs) {
-    return attrs.toBuilder().set(GrpclbConstants.ATTR_LB_ADDRS, lbAddrs).build();
-  }
+  @EquivalentAddressGroup.Attr
+  public static Attributes.Key<String> ATTR_LB_ADDR_AUTHORITY =
+      GrpclbConstants.ATTR_LB_ADDR_AUTHORITY;
 }