feat: Add TableDataService#getRawTableLocationProvider

engine/table/src/main/java/io/deephaven/engine/table/impl/locations/TableDataService.java

@@ -20,6 +20,24 @@ public interface TableDataService {
     @NotNull
     TableLocationProvider getTableLocationProvider(@NotNull TableKey tableKey);
 
+    /**
+     * Request the raw {@link TableLocationProvider} from this service that will provide the {@link TableLocation} for
+     * {@code tableKey} and {@code tableLocationKey} if the location may exist. A raw {@link TableLocationProvider} does
+     * not compose multiple {@link TableLocationProvider TableLocationProviders} or delegate to other implementations.
+     *
+     * @param tableKey The {@link TableKey} to lookup
+     * @param tableLocationKey The {@link TableLocationKey} to lookup
+     * @return A raw {@link TableLocationProvider} for the specified {@link TableKey} and {@link TableLocationKey}, or
+     *         {@code null} if either key is not present
+     * @implSpec Non-raw {@link TableDataService TableDataServices} must implement this method.
+     * @throws TableDataException if the TableLocationKey is provided by more than one TableLocationProvider
+     */
+    @Nullable
+    default TableLocationProvider getRawTableLocationProvider(@NotNull TableKey tableKey,
+            @NotNull TableLocationKey tableLocationKey) {
+        return getTableLocationProvider(tableKey);
+    }
+
     /**
      * Forget all state for subsequent requests for all tables.
      */

engine/table/src/main/java/io/deephaven/engine/table/impl/locations/impl/CompositeTableDataService.java

@@ -53,6 +53,37 @@ public CompositeTableDataService(@NotNull String name, @NotNull final ServiceSel
         this.serviceSelector = Require.neqNull(serviceSelector, "serviceSelector");
     }
 
+    @Override
+    @Nullable
+    public TableLocationProvider getRawTableLocationProvider(@NotNull TableKey tableKey,
+            @NotNull TableLocationKey tableLocationKey) {
+        final TableDataService[] services = serviceSelector.call(tableKey);
+
+        if (services == null || services.length == 0) {
+            return null;
+        }
+
+        TableLocationProvider tlp = null;
+        for (final TableDataService service : services) {
+            final TableLocationProvider tlpCandidate = service.getRawTableLocationProvider(tableKey, tableLocationKey);
+            if (tlpCandidate == null) {
+                continue;
+            }
+
+            if (tlpCandidate.hasTableLocationKey(tableLocationKey)) {
+                if (tlp != null) {
+                    throw new TableDataException(
+                            "TableDataService elements " + tlpCandidate.getName() + " and " + tlp.getName()
+                                    + " both contain " + tableLocationKey + ". Full TableDataService configuration:\n"
+                                    + Formatter.formatTableDataService(CompositeTableDataService.this.toString()));
+                }
+                tlp = tlpCandidate;
+            }
+        }
+
+        return tlp;
+    }
+
     @Override
     public void reset() {
         super.reset();

engine/table/src/main/java/io/deephaven/engine/table/impl/locations/impl/FilteredTableDataService.java

@@ -47,6 +47,17 @@ public FilteredTableDataService(@NotNull final TableDataService serviceToFilter,
         this.locationKeyFilter = Require.neqNull(locationKeyFilter, "locationKeyFilter");
     }
 
+    @Override
+    @Nullable
+    public TableLocationProvider getRawTableLocationProvider(@NotNull TableKey tableKey,
+            @NotNull TableLocationKey tableLocationKey) {
+        if (!locationKeyFilter.accept(tableLocationKey)) {
+            return null;
+        }
+
+        return serviceToFilter.getRawTableLocationProvider(tableKey, tableLocationKey);
+    }
+
     @Override
     public void reset() {
         super.reset();