drainer

type Config struct {
	WorkerName    string         `long:"name" required:"true" description:"The name of the worker you wish to drain."`
	BeaconPidFile string         `long:"beacon-pid-file" description:"Path to beacon pid file."`
	IsShutdown    bool           `long:"shutdown" description:"Whether worker is about to shutdown."`
	Timeout       *time.Duration `long:"timeout" description:"Maximum time to wait for draining to finish."`
	TSAConfig     tsa.Config     `group:"TSA Configuration" namespace:"tsa" required:"true"`
}

type Drainer struct {
	BeaconClient beacon.BeaconClient
	IsShutdown   bool
	WatchProcess WatchProcess
	WaitInterval time.Duration
	Clock        clock.Clock
	Timeout      *time.Duration
}

type WatchProcess interface {
	IsRunning(lager.Logger) (bool, error)
}