Files
crawlab/core/models/service/interface.go
2024-06-14 15:59:48 +08:00

106 lines
8.0 KiB
Go

package service
import (
"github.com/crawlab-team/crawlab/core/interfaces"
"github.com/crawlab-team/crawlab/core/models/models"
"github.com/crawlab-team/crawlab/db/mongo"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
)
type ModelService interface {
interfaces.ModelService
DropAll() (err error)
GetNodeById(id primitive.ObjectID) (res *models.Node, err error)
GetNode(query bson.M, opts *mongo.FindOptions) (res *models.Node, err error)
GetNodeList(query bson.M, opts *mongo.FindOptions) (res []models.Node, err error)
GetNodeByKey(key string, opts *mongo.FindOptions) (res *models.Node, err error)
GetProjectById(id primitive.ObjectID) (res *models.Project, err error)
GetProject(query bson.M, opts *mongo.FindOptions) (res *models.Project, err error)
GetProjectList(query bson.M, opts *mongo.FindOptions) (res []models.Project, err error)
GetArtifactById(id primitive.ObjectID) (res *models.Artifact, err error)
GetArtifact(query bson.M, opts *mongo.FindOptions) (res *models.Artifact, err error)
GetArtifactList(query bson.M, opts *mongo.FindOptions) (res []models.Artifact, err error)
GetTagById(id primitive.ObjectID) (res *models.Tag, err error)
GetTag(query bson.M, opts *mongo.FindOptions) (res *models.Tag, err error)
GetTagList(query bson.M, opts *mongo.FindOptions) (res []models.Tag, err error)
GetTagIds(colName string, tags []interfaces.Tag) (tagIds []primitive.ObjectID, err error)
UpdateTagsById(colName string, id primitive.ObjectID, tags []interfaces.Tag) (tagIds []primitive.ObjectID, err error)
UpdateTags(colName string, query bson.M, tags []interfaces.Tag) (tagIds []primitive.ObjectID, err error)
GetJobById(id primitive.ObjectID) (res *models.Job, err error)
GetJob(query bson.M, opts *mongo.FindOptions) (res *models.Job, err error)
GetJobList(query bson.M, opts *mongo.FindOptions) (res []models.Job, err error)
GetScheduleById(id primitive.ObjectID) (res *models.Schedule, err error)
GetSchedule(query bson.M, opts *mongo.FindOptions) (res *models.Schedule, err error)
GetScheduleList(query bson.M, opts *mongo.FindOptions) (res []models.Schedule, err error)
GetUserById(id primitive.ObjectID) (res *models.User, err error)
GetUser(query bson.M, opts *mongo.FindOptions) (res *models.User, err error)
GetUserList(query bson.M, opts *mongo.FindOptions) (res []models.User, err error)
GetUserByUsername(username string, opts *mongo.FindOptions) (res *models.User, err error)
GetUserByUsernameWithPassword(username string, opts *mongo.FindOptions) (res *models.User, err error)
GetSettingById(id primitive.ObjectID) (res *models.Setting, err error)
GetSetting(query bson.M, opts *mongo.FindOptions) (res *models.Setting, err error)
GetSettingList(query bson.M, opts *mongo.FindOptions) (res []models.Setting, err error)
GetSettingByKey(key string, opts *mongo.FindOptions) (res *models.Setting, err error)
GetSpiderById(id primitive.ObjectID) (res *models.Spider, err error)
GetSpider(query bson.M, opts *mongo.FindOptions) (res *models.Spider, err error)
GetSpiderList(query bson.M, opts *mongo.FindOptions) (res []models.Spider, err error)
GetTaskById(id primitive.ObjectID) (res *models.Task, err error)
GetTask(query bson.M, opts *mongo.FindOptions) (res *models.Task, err error)
GetTaskList(query bson.M, opts *mongo.FindOptions) (res []models.Task, err error)
GetTokenById(id primitive.ObjectID) (res *models.Token, err error)
GetToken(query bson.M, opts *mongo.FindOptions) (res *models.Token, err error)
GetTokenList(query bson.M, opts *mongo.FindOptions) (res []models.Token, err error)
GetVariableById(id primitive.ObjectID) (res *models.Variable, err error)
GetVariable(query bson.M, opts *mongo.FindOptions) (res *models.Variable, err error)
GetVariableList(query bson.M, opts *mongo.FindOptions) (res []models.Variable, err error)
GetVariableByKey(key string, opts *mongo.FindOptions) (res *models.Variable, err error)
GetTaskQueueItemById(id primitive.ObjectID) (res *models.TaskQueueItem, err error)
GetTaskQueueItem(query bson.M, opts *mongo.FindOptions) (res *models.TaskQueueItem, err error)
GetTaskQueueItemList(query bson.M, opts *mongo.FindOptions) (res []models.TaskQueueItem, err error)
GetTaskStatById(id primitive.ObjectID) (res *models.TaskStat, err error)
GetTaskStat(query bson.M, opts *mongo.FindOptions) (res *models.TaskStat, err error)
GetTaskStatList(query bson.M, opts *mongo.FindOptions) (res []models.TaskStat, err error)
GetSpiderStatById(id primitive.ObjectID) (res *models.SpiderStat, err error)
GetSpiderStat(query bson.M, opts *mongo.FindOptions) (res *models.SpiderStat, err error)
GetSpiderStatList(query bson.M, opts *mongo.FindOptions) (res []models.SpiderStat, err error)
GetDataSourceById(id primitive.ObjectID) (res *models.DataSource, err error)
GetDataSource(query bson.M, opts *mongo.FindOptions) (res *models.DataSource, err error)
GetDataSourceList(query bson.M, opts *mongo.FindOptions) (res []models.DataSource, err error)
GetDataCollectionById(id primitive.ObjectID) (res *models.DataCollection, err error)
GetDataCollection(query bson.M, opts *mongo.FindOptions) (res *models.DataCollection, err error)
GetDataCollectionList(query bson.M, opts *mongo.FindOptions) (res []models.DataCollection, err error)
GetDataCollectionByName(name string, opts *mongo.FindOptions) (res *models.DataCollection, err error)
GetPasswordById(id primitive.ObjectID) (res *models.Password, err error)
GetPassword(query bson.M, opts *mongo.FindOptions) (res *models.Password, err error)
GetPasswordList(query bson.M, opts *mongo.FindOptions) (res []models.Password, err error)
GetExtraValueById(id primitive.ObjectID) (res *models.ExtraValue, err error)
GetExtraValue(query bson.M, opts *mongo.FindOptions) (res *models.ExtraValue, err error)
GetExtraValueList(query bson.M, opts *mongo.FindOptions) (res []models.ExtraValue, err error)
GetExtraValueByObjectIdModel(oid primitive.ObjectID, m string, opts *mongo.FindOptions) (res *models.ExtraValue, err error)
GetGitById(id primitive.ObjectID) (res *models.Git, err error)
GetGit(query bson.M, opts *mongo.FindOptions) (res *models.Git, err error)
GetGitList(query bson.M, opts *mongo.FindOptions) (res []models.Git, err error)
GetRoleById(id primitive.ObjectID) (res *models.Role, err error)
GetRole(query bson.M, opts *mongo.FindOptions) (res *models.Role, err error)
GetRoleList(query bson.M, opts *mongo.FindOptions) (res []models.Role, err error)
GetRoleByName(name string, opts *mongo.FindOptions) (res *models.Role, err error)
GetRoleByKey(key string, opts *mongo.FindOptions) (res *models.Role, err error)
GetUserRoleById(id primitive.ObjectID) (res *models.UserRole, err error)
GetUserRole(query bson.M, opts *mongo.FindOptions) (res *models.UserRole, err error)
GetUserRoleList(query bson.M, opts *mongo.FindOptions) (res []models.UserRole, err error)
GetUserRoleListByUserId(id primitive.ObjectID, opts *mongo.FindOptions) (res []models.UserRole, err error)
GetUserRoleListByRoleId(id primitive.ObjectID, opts *mongo.FindOptions) (res []models.UserRole, err error)
GetPermissionById(id primitive.ObjectID) (res *models.Permission, err error)
GetPermission(query bson.M, opts *mongo.FindOptions) (res *models.Permission, err error)
GetPermissionList(query bson.M, opts *mongo.FindOptions) (res []models.Permission, err error)
GetPermissionByKey(key string, opts *mongo.FindOptions) (res *models.Permission, err error)
GetRolePermission(query bson.M, opts *mongo.FindOptions) (res *models.RolePermission, err error)
GetRolePermissionList(query bson.M, opts *mongo.FindOptions) (res []models.RolePermission, err error)
GetRolePermissionListByRoleId(id primitive.ObjectID, opts *mongo.FindOptions) (res []models.RolePermission, err error)
GetRolePermissionListByPermissionId(id primitive.ObjectID, opts *mongo.FindOptions) (res []models.RolePermission, err error)
GetEnvironmentById(id primitive.ObjectID) (res *models.Environment, err error)
GetEnvironment(query bson.M, opts *mongo.FindOptions) (res *models.Environment, err error)
GetEnvironmentList(query bson.M, opts *mongo.FindOptions) (res []models.Environment, err error)
}