feat: added modules

core/models/service/binder_basic.go

@@ -0,0 +1,92 @@
+package service
+
+import (
+	"github.com/crawlab-team/crawlab-db/mongo"
+	"github.com/crawlab-team/crawlab/core/errors"
+	"github.com/crawlab-team/crawlab/core/interfaces"
+	"github.com/crawlab-team/crawlab/core/models/models"
+)
+
+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
+}