luxolus
/
open-consul
2
0
Fork 0
Code Issues 1 Pull Requests Packages Releases Wiki Activity

command: maint command works

This commit is contained in:
Ryan Uber 2015-01-21 13:11:23 -08:00
parent d10529c233
commit 8081403ad9
1 changed files with 23 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
command/maint.go
View File

@ -102,7 +102,7 @@ func (c *MaintCommand) Run(args []string) int {
if enable { if enable {
// Enable node maintenance // Enable node maintenance
if serviceID == "" { if serviceID == "" {
if err := agent.EnableNodeMaintenance(); err != nil { if err := agent.EnableNodeMaintenance(reason); err != nil {
c.Ui.Error(fmt.Sprintf("Error enabling node maintenance: %s", err)) c.Ui.Error(fmt.Sprintf("Error enabling node maintenance: %s", err))
return 1 return 1
} }
@ -111,11 +111,32 @@ func (c *MaintCommand) Run(args []string) int {
} }
// Enable service maintenance // Enable service maintenance
if err := agent.EnableServiceMaintenance(serviceID); err != nil { if err := agent.EnableServiceMaintenance(serviceID, reason); err != nil {
c.Ui.Error(fmt.Sprintf("Error enabling service maintenance: %s", err)) c.Ui.Error(fmt.Sprintf("Error enabling service maintenance: %s", err))
return 1 return 1
} }
c.Ui.Output(fmt.Sprintf("Service maintenance is now enabled for %q", serviceID)) c.Ui.Output(fmt.Sprintf("Service maintenance is now enabled for %q", serviceID))
return 0
}
if disable {
// Disable node maintenance
if serviceID == "" {
if err := agent.DisableNodeMaintenance(); err != nil {
c.Ui.Error(fmt.Sprintf("Error disabling node maintenance: %s", err))
return 1
}
c.Ui.Output("Node maintenance is now disabled")
return 0
}
// Disable service maintenance
if err := agent.DisableServiceMaintenance(serviceID); err != nil {
c.Ui.Error(fmt.Sprintf("Error disabling service maintenance: %s", err))
return 1
}
c.Ui.Output(fmt.Sprintf("Service maintenance is now disabled for %q", serviceID))
return 0
} }
return 0 return 0