mirror of
https://github.com/crawlab-team/crawlab.git
synced 2026-01-22 17:31:03 +01:00
93 lines
2.5 KiB
Go
93 lines
2.5 KiB
Go
package service
|
|
|
|
import (
|
|
"github.com/crawlab-team/crawlab/core/errors"
|
|
"github.com/crawlab-team/crawlab/core/interfaces"
|
|
"github.com/crawlab-team/crawlab/core/models/models"
|
|
"github.com/crawlab-team/crawlab/db/mongo"
|
|
)
|
|
|
|
func NewBasicBinder(id interfaces.ModelId, fr *mongo.FindResult) (b interfaces.ModelBinder) {
|
|
return &BasicBinder{
|
|
id: id,
|
|
fr: fr,
|
|
m: models.NewModelMap(),
|
|
}
|
|
}
|
|
|
|
type BasicBinder struct {
|
|
id interfaces.ModelId
|
|
fr *mongo.FindResult
|
|
m *models.ModelMap
|
|
}
|
|
|
|
func (b *BasicBinder) Bind() (res interfaces.Model, err error) {
|
|
m := b.m
|
|
|
|
switch b.id {
|
|
case interfaces.ModelIdArtifact:
|
|
return b.Process(&m.Artifact)
|
|
case interfaces.ModelIdTag:
|
|
return b.Process(&m.Tag)
|
|
case interfaces.ModelIdNode:
|
|
return b.Process(&m.Node)
|
|
case interfaces.ModelIdProject:
|
|
return b.Process(&m.Project)
|
|
case interfaces.ModelIdSpider:
|
|
return b.Process(&m.Spider)
|
|
case interfaces.ModelIdTask:
|
|
return b.Process(&m.Task)
|
|
case interfaces.ModelIdJob:
|
|
return b.Process(&m.Job)
|
|
case interfaces.ModelIdSchedule:
|
|
return b.Process(&m.Schedule)
|
|
case interfaces.ModelIdUser:
|
|
return b.Process(&m.User)
|
|
case interfaces.ModelIdSetting:
|
|
return b.Process(&m.Setting)
|
|
case interfaces.ModelIdToken:
|
|
return b.Process(&m.Token)
|
|
case interfaces.ModelIdVariable:
|
|
return b.Process(&m.Variable)
|
|
case interfaces.ModelIdTaskQueue:
|
|
return b.Process(&m.TaskQueueItem)
|
|
case interfaces.ModelIdTaskStat:
|
|
return b.Process(&m.TaskStat)
|
|
case interfaces.ModelIdSpiderStat:
|
|
return b.Process(&m.SpiderStat)
|
|
case interfaces.ModelIdDataSource:
|
|
return b.Process(&m.DataSource)
|
|
case interfaces.ModelIdDataCollection:
|
|
return b.Process(&m.DataCollection)
|
|
case interfaces.ModelIdResult:
|
|
return b.Process(&m.Result)
|
|
case interfaces.ModelIdPassword:
|
|
return b.Process(&m.Password)
|
|
case interfaces.ModelIdExtraValue:
|
|
return b.Process(&m.ExtraValue)
|
|
case interfaces.ModelIdGit:
|
|
return b.Process(&m.Git)
|
|
case interfaces.ModelIdRole:
|
|
return b.Process(&m.Role)
|
|
case interfaces.ModelIdUserRole:
|
|
return b.Process(&m.UserRole)
|
|
case interfaces.ModelIdPermission:
|
|
return b.Process(&m.Permission)
|
|
case interfaces.ModelIdRolePermission:
|
|
return b.Process(&m.RolePermission)
|
|
case interfaces.ModelIdEnvironment:
|
|
return b.Process(&m.Environment)
|
|
case interfaces.ModelIdDependencySetting:
|
|
return b.Process(&m.DependencySetting)
|
|
default:
|
|
return nil, errors.ErrorModelInvalidModelId
|
|
}
|
|
}
|
|
|
|
func (b *BasicBinder) Process(d interfaces.Model) (res interfaces.Model, err error) {
|
|
if err := b.fr.One(d); err != nil {
|
|
return nil, err
|
|
}
|
|
return d, nil
|
|
}
|