jsonpath support mysql syntax

alibaba · Nov 5, 2020 · 13f3f2b · 13f3f2b
1 parent 1ccd0f1
commit 13f3f2b
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/src/main/java/com/alibaba/fastjson/JSONPath.java b/src/main/java/com/alibaba/fastjson/JSONPath.java
@@ -132,7 +132,9 @@ public Object extract(DefaultJSONParser parser) {
         }
 
         Segment lastSegment = segments[segments.length - 1];
-        if (lastSegment instanceof TypeSegment || lastSegment instanceof FloorSegment) {
+        if (lastSegment instanceof TypeSegment
+                || lastSegment instanceof FloorSegment
+                || lastSegment instanceof MultiIndexSegment) {
             return eval(
                     parser.parse());
         }
@@ -1021,6 +1023,12 @@ Object parseArrayAccessFilter(boolean acceptBracket) {
                 }
 
                 if (acceptBracket && ch == ']') {
+                    if (isEOF()) {
+                        if (propertyName.equals("last")) {
+                            return new MultiIndexSegment(new int[]{-1});
+                        }
+                    }
+
                     next();
                     Filter filter = new NotNullSegement(propertyName, false);
                     while (ch == ' ') {

diff --git a/src/test/java/com/alibaba/json/bvt/path/JSONPath_17.java b/src/test/java/com/alibaba/json/bvt/path/JSONPath_17.java
@@ -19,4 +19,8 @@ public void test_for_jsonpath() throws Exception {
         String oupput = JSON.parse(input).toString();
         assertEquals(obj, JSON.parse(oupput));
     }
+
+    public void test_for_jsonpath_1() throws Exception {
+        assertEquals("[5]", JSONPath.extract("[1, 2, 3, 4, 5]", "$[last]").toString());
+    }
 }