From 13db456b0acc9a315f0b823adc4f607ec7b0c0ec Mon Sep 17 00:00:00 2001
From: Martin Zink <martinzink@apache.org>
Date: Mon, 20 Jan 2025 14:07:35 +0100
Subject: [PATCH] NIFI-14176 Add runstate information to C2 FlowInfo

---
 .../apache/nifi/c2/protocol/api/ProcessorStatus.java   | 10 ++++++++++
 .../org/apache/nifi/minifi/c2/C2NifiClientService.java |  1 +
 2 files changed, 11 insertions(+)

diff --git a/c2/c2-protocol/c2-protocol-api/src/main/java/org/apache/nifi/c2/protocol/api/ProcessorStatus.java b/c2/c2-protocol/c2-protocol-api/src/main/java/org/apache/nifi/c2/protocol/api/ProcessorStatus.java
index a172200138c3..e778f38d8317 100644
--- a/c2/c2-protocol/c2-protocol-api/src/main/java/org/apache/nifi/c2/protocol/api/ProcessorStatus.java
+++ b/c2/c2-protocol/c2-protocol-api/src/main/java/org/apache/nifi/c2/protocol/api/ProcessorStatus.java
@@ -37,6 +37,7 @@ public class ProcessorStatus implements Serializable {
     private long processingNanos;
     private int activeThreadCount;
     private int terminatedThreadCount;
+    private String runState;
 
     @Schema(description = "The id of the processor")
     public String getId() {
@@ -145,4 +146,13 @@ public int getTerminatedThreadCount() {
     public void setTerminatedThreadCount(int terminatedThreadCount) {
         this.terminatedThreadCount = terminatedThreadCount;
     }
+
+    @Schema(description = "The state of the processor")
+    public String getRunState() {
+        return runState;
+    }
+
+    public void setRunState(String runState) {
+        this.runState = runState;
+    }
 }
diff --git a/minifi/minifi-nar-bundles/minifi-framework-bundle/minifi-framework/minifi-framework-core/src/main/java/org/apache/nifi/minifi/c2/C2NifiClientService.java b/minifi/minifi-nar-bundles/minifi-framework-bundle/minifi-framework/minifi-framework-core/src/main/java/org/apache/nifi/minifi/c2/C2NifiClientService.java
index d8c055a23d18..5cb1f2f17d0b 100644
--- a/minifi/minifi-nar-bundles/minifi-framework-bundle/minifi-framework/minifi-framework-core/src/main/java/org/apache/nifi/minifi/c2/C2NifiClientService.java
+++ b/minifi/minifi-nar-bundles/minifi-framework-bundle/minifi-framework/minifi-framework-core/src/main/java/org/apache/nifi/minifi/c2/C2NifiClientService.java
@@ -406,6 +406,7 @@ private ProcessorStatus convertProcessorStatus(org.apache.nifi.controller.status
         result.setProcessingNanos(processorStatus.getProcessingNanos());
         result.setActiveThreadCount(processorStatus.getActiveThreadCount());
         result.setTerminatedThreadCount(processorStatus.getTerminatedThreadCount());
+        result.setRunState(processorStatus.getRunStatus().toString());
         return result;
     }
 }
\ No newline at end of file