Merge 958d4f3 into b8a09d0

onehouseinc · Dec 24, 2024 · e12958d · e12958d
2 parents b8a09d0 + 958d4f3
commit e12958d
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/lakeview/src/main/java/ai/onehouse/api/OnehouseApiClient.java b/lakeview/src/main/java/ai/onehouse/api/OnehouseApiClient.java
@@ -5,6 +5,7 @@
 import static ai.onehouse.constants.ApiConstants.GET_TABLE_METRICS_CHECKPOINT;
 import static ai.onehouse.constants.ApiConstants.INITIALIZE_TABLE_METRICS_CHECKPOINT;
 import static ai.onehouse.constants.ApiConstants.LINK_UID_KEY;
+import static ai.onehouse.constants.ApiConstants.MAINTENANCE_MODE_KEY;
 import static ai.onehouse.constants.ApiConstants.ONEHOUSE_API_ENDPOINT;
 import static ai.onehouse.constants.ApiConstants.ONEHOUSE_API_KEY;
 import static ai.onehouse.constants.ApiConstants.ONEHOUSE_API_SECRET_KEY;
@@ -114,6 +115,9 @@ Headers getHeaders(OnehouseClientConfig onehouseClientConfig) {
     if (StringUtils.isNotEmpty(onehouseClientConfig.getRegion())) {
       headersBuilder.add(ONEHOUSE_REGION_KEY, onehouseClientConfig.getRegion());
     }
+    if (onehouseClientConfig.getMaintenanceMode() != null && onehouseClientConfig.getMaintenanceMode().equals(Boolean.TRUE)) {
+      headersBuilder.add(MAINTENANCE_MODE_KEY, Boolean.TRUE.toString());
+    }
     return headersBuilder.build();
   }
 

diff --git a/lakeview/src/main/java/ai/onehouse/config/models/common/OnehouseClientConfig.java b/lakeview/src/main/java/ai/onehouse/config/models/common/OnehouseClientConfig.java
@@ -20,4 +20,5 @@ public class OnehouseClientConfig {
   @Nullable private String apiSecret;
   @Nullable private String userId;
   @Nullable private String file;
+  @Nullable private Boolean maintenanceMode;
 }
diff --git a/lakeview/src/main/java/ai/onehouse/constants/ApiConstants.java b/lakeview/src/main/java/ai/onehouse/constants/ApiConstants.java
@@ -26,6 +26,7 @@ private ApiConstants() {}
   public static final String LINK_UID_KEY = "x-onehouse-link-uid";
   public static final String ONEHOUSE_REGION_KEY = "x-onehouse-region";
   public static final String ONEHOUSE_USER_UUID_KEY = "x-onehouse-uuid";
+  public static final String MAINTENANCE_MODE_KEY = "x-maintenance-mode";
   // using mapping from:
   // https://chromium.googlesource.com/external/github.com/grpc/grpc/+/refs/tags/v1.21.4-pre1/doc/statuscodes.md
   public static final List<Integer> ACCEPTABLE_HTTP_FAILURE_STATUS_CODES =