98cbe6a67d
Tweak the logging a little bit so it's less noisy, properly warns when the Shaman checkout dir cannot be removed, and optimise the database query a bit (by just fetching the one field that's needed, instead of the entire job). Deletion still works the same.
60 lines
2.1 KiB
Go
60 lines
2.1 KiB
Go
package job_deleter
|
|
|
|
// SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"projects.blender.org/studio/flamenco/internal/manager/eventbus"
|
|
"projects.blender.org/studio/flamenco/internal/manager/local_storage"
|
|
"projects.blender.org/studio/flamenco/internal/manager/persistence"
|
|
"projects.blender.org/studio/flamenco/pkg/api"
|
|
"projects.blender.org/studio/flamenco/pkg/shaman"
|
|
)
|
|
|
|
// Generate mock implementations of these interfaces.
|
|
//go:generate go run github.com/golang/mock/mockgen -destination mocks/interfaces_mock.gen.go -package mocks projects.blender.org/studio/flamenco/internal/manager/job_deleter PersistenceService,Storage,ChangeBroadcaster,Shaman
|
|
|
|
type PersistenceService interface {
|
|
FetchJob(ctx context.Context, jobUUID string) (*persistence.Job, error)
|
|
FetchJobShamanCheckoutID(ctx context.Context, jobUUID string) (string, error)
|
|
|
|
RequestJobDeletion(ctx context.Context, j *persistence.Job) error
|
|
RequestJobMassDeletion(ctx context.Context, lastUpdatedMax time.Time) ([]string, error)
|
|
|
|
// FetchJobsDeletionRequested returns the UUIDs of to-be-deleted jobs.
|
|
FetchJobsDeletionRequested(ctx context.Context) ([]string, error)
|
|
DeleteJob(ctx context.Context, jobUUID string) error
|
|
|
|
RequestIntegrityCheck()
|
|
}
|
|
|
|
// PersistenceService should be a subset of persistence.DB
|
|
var _ PersistenceService = (*persistence.DB)(nil)
|
|
|
|
type Storage interface {
|
|
// RemoveJobStorage removes from disk the directory for storing job-related files.
|
|
RemoveJobStorage(ctx context.Context, jobUUID string) error
|
|
}
|
|
|
|
var _ Storage = (*local_storage.StorageInfo)(nil)
|
|
|
|
type ChangeBroadcaster interface {
|
|
// BroadcastJobUpdate sends the job update to SocketIO clients.
|
|
BroadcastJobUpdate(jobUpdate api.EventJobUpdate)
|
|
}
|
|
|
|
// ChangeBroadcaster should be a subset of eventbus.Broker
|
|
var _ ChangeBroadcaster = (*eventbus.Broker)(nil)
|
|
|
|
type Shaman interface {
|
|
// IsEnabled returns whether this Shaman service is enabled or not.
|
|
IsEnabled() bool
|
|
|
|
// EraseCheckout deletes the symlinks and the directory structure that makes up the checkout.
|
|
EraseCheckout(checkoutID string) error
|
|
}
|
|
|
|
var _ Shaman = (*shaman.Server)(nil)
|