114 lines
3.8 KiB
Go
114 lines
3.8 KiB
Go
package booking
|
|
|
|
import (
|
|
"time"
|
|
|
|
"cloud.o-forge.io/core/oc-lib/dbs"
|
|
"cloud.o-forge.io/core/oc-lib/dbs/mongo"
|
|
"cloud.o-forge.io/core/oc-lib/logs"
|
|
"cloud.o-forge.io/core/oc-lib/models/utils"
|
|
"cloud.o-forge.io/core/oc-lib/models/workflow_execution"
|
|
"cloud.o-forge.io/core/oc-lib/tools"
|
|
)
|
|
|
|
type bookingMongoAccessor struct {
|
|
utils.AbstractAccessor // AbstractAccessor contains the basic fields of an accessor (model, caller)
|
|
}
|
|
|
|
// New creates a new instance of the bookingMongoAccessor
|
|
func New(t tools.DataType, peerID string, groups []string, caller *tools.HTTPCaller) *bookingMongoAccessor {
|
|
return &bookingMongoAccessor{
|
|
AbstractAccessor: utils.AbstractAccessor{
|
|
Logger: logs.CreateLogger(t.String()), // Create a logger with the data type
|
|
Caller: caller,
|
|
PeerID: peerID,
|
|
Groups: groups, // Set the caller
|
|
Type: t.String(),
|
|
},
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Nothing special here, just the basic CRUD operations
|
|
*/
|
|
func (wfa *bookingMongoAccessor) DeleteOne(id string) (utils.DBObject, int, error) {
|
|
return wfa.GenericDeleteOne(id, wfa)
|
|
}
|
|
|
|
func (wfa *bookingMongoAccessor) UpdateOne(set utils.DBObject, id string) (utils.DBObject, int, error) {
|
|
return wfa.GenericUpdateOne(set, id, wfa, &Booking{})
|
|
}
|
|
|
|
func (wfa *bookingMongoAccessor) StoreOne(data utils.DBObject) (utils.DBObject, int, error) {
|
|
return wfa.GenericStoreOne(data, wfa)
|
|
}
|
|
|
|
func (wfa *bookingMongoAccessor) CopyOne(data utils.DBObject) (utils.DBObject, int, error) {
|
|
return wfa.GenericStoreOne(data, wfa)
|
|
}
|
|
|
|
func (wfa *bookingMongoAccessor) LoadOne(id string) (utils.DBObject, int, error) {
|
|
var workflow Booking
|
|
res_mongo, code, err := mongo.MONGOService.LoadOne(id, wfa.GetType())
|
|
if err != nil {
|
|
wfa.Logger.Error().Msg("Could not retrieve " + id + " from db. Error: " + err.Error())
|
|
return nil, code, err
|
|
}
|
|
res_mongo.Decode(&workflow)
|
|
if workflow.State == workflow_execution.SCHEDULED && time.Now().UTC().After(*workflow.ExecDate) {
|
|
workflow.State = workflow_execution.FORGOTTEN
|
|
wfa.GenericRawUpdateOne(&workflow, id, wfa)
|
|
}
|
|
return &workflow, 200, nil
|
|
}
|
|
|
|
func (wfa *bookingMongoAccessor) LoadAll() ([]utils.ShallowDBObject, int, error) {
|
|
objs := []utils.ShallowDBObject{}
|
|
res_mongo, code, err := mongo.MONGOService.LoadAll(wfa.GetType())
|
|
if err != nil {
|
|
wfa.Logger.Error().Msg("Could not retrieve any from db. Error: " + err.Error())
|
|
return nil, code, err
|
|
}
|
|
var results []Booking
|
|
if err = res_mongo.All(mongo.MngoCtx, &results); err != nil {
|
|
return nil, 404, err
|
|
}
|
|
for _, r := range results {
|
|
if r.State == workflow_execution.SCHEDULED && time.Now().UTC().After(*r.ExecDate) {
|
|
r.State = workflow_execution.FORGOTTEN
|
|
wfa.GenericRawUpdateOne(&r, r.UUID, wfa)
|
|
}
|
|
objs = append(objs, &r.AbstractObject) // Warning only AbstractObject is returned
|
|
}
|
|
return objs, 200, nil
|
|
}
|
|
|
|
// Search is a function that searches for a booking in the database
|
|
func (wfa *bookingMongoAccessor) Search(filters *dbs.Filters, search string) ([]utils.ShallowDBObject, int, error) {
|
|
objs := []utils.ShallowDBObject{}
|
|
if (filters == nil || len(filters.And) == 0 || len(filters.Or) == 0) && search != "" {
|
|
filters = &dbs.Filters{
|
|
Or: map[string][]dbs.Filter{ // filter by name if no filters are provided
|
|
"abstractobject.name": {{Operator: dbs.LIKE.String(), Value: search}},
|
|
},
|
|
}
|
|
}
|
|
res_mongo, code, err := mongo.MONGOService.Search(filters, wfa.GetType())
|
|
if err != nil {
|
|
wfa.Logger.Error().Msg("Could not store to db. Error: " + err.Error())
|
|
return nil, code, err
|
|
}
|
|
var results []Booking
|
|
if err = res_mongo.All(mongo.MngoCtx, &results); err != nil {
|
|
return nil, 404, err
|
|
}
|
|
for _, r := range results {
|
|
if r.State == workflow_execution.SCHEDULED && time.Now().UTC().After(*r.ExecDate) {
|
|
r.State = workflow_execution.FORGOTTEN
|
|
wfa.GenericRawUpdateOne(&r, r.UUID, wfa)
|
|
}
|
|
objs = append(objs, &r)
|
|
}
|
|
return objs, 200, nil
|
|
}
|