48 lines
2.3 KiB
Go
48 lines
2.3 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"remote-task-excutor-cli/pkg/models"
|
|
)
|
|
|
|
// AuthService 处理认证相关操作
|
|
type AuthService interface {
|
|
GetToken(ctx context.Context) (*models.AuthData, error)
|
|
GetClusterID(ctx context.Context, label string) (string, error)
|
|
}
|
|
|
|
type PreparationService interface {
|
|
PrepareAll(ctx context.Context, authData *models.AuthData, config *models.RunConfig,
|
|
clusterID string) (*models.BindResultSet, error)
|
|
PrepareCode(ctx context.Context, authData *models.AuthData, config *models.RunConfig, clientID string) (int, error)
|
|
PrepareDataset(ctx context.Context, authData *models.AuthData, config models.DataResourceConfig, clientID string) (int, error)
|
|
PrepareModel(ctx context.Context, authData *models.AuthData, config models.DataResourceConfig, clientID string) (int, error)
|
|
}
|
|
|
|
type TaskService interface {
|
|
SubmitTask(ctx context.Context, authData *models.AuthData, config *models.RunConfig,
|
|
clusterID string, bindResultSet *models.BindResultSet) (string, error)
|
|
GetTaskStatus(ctx context.Context, authData *models.AuthData, jobSetID string) (*models.TaskDetailResponse, error)
|
|
GetTaskResult(ctx context.Context, authData *models.AuthData, path string, jobSetID string) error
|
|
//GetTaskLogs(ctx context.Context, authData *models.AuthData, jobSetID string) (string, error)
|
|
}
|
|
|
|
type InferenceService interface {
|
|
SubmitTask(ctx context.Context, authData *models.AuthData, config *models.RunConfig,
|
|
clusterID string, bindResultSet *models.BindResultSet) (string, error)
|
|
SubmitInferenceTask(ctx context.Context, authData *models.AuthData,
|
|
request *models.InferenceSubmitTaskRequest, clusterID string) (*models.InferenceSubmitTaskResponse, error)
|
|
GetTaskStatus(ctx context.Context, authData *models.AuthData, jobSetID string) (*models.InferenceTaskDetailResponse, error)
|
|
StopTask(ctx context.Context, authData *models.AuthData, jobSetID string) error
|
|
}
|
|
|
|
type LogService interface {
|
|
RecordLogToFile(ctx context.Context, authData *models.AuthData, jobSetID string, done chan bool) error
|
|
}
|
|
|
|
type UpDownService interface {
|
|
Upload(ctx context.Context, authData *models.AuthData, filesConfig *models.UploadFileConfig,
|
|
info models.PackageCreateLoadInfo) (*models.PackageCreateResponse, error)
|
|
DownLoad(ctx context.Context, authData *models.AuthData, path string, packageID int) (string, error)
|
|
}
|