resource as resource named
This commit is contained in:
@@ -3,12 +3,21 @@ package utils
|
||||
import (
|
||||
"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/resources"
|
||||
"github.com/go-playground/validator/v10"
|
||||
"github.com/google/uuid"
|
||||
"github.com/rs/zerolog"
|
||||
)
|
||||
|
||||
var Validate = validator.New(validator.WithRequiredStructEnabled())
|
||||
var validate = validator.New(validator.WithRequiredStructEnabled())
|
||||
|
||||
type AbstractObject struct {
|
||||
UUID string `json:"id" required:"true" bson:"_id" validate:"required"`
|
||||
Name string `json:"name" required:"true" bson:"name" validate:"required"`
|
||||
}
|
||||
|
||||
func (r *AbstractObject) GenerateID() {
|
||||
r.UUID = uuid.New().String()
|
||||
}
|
||||
|
||||
type AbstractAccessor struct {
|
||||
Logger zerolog.Logger
|
||||
@@ -19,10 +28,26 @@ func (dma *AbstractAccessor) GetType() string {
|
||||
return dma.Type
|
||||
}
|
||||
|
||||
func (dma *AbstractAccessor) SetLogger(t resources.ResourceType) {
|
||||
func (dma *AbstractAccessor) SetLogger(t DataType) {
|
||||
dma.Logger = logs.CreateLogger(t.String(), "")
|
||||
dma.Type = t.String()
|
||||
}
|
||||
|
||||
func (wfa *AbstractAccessor) GenericStoreOne(data DBObject, accessor Accessor) (DBObject, error) {
|
||||
err := validate.Struct(data)
|
||||
wfa.Logger.Error().Msg("Could not store " + data.GetName() + " to db. Error: " + err.Error())
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
data.GenerateID()
|
||||
id, err := mongo.MONGOService.StoreOne(data, wfa.GetType())
|
||||
if err != nil {
|
||||
wfa.Logger.Error().Msg("Could not store " + data.GetName() + " to db. Error: " + err.Error())
|
||||
return nil, err
|
||||
}
|
||||
return accessor.LoadOne(id)
|
||||
}
|
||||
|
||||
func (dma *AbstractAccessor) GenericDeleteOne(id string, accessor Accessor) (DBObject, error) {
|
||||
res, err := accessor.LoadOne(id)
|
||||
if err != nil {
|
||||
|
36
models/utils/enums.go
Normal file
36
models/utils/enums.go
Normal file
@@ -0,0 +1,36 @@
|
||||
package utils
|
||||
|
||||
type DataType int
|
||||
|
||||
const (
|
||||
INVALID DataType = iota
|
||||
DATA_RESOURCE
|
||||
PROCESSING_RESOURCE
|
||||
STORAGE_RESOURCE
|
||||
DATACENTER_RESOURCE
|
||||
WORKFLOW_RESOURCE
|
||||
WORKFLOW
|
||||
)
|
||||
|
||||
var str = [...]string{
|
||||
"invalid",
|
||||
"data_resource",
|
||||
"processing_resource",
|
||||
"storage_resource",
|
||||
"datacenter_resource",
|
||||
"workflow_resource",
|
||||
"workflow",
|
||||
}
|
||||
|
||||
func FromInt(i int) string {
|
||||
return str[i]
|
||||
}
|
||||
|
||||
func (d DataType) String() string {
|
||||
return str[d]
|
||||
}
|
||||
|
||||
// EnumIndex - Creating common behavior - give the type a EnumIndex functio
|
||||
func (d DataType) EnumIndex() int {
|
||||
return int(d)
|
||||
}
|
@@ -1,8 +1,7 @@
|
||||
package utils
|
||||
|
||||
import "cloud.o-forge.io/core/oc-lib/models/resources"
|
||||
|
||||
type DBObject interface {
|
||||
GenerateID()
|
||||
GetName() string
|
||||
Deserialize(j map[string]interface{}) DBObject
|
||||
Serialize() map[string]interface{}
|
||||
@@ -10,7 +9,7 @@ type DBObject interface {
|
||||
}
|
||||
|
||||
type Accessor interface {
|
||||
SetLogger(t resources.ResourceType)
|
||||
SetLogger(t DataType)
|
||||
GetType() string
|
||||
LoadOne(id string) (DBObject, error)
|
||||
DeleteOne(id string) (DBObject, error)
|
||||
|
Reference in New Issue
Block a user