wwctl should return non-zero exit code on wwctl power sub-commands
Signed-off-by: xu yang <xyang@ciq.com>
This commit is contained in:
95
internal/app/wwctl/power/soft/main.go
Normal file
95
internal/app/wwctl/power/soft/main.go
Normal file
@@ -0,0 +1,95 @@
|
||||
package soft
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
|
||||
"github.com/spf13/cobra"
|
||||
"github.com/warewulf/warewulf/internal/pkg/batch"
|
||||
"github.com/warewulf/warewulf/internal/pkg/hostlist"
|
||||
"github.com/warewulf/warewulf/internal/pkg/node"
|
||||
"github.com/warewulf/warewulf/internal/pkg/power"
|
||||
"github.com/warewulf/warewulf/internal/pkg/wwlog"
|
||||
)
|
||||
|
||||
func CobraRunE(cmd *cobra.Command, args []string) error {
|
||||
var returnErr error = nil
|
||||
|
||||
nodeDB, err := node.New()
|
||||
if err != nil {
|
||||
return fmt.Errorf("could not open node configuration: %s", err)
|
||||
}
|
||||
|
||||
nodes, err := nodeDB.FindAllNodes()
|
||||
if err != nil {
|
||||
return fmt.Errorf("could not get nodeList: %s", err)
|
||||
}
|
||||
|
||||
if len(args) > 0 {
|
||||
nodes = node.FilterByName(nodes, hostlist.Expand(args))
|
||||
} else {
|
||||
//nolint:errcheck
|
||||
cmd.Usage()
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
if len(nodes) == 0 {
|
||||
return fmt.Errorf("no nodes found")
|
||||
}
|
||||
|
||||
batchpool := batch.New(50)
|
||||
jobcount := len(nodes)
|
||||
results := make(chan power.IPMI, jobcount)
|
||||
|
||||
for _, node := range nodes {
|
||||
|
||||
if node.Ipmi.Ipaddr.Get() == "" {
|
||||
wwlog.Error("%s: No IPMI IP address", node.Id.Get())
|
||||
continue
|
||||
}
|
||||
var ipmiInterface = "lan"
|
||||
if node.Ipmi.Interface.Get() != "" {
|
||||
ipmiInterface = node.Ipmi.Interface.Get()
|
||||
}
|
||||
var ipmiPort = "623"
|
||||
if node.Ipmi.Port.Get() != "" {
|
||||
ipmiPort = node.Ipmi.Port.Get()
|
||||
}
|
||||
ipmiCmd := power.IPMI{
|
||||
NodeName: node.Id.Get(),
|
||||
HostName: node.Ipmi.Ipaddr.Get(),
|
||||
Port: ipmiPort,
|
||||
User: node.Ipmi.UserName.Get(),
|
||||
Password: node.Ipmi.Password.Get(),
|
||||
Interface: ipmiInterface,
|
||||
AuthType: "MD5",
|
||||
}
|
||||
|
||||
batchpool.Submit(func() {
|
||||
//nolint:errcheck
|
||||
ipmiCmd.PowerSoft()
|
||||
results <- ipmiCmd
|
||||
})
|
||||
|
||||
}
|
||||
|
||||
batchpool.Run()
|
||||
|
||||
close(results)
|
||||
|
||||
for result := range results {
|
||||
|
||||
out, err := result.Result()
|
||||
|
||||
if err != nil {
|
||||
wwlog.Error("%s: %s", result.NodeName, out)
|
||||
returnErr = err
|
||||
continue
|
||||
}
|
||||
|
||||
fmt.Printf("%s: %s\n", result.NodeName, out)
|
||||
|
||||
}
|
||||
|
||||
return returnErr
|
||||
}
|
||||
Reference in New Issue
Block a user