blob: 6281dcae570675809c12e79ada74e9fff1af75b5 [file] [log] [blame]
Licensed to the Apache Software Foundation (ASF) under one or more
contributor license agreements. See the NOTICE file distributed with
this work for additional information regarding copyright ownership.
The ASF licenses this file to You 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.
This document is intended to provide a walk-through of what is necessary
in order to add a new function to the Expression Language. Doing so requires
a handful of steps, so we will outline each of those steps here, in the order
that they must be done. While this documentation is fairly verbose, it is often
the case that reading the documentation takes longer than performing the tasks
outlined by the documentation.
1) In order to make the nifi-expression-language Maven module compile in your IDE, you may need to add the ANTLR-generated sources to your IDE's classpath.
This can be done using Eclipse, as follows:
- Right-click on the nifi-expression-language project
- Go to "Properties" on the context menu
- Go to the "Java Build Path" item in the left tree and choose the "Source" tab.
- Click "Add Folder..."
- Add the target/generated-sources/antlr3 folder. If this folder does not exist, first build the project from Maven and then
right-click on the nifi-expression-language project in Eclipse and click Refresh.
- Click OK to close all dialogs.
2) Add the method name to the Tokens for the Lexer
- Open the src/main/antlr3/org/apache/nifi/attribute/expression/language/antlr/AttributeExpressionLexer.g file
- Add the function name to the list of tokens in this file. These functions are grouped by the number of arguments
that they take. This grouping mechanism could probably be made better, perhaps grouping by the type of function
provided. However, for now, it is best to keep some sort of structure, at least. If the function has optional
arguments, the function should be grouped by the maximum number of arguments that it takes (for example, the
substring function can take 1 or 2 arguments, so it is grouped with the '2 argument functions').
The syntax to use is:
<Token Name> : '<function name>';
The Token Name should be all-caps and words should be separated by underscores. The Token Name is what will be used to
identify the token when ANTLR parses an Expression. The function name should use camel case starting with a lower-case
letter. This is the name of the function as it will be referenced in the Expression Language.
- Save the AttributeExpressionLexer.g file
3) Add the method to the grammar
- Open the src/main/antlr3/org/apache/nifi/attribute/expression/language/antlr/AttributeExpressionParser.g file
- Starting around line 75, the functions are defined, grouped by the type of value returned. We can add the new function
into the grammar here. Please see the ANTLR documentation for syntax on the grammar used. Note that this is ANTLR 3, NOT ANTLR 4.
The idea here is to spell out the syntax that should be used for the function. So generally, we do this by specifying the function name,
"LPAREN!" (which indicates a left parenthesis and the ! indicates that we do not want this passed to us when obtaining the parsed tokens),
and then a list of arguments that are separated by "COMMA!" (again, indicating a comma character and that we do not want the token passed
to us when we are looking at parsed tokens). We then end with the matching "RPAREN!".
- Save this file.
4) Rebuild via Maven
- In order to make sure that we now can reference the tokens that are generated for our new function, we need to rebuild via Maven.
We can do this by building just the nifi-expression-language project, rather than rebuilding the entire NiFi code base.
- If necessary, right-click on the nifi-expression-language project in your IDE and refresh / update project from new Maven build.
This is generally necessary when using Eclipse.
5) Add the logic for the function
- In the src/main/java/org/apache/nifi/attribute/expression/language/evaluation/function package directory, we will need to create a new
class that is capable of implementing the logic of the new function. Create a class using the standard naming convention of
<function name>Evaluator and extends the appropriate abstract evaluator. If the function will return a String, the evaluator should extend
StringEvaluator. If the function will return a boolean, the evaluator should extend BooleanEvaluator. There are also evaluators for Date
and Number return types.
- Generally the constructor for the evaluator will take an Evaluator for the "Subject" and an Evaluator for each argument. The subject is the
value that the function will be evaluated against. The substring function, for instance, takes a subject of type String. Thinking in terms of
Java, the "subject" is the object on which the function is being called. It is important to take Evaluator objects and not just a String,
for instance, as we have to ensure that we determine that actual values to use dynamically at runtime.
- Implement the functionality as appropriate by implementing the abstract methods provided by the abstract Evaluator that is being extended by
your newly created Evaluator.
- The Evaluator need not be thread-safe. The existing Evaluators are numerous and provide great examples for understanding the API.
6) Add the logic to the query parser
- Generally, when using ANTLR, the preferred method to parse the input is to use a Tree Walker. However, this is far less intuitive for many
Java developers (including those of us who wrote the Expression Language originally). As a result, we instead use ANTLR to tokenize and parse the
input and then obtain an Abstract Syntax Tree and process this "manually" in Java code. This occurs in the Query class.
- We can add the function into our parsing logic by updating the #buildFunctionEvaluator method of the org.apache.nifi.attribute.expression.language.Query class.
A static import will likely need to be added to the Query class in order to reference the new token. The token can then be added to the existing
'case' statement, which will return a new instance of the Evaluator that was just added.
7) Add Unit Tests!
- Unit tests are critical for the Expression Language. These expressions can be used throughout the entire application and it is important that each function
perform its task properly. Otherwise, incorrect routing decisions could be made, or data could become corrupted as a result.
- Each function should have its battery of unit tests added to the TestQuery class. This class includes a convenience method named #verifyEquals that is
used to ensure that the Expression returns the same value, regardless of how it is compiled and evaluated.
8) Add Documentation!
- The documentation for each function is provided in the nifi-docs module, under src/main/asciidoc/expression-language-guide.adoc.
The format of the document is crucial to maintain, as this document is not only rendered as HTML in the NiFi Documentation page, but the
CSS classes that are used in the rendered docs are also made use of by the NiFi UI. When a user is entering an Expression Language expression and
presses Ctrl+Space, the UI provides auto-completion information as well as inline documentation for each function. This information is pulled
directly from the HTML that is generated from this expression-language-guide file.
- Rebuild NiFi and run the application. Add an UpdateAttribute Processor to the graph and add a new property. For the value, type the Expression Language
opening tokens ${ and then press Ctrl+Space to ensure that the function and its documentation is presented as expected. Most functions that are added
will require a Subject. In order to see the function, then, you will need to provide a subject, such as typing "${myVariable:" (without the quotes)
and then press Ctrl+Space. This step is important, as it is quite easy to make a mistake when creating the documentation using a free-form text editor,
and this will ensure that users receive a very consistent and quality experience when using the new function.