service

type CPTaskWatcher interface {
	datastructure.Observer[models.TaskEventBus]

	AddTaskToWatch(taskID primitive.ObjectID)
	WatcherLoop(ctx context.Context)
}

type ContainerService struct {
	// contains filtered or unexported fields
}

type DynamicScalingService struct {
	// contains filtered or unexported fields
}

type IContainer interface {
	StartContainer(ctx context.Context, imageUrl string) (container.IContainer, error)
	GetContainerIdShort() []string
	GetContainerCoolDownState() datastructure.Buffer[string, time.Time]
	GetContainerBuffer() datastructure.Buffer[string, container.IContainer]
	DownContainer(ctx context.Context, container container.IContainer) error
	AddContainerTakeDownTimer(containerId string) error
	RemoveContainerTakeDownTimer(containerImage string)
}

type IDynamicScaling interface {
	CheckResourceUsageLimit(ctx context.Context) (*models.CheckResourceReport, error)
	CheckResourceUsageLimitWithTimeBuffer(ctx context.Context) (*models.CheckResourceReport, error)
}

type IResourceMonitor interface {
	GetResourceUsage() ([]models.ContainerResourceUsage, error)
	GetSystemMemUsage() (*models.MemoryUsage, error)
	GetSystemCpuUsage(ctx context.Context) (*models.CpuUsage, error)
	CalculateAvailableResource(ctx context.Context) (*models.AvailableResource, error)
}

type Job interface {
	GetJob(ctx context.Context, id primitive.ObjectID) (*models.Job, error)
	CreateJob(ctx context.Context, name, imageURL string, isExperiment bool, distributionLogic models.DistributorName) (*models.Job, error)
	ListJob(ctx context.Context) ([]models.Job, error)
	ListJobReadyToDistribute(ctx context.Context) ([]models.Job, error)
	UpdateJobStatusToFinish(ctx context.Context, id *primitive.ObjectID) error
	UpdateJobStatusToDistributing(ctx context.Context, id *primitive.ObjectID) error
	UpdateJobStatusToExperimenting(ctx context.Context, id *primitive.ObjectID) error
	UpdateJobToFailed(ctx context.Context, id *primitive.ObjectID, message string, inputErr error) error
}

type ResourceMonitor struct {
	// contains filtered or unexported fields
}

type Task interface {
	GetAvailableTask(ctx context.Context, jobIDs []models.Job) (*models.Job, []models.Task, error)
	UpdateTaskAfterDistribution(ctx context.Context, successTasks []models.Task, errorTasks []models.DistributeError) error
	UpdateTaskWorkOnFailure(ctx context.Context, taskID primitive.ObjectID, nodeID string, errMessage string) error
	UpdateTaskSuccess(ctx context.Context, taskID primitive.ObjectID, nodeID string, result []byte, averageCPUUsage float32, averageMemoryUsage float64) error
	UpdateTaskWaitTimeout(ctx context.Context, taskID primitive.ObjectID) error
	CreateTask(ctx context.Context, job *models.Job, taskAttributes [][]byte, isExperiment bool) ([]models.Task, error)
	GetTaskByJob(ctx context.Context, job *models.Job) ([]models.Task, error)
	GetTaskByID(ctx context.Context, taskID primitive.ObjectID) (*models.Task, error)
	GetAverageResourceUsage(ctx context.Context, jobID *primitive.ObjectID) (*models.TaskResourceUsage, error)
	UpdateTaskToBeReadyToBeDistributed(ctx context.Context, jobID *primitive.ObjectID) error
	CountUnfinishedTaskByJobID(ctx context.Context, jobID *primitive.ObjectID) (int64, error)
	UpdateAllTaskToWorkOnFailure(ctx context.Context, job *models.Job, jobErrorMessage string) error
	GetAllDistributedTask(ctx context.Context) ([]models.Task, error)
}