Make HAManager configurable by Blip users

cashapp · Dec 18, 2023 · 5ef4785 · 5ef4785
1 parent 6d6adfc
commit 5ef4785
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/ha/ha.go b/ha/ha.go
@@ -2,6 +2,8 @@
 
 package ha
 
+import "sync"
+
 type Manager interface {
 	Standby() bool
 }
@@ -15,3 +17,12 @@ func (d disabled) Standby() bool {
 }
 
 var Disabled = disabled{}
+
+// HAManagerImpl is the Manager that the public packages below use.
+// Defaults to Disabled, SetHAManager (before Server.Run) to override.
+var HAManagerImpl Manager = Disabled
+var once sync.Once
+
+func SetHAManager(m Manager) {
+	once.Do(func() { HAManagerImpl = m })
+}
diff --git a/monitor/monitor.go b/monitor/monitor.go
@@ -422,7 +422,7 @@ func (m *Monitor) startup() error {
 			Config:    m.cfg.Plans.Change,
 			DB:        m.db,
 			LCO:       m.lco,
-			HA:        ha.Disabled,
+			HA:        ha.HAManagerImpl,
 		})
 
 		m.wg.Add(1)