dragonfly/cdn/supervisor/task_mgr.go

46 lines
1.5 KiB
Go

/*
* Copyright 2020 The Dragonfly Authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
//go:generate mockgen -destination ./mock/mock_task_mgr.go -package mock d7y.io/dragonfly/v2/cdn/supervisor SeedTaskMgr
package supervisor
import (
"context"
"d7y.io/dragonfly/v2/cdn/types"
)
// SeedTaskMgr as an interface defines all operations against SeedTask.
// A SeedTask will store some meta info about the taskFile, pieces and something else.
// A seedTask corresponds to three files on the disk, which are identified by taskId, the data file meta file piece file
type SeedTaskMgr interface {
// Register register seed task
Register(context.Context, *types.SeedTask) (pieceCh <-chan *types.SeedPiece, err error)
// Get get task Info with specified taskId.
Get(string) (*types.SeedTask, error)
// Exist check task existence with specified taskId.
Exist(string) (*types.SeedTask, bool)
// Delete delete a task.
Delete(string) error
// GetPieces
GetPieces(context.Context, string) (pieces []*types.SeedPiece, err error)
}