Update getInfo() in StorageSplit to getSplitInfo()

snowlift · Jun 3, 2024 · 993c0ce · 993c0ce
1 parent ec0a00b
commit 993c0ce
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/main/java/org/ebyhr/trino/storage/StorageSplit.java b/src/main/java/org/ebyhr/trino/storage/StorageSplit.java
@@ -15,10 +15,12 @@
 
 import com.fasterxml.jackson.annotation.JsonCreator;
 import com.fasterxml.jackson.annotation.JsonProperty;
+import com.google.common.collect.ImmutableMap;
 import io.trino.spi.HostAddress;
 import io.trino.spi.connector.ConnectorSplit;
 
 import java.util.List;
+import java.util.Map;
 
 import static java.util.Objects.requireNonNull;
 
@@ -75,9 +77,13 @@ public List<HostAddress> getAddresses()
     }
 
     @Override
-    public Object getInfo()
+    public Map<String, String> getSplitInfo()
     {
-        return this;
+        return ImmutableMap.<String, String>builder()
+                .put("mode", mode.name())
+                .put("schemaName", schemaName)
+                .put("tableName", tableName)
+                .buildOrThrow();
     }
 
     public enum Mode