core/oc-lib
6
0
Fork 0
Code Issues 7 Pull Requests Packages Projects Releases Wiki Activity

validator

Browse Source
This commit is contained in:
mr
2024-07-18 18:14:12 +02:00
parent cc4cc49617
commit 4f4e2949d6
9 changed files with 47 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
models/resources/data/data_mongo_accessor.go
View File

@@ -18,6 +18,10 @@ func (dma *DataMongoAccessor) UpdateOne(set map[string]interface{}, id string) (
}
func (dma *DataMongoAccessor) StoreOne(data utils.DBObject) (utils.DBObject, error) {
err := dma.Validate(data.(*Data))
if err != nil {
return nil, err
}
id, err := mongo.MONGOService.StoreOne(data.(*Data), dma.GetType())
if err != nil {
dma.Logger.Error().Msg("Could not store " + data.GetName() + " to db. Error: " + err.Error())

4
models/resources/datacenter/datacenter_mongo_accessor.go
View File

@@ -18,6 +18,10 @@ func (dca *DatacenterMongoAccessor) UpdateOne(set map[string]interface{}, id str
}
func (dca *DatacenterMongoAccessor) StoreOne(data utils.DBObject) (utils.DBObject, error) {
err := dca.Validate(data.(*Datacenter))
if err != nil {
return nil, err
}
id, err := mongo.MONGOService.StoreOne(data.(*Datacenter), dca.GetType())
if err != nil {
dca.Logger.Error().Msg("Could not store " + data.GetName() + " to db. Error: " + err.Error())

4
models/resources/processing/processing_mongo_accessor.go
View File

@@ -18,6 +18,10 @@ func (pma *ProcessingMongoAccessor) UpdateOne(set map[string]interface{}, id str
}
func (pma *ProcessingMongoAccessor) StoreOne(data utils.DBObject) (utils.DBObject, error) {
err := pma.Validate(data.(*Processing))
if err != nil {
return nil, err
}
id, err := mongo.MONGOService.StoreOne(data.(*Processing), pma.GetType())
if err != nil {
pma.Logger.Error().Msg("Could not store " + data.GetName() + " to db. Error: " + err.Error())

4
models/resources/storage/storage_mongo_accessor.go
View File

@@ -18,6 +18,10 @@ func (sma *StorageMongoAccessor) UpdateOne(set map[string]interface{}, id string
}
func (sma *StorageMongoAccessor) StoreOne(data utils.DBObject) (utils.DBObject, error) {
err := sma.Validate(data.(*Storage))
if err != nil {
return nil, err
}
id, err := mongo.MONGOService.StoreOne(data.(*Storage), sma.GetType())
if err != nil {
sma.Logger.Error().Msg("Could not store " + data.GetName() + " to db. Error: " + err.Error())

4
models/resources/workflow/workflow_mongo_accessor.go
View File

@@ -18,6 +18,10 @@ func (wfa *WorkflowMongoAccessor) UpdateOne(set map[string]interface{}, id strin
}
func (wfa *WorkflowMongoAccessor) StoreOne(data utils.DBObject) (utils.DBObject, error) {
err := wfa.Validate(data.(*Workflow))
if err != nil {
return nil, err
}
id, err := mongo.MONGOService.StoreOne(data.(*Workflow), wfa.GetType())
if err != nil {
wfa.Logger.Error().Msg("Could not store " + data.GetName() + " to db. Error: " + err.Error())

7
models/utils/abstracts.go
View File

@@ -4,9 +4,12 @@ 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/rs/zerolog"
)
var validate = validator.New(validator.WithRequiredStructEnabled())
type AbstractAccessor struct {
Logger zerolog.Logger
Type string
@@ -16,6 +19,10 @@ func (dma *AbstractAccessor) GetType() string {
return dma.Type
}
func (dma *AbstractAccessor) Validate(d DBObject) error {
return validate.Struct(d)
}
func (dma *AbstractAccessor) SetLogger(t resources.ResourceType) {
dma.Logger = logs.CreateLogger(t.String(), "")
dma.Type = t.String()