restart.go

package main

import (
	"log"
	"os"
	"strings"
)

var cmdRestart = &Command{
	Run:             maybeMessage(runRestart),
	Usage:           "restart [<type or name>]",
	NeedsApp:        true,
	OptionalMessage: true,
	Category:        "dyno",
	Short:           "restart dynos (or stop a dyno started with 'emp run')",
	Long: `
Restart all app dynos, all dynos of a specific type, or a single dyno. If used
on a dyno started using 'emp run' this will effectively stop it.

Examples:

    $ emp restart
    Restarted all dynos on myapp.

    $ emp restart web
    Restarted web dynos on myapp.

    $ emp restart web.1
    Restarted web.1 dyno on myapp.
`,
}

func runRestart(cmd *Command, args []string) {
	appname := mustApp()
	if len(args) > 1 {
		cmd.PrintUsage()
		os.Exit(2)
	}
	message := getMessage()

	target := "all"
	if len(args) == 1 {
		target = args[0]
		must(client.DynoRestart(appname, target, message))
	} else {
		must(client.DynoRestartAll(appname, message))
	}

	switch {
	case strings.Contains(target, "."):
		log.Printf("Restarted %s dyno for %s.", target, appname)
	default:
		log.Printf("Restarted %s dynos for %s.", target, appname)
	}
}