mirror of https://github.com/apache/nifi.git
106 lines
8.5 KiB
Plaintext
106 lines
8.5 KiB
Plaintext
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.
|
|
|