Internal change

PiperOrigin-RevId: 875062474

Author: DeviceInfra

src/java/com/google/devtools/mobileharness/shared/util/comm/stub/BUILD

@@ -69,6 +69,7 @@ java_library(
     runtime_deps = [
     ],
     deps = [
+        ":grpc_address_util",
         "//src/java/com/google/devtools/mobileharness/shared/constant/closeable",
         "@grpc-java//core",
         "@grpc-java//stub",
@@ -128,3 +129,8 @@ java_library(
         "@maven//:javax_inject_jsr330_api",
     ],
 )
+
+java_library(
+    name = "grpc_address_util",
+    srcs = ["GrpcAddressUtil.java"],
+)

src/java/com/google/devtools/mobileharness/shared/util/comm/stub/GrpcAddressUtil.java

@@ -0,0 +1,37 @@
+/*
+ * Copyright 2022 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *    https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.google.devtools.mobileharness.shared.util.comm.stub;
+
+/** Utility class for gRPC addresses. */
+public final class GrpcAddressUtil {
+
+  private GrpcAddressUtil() {}
+
+  public static String getGrpcTarget(String address) {
+    if (address.startsWith("dns:///")) {
+      return address;
+    }
+    return "dns:///" + address;
+  }
+
+  public static String getGrpcAuthority(String address) {
+    if (address.startsWith("dns:///")) {
+      return address.substring("dns:///".length());
+    }
+    return address;
+  }
+}