lifecyclemgr

type LaunchableTaskInfo struct {
	*task.TaskInfo
	// ConfigAddOn is the task config add on.
	ConfigAddOn *models.ConfigAddOn
	// Spec is the pod spec for the pod to be launched.
	Spec *pbpod.PodSpec
}

type Lockable interface {
	// LockKill locks kill operations including executor shutdown
	LockKill()
	// UnlockKill unlocks kill operations including executor shutdown
	UnlockKill()
	// LockLaunch locks launch operations
	LockLaunch()
	// UnlockLaunch unlocks launch operations
	UnlockLaunch()
}

type Manager interface {
	Lockable

	// Launch will launch tasks/pods using their config/spec on the specified
	// host using the acquired leaseID.
	Launch(
		ctx context.Context,
		leaseID string,
		hostname string,
		agentID string,
		tasks map[string]*LaunchableTaskInfo,
		rateLimiter *rate.Limiter,
	) error
	// Kill will kill tasks/pods using their ID.
	Kill(
		ctx context.Context,
		id string,
		hostToReserve string,
		rateLimiter *rate.Limiter,
	) error
	// ShutdownExecutor will shutdown the underlying mesos executor. This will
	// be a no-op for v1 LifecycleMgr.
	ShutdownExecutor(
		ctx context.Context,
		id string,
		agentID string,
		rateLimiter *rate.Limiter,
	) error

	// TerminateLease will be called to terminate the acquired lease on
	// hostmgr in case of any errors. This will ensure that the hosts that
	// are leased are not freed up for placement in case we cannot place the
	// current set of tasks on them.
	TerminateLease(
		ctx context.Context,
		hostname string,
		agentID string,
		leaseID string,
	) error

	// GetTasksOnDrainingHosts gets the taskIDs of the tasks on the
	// hosts in DRAINING state
	GetTasksOnDrainingHosts(
		ctx context.Context,
		limit uint32,
		timeout uint32,
	) ([]string, error)
}

type Metrics struct {
	Launch          tally.Counter
	LaunchFail      tally.Counter
	LaunchRetry     tally.Counter
	LaunchDuration  tally.Timer
	LaunchRateLimit tally.Counter

	Kill          tally.Counter
	KillFail      tally.Counter
	Shutdown      tally.Counter
	ShutdownFail  tally.Counter
	KillRateLimit tally.Counter

	TerminateLease     tally.Counter
	TerminateLeaseFail tally.Counter

	GetTasksOnDrainingHosts     tally.Counter
	GetTasksOnDrainingHostsFail tally.Counter
}