JSONPath add static method 'remove', JSONPath.set & JSONPath.remove s…

…upport more syntax
majay123 · May 4, 2022 · f5057e6 · f5057e6
1 parent 7f4cfc8
commit f5057e6
Show file tree

Hide file tree

Showing 4 changed files with 169 additions and 1 deletion.
diff --git a/core/src/main/java/com/alibaba/fastjson2/JSONPath.java b/core/src/main/java/com/alibaba/fastjson2/JSONPath.java
@@ -54,7 +54,13 @@ public static Object eval(Object rootObject, String path) {
     }
 
     public static void set(Object rootObject, String path, Object value) {
-        JSONPath.of(path).set(rootObject, value);
+        JSONPath.of(path)
+                .set(rootObject, value);
+    }
+
+    public static void remove(Object rootObject, String path) {
+        JSONPath.of(path)
+                .remove(rootObject);
     }
 
     public static Map<String, Object> paths(Object javaObject) {
@@ -1184,6 +1190,25 @@ public boolean apply(Object fieldValue) {
                     throw new UnsupportedOperationException();
             }
         }
+
+        public void set(Context context, Object value) {
+            Object object = context.parent == null
+                    ? context.root
+                    : context.parent.value;
+
+            if (object instanceof List) {
+                List list = (List) object;
+                for (int i = 0; i < list.size(); i++) {
+                    Object item = list.get(i);
+                    if (apply(context, item)) {
+                        list.set(i, value);
+                    }
+                }
+                return;
+            }
+
+            throw new JSONException("UnsupportedOperation ");
+        }
     }
 
     static final class NameDecimalOpSegment extends NameFilter {
@@ -4067,6 +4092,32 @@ public void set(Context ctx, Object value) {
             throw new JSONException("UnsupportedOperation");
         }
 
+        @Override
+        public boolean remove(Context ctx) {
+            Object object = ctx.parent == null
+                    ? ctx.root
+                    : ctx.parent.value;
+
+            if (object instanceof java.util.List) {
+                List list = (List) object;
+                if (index >= 0) {
+                    if (index < list.size()) {
+                        list.remove(index);
+                        return true;
+                    }
+                } else {
+                    int itemIndex = list.size() + this.index;
+                    if (itemIndex >= 0) {
+                        list.remove(itemIndex);
+                        return true;
+                    }
+                }
+                return false;
+            }
+
+            throw new JSONException("UnsupportedOperation");
+        }
+
         @Override
         public void setInt(Context ctx, int value) {
             Object object = ctx.parent == null
@@ -4473,6 +4524,58 @@ public void accept(JSONReader jsonReader, Context ctx) {
             ctx.value = array;
             ctx.eval = true;
         }
+
+        public void set(Context context, Object value) {
+            Object object = context.parent == null
+                    ? context.root
+                    : context.parent.value;
+
+            if (object instanceof java.util.List) {
+                List list = (List) object;
+                for (int i = 0, size = list.size(); i < size; i++) {
+                    boolean match;
+                    if (begin >= 0) {
+                        match = i >= begin && i < end;
+                    } else {
+                        int ni = i - size;
+                        match = ni >= begin && ni < end;
+                    }
+                    if (match) {
+                        list.set(i, value);
+                    }
+                }
+                return;
+            }
+
+            throw new JSONException("UnsupportedOperation " + getClass());
+        }
+
+        public boolean remove(Context context) {
+            Object object = context.parent == null
+                    ? context.root
+                    : context.parent.value;
+
+            if (object instanceof java.util.List) {
+                List list = (List) object;
+                int removeCount = 0;
+                for (int size = list.size(), i = size - 1; i >= 0; i--) {
+                    boolean match;
+                    if (begin >= 0) {
+                        match = i >= begin && i < end;
+                    } else {
+                        int ni = i - size;
+                        match = ni >= begin && ni < end;
+                    }
+                    if (match) {
+                        list.remove(i);
+                        removeCount++;
+                    }
+                }
+                return removeCount > 0;
+            }
+
+            throw new JSONException("UnsupportedOperation " + getClass());
+        }
     }
 
     static final class MultiIndexSegment extends Segment {

diff --git a/core/src/main/java/com/alibaba/fastjson2/JSONReader.java b/core/src/main/java/com/alibaba/fastjson2/JSONReader.java
@@ -1209,6 +1209,7 @@ public List readArray() {
                 case '{':
                     val = readObject();
                     break;
+                case '\'':
                 case '"':
                     val = readString();
                     break;

diff --git a/core/src/test/java/com/alibaba/fastjson2/jsonpath/JSONPathRemoveTest.java b/core/src/test/java/com/alibaba/fastjson2/jsonpath/JSONPathRemoveTest.java
@@ -0,0 +1,32 @@
+package com.alibaba.fastjson2.jsonpath;
+
+import com.alibaba.fastjson2.JSON;
+import com.alibaba.fastjson2.JSONArray;
+import com.alibaba.fastjson2.JSONObject;
+import com.alibaba.fastjson2.JSONPath;
+import org.junit.jupiter.api.Test;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+
+public class JSONPathRemoveTest {
+    @Test
+    public void test() {
+        JSONArray array = JSON.parseArray("[0,1,2,3]");
+        JSONPath.remove(array, "$[?(@ == 1)]");
+        assertEquals("[0,2,3]", array.toJSONString());
+    }
+
+    @Test
+    public void test1() {
+        JSONObject object = JSON.parseObject("{'store':{'book':['x0','x1']}}");
+        JSONPath.remove(object, "$.store.book[0]");
+        assertEquals("{\"store\":{\"book\":[\"x1\"]}}", object.toJSONString());
+    }
+
+    @Test
+    public void test2() {
+        JSONObject object = JSON.parseObject("{'store':{'book':['x0','x1','x2','x3']}}");
+        JSONPath.remove(object, "$.store.book[0:2]");
+        assertEquals("{\"store\":{\"book\":[\"x2\",\"x3\"]}}", object.toJSONString());
+    }
+}
diff --git a/core/src/test/java/com/alibaba/fastjson2/jsonpath/JSONPathSetTest1.java b/core/src/test/java/com/alibaba/fastjson2/jsonpath/JSONPathSetTest1.java
@@ -0,0 +1,32 @@
+package com.alibaba.fastjson2.jsonpath;
+
+import com.alibaba.fastjson2.JSON;
+import com.alibaba.fastjson2.JSONArray;
+import com.alibaba.fastjson2.JSONObject;
+import com.alibaba.fastjson2.JSONPath;
+import org.junit.jupiter.api.Test;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+
+public class JSONPathSetTest1 {
+    @Test
+    public void test() {
+        JSONArray array = JSON.parseArray("[0,1,2,3]");
+        JSONPath.set(array, "$[?(@ == 1)]", 9);
+        assertEquals("[0,9,2,3]", array.toJSONString());
+    }
+
+    @Test
+    public void test1() {
+        JSONObject object = JSON.parseObject("{'store':{'book':['x0','x1']}}");
+        JSONPath.set(object, "$.store.book[0]", "a");
+        assertEquals("{\"store\":{\"book\":[\"a\",\"x1\"]}}", object.toJSONString());
+    }
+
+    @Test
+    public void test2() {
+        JSONObject object = JSON.parseObject("{'store':{'book':['x0','x1','x2','x3']}}");
+        JSONPath.set(object, "$.store.book[0:2]", "a");
+        assertEquals("{\"store\":{\"book\":[\"a\",\"a\",\"x2\",\"x3\"]}}", object.toJSONString());
+    }
+}