mirror of
https://github.com/crawlab-team/crawlab.git
synced 2026-01-22 17:31:03 +01:00
119 lines
2.7 KiB
Go
119 lines
2.7 KiB
Go
package ds
|
|
|
|
import (
|
|
"github.com/crawlab-team/crawlab/core/constants"
|
|
"github.com/crawlab-team/crawlab/core/interfaces"
|
|
"github.com/crawlab-team/crawlab/core/models/models"
|
|
"github.com/crawlab-team/crawlab/core/models/service"
|
|
"github.com/crawlab-team/crawlab/core/utils"
|
|
utils2 "github.com/crawlab-team/crawlab/core/utils"
|
|
"github.com/crawlab-team/crawlab/db/generic"
|
|
"github.com/crawlab-team/crawlab/db/mongo"
|
|
"go.mongodb.org/mongo-driver/bson/primitive"
|
|
mongo2 "go.mongodb.org/mongo-driver/mongo"
|
|
"time"
|
|
)
|
|
|
|
type MongoService struct {
|
|
// dependencies
|
|
modelSvc service.ModelService
|
|
|
|
// internals
|
|
dc *models.DataCollection // models.DataCollection
|
|
ds *models.DataSource // models.DataSource
|
|
c *mongo2.Client
|
|
db *mongo2.Database
|
|
col *mongo.Col
|
|
t time.Time
|
|
}
|
|
|
|
func (svc *MongoService) Insert(records ...interface{}) (err error) {
|
|
_, err = svc.col.InsertMany(records)
|
|
return err
|
|
}
|
|
|
|
func (svc *MongoService) List(query generic.ListQuery, opts *generic.ListOptions) (results []interface{}, err error) {
|
|
var docs []models.Result
|
|
if err := svc.col.Find(utils.GetMongoQuery(query), utils.GetMongoOpts(opts)).All(&docs); err != nil {
|
|
return nil, err
|
|
}
|
|
for i := range docs {
|
|
results = append(results, &docs[i])
|
|
}
|
|
return results, nil
|
|
}
|
|
|
|
func (svc *MongoService) Count(query generic.ListQuery) (n int, err error) {
|
|
return svc.col.Count(utils.GetMongoQuery(query))
|
|
}
|
|
|
|
func NewDataSourceMongoService(colId primitive.ObjectID, dsId primitive.ObjectID) (svc2 interfaces.ResultService, err error) {
|
|
// service
|
|
svc := &MongoService{}
|
|
|
|
// dependency injection
|
|
svc.modelSvc, err = service.GetService()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// data source
|
|
if dsId.IsZero() {
|
|
svc.ds = &models.DataSource{}
|
|
} else {
|
|
svc.ds, err = svc.modelSvc.GetDataSourceById(dsId)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
// data source defaults
|
|
if svc.ds.Host == "" {
|
|
svc.ds.Host = constants.DefaultHost
|
|
}
|
|
if svc.ds.Port == 0 {
|
|
svc.ds.Port = constants.DefaultMongoPort
|
|
}
|
|
|
|
// data source password
|
|
pwd, err := svc.modelSvc.GetPasswordById(svc.ds.Id)
|
|
if err == nil {
|
|
svc.ds.Password, err = utils.DecryptAES(pwd.Password)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
// data collection
|
|
svc.dc, err = svc.modelSvc.GetDataCollectionById(colId)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// mongo client
|
|
svc.c, err = utils2.GetMongoClient(svc.ds)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// mongo database
|
|
svc.db = mongo.GetMongoDb(svc.ds.Database, mongo.WithDbClient(svc.c))
|
|
|
|
// mongo col
|
|
svc.col = mongo.GetMongoColWithDb(svc.dc.Name, svc.db)
|
|
|
|
return svc, nil
|
|
}
|
|
|
|
func (svc *MongoService) Index(fields []string) {
|
|
// TODO: implement me
|
|
}
|
|
|
|
func (svc *MongoService) SetTime(t time.Time) {
|
|
svc.t = t
|
|
}
|
|
|
|
func (svc *MongoService) GetTime() (t time.Time) {
|
|
return svc.t
|
|
}
|