kemko/nomad
1
0
Fork 0
mirror of https://github.com/kemko/nomad.git synced 2026-01-09 03:45:41 +03:00
Code Issues Packages Projects Releases Wiki Activity

Allow short identifiers for node-drain command

Browse Source
This commit is contained in:
Ivo Verberk
2015-12-20 18:02:10 +01:00
parent 5d3bd1b6f0
commit 23bfbbf66d
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
command/node_drain.go
View File

@@ -68,8 +68,15 @@ func (c *NodeDrainCommand) Run(args []string) int {
return 1
}
// Check if node exists
node, _, err := client.Nodes().Info(nodeID, nil)
if err != nil {
c.Ui.Error(fmt.Sprintf("Error querying node info: %s", err))
return 1
}
// Toggle node draining
if _, err := client.Nodes().ToggleDrain(nodeID, enable, nil); err != nil {
if _, err := client.Nodes().ToggleDrain(node.ID, enable, nil); err != nil {
c.Ui.Error(fmt.Sprintf("Error toggling drain mode: %s", err))
return 1
}