feat: Add TableDataService#getRawTableLocationProvider

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

@@ -20,6 +20,26 @@ public interface TableDataService {
     @NotNull
     TableLocationProvider getTableLocationProvider(@NotNull TableKey tableKey);
 
+    /**
+     * Request the single raw {@link TableLocationProvider} from this service that provides the {@link TableLocation}
+     * for {@code tableKey} and {@code tableLocationKey}. 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 The raw {@link TableLocationProvider} that provides the {@link TableLocation} for {@code tableKey} and
+     *         {@code tableLocationKey}, or {@code null} if there is none
+     * @implSpec Non-raw {@link TableDataService TableDataServices} must implement this method to provide a single raw
+     *           {@link TableLocationProvider} or throw a {@link TableDataException} if there is ambiguity.
+     * @throws TableDataException If the {@link TableLocation} is provided by more than one
+     *         {@link TableLocationProvider}
+     */
+    @Nullable
+    default TableLocationProvider getRawTableLocationProvider(@NotNull final TableKey tableKey,
+            @NotNull final 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 final TableKey tableKey,
+            @NotNull final 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 final TableKey tableKey,
+            @NotNull final TableLocationKey tableLocationKey) {
+        if (!locationKeyFilter.accept(tableLocationKey)) {
+            return null;
+        }
+
+        return serviceToFilter.getRawTableLocationProvider(tableKey, tableLocationKey);
+    }
+
     @Override
     public void reset() {
         super.reset();