From 993c0ce6f8defce4dc837e30d9d5bd85ecde6321 Mon Sep 17 00:00:00 2001
From: weixubin <weixubin814@gmail.com>
Date: Mon, 3 Jun 2024 11:14:11 +0800
Subject: [PATCH] Update getInfo() in StorageSplit to getSplitInfo()

---
 .../java/org/ebyhr/trino/storage/StorageSplit.java     | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/main/java/org/ebyhr/trino/storage/StorageSplit.java b/src/main/java/org/ebyhr/trino/storage/StorageSplit.java
index 74d2a55..969bf81 100644
--- 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