Apache
/
nifi
mirror of https://github.com/apache/nifi.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

NIFI-4051: Ensure that we use the appropriate Lexical syntax for SQL query validation as we do when running 

Signed-off-by: Pierre Villard <pierre.villard.fr@gmail.com>

This closes #2054.
This commit is contained in:
Mark Payne 2017-08-03 16:56:10 -04:00 committed by Pierre Villard
parent 40cde0466a
commit 969bbe654c
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
nifi-nar-bundles/nifi-standard-bundle/nifi-standard-processors/src/main/java/org/apache/nifi/processors/standard/QueryRecord.java
View File

@ -46,6 +46,7 @@ import org.apache.calcite.config.Lex;
import org.apache.calcite.jdbc.CalciteConnection; import org.apache.calcite.jdbc.CalciteConnection;
import org.apache.calcite.schema.SchemaPlus; import org.apache.calcite.schema.SchemaPlus;
import org.apache.calcite.sql.parser.SqlParser; import org.apache.calcite.sql.parser.SqlParser;
import org.apache.calcite.sql.parser.SqlParser.Config;
import org.apache.nifi.annotation.behavior.DynamicProperty; import org.apache.nifi.annotation.behavior.DynamicProperty;
import org.apache.nifi.annotation.behavior.DynamicRelationship; import org.apache.nifi.annotation.behavior.DynamicRelationship;
import org.apache.nifi.annotation.behavior.EventDriven; import org.apache.nifi.annotation.behavior.EventDriven;
@ -546,7 +547,12 @@ public class QueryRecord extends AbstractProcessor {
} }
final String substituted = context.newPropertyValue(input).evaluateAttributeExpressions().getValue(); final String substituted = context.newPropertyValue(input).evaluateAttributeExpressions().getValue();
final SqlParser parser = SqlParser.create(substituted);
final Config config = SqlParser.configBuilder()
.setLex(Lex.MYSQL_ANSI)
.build();
final SqlParser parser = SqlParser.create(substituted, config);
try { try {
parser.parseStmt(); parser.parseStmt();
return new ValidationResult.Builder() return new ValidationResult.Builder()