@@ -47,16 +47,19 @@ def find_reporting_task(name: str, port: str):
47
47
return None
48
48
49
49
50
- def create_reporting_task (name : str , port : str , version : str ):
50
+ def create_reporting_task (name : str , port : str ):
51
51
"""Create a ReportingTask"""
52
+ nifi_version = nipyapi .system .get_system_diagnostics ().system_diagnostics .aggregate_snapshot .version_info .ni_fi_version
52
53
task = nipyapi .nifi .models .reporting_task_entity .ReportingTaskEntity (
53
54
revision = nipyapi .nifi .models .revision_dto .RevisionDTO (version = 0 ),
54
55
disconnected_node_acknowledged = False ,
55
56
component = nipyapi .nifi .models .reporting_task_dto .ReportingTaskDTO (
56
57
name = name ,
57
58
type = "org.apache.nifi.reporting.prometheus.PrometheusReportingTask" ,
58
59
bundle = nipyapi .nifi .models .bundle_dto .BundleDTO (
59
- group = "org.apache.nifi" , artifact = "nifi-prometheus-nar" , version = version
60
+ group = "org.apache.nifi" ,
61
+ artifact = "nifi-prometheus-nar" ,
62
+ version = nifi_version ,
60
63
),
61
64
properties = {
62
65
"prometheus-reporting-task-metrics-endpoint-port" : port ,
@@ -158,9 +161,7 @@ def main():
158
161
reporting_task = find_reporting_task (name = task_name , port = port )
159
162
160
163
if reporting_task is None :
161
- reporting_task = create_reporting_task (
162
- name = task_name , port = port , version = args ["nifi_version" ]
163
- )
164
+ reporting_task = create_reporting_task (name = task_name , port = port )
164
165
print (
165
166
get_reporting_task_name (task = reporting_task )
166
167
+ " [%s] -> CREATED" % reporting_task .id
0 commit comments