use checkNotNull

apache · Dec 21, 2024 · 04426e8 · 04426e8
1 parent 6284af8
commit 04426e8
Showing 1 changed file with 11 additions and 15 deletions.
diff --git a/service/common/src/main/java/org/apache/polaris/service/catalog/BasePolarisCatalog.java b/service/common/src/main/java/org/apache/polaris/service/catalog/BasePolarisCatalog.java
@@ -353,7 +353,7 @@ protected String defaultWarehouseLocation(TableIdentifier tableIdentifier) {
             "Namespace does not exist: %s", tableIdentifier.namespace());
       }
       List<PolarisEntity> namespacePath = resolvedNamespace.getRawFullPath();
-      String namespaceLocation = resolveLocationForPath(namespacePath);
+      String namespaceLocation = resolveLocationForPath(callContext, namespacePath);
       return SLASH.join(namespaceLocation, tableIdentifier.name());
     }
   }
@@ -534,13 +534,13 @@ private String resolveNamespaceLocation(Namespace namespace, Map<String, String>
               ? getResolvedParentNamespace(namespace).getRawFullPath()
               : List.of(resolvedEntityView.getResolvedReferenceCatalogEntity().getRawLeafEntity());
 
-      String parentLocation = resolveLocationForPath(parentPath);
+      String parentLocation = resolveLocationForPath(callContext, parentPath);
 
       return parentLocation + "/" + namespace.level(namespace.length() - 1);
     }
   }
 
-  private static @Nonnull String resolveLocationForPath(List<PolarisEntity> parentPath) {
+  private static @Nonnull String resolveLocationForPath(@Nonnull CallContext callContext, List<PolarisEntity> parentPath) {
     // always take the first object. If it has the base-location, stop there
     AtomicBoolean foundBaseLocation = new AtomicBoolean(false);
     return parentPath.reversed().stream()
@@ -553,20 +553,18 @@ private String resolveNamespaceLocation(Namespace namespace, Map<String, String>
         .toList()
         .reversed()
         .stream()
-        .map(this::baseLocation)
+        .map(entity -> baseLocation(callContext, entity))
         .map(BasePolarisCatalog::stripLeadingTrailingSlash)
         .collect(Collectors.joining("/"));
   }
 
-  private static @Nullable String baseLocation(PolarisEntity entity) {
+  private static @Nullable String baseLocation(@Nonnull CallContext callContext, PolarisEntity entity) {
     if (entity.getType().equals(PolarisEntityType.CATALOG)) {
       CatalogEntity catEntity = CatalogEntity.of(entity);
       String catalogDefaultBaseLocation = catEntity.getDefaultBaseLocation();
-      if (catalogDefaultBaseLocation == null) {
-        LOGGER.debug(
-                "Tried to resolve location with catalog with null default base location. Catalog = {}",
-                catEntity);
-      }
+      callContext
+              .getPolarisCallContext().getDiagServices().checkNotNull(catalogDefaultBaseLocation, "Tried to resolve location with catalog with null default base location", "catalog = {}",
+                      catEntity);
       return catalogDefaultBaseLocation;
     } else {
       String baseLocation =
@@ -575,11 +573,9 @@ private String resolveNamespaceLocation(Namespace namespace, Map<String, String>
         return baseLocation;
       } else {
         String entityName = entity.getName();
-        if (entityName == null) {
-          LOGGER.debug(
-                  "Tried to resolve location with entity without base location or name. entity = {}",
-                  entity);
-        }
+        callContext
+                .getPolarisCallContext().getDiagServices().checkNotNull(entityName, "Tried to resolve location with entity without base location or name", "entity = {}",
+                        entity);
         return entityName;
       }
     }