1 files changed, 29 insertions, 0 deletions
diff --git a/app/cluster.go b/app/cluster.go
new file mode 100644
index 000000000..020e57c61
--- /dev/null
+++ b/app/cluster.go
@@ -0,0 +1,29 @@
+// Copyright (c) 2018-present Mattermost, Inc. All Rights Reserved.
+// See License.txt for license information.
+
+package app
+
+import "github.com/mattermost/mattermost-server/model"
+
+// Registers a given function to be called when the cluster leader may have changed. Returns a unique ID for the
+// listener which can later be used to remove it. If clustering is not enabled in this build, the callback will never
+// be called.
+func (a *App) AddClusterLeaderChangedListener(listener func()) string {
+	id := model.NewId()
+	a.clusterLeaderListeners.Store(id, listener)
+	return id
+}
+
+// Removes a listener function by the unique ID returned when AddConfigListener was called
+func (a *App) RemoveClusterLeaderChangedListener(id string) {
+	a.clusterLeaderListeners.Delete(id)
+}
+
+func (a *App) InvokeClusterLeaderChangedListeners() {
+	a.Go(func() {
+		a.clusterLeaderListeners.Range(func(_, listener interface{}) bool {
+			listener.(func())()
+			return true
+		})
+	})
+}