Merge pull request #137 from Keyruu/main

feat: add is_suspended to host info metric

Author: nlewo

internal/manager/manager.go

@@ -224,7 +224,7 @@ func (m *Manager) Run(ctx context.Context) {
 	if lastDpl != nil {
 		m.needToReboot = m.executor.NeedToReboot(lastDpl.Generation.OutPath, lastDpl.Operation)
 	}
-	m.prometheus.SetHostInfo(m.needToReboot)
+	m.prometheus.SetHostInfo(m.needToReboot, m.isSuspended)
 
 	m.FetchAndBuild(ctx)
 	m.deployer.Run(ctx)
@@ -256,7 +256,7 @@ func (m *Manager) Run(ctx context.Context) {
 				e := &protobuf.Event_RebootRequired{Deployment: dpl}
 				m.broker.Publish(&protobuf.Event{Type: &protobuf.Event_RebootRequired_{RebootRequired: e}})
 			}
-			m.prometheus.SetHostInfo(m.needToReboot)
+			m.prometheus.SetHostInfo(m.needToReboot, m.isSuspended)
 			if dpl.RestartComin.GetValue() {
 				// TODO: stop contexts
 				logrus.Infof("manager: comin needs to be restarted")

internal/prometheus/prometheus.go

@@ -32,7 +32,7 @@ func New() Prometheus {
 	hostInfo := prometheus.NewGaugeVec(prometheus.GaugeOpts{
 		Name: "comin_host_info",
 		Help: "Info of the host.",
-	}, []string{"need_to_reboot"})
+	}, []string{"is_suspended", "need_to_reboot"})
 	promReg.MustRegister(buildInfo)
 	promReg.MustRegister(deploymentInfo)
 	promReg.MustRegister(fetchCounter)
@@ -68,13 +68,17 @@ func (m Prometheus) SetDeploymentInfo(commitId, status string) {
 	m.deploymentInfo.With(prometheus.Labels{"commit_id": commitId, "status": status}).Set(1)
 }
 
-func (m Prometheus) SetHostInfo(needToReboot bool) {
-	m.hostInfo.Reset()
-	var value string
-	if needToReboot {
-		value = "1"
-	} else {
-		value = "0"
+func boolToString(b bool) string {
+	if b {
+		return "1"
 	}
-	m.hostInfo.With(prometheus.Labels{"need_to_reboot": value}).Set(1)
+	return "0"
+}
+
+func (m Prometheus) SetHostInfo(needToReboot bool, isSuspended bool) {
+	m.hostInfo.Reset()
+	m.hostInfo.With(prometheus.Labels{
+		"need_to_reboot": boolToString(needToReboot),
+		"is_suspended":   boolToString(isSuspended),
+	}).Set(1)
 }