From 8cf34c3ea5a86a18d3bc7496be7319df0b051999 Mon Sep 17 00:00:00 2001
From: Matt Gilman <matt.c.gilman@gmail.com>
Date: Tue, 2 Feb 2016 13:16:10 -0500
Subject: [PATCH] NIFI-1457: - Update heartbeat bean when primary state
 changes.

---
 .../main/java/org/apache/nifi/controller/FlowController.java   | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/nifi-nar-bundles/nifi-framework-bundle/nifi-framework/nifi-framework-core/src/main/java/org/apache/nifi/controller/FlowController.java b/nifi-nar-bundles/nifi-framework-bundle/nifi-framework/nifi-framework-core/src/main/java/org/apache/nifi/controller/FlowController.java
index 04b48dd1da..840df947cd 100644
--- a/nifi-nar-bundles/nifi-framework-bundle/nifi-framework/nifi-framework-core/src/main/java/org/apache/nifi/controller/FlowController.java
+++ b/nifi-nar-bundles/nifi-framework-bundle/nifi-framework/nifi-framework-core/src/main/java/org/apache/nifi/controller/FlowController.java
@@ -3108,6 +3108,9 @@ public class FlowController implements EventAccess, ControllerServiceProvider, R
             // update primary
             this.primary = primary;
             eventDrivenWorkerQueue.setPrimary(primary);
+
+            // update the heartbeat bean
+            this.heartbeatBeanRef.set(new HeartbeatBean(rootGroup, primary, connected));
         } finally {
             rwLock.writeLock().unlock();
         }