expose ipmi.PowerCycleMachine (#47)

metal-stack · Jul 21, 2021 · d0085a7 · d0085a7
1 parent ec42cc2
commit d0085a7
Show file tree

Hide file tree

Showing 4 changed files with 25 additions and 0 deletions.
diff --git a/cmd/metalcore/initConsumer.go b/cmd/metalcore/initConsumer.go
@@ -67,6 +67,8 @@ func (s *Server) initConsumer() error {
 					s.EventHandler().PowerOffMachine(evt.Cmd.TargetMachineID)
 				case domain.MachineResetCmd:
 					s.EventHandler().PowerResetMachine(evt.Cmd.TargetMachineID)
+				case domain.MachineCycleCmd:
+					s.EventHandler().PowerCycleMachine(evt.Cmd.TargetMachineID)
 				case domain.MachineBiosCmd:
 					s.EventHandler().PowerBootBiosMachine(evt.Cmd.TargetMachineID)
 				case domain.MachineDiskCmd:

diff --git a/internal/event/powerControl.go b/internal/event/powerControl.go
@@ -63,6 +63,25 @@ func (h *eventHandler) PowerResetMachine(machineID string) {
 	}
 }
 
+func (h *eventHandler) PowerCycleMachine(machineID string) {
+	ipmiCfg, err := h.APIClient().IPMIConfig(machineID)
+	if err != nil {
+		zapup.MustRootLogger().Error("Unable to read IPMI connection details",
+			zap.String("machine", machineID),
+			zap.Error(err),
+		)
+		return
+	}
+
+	err = ipmi.PowerCycleMachine(ipmiCfg)
+	if err != nil {
+		zapup.MustRootLogger().Error("Unable to power cycle machine",
+			zap.String("machine", machineID),
+			zap.Error(err),
+		)
+	}
+}
+
 func (h *eventHandler) PowerOnChassisIdentifyLED(machineID, description string) {
 	ipmiCfg, err := h.APIClient().IPMIConfig(machineID)
 	if err != nil {

diff --git a/pkg/domain/model.go b/pkg/domain/model.go
@@ -21,6 +21,7 @@ const (
 	MachineOnCmd             MachineCommand = "ON"
 	MachineOffCmd            MachineCommand = "OFF"
 	MachineResetCmd          MachineCommand = "RESET"
+	MachineCycleCmd          MachineCommand = "CYCLE"
 	MachineBiosCmd           MachineCommand = "BIOS"
 	MachineDiskCmd           MachineCommand = "DISK"
 	MachinePxeCmd            MachineCommand = "PXE"
@@ -89,6 +90,7 @@ type EventHandler interface {
 	PowerOnMachine(machineID string)
 	PowerOffMachine(machineID string)
 	PowerResetMachine(machineID string)
+	PowerCycleMachine(machineID string)
 	PowerBootBiosMachine(machineID string)
 	PowerBootDiskMachine(machineID string)
 	PowerBootPxeMachine(machineID string)

diff --git a/test/helpers.go b/test/helpers.go
@@ -40,6 +40,8 @@ func (h *noopEventHandler) PowerOffMachine(machineID string) {}
 
 func (h *noopEventHandler) PowerResetMachine(machineID string) {}
 
+func (h *noopEventHandler) PowerCycleMachine(machineID string) {}
+
 func (h *noopEventHandler) PowerBootBiosMachine(machineID string) {}
 
 func (h *noopEventHandler) PowerBootDiskMachine(machineID string) {}