From 1bddac5f91b5b7742c6082da2ae0c6667523aaa4 Mon Sep 17 00:00:00 2001
From: Lawrence Qiu <lawrenceqiu@google.com>
Date: Wed, 3 Jan 2024 22:04:36 +0000
Subject: [PATCH] feat: Add Universe Domain to ClientSettings (#2331)

* feat: Add Universe Domain to ClientSettings

* chore: Revert adding builder's getters

* chore: Udpate tests for ClientSettingsTest
---
 .../java/com/google/api/gax/rpc/ClientSettings.java   | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/gax-java/gax/src/main/java/com/google/api/gax/rpc/ClientSettings.java b/gax-java/gax/src/main/java/com/google/api/gax/rpc/ClientSettings.java
index 04b2c9f55b..25929756f5 100644
--- a/gax-java/gax/src/main/java/com/google/api/gax/rpc/ClientSettings.java
+++ b/gax-java/gax/src/main/java/com/google/api/gax/rpc/ClientSettings.java
@@ -93,6 +93,10 @@ public final ApiClock getClock() {
     return stubSettings.getClock();
   }
 
+  public final String getUniverseDomain() {
+    return stubSettings.getUniverseDomain();
+  }
+
   public final String getEndpoint() {
     return stubSettings.getEndpoint();
   }
@@ -125,6 +129,7 @@ public String toString() {
         .add("headerProvider", getHeaderProvider())
         .add("internalHeaderProvider", getInternalHeaderProvider())
         .add("clock", getClock())
+        .add("universeDomain", getUniverseDomain())
         .add("endpoint", getEndpoint())
         .add("quotaProjectId", getQuotaProjectId())
         .add("watchdogProvider", getWatchdogProvider())
@@ -241,6 +246,12 @@ public B setClock(ApiClock clock) {
       return self();
     }
 
+    /** Sets the Universe Domain to configure the resolved endpoint */
+    public B setUniverseDomain(String universeDomain) {
+      stubSettings.setUniverseDomain(universeDomain);
+      return self();
+    }
+
     public B setEndpoint(String endpoint) {
       stubSettings.setEndpoint(endpoint);
       return self();