org.jsfr.json.compiler.JsonPathCompiler Maven / Gradle / Ivy
/*
* The MIT License
*
* Copyright (c) 2017 WANG Lingsong
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
* THE SOFTWARE.
*/
package org.jsfr.json.compiler;
import org.antlr.v4.runtime.BailErrorStrategy;
import org.antlr.v4.runtime.CharStreams;
import org.antlr.v4.runtime.CommonTokenStream;
import org.antlr.v4.runtime.tree.ParseTree;
import org.antlr.v4.runtime.tree.TerminalNode;
import org.jsfr.json.filter.*;
import org.jsfr.json.path.JsonPath;
import java.math.BigDecimal;
/**
* Created by Leo on 2015/4/1.
*/
public class JsonPathCompiler extends JsonPathBaseVisitor {
private JsonPath.Builder pathBuilder;
private FilterBuilder filterBuilder;
private JsonPath.Builder filterPathBuilder;
@Override
public Void visitPath(JsonPathParser.PathContext ctx) {
pathBuilder = JsonPath.Builder.start();
return super.visitPath(ctx);
}
private JsonPath.Builder currentPathBuilder() {
return filterPathBuilder != null ? filterPathBuilder : pathBuilder;
}
@Override
public Void visitSearchChild(JsonPathParser.SearchChildContext ctx) {
currentPathBuilder().scan().child(ctx.KEY().getText());
return super.visitSearchChild(ctx);
}
@Override
public Void visitSearch(JsonPathParser.SearchContext ctx) {
currentPathBuilder().scan();
return super.visitSearch(ctx);
}
@Override
public Void visitChildNode(JsonPathParser.ChildNodeContext ctx) {
currentPathBuilder().child(ctx.KEY().getText());
return super.visitChildNode(ctx);
}
@Override
public Void visitChildrenNode(JsonPathParser.ChildrenNodeContext ctx) {
int i = 0;
String[] strings = new String[ctx.QUOTED_STRING().size()];
for (TerminalNode node : ctx.QUOTED_STRING()) {
String quotedString = node.getText();
strings[i++] = removeQuote(quotedString);
}
currentPathBuilder().children(strings);
return super.visitChildren(ctx);
}
private String removeQuote(String quotedString) {
return quotedString.substring(1, quotedString.length() - 1);
}
@Override
public Void visitIndex(JsonPathParser.IndexContext ctx) {
currentPathBuilder().index(Integer.parseInt(ctx.NUM().getText()));
return super.visitIndex(ctx);
}
@Override
public Void visitIndexes(JsonPathParser.IndexesContext ctx) {
int i = 0;
Integer[] keys = new Integer[ctx.NUM().size()];
for (TerminalNode key : ctx.NUM()) {
keys[i++] = Integer.parseInt(key.getText());
}
currentPathBuilder().indexes(keys);
return super.visitIndexes(ctx);
}
@Override
public Void visitSlicing(JsonPathParser.SlicingContext ctx) {
Integer left = null;
Integer right;
Integer temp = null;
for (ParseTree node : ctx.children) {
if (node instanceof TerminalNode) {
TerminalNode tNode = (TerminalNode) node;
if (((TerminalNode) node).getSymbol().getType() == JsonPathParser.COLON) {
left = temp;
temp = null;
} else if (tNode.getSymbol().getType() == JsonPathParser.NUM) {
temp = Integer.parseInt(tNode.getText());
}
}
}
right = temp;
pathBuilder.slicing(left, right);
return super.visitSlicing(ctx);
}
@Override
public Void visitAnyChild(JsonPathParser.AnyChildContext ctx) {
currentPathBuilder().anyChild();
return super.visitAnyChild(ctx);
}
@Override
public Void visitAnyIndex(JsonPathParser.AnyIndexContext ctx) {
currentPathBuilder().anyIndex();
return super.visitAnyIndex(ctx);
}
@Override
public Void visitAny(JsonPathParser.AnyContext ctx) {
currentPathBuilder().any();
return super.visitAny(ctx);
}
@Override
public Void visitFilter(JsonPathParser.FilterContext ctx) {
pathBuilder.any();
filterBuilder = new FilterBuilder();
Void rst = super.visitFilter(ctx);
pathBuilder.withFilter(filterBuilder.build());
return rst;
}
@Override
public Void visitFilterExpr(JsonPathParser.FilterExprContext ctx) {
Void rst;
if (ctx.AndOperator() != null) {
filterBuilder.startAndPredicate();
rst = super.visitFilterExpr(ctx);
filterBuilder.endAndPredicate();
} else if (ctx.OrOperator() != null) {
filterBuilder.startOrPredicate();
rst = super.visitFilterExpr(ctx);
filterBuilder.endOrPredicate();
} else {
rst = super.visitFilterExpr(ctx);
}
return rst;
}
@Override
public Void visitFilterEqualNum(JsonPathParser.FilterEqualNumContext ctx) {
// JsonPath relativePath = JsonPath.Builder.start().child(ctx.KEY().getText()).build();
filterPathBuilder = JsonPath.Builder.start();
Void rst = super.visitFilterEqualNum(ctx);
filterBuilder.append(new EqualityNumPredicate(filterPathBuilder.build(), new BigDecimal(ctx.NUM().getText())));
filterPathBuilder = null;
return rst;
}
@Override
public Void visitFilterExist(JsonPathParser.FilterExistContext ctx) {
filterPathBuilder = JsonPath.Builder.start();
Void rst = super.visitFilterExist(ctx);
filterBuilder.append(new ExistencePredicate(filterPathBuilder.build()));
filterPathBuilder = null;
return rst;
}
@Override
public Void visitFilterGtNum(JsonPathParser.FilterGtNumContext ctx) {
filterPathBuilder = JsonPath.Builder.start();
Void rst = super.visitFilterGtNum(ctx);
filterBuilder.append(new GreaterThanNumPredicate(filterPathBuilder.build(), new BigDecimal(ctx.NUM().getText())));
filterPathBuilder = null;
return rst;
}
@Override
public Void visitFilterLtNum(JsonPathParser.FilterLtNumContext ctx) {
filterPathBuilder = JsonPath.Builder.start();
Void rst = super.visitFilterLtNum(ctx);
filterBuilder.append(new LessThanNumPredicate(filterPathBuilder.build(), new BigDecimal(ctx.NUM().getText())));
filterPathBuilder = null;
return rst;
}
@Override
public Void visitFilterEqualStr(JsonPathParser.FilterEqualStrContext ctx) {
filterPathBuilder = JsonPath.Builder.start();
Void rst = super.visitFilterEqualStr(ctx);
filterBuilder.append(new EqualityStrPredicate(filterPathBuilder.build(), removeQuote(ctx.QUOTED_STRING().getText())));
filterPathBuilder = null;
return rst;
}
public static JsonPath[] compile(String... paths) {
JsonPath[] jsonPaths = new JsonPath[paths.length];
for (int i = 0; i < paths.length; i++) {
jsonPaths[i] = compile(paths[i]);
}
return jsonPaths;
}
public static JsonPath compile(String path) {
JsonPathLexer lexer = new JsonPathLexer(CharStreams.fromString(path));
CommonTokenStream tokens = new CommonTokenStream(lexer);
JsonPathParser parser = new JsonPathParser(tokens);
parser.setErrorHandler(new BailErrorStrategy());
JsonPathParser.PathContext tree = parser.path();
JsonPathCompiler compiler = new JsonPathCompiler();
compiler.visit(tree);
return compiler.pathBuilder.build();
}
// public static void main(String[] s) {
// JsonPath path = compile("$..abc.c.d[1].e[2,3,6]");
// System.out.println(path);
// }
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy