org.neo4j.cypher.internal.parser.javacc.ParseExceptions Maven / Gradle / Ivy
/*
* Copyright (c) "Neo4j"
* Neo4j Sweden AB [https://neo4j.com]
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.neo4j.cypher.internal.parser.javacc;
import java.util.Arrays;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;
public class ParseExceptions extends RuntimeException {
public static List expected(int[][] expectedTokenSequences, String[] tokenImage) {
HashMap tokenCount = new HashMap<>();
Arrays.stream(expectedTokenSequences)
.flatMapToInt(Arrays::stream)
.boxed()
.forEach((token) -> {
tokenCount.put(token, tokenCount.getOrDefault(token, 0L) + 1L);
});
List strings = processExpectedList(tokenCount, tokenImage);
Collections.sort(strings);
return strings;
}
public static List processExpectedList(Map expectedTokens, String[] tokenImage) {
long identifiers = expectedTokens.getOrDefault(CypherConstants.IDENTIFIER, 0L);
long plusCount = expectedTokens.getOrDefault(CypherConstants.PLUS, 0L);
long expressions = Math.min(identifiers, plusCount);
if (identifiers > 0) {
filterTokenSet(expectedTokens, IdentifierTokens.getIdentifierTokens(), identifiers);
}
if (expressions > 0) {
filterTokenSet(expectedTokens, ExpressionTokens.getExpressionTokens(), expressions);
}
List expectedMessage = expectedTokens.keySet().stream()
.map(token -> {
String image = tokenImage[token];
return image.equals("\"$\"") ? "a parameter" : image;
})
.collect(Collectors.toList());
if (identifiers - expressions > 0) {
expectedMessage.add("an identifier");
}
if (expressions > 0) {
expectedMessage.add("an expression");
}
return expectedMessage;
}
private static Map filterTokenSet(
Map expectedTokens, Set tokens, long quantitiy) {
for (Integer token : tokens) {
if (expectedTokens.containsKey(token)) {
long newCount = expectedTokens.get(token) - quantitiy;
if (newCount > 0) {
expectedTokens.replace(token, newCount);
} else {
expectedTokens.remove(token);
}
}
}
return expectedTokens;
}
}