add custom GridSampleDimensionSerializer

apache · Apr 26, 2024 · f5b8923 · f5b8923
1 parent b1103bf
commit f5b8923
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/common/src/main/java/org/apache/sedona/common/raster/serde/Serde.java b/common/src/main/java/org/apache/sedona/common/raster/serde/Serde.java
@@ -183,14 +183,17 @@ public static byte[] serializeGridSampleDimension(GridSampleDimension sampleDime
         Kryo kryo = kryos.get();
         ByteArrayOutputStream baos = new ByteArrayOutputStream();
         Output output = new Output(baos);
-        kryo.writeClassAndObject(output, sampleDimension);
+        GridSampleDimensionSerializer serializer = new GridSampleDimensionSerializer();
+        serializer.write(kryo, output, sampleDimension);
         output.close();
         return baos.toByteArray();
     }
 
     public static GridSampleDimension deserializeGridSampleDimension(byte[] data) {
         Kryo kryo = kryos.get();
         Input input = new Input(new ByteArrayInputStream(data));
-        return (GridSampleDimension) kryo.readClassAndObject(input);
+        GridSampleDimensionSerializer serializer = new GridSampleDimensionSerializer();
+        return serializer.read(kryo, input, GridSampleDimension.class);
     }
+
 }