db

type ActualLRPDB interface {
	ActualLRPGroups(logger lager.Logger, filter models.ActualLRPFilter) ([]*models.ActualLRPGroup, error)
	ActualLRPGroupsByProcessGuid(logger lager.Logger, processGuid string) ([]*models.ActualLRPGroup, error)
	ActualLRPGroupByProcessGuidAndIndex(logger lager.Logger, processGuid string, index int32) (*models.ActualLRPGroup, error)

	ClaimActualLRP(logger lager.Logger, processGuid string, index int32, instanceKey *models.ActualLRPInstanceKey) error
	StartActualLRP(logger lager.Logger, key *models.ActualLRPKey, instanceKey *models.ActualLRPInstanceKey, netInfo *models.ActualLRPNetInfo) error
	CrashActualLRP(logger lager.Logger, key *models.ActualLRPKey, instanceKey *models.ActualLRPInstanceKey, errorMessage string) error
	FailActualLRP(logger lager.Logger, key *models.ActualLRPKey, errorMessage string) error
	RemoveActualLRP(logger lager.Logger, processGuid string, index int32) error
	RetireActualLRP(logger lager.Logger, key *models.ActualLRPKey) error
}

type CompleteTaskWork func(logger lager.Logger, taskDB TaskDB, task *models.Task) func()

type DB interface {
	DomainDB
	EncryptionDB
	EvacuationDB
	EventDB
	LRPDB
	TaskDB
	VersionDB
}

type DesiredLRPDB interface {
	DesiredLRPs(logger lager.Logger, filter models.DesiredLRPFilter) ([]*models.DesiredLRP, error)
	DesiredLRPByProcessGuid(logger lager.Logger, processGuid string) (*models.DesiredLRP, error)

	DesiredLRPSchedulingInfos(logger lager.Logger, filter models.DesiredLRPFilter) ([]*models.DesiredLRPSchedulingInfo, error)

	DesireLRP(logger lager.Logger, desiredLRP *models.DesiredLRP) error
	UpdateDesiredLRP(logger lager.Logger, processGuid string, update *models.DesiredLRPUpdate) error
	RemoveDesiredLRP(logger lager.Logger, processGuid string) error
}

type DomainDB interface {
	Domains(logger lager.Logger) ([]string, error)
	UpsertDomain(lgger lager.Logger, domain string, ttl uint32) error
}

type EncryptionDB interface {
	EncryptionKeyLabel(logger lager.Logger) (string, error)
	SetEncryptionKeyLabel(logger lager.Logger, encryptionKeyLabel string) error
}

type EvacuationDB interface {
	EvacuateClaimedActualLRP(lager.Logger, *models.ActualLRPKey, *models.ActualLRPInstanceKey) (bool, error)
	EvacuateRunningActualLRP(lager.Logger, *models.ActualLRPKey, *models.ActualLRPInstanceKey, *models.ActualLRPNetInfo, uint64) (bool, error)
	EvacuateStoppedActualLRP(lager.Logger, *models.ActualLRPKey, *models.ActualLRPInstanceKey) (bool, error)
	EvacuateCrashedActualLRP(lager.Logger, *models.ActualLRPKey, *models.ActualLRPInstanceKey, string) (bool, error)
	RemoveEvacuatingActualLRP(lager.Logger, *models.ActualLRPKey, *models.ActualLRPInstanceKey) error
}

type EventDB interface {
	WatchForActualLRPChanges(lager.Logger,
		func(created *models.ActualLRPGroup),
		func(changed *models.ActualLRPChange),
		func(deleted *models.ActualLRPGroup)) (chan<- bool, <-chan error)
	WatchForDesiredLRPChanges(lager.Logger,
		func(created *models.DesiredLRP),
		func(changed *models.DesiredLRPChange),
		func(deleted *models.DesiredLRP)) (chan<- bool, <-chan error)
	WatchForTaskChanges(lager.Logger,
		func(created *models.Task),
		func(changed *models.TaskChange),
		func(deleted *models.Task)) (chan<- bool, <-chan error)
}

type LRPDB interface {
	ActualLRPDB
	DesiredLRPDB

	ConvergeLRPs(logger lager.Logger)

	// Exposed For Test
	GatherAndPruneLRPs(logger lager.Logger) (*models.ConvergenceInput, error)
}

type TaskDB interface {
	Tasks(logger lager.Logger, filter models.TaskFilter) ([]*models.Task, error)
	TaskByGuid(logger lager.Logger, taskGuid string) (*models.Task, error)

	DesireTask(logger lager.Logger, taskDefinition *models.TaskDefinition, taskGuid, domain string) error
	StartTask(logger lager.Logger, taskGuid, cellId string) (bool, error)
	CancelTask(logger lager.Logger, taskGuid string) error
	FailTask(logger lager.Logger, taskGuid, failureReason string) error
	CompleteTask(logger lager.Logger, taskGuid, cellId string, failed bool, failureReason, result string) error
	ResolvingTask(logger lager.Logger, taskGuid string) error
	DeleteTask(logger lager.Logger, taskGuid string) error

	ConvergeTasks(
		logger lager.Logger,
		kickTaskDuration, expirePendingTaskDuration, expireCompletedTaskDuration time.Duration,
	)
}

type VersionDB interface {
	Version(logger lager.Logger) (*models.Version, error)
	SetVersion(logger lager.Logger, version *models.Version) error
}