Fix serialization of LineMode for LineChartOp (#3199)

Same reason as #3132
Texera · Jan 14, 2025 · 3f66f4e · 3f66f4e
1 parent 8ee3570
commit 3f66f4e
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/...-operator/src/main/scala/edu/uci/ics/amber/operator/visualization/lineChart/LineMode.java b/...-operator/src/main/scala/edu/uci/ics/amber/operator/visualization/lineChart/LineMode.java
@@ -1,5 +1,6 @@
 package edu.uci.ics.amber.operator.visualization.lineChart;
 
+import com.fasterxml.jackson.annotation.JsonCreator;
 import com.fasterxml.jackson.annotation.JsonValue;
 
 public enum LineMode {
@@ -12,6 +13,17 @@ public enum LineMode {
         this.mode = mode;
     }
 
+    // Handle custom deserialization for enum
+    @JsonCreator
+    public static LineMode fromString(String value) {
+        for (LineMode mode : LineMode.values()) {
+            if (mode.mode.equalsIgnoreCase(value)) {
+                return mode;
+            }
+        }
+        throw new IllegalArgumentException("Unknown line mode: " + value);
+    }
+
     @JsonValue
     public String getMode() {
         return mode;