nifi: show state

noharm-ai · Jan 20, 2025 · 95be4f0 · 95be4f0
1 parent e370418
commit 95be4f0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/models/enums.py b/models/enums.py
@@ -127,6 +127,7 @@ class NifiQueueActionTypeEnum(Enum):
     CLEAR_QUEUE = "CLEAR_QUEUE"
     LIST_QUEUE = "LIST_QUEUE"
     CLEAR_STATE = "CLEAR_STATE"
+    VIEW_STATE = "VIEW_STATE"
     TERMINATE_PROCESS = "TERMINATE_PROCESS"
     CUSTOM_CALLBACK = "CUSTOM_CALLBACK"
     REFRESH_TEMPLATE = "REFRESH_TEMPLATE"

diff --git a/services/admin/admin_integration_remote_service.py b/services/admin/admin_integration_remote_service.py
@@ -233,6 +233,9 @@ def _get_new_queue(id_processor: str, action_type: str, data: dict):
     if NifiQueueActionTypeEnum.CLEAR_STATE.value == action_type:
         queue.url = f"nifi-api/processors/{escape(id_processor)}/state/clear-requests"
         queue.method = "POST"
+    if NifiQueueActionTypeEnum.VIEW_STATE.value == action_type:
+        queue.url = f"nifi-api/processors/{escape(id_processor)}/state"
+        queue.method = "GET"
     elif NifiQueueActionTypeEnum.SET_STATE.value == action_type:
         queue.url = f"nifi-api/processors/{escape(id_processor)}/diagnostics"
         queue.method = "GET"