merge related api change fix

nbauernfeind · Oct 24, 2024 · 6ca321c · 6ca321c
1 parent ab74e91
commit 6ca321c
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/...ns/barrage/src/main/java/io/deephaven/extensions/barrage/util/PythonTableDataService.java b/...ns/barrage/src/main/java/io/deephaven/extensions/barrage/util/PythonTableDataService.java
@@ -18,6 +18,7 @@
 import io.deephaven.engine.table.Table;
 import io.deephaven.engine.table.TableDefinition;
 import io.deephaven.engine.table.impl.PartitionAwareSourceTable;
+import io.deephaven.engine.table.impl.TableUpdateMode;
 import io.deephaven.engine.table.impl.chunkboxer.ChunkBoxer;
 import io.deephaven.engine.table.impl.locations.*;
 import io.deephaven.engine.table.impl.locations.impl.*;
@@ -443,7 +444,7 @@ private class TableLocationProviderImpl extends AbstractTableLocationProvider {
         private Subscription subscription = null;
 
         private TableLocationProviderImpl(@NotNull final TableKeyImpl tableKey) {
-            super(tableKey, true);
+            super(tableKey, true, TableUpdateMode.ADD_ONLY, TableUpdateMode.ADD_REMOVE);
             final BarrageUtil.ConvertedArrowSchema[] schemas = backend.getTableSchema(tableKey);
 
             final TableDefinition partitioningDef = schemas[0].tableDef;
@@ -483,15 +484,15 @@ private TableLocationProviderImpl(@NotNull final TableKeyImpl tableKey) {
         @Override
         public void refresh() {
             TableKeyImpl key = (TableKeyImpl) getKey();
-            backend.getTableLocations(tableDefinition, key, this::handleTableLocationKey);
+            backend.getTableLocations(tableDefinition, key, this::handleTableLocationKeyAdded);
         }
 
         @Override
         protected void activateUnderlyingDataSource() {
             TableKeyImpl key = (TableKeyImpl) getKey();
             final Subscription localSubscription = subscription = new Subscription();
             localSubscription.cancellationCallback = backend.subscribeToTableLocations(
-                    tableDefinition, key, this::handleTableLocationKey);
+                    tableDefinition, key, this::handleTableLocationKeyAdded);
             activationSuccessful(localSubscription);
         }