Compare commits
No commits in common. "main" and "issue#4" have entirely different histories.
@ -16,7 +16,6 @@ type Config struct {
|
|||||||
Port string
|
Port string
|
||||||
LokiUrl string
|
LokiUrl string
|
||||||
LogLevel string
|
LogLevel string
|
||||||
Whitelist bool
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c Config) GetUrl() string {
|
func (c Config) GetUrl() string {
|
||||||
@ -38,11 +37,19 @@ func GetConfig() *Config {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func SetConfig(mongoUrl string, database string, natsUrl string, lokiUrl string, logLevel string) *Config {
|
func SetConfig(mongoUrl string, database string, natsUrl string, lokiUrl string, logLevel string) *Config {
|
||||||
|
/*once.Do(func() {
|
||||||
|
instance = &Config{
|
||||||
|
MongoUrl: mongoUrl,
|
||||||
|
MongoDatabase: database,
|
||||||
|
NATSUrl: natsUrl,
|
||||||
|
LokiUrl: lokiUrl,
|
||||||
|
LogLevel: logLevel,
|
||||||
|
}
|
||||||
|
})*/
|
||||||
GetConfig().MongoUrl = mongoUrl
|
GetConfig().MongoUrl = mongoUrl
|
||||||
GetConfig().MongoDatabase = database
|
GetConfig().MongoDatabase = database
|
||||||
GetConfig().NATSUrl = natsUrl
|
GetConfig().NATSUrl = natsUrl
|
||||||
GetConfig().LokiUrl = lokiUrl
|
GetConfig().LokiUrl = lokiUrl
|
||||||
GetConfig().LogLevel = logLevel
|
GetConfig().LogLevel = logLevel
|
||||||
GetConfig().Whitelist = true
|
|
||||||
return GetConfig()
|
return GetConfig()
|
||||||
}
|
}
|
||||||
|
@ -48,7 +48,7 @@ func (m *MongoDB) Init(collections []string, config MongoConf) {
|
|||||||
mngoCollections = collections
|
mngoCollections = collections
|
||||||
mngoConfig = config
|
mngoConfig = config
|
||||||
if err := m.createClient(config.GetUrl(), false); err != nil {
|
if err := m.createClient(config.GetUrl(), false); err != nil {
|
||||||
// m.Logger.Error().Msg(err.Error())
|
m.Logger.Error().Msg(err.Error())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -170,12 +170,12 @@ func (m *MongoDB) DeleteOne(id string, collection_name string) (int64, int, erro
|
|||||||
filter := bson.M{"_id": id}
|
filter := bson.M{"_id": id}
|
||||||
targetDBCollection := CollectionMap[collection_name]
|
targetDBCollection := CollectionMap[collection_name]
|
||||||
opts := options.Delete().SetHint(bson.D{{Key: "_id", Value: 1}})
|
opts := options.Delete().SetHint(bson.D{{Key: "_id", Value: 1}})
|
||||||
MngoCtx, cancel = context.WithTimeout(context.Background(), 5*time.Second)
|
MngoCtx, cancel = context.WithTimeout(context.Background(), 10*time.Second)
|
||||||
//defer cancel()
|
defer cancel()
|
||||||
|
|
||||||
result, err := targetDBCollection.DeleteOne(MngoCtx, filter, opts)
|
result, err := targetDBCollection.DeleteOne(MngoCtx, filter, opts)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
// m.Logger.Error().Msg("Couldn't insert resource: " + err.Error())
|
m.Logger.Error().Msg("Couldn't insert resource: " + err.Error())
|
||||||
return 0, 404, err
|
return 0, 404, err
|
||||||
}
|
}
|
||||||
return result.DeletedCount, 200, nil
|
return result.DeletedCount, 200, nil
|
||||||
@ -191,12 +191,12 @@ func (m *MongoDB) DeleteMultiple(f map[string]interface{}, collection_name strin
|
|||||||
}
|
}
|
||||||
targetDBCollection := CollectionMap[collection_name]
|
targetDBCollection := CollectionMap[collection_name]
|
||||||
opts := options.Delete().SetHint(bson.D{{Key: "_id", Value: 1}})
|
opts := options.Delete().SetHint(bson.D{{Key: "_id", Value: 1}})
|
||||||
MngoCtx, cancel = context.WithTimeout(context.Background(), 5*time.Second)
|
MngoCtx, cancel = context.WithTimeout(context.Background(), 10*time.Second)
|
||||||
//defer cancel()
|
defer cancel()
|
||||||
|
|
||||||
result, err := targetDBCollection.DeleteMany(MngoCtx, filter, opts)
|
result, err := targetDBCollection.DeleteMany(MngoCtx, filter, opts)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
// m.Logger.Error().Msg("Couldn't insert resource: " + err.Error())
|
m.Logger.Error().Msg("Couldn't insert resource: " + err.Error())
|
||||||
return 0, 404, err
|
return 0, 404, err
|
||||||
}
|
}
|
||||||
return result.DeletedCount, 200, nil
|
return result.DeletedCount, 200, nil
|
||||||
@ -214,11 +214,11 @@ func (m *MongoDB) UpdateMultiple(set interface{}, filter map[string]interface{},
|
|||||||
f = append(f, bson.E{Key: k, Value: v})
|
f = append(f, bson.E{Key: k, Value: v})
|
||||||
}
|
}
|
||||||
targetDBCollection := CollectionMap[collection_name]
|
targetDBCollection := CollectionMap[collection_name]
|
||||||
MngoCtx, cancel = context.WithTimeout(context.Background(), 5*time.Second)
|
MngoCtx, cancel = context.WithTimeout(context.Background(), 50*time.Second)
|
||||||
//defer cancel()
|
defer cancel()
|
||||||
res, err := targetDBCollection.UpdateMany(MngoCtx, f, dbs.InputToBson(doc, true))
|
res, err := targetDBCollection.UpdateMany(MngoCtx, f, dbs.InputToBson(doc, true))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
// m.Logger.Error().Msg("Couldn't update resource: " + err.Error())
|
m.Logger.Error().Msg("Couldn't update resource: " + err.Error())
|
||||||
return 0, 404, err
|
return 0, 404, err
|
||||||
}
|
}
|
||||||
return res.UpsertedCount, 200, nil
|
return res.UpsertedCount, 200, nil
|
||||||
@ -233,11 +233,11 @@ func (m *MongoDB) UpdateOne(set interface{}, id string, collection_name string)
|
|||||||
bson.Unmarshal(b, &doc)
|
bson.Unmarshal(b, &doc)
|
||||||
filter := bson.M{"_id": id}
|
filter := bson.M{"_id": id}
|
||||||
targetDBCollection := CollectionMap[collection_name]
|
targetDBCollection := CollectionMap[collection_name]
|
||||||
MngoCtx, cancel = context.WithTimeout(context.Background(), 5*time.Second)
|
MngoCtx, cancel = context.WithTimeout(context.Background(), 50*time.Second)
|
||||||
//defer cancel()
|
defer cancel()
|
||||||
_, err := targetDBCollection.UpdateOne(MngoCtx, filter, dbs.InputToBson(doc, true))
|
_, err := targetDBCollection.UpdateOne(MngoCtx, filter, dbs.InputToBson(doc, true))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
// m.Logger.Error().Msg("Couldn't update resource: " + err.Error())
|
m.Logger.Error().Msg("Couldn't update resource: " + err.Error())
|
||||||
return "", 404, err
|
return "", 404, err
|
||||||
}
|
}
|
||||||
return id, 200, nil
|
return id, 200, nil
|
||||||
@ -252,12 +252,12 @@ func (m *MongoDB) StoreOne(obj interface{}, id string, collection_name string) (
|
|||||||
bson.Unmarshal(b, &doc)
|
bson.Unmarshal(b, &doc)
|
||||||
doc["_id"] = id
|
doc["_id"] = id
|
||||||
targetDBCollection := CollectionMap[collection_name]
|
targetDBCollection := CollectionMap[collection_name]
|
||||||
MngoCtx, cancel = context.WithTimeout(context.Background(), 5*time.Second)
|
MngoCtx, cancel = context.WithTimeout(context.Background(), 10*time.Second)
|
||||||
//defer cancel()
|
defer cancel()
|
||||||
|
|
||||||
_, err := targetDBCollection.InsertOne(MngoCtx, doc)
|
_, err := targetDBCollection.InsertOne(MngoCtx, doc)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
// m.Logger.Error().Msg("Couldn't insert resource: " + err.Error())
|
m.Logger.Error().Msg("Couldn't insert resource: " + err.Error())
|
||||||
return "", 409, err
|
return "", 409, err
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -270,12 +270,12 @@ func (m *MongoDB) LoadOne(id string, collection_name string) (*mongo.SingleResul
|
|||||||
}
|
}
|
||||||
filter := bson.M{"_id": id}
|
filter := bson.M{"_id": id}
|
||||||
targetDBCollection := CollectionMap[collection_name]
|
targetDBCollection := CollectionMap[collection_name]
|
||||||
MngoCtx, cancel = context.WithTimeout(context.Background(), 5*time.Second)
|
MngoCtx, cancel = context.WithTimeout(context.Background(), 10*time.Second)
|
||||||
//defer cancel()
|
defer cancel()
|
||||||
|
|
||||||
res := targetDBCollection.FindOne(MngoCtx, filter)
|
res := targetDBCollection.FindOne(MngoCtx, filter)
|
||||||
if res.Err() != nil {
|
if res.Err() != nil {
|
||||||
// m.Logger.Error().Msg("Couldn't find resource " + id + ". Error : " + res.Err().Error())
|
m.Logger.Error().Msg("Couldn't find resource " + id + ". Error : " + res.Err().Error())
|
||||||
err := res.Err()
|
err := res.Err()
|
||||||
return nil, 404, err
|
return nil, 404, err
|
||||||
}
|
}
|
||||||
@ -287,7 +287,7 @@ func (m *MongoDB) Search(filters *dbs.Filters, collection_name string) (*mongo.C
|
|||||||
return nil, 503, err
|
return nil, 503, err
|
||||||
}
|
}
|
||||||
opts := options.Find()
|
opts := options.Find()
|
||||||
opts.SetLimit(1000)
|
opts.SetLimit(100)
|
||||||
targetDBCollection := CollectionMap[collection_name]
|
targetDBCollection := CollectionMap[collection_name]
|
||||||
orList := bson.A{}
|
orList := bson.A{}
|
||||||
andList := bson.A{}
|
andList := bson.A{}
|
||||||
@ -313,8 +313,8 @@ func (m *MongoDB) Search(filters *dbs.Filters, collection_name string) (*mongo.C
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
MngoCtx, cancel = context.WithTimeout(context.Background(), 5*time.Second)
|
MngoCtx, cancel = context.WithTimeout(context.Background(), 10*time.Second)
|
||||||
// defer cancel()
|
defer cancel()
|
||||||
if cursor, err := targetDBCollection.Find(
|
if cursor, err := targetDBCollection.Find(
|
||||||
MngoCtx,
|
MngoCtx,
|
||||||
f,
|
f,
|
||||||
@ -336,12 +336,12 @@ func (m *MongoDB) LoadFilter(filter map[string]interface{}, collection_name stri
|
|||||||
}
|
}
|
||||||
targetDBCollection := CollectionMap[collection_name]
|
targetDBCollection := CollectionMap[collection_name]
|
||||||
|
|
||||||
MngoCtx, cancel = context.WithTimeout(context.Background(), 5*time.Second)
|
MngoCtx, cancel = context.WithTimeout(context.Background(), 10*time.Second)
|
||||||
//defer cancel()
|
defer cancel()
|
||||||
|
|
||||||
res, err := targetDBCollection.Find(MngoCtx, f)
|
res, err := targetDBCollection.Find(MngoCtx, f)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
// m.Logger.Error().Msg("Couldn't find any resources. Error : " + err.Error())
|
m.Logger.Error().Msg("Couldn't find any resources. Error : " + err.Error())
|
||||||
return nil, 404, err
|
return nil, 404, err
|
||||||
}
|
}
|
||||||
return res, 200, nil
|
return res, 200, nil
|
||||||
@ -353,12 +353,12 @@ func (m *MongoDB) LoadAll(collection_name string) (*mongo.Cursor, int, error) {
|
|||||||
}
|
}
|
||||||
targetDBCollection := CollectionMap[collection_name]
|
targetDBCollection := CollectionMap[collection_name]
|
||||||
|
|
||||||
MngoCtx, cancel = context.WithTimeout(context.Background(), 5*time.Second)
|
MngoCtx, cancel = context.WithTimeout(context.Background(), 10*time.Second)
|
||||||
//defer cancel()
|
defer cancel()
|
||||||
|
|
||||||
res, err := targetDBCollection.Find(MngoCtx, bson.D{})
|
res, err := targetDBCollection.Find(MngoCtx, bson.D{})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
// m.Logger.Error().Msg("Couldn't find any resources. Error : " + err.Error())
|
m.Logger.Error().Msg("Couldn't find any resources. Error : " + err.Error())
|
||||||
return nil, 404, err
|
return nil, 404, err
|
||||||
}
|
}
|
||||||
return res, 200, nil
|
return res, 200, nil
|
||||||
|
@ -7,7 +7,7 @@ abstract Resource{
|
|||||||
+icon: string
|
+icon: string
|
||||||
+description: string
|
+description: string
|
||||||
+graphic: GraphicElement
|
+graphic: GraphicElement
|
||||||
+element: DataResource/ProcessingResource/StorageResource/Workflow/ComputeResource
|
+element: DataResource/ProcessingResource/StorageResource/Workflow/DatacenterResource
|
||||||
}
|
}
|
||||||
|
|
||||||
class DataResource {
|
class DataResource {
|
||||||
@ -31,7 +31,7 @@ class StorageResource {
|
|||||||
+capacity: int
|
+capacity: int
|
||||||
}
|
}
|
||||||
|
|
||||||
class ComputeResource {
|
class DatacenterResource {
|
||||||
+UUID: int
|
+UUID: int
|
||||||
+name: string
|
+name: string
|
||||||
|
|
||||||
@ -96,7 +96,7 @@ class UserWorkflows {
|
|||||||
|
|
||||||
class DatacenterWorkflows {
|
class DatacenterWorkflows {
|
||||||
+UUID: int
|
+UUID: int
|
||||||
+compute: ComputeResource
|
+datacenter: DatacenterResource
|
||||||
+workflows: Workflow[]
|
+workflows: Workflow[]
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -159,7 +159,7 @@ DatacenterWorkflows "1" o-- "0..*" Workflow
|
|||||||
Resource<|-- DataResource
|
Resource<|-- DataResource
|
||||||
Resource<|-- ProcessingResource
|
Resource<|-- ProcessingResource
|
||||||
Resource<|-- StorageResource
|
Resource<|-- StorageResource
|
||||||
Resource<|-- ComputeResource
|
Resource<|-- DatacenterResource
|
||||||
Resource<|-- Workflow
|
Resource<|-- Workflow
|
||||||
|
|
||||||
ResourceSet "1" o-- "0..*" Ressource
|
ResourceSet "1" o-- "0..*" Ressource
|
||||||
|
@ -1,325 +0,0 @@
|
|||||||
@startuml
|
|
||||||
|
|
||||||
class AbstractObject {
|
|
||||||
ID string
|
|
||||||
Name string
|
|
||||||
IsDraft bool // is consider as a draft
|
|
||||||
UpdateDate date
|
|
||||||
LastPeerWriter string
|
|
||||||
CreatorID string
|
|
||||||
AccessMode int // public or private
|
|
||||||
}
|
|
||||||
|
|
||||||
AbstractObject ^-- AbstractResource
|
|
||||||
AbstractObject ^-- Order
|
|
||||||
AbstractObject ^-- Booking
|
|
||||||
AbstractObject ^-- BuyingStatus
|
|
||||||
AbstractObject ^-- WorkflowExecution
|
|
||||||
AbstractObject ^-- Workflow
|
|
||||||
|
|
||||||
class AbstractResource {
|
|
||||||
Logo string
|
|
||||||
Description string
|
|
||||||
ShortDescription string
|
|
||||||
Owners []string
|
|
||||||
UsageRestrictions string
|
|
||||||
|
|
||||||
VerifyAuth(request) bool
|
|
||||||
}
|
|
||||||
|
|
||||||
AbstractResource "1 " --* "many " ResourceInstanceITF
|
|
||||||
AbstractCustomizedResource "1 " --* "1 " ResourceInstanceITF
|
|
||||||
|
|
||||||
AbstractResource ^-- ComputeResource
|
|
||||||
AbstractResource ^-- DataResource
|
|
||||||
AbstractResource ^-- ProcessingResource
|
|
||||||
AbstractResource ^-- StorageResource
|
|
||||||
AbstractResource ^-- WorkflowResource
|
|
||||||
class ComputeResource {
|
|
||||||
Architecture string
|
|
||||||
Infrastructure string
|
|
||||||
}
|
|
||||||
class DataResource {
|
|
||||||
Type string
|
|
||||||
Quality string
|
|
||||||
OpenData bool
|
|
||||||
Static bool
|
|
||||||
UpdatePeriod date
|
|
||||||
PersonalData bool
|
|
||||||
AnonymizedPersonalData bool
|
|
||||||
SizeGB float64
|
|
||||||
Licence string
|
|
||||||
Example string
|
|
||||||
}
|
|
||||||
ProcessingResource "1 " *-- "1 " ProcessingUsage
|
|
||||||
class ProcessingUsage {
|
|
||||||
CPUs map[string]CPU
|
|
||||||
GPUs map[string]GPU
|
|
||||||
RAM RAM
|
|
||||||
StorageGB float64
|
|
||||||
Hypothesis string
|
|
||||||
ScalingModel string
|
|
||||||
}
|
|
||||||
|
|
||||||
class ProcessingResource {
|
|
||||||
Infrastructure string
|
|
||||||
Service bool
|
|
||||||
Usage ProcessingUsage
|
|
||||||
OpenSource bool
|
|
||||||
License string
|
|
||||||
Maturity string
|
|
||||||
}
|
|
||||||
class StorageResource {
|
|
||||||
Type string
|
|
||||||
Accronym string
|
|
||||||
}
|
|
||||||
WorkflowResource "1 " --* "many " ComputeResource
|
|
||||||
WorkflowResource "1 " --* "many " DataResource
|
|
||||||
WorkflowResource "1 " --* "many " ProcessingResource
|
|
||||||
WorkflowResource "1 " --* "many " StorageResource
|
|
||||||
class WorkflowResource {
|
|
||||||
WorkflowID string
|
|
||||||
}
|
|
||||||
|
|
||||||
class ExploitResourceSet {}
|
|
||||||
|
|
||||||
AbstractCustomizedResource --^ AbstractResource
|
|
||||||
AbstractCustomizedResource --* ExploitResourceSet
|
|
||||||
ExploitResourceSet ^-- CustomizedComputeResource
|
|
||||||
ExploitResourceSet ^-- CustomizedDataResource
|
|
||||||
ExploitResourceSet ^-- CustomizedProcessingResource
|
|
||||||
ExploitResourceSet ^-- CustomizedStorageResource
|
|
||||||
ExploitResourceSet ^-- CustomizedWorkflowResource
|
|
||||||
class AbstractCustomizedResource {
|
|
||||||
// A customized resource is an
|
|
||||||
// extended abstract resource not use in catalog
|
|
||||||
ExplicitBookingDurationS float64
|
|
||||||
UsageStart date
|
|
||||||
UsageEnd date
|
|
||||||
SelectedPricing string
|
|
||||||
}
|
|
||||||
class CustomizedComputeResource {
|
|
||||||
CPUsLocated map[string]float64
|
|
||||||
GPUsLocated map[string]float64
|
|
||||||
RAMLocated float64
|
|
||||||
}
|
|
||||||
class CustomizedDataResource {
|
|
||||||
StorageGB float64
|
|
||||||
}
|
|
||||||
class CustomizedProcessingResource {
|
|
||||||
Container Container
|
|
||||||
}
|
|
||||||
class CustomizedStorageResource {
|
|
||||||
StorageGB bool
|
|
||||||
}
|
|
||||||
class CustomizedWorkflowResource {}
|
|
||||||
|
|
||||||
interface ResourceInstanceITF {
|
|
||||||
GetID() string
|
|
||||||
VerifyPartnership() bool // eval if there is one partnership per peer groups in every instance
|
|
||||||
GetPeerGroups() []ResourcePartnerITF, []map[string][]string
|
|
||||||
ClearPeerGroups()
|
|
||||||
}
|
|
||||||
|
|
||||||
ResourceInstanceITF -- ResourceInstance
|
|
||||||
ResourceInstance ^-- ComputeResourceInstance
|
|
||||||
ResourceInstance ^-- StorageResourceInstance
|
|
||||||
ResourceInstance "many " --* "1 " ResourcePartnerITF
|
|
||||||
class ResourceInstance {
|
|
||||||
ID string
|
|
||||||
Location Geopoint
|
|
||||||
Country CountryCode
|
|
||||||
AccessProtocol string
|
|
||||||
}
|
|
||||||
class ComputeResourceInstance {
|
|
||||||
SecurityLevel string
|
|
||||||
PowerSource string
|
|
||||||
CPUs map[string]CPU
|
|
||||||
GPUs map[string]GPU
|
|
||||||
RAM RAM
|
|
||||||
}
|
|
||||||
class StorageResourceInstance {
|
|
||||||
Local bool
|
|
||||||
SecurityLevel string
|
|
||||||
SizeType string
|
|
||||||
SizeGB int
|
|
||||||
Encryption bool
|
|
||||||
Redundancy string
|
|
||||||
Throughput string
|
|
||||||
}
|
|
||||||
|
|
||||||
ResourcePartnerITF -- ResourcePartnership
|
|
||||||
ResourcePartnership ^-- ComputeResourcePartnership
|
|
||||||
ResourcePartnership ^-- DataResourcePartnership
|
|
||||||
ResourcePartnership ^-- StorageResourcePartnership
|
|
||||||
|
|
||||||
interface ResourcePartnerITF {
|
|
||||||
GetPricing(id string) PricingProfileITF
|
|
||||||
GetPeerGroups() []ResourcePartnerITF, []map[string][]string
|
|
||||||
ClearPeerGroups()
|
|
||||||
}
|
|
||||||
|
|
||||||
ResourcePartnership "many " --* "1 " PricingProfileITF
|
|
||||||
class ResourcePartnership{
|
|
||||||
Namespace string
|
|
||||||
PeerGroups map[string][]string
|
|
||||||
}
|
|
||||||
class ComputeResourcePartnership {
|
|
||||||
MaxAllowedCPUsCores map[string]int
|
|
||||||
MaxAllowedGPUsMemoryGB map[string]float64
|
|
||||||
RAMSizeGB float64
|
|
||||||
}
|
|
||||||
class DataResourcePartnership {
|
|
||||||
MaxDownloadableGBAllowed float64
|
|
||||||
PersonalDataAllowed bool
|
|
||||||
AnonymizedPersonalDataAllowed bool
|
|
||||||
}
|
|
||||||
class StorageResourcePartnership {
|
|
||||||
MaxSizeGBAllowed float64
|
|
||||||
OnlyEncryptedAllowed bool
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
RefundType -- AccessPricingProfile
|
|
||||||
enum RefundType {
|
|
||||||
REFUND_DEAD_END
|
|
||||||
REFUND_ON_ERROR
|
|
||||||
REFUND_ON_EARLY_END
|
|
||||||
}
|
|
||||||
PricingProfileITF -- AccessPricingProfile
|
|
||||||
PricingProfileITF -- ExploitPricingProfile
|
|
||||||
PricingProfileITF -- WorkflowResourcePricingProfile
|
|
||||||
AccessPricingProfile ^-- DataResourcePricingProfile
|
|
||||||
AccessPricingProfile ^-- ProcessingResourcePricingProfile
|
|
||||||
ExploitPricingProfile ^-- ComputeResourcePricingProfile
|
|
||||||
ExploitPricingProfile ^-- StorageResourcePricingProfile
|
|
||||||
interface PricingProfileITF {
|
|
||||||
GetPrice(quantity float64, val float64, start date, end date, request) float64
|
|
||||||
IsPurchased() bool
|
|
||||||
}
|
|
||||||
class AccessPricingProfile {
|
|
||||||
ID string
|
|
||||||
Pricing PricingStrategy
|
|
||||||
DefaultRefundType RefundType
|
|
||||||
RefundRatio int // percentage of refund on price
|
|
||||||
}
|
|
||||||
class DataResourcePricingProfile {}
|
|
||||||
class ProcessingResourcePricingProfile {}
|
|
||||||
|
|
||||||
ExploitPrivilegeStrategy -- ExploitPricingProfile
|
|
||||||
enum ExploitPrivilegeStrategy {
|
|
||||||
BASIC
|
|
||||||
GARANTED_ON_DELAY
|
|
||||||
GARANTED
|
|
||||||
}
|
|
||||||
|
|
||||||
AccessPricingProfile --* PricingStrategy
|
|
||||||
AccessPricingProfile ^-- ExploitPricingProfile
|
|
||||||
class ExploitPricingProfile {
|
|
||||||
AdditionnalRefundTypes RefundTypeint
|
|
||||||
PrivilegeStrategy ExploitPrivilegeStrategy
|
|
||||||
GarantedDelaySecond int
|
|
||||||
Exceeding bool
|
|
||||||
ExceedingRatio int // percentage of Exceeding based on price
|
|
||||||
}
|
|
||||||
class ComputeResourcePricingProfile {
|
|
||||||
OverrideCPUsPrices map[string]float64
|
|
||||||
OverrideGPUsPrices map[string]float64
|
|
||||||
OverrideRAMPrice float64
|
|
||||||
}
|
|
||||||
class StorageResourcePricingProfile {}
|
|
||||||
WorkflowResourcePricingProfile "1 " --* "many " ExploitResourceSet
|
|
||||||
|
|
||||||
class WorkflowResourcePricingProfile {
|
|
||||||
ID string
|
|
||||||
}
|
|
||||||
|
|
||||||
BuyingStrategy -- PricingStrategy
|
|
||||||
enum BuyingStrategy {
|
|
||||||
UNLIMITED
|
|
||||||
SUBSCRIPTION
|
|
||||||
PAY_PER_USE
|
|
||||||
}
|
|
||||||
Strategy -- TimePricingStrategy
|
|
||||||
Strategy "0-1 " *-- " " PricingStrategy
|
|
||||||
interface Strategy {
|
|
||||||
GetStrategy () string
|
|
||||||
GetStrategyValue() int
|
|
||||||
}
|
|
||||||
enum TimePricingStrategy {
|
|
||||||
ONCE
|
|
||||||
PER_SECOND
|
|
||||||
PER_MINUTE
|
|
||||||
PER_HOUR
|
|
||||||
PER_DAY
|
|
||||||
PER_WEEK
|
|
||||||
PER_MONTH
|
|
||||||
}
|
|
||||||
|
|
||||||
class PricingStrategy {
|
|
||||||
Price float64
|
|
||||||
BuyingStrategy
|
|
||||||
TimePricingStrategy TimePricingStrategy
|
|
||||||
OverrideStrategy Strategy
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
PeerOrder "many " *-- "1 " Order
|
|
||||||
PeerItemOrder "many " *-- "1 " PeerOrder
|
|
||||||
PricedItemITF "many " *-- "1 " PeerItemOrder
|
|
||||||
|
|
||||||
PricedItemITF -- AbstractCustomizedResource
|
|
||||||
|
|
||||||
class Order {
|
|
||||||
OrderBy string
|
|
||||||
WorkflowExecutionIDs []string
|
|
||||||
Status string
|
|
||||||
Total float64
|
|
||||||
}
|
|
||||||
class PeerOrder {
|
|
||||||
PeerID string
|
|
||||||
Error string
|
|
||||||
Status string
|
|
||||||
BillingAddress string
|
|
||||||
Total float64
|
|
||||||
}
|
|
||||||
class PeerItemOrder {
|
|
||||||
Quantity int
|
|
||||||
BuyingStatus string
|
|
||||||
}
|
|
||||||
|
|
||||||
class BuyingStatus {}
|
|
||||||
|
|
||||||
WorkflowExecution "many " --* "1 " Workflow
|
|
||||||
Workflow "1 " --* "many " WorkflowScheduler
|
|
||||||
WorkflowScheduler "1 " --* "many " WorkflowExecution
|
|
||||||
|
|
||||||
class WorkflowExecution {
|
|
||||||
ExecDate date
|
|
||||||
EndDate date
|
|
||||||
State string
|
|
||||||
WorkflowID string
|
|
||||||
|
|
||||||
ToBookings() []Booking
|
|
||||||
}
|
|
||||||
class WorkflowScheduler* {
|
|
||||||
Message string
|
|
||||||
Warning string
|
|
||||||
Start date
|
|
||||||
End date
|
|
||||||
DurationS float64
|
|
||||||
Cron string
|
|
||||||
|
|
||||||
Schedules(workflowID string, request) []WorkflowExecution
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
Workflow "1 " --* "many " ExploitResourceSet
|
|
||||||
|
|
||||||
class Workflow {}
|
|
||||||
|
|
||||||
interface PricedItemITF {
|
|
||||||
getPrice(request) float64, error
|
|
||||||
}
|
|
||||||
|
|
||||||
@enduml
|
|
@ -1,29 +0,0 @@
|
|||||||
@startuml
|
|
||||||
user -> client : schedule
|
|
||||||
client -> OrderAPIP1 : check book
|
|
||||||
OrderAPIP1 -> datacenterAPIP2 : check book
|
|
||||||
datacenterAPIP2 -> OrderAPIP1 : send ok
|
|
||||||
OrderAPIP1 -> datacenterAPIP2 : generate draft book
|
|
||||||
OrderAPIP1 -> client : send ok
|
|
||||||
client -> OrderAPIP1 : send scheduler
|
|
||||||
OrderAPIP1 -> OrderAPIP1 : draft executions
|
|
||||||
OrderAPIP1 -> OrderAPIP1 : draft order
|
|
||||||
OrderAPIP1 -> client : send drafted order
|
|
||||||
client -> user :
|
|
||||||
user -> client : select pricing profile
|
|
||||||
client -> OrderAPIP1 : update order
|
|
||||||
OrderAPIP1 -> datacenterAPIP2 : check book
|
|
||||||
datacenterAPIP2 -> OrderAPIP1 : send ok
|
|
||||||
OrderAPIP1 -> datacenterAPIP2 : generate draft book
|
|
||||||
OrderAPIP1 -> client : send order
|
|
||||||
user -> client : order
|
|
||||||
client -> OrderAPIP1 : order
|
|
||||||
OrderAPIP1 -> PaymentAPIBCP1 : send payment
|
|
||||||
PaymentAPIBCP1 -> OrderAPIP1 : send ok
|
|
||||||
OrderAPIP1 -> datacenterAPIP2 : undraft booking
|
|
||||||
OrderAPIP1 -> OrderAPIP1 : undraft execution
|
|
||||||
OrderAPIP1 -> OrderAPIP1 : undraft order
|
|
||||||
OrderAPIP1 -> client : send ok
|
|
||||||
client -> client : redirect
|
|
||||||
@enduml
|
|
||||||
|
|
338
entrypoint.go
338
entrypoint.go
@ -1,11 +1,8 @@
|
|||||||
package oclib
|
package oclib
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/base64"
|
|
||||||
"encoding/json"
|
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"net/http"
|
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
"runtime/debug"
|
"runtime/debug"
|
||||||
@ -15,14 +12,15 @@ import (
|
|||||||
"cloud.o-forge.io/core/oc-lib/dbs/mongo"
|
"cloud.o-forge.io/core/oc-lib/dbs/mongo"
|
||||||
"cloud.o-forge.io/core/oc-lib/logs"
|
"cloud.o-forge.io/core/oc-lib/logs"
|
||||||
"cloud.o-forge.io/core/oc-lib/models"
|
"cloud.o-forge.io/core/oc-lib/models"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/booking"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/collaborative_area"
|
"cloud.o-forge.io/core/oc-lib/models/collaborative_area"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/collaborative_area/rules/rule"
|
"cloud.o-forge.io/core/oc-lib/models/collaborative_area/rules/rule"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/live"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/order"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/peer"
|
"cloud.o-forge.io/core/oc-lib/models/peer"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/resources"
|
"cloud.o-forge.io/core/oc-lib/models/resource_model"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/resources/purchase_resource"
|
"cloud.o-forge.io/core/oc-lib/models/resources/compute"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/resources/data"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/resources/processing"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/resources/storage"
|
||||||
|
w "cloud.o-forge.io/core/oc-lib/models/resources/workflow"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
"cloud.o-forge.io/core/oc-lib/models/utils"
|
||||||
w2 "cloud.o-forge.io/core/oc-lib/models/workflow"
|
w2 "cloud.o-forge.io/core/oc-lib/models/workflow"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/workflow_execution"
|
"cloud.o-forge.io/core/oc-lib/models/workflow_execution"
|
||||||
@ -50,13 +48,9 @@ const (
|
|||||||
WORKSPACE = tools.WORKSPACE
|
WORKSPACE = tools.WORKSPACE
|
||||||
WORKFLOW_EXECUTION = tools.WORKFLOW_EXECUTION
|
WORKFLOW_EXECUTION = tools.WORKFLOW_EXECUTION
|
||||||
PEER = tools.PEER
|
PEER = tools.PEER
|
||||||
COLLABORATIVE_AREA = tools.COLLABORATIVE_AREA
|
SHARED_WORKSPACE = tools.COLLABORATIVE_AREA
|
||||||
RULE = tools.RULE
|
RULE = tools.RULE
|
||||||
BOOKING = tools.BOOKING
|
BOOKING = tools.BOOKING
|
||||||
ORDER = tools.ORDER
|
|
||||||
LIVE_DATACENTER = tools.LIVE_DATACENTER
|
|
||||||
LIVE_STORAGE = tools.LIVE_STORAGE
|
|
||||||
PURCHASE_RESOURCE = tools.PURCHASE_RESOURCE
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// will turn into standards api hostnames
|
// will turn into standards api hostnames
|
||||||
@ -124,49 +118,6 @@ func InitDaemon(appName string) {
|
|||||||
beego.BConfig.WebConfig.StaticDir["/swagger"] = "swagger"
|
beego.BConfig.WebConfig.StaticDir["/swagger"] = "swagger"
|
||||||
}
|
}
|
||||||
|
|
||||||
type IDTokenClaims struct {
|
|
||||||
UserID string `json:"user_id"`
|
|
||||||
PeerID string `json:"peer_id"`
|
|
||||||
Groups []string `json:"groups"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// SessionClaims struct
|
|
||||||
type SessionClaims struct {
|
|
||||||
AccessToken map[string]interface{} `json:"access_token"`
|
|
||||||
IDToken IDTokenClaims `json:"id_token"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// Claims struct
|
|
||||||
type Claims struct {
|
|
||||||
Session SessionClaims `json:"session"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func ExtractTokenInfo(request http.Request) (string, string, []string) {
|
|
||||||
reqToken := request.Header.Get("Authorization")
|
|
||||||
splitToken := strings.Split(reqToken, "Bearer ")
|
|
||||||
if len(splitToken) < 2 {
|
|
||||||
reqToken = ""
|
|
||||||
} else {
|
|
||||||
reqToken = splitToken[1]
|
|
||||||
}
|
|
||||||
if reqToken != "" {
|
|
||||||
token := strings.Split(reqToken, ".")
|
|
||||||
if len(token) > 2 {
|
|
||||||
bytes, err := base64.StdEncoding.DecodeString(token[2])
|
|
||||||
if err != nil {
|
|
||||||
return "", "", []string{}
|
|
||||||
}
|
|
||||||
var c Claims
|
|
||||||
err = json.Unmarshal(bytes, &c)
|
|
||||||
if err != nil {
|
|
||||||
return "", "", []string{}
|
|
||||||
}
|
|
||||||
return c.Session.IDToken.UserID, c.Session.IDToken.PeerID, c.Session.IDToken.Groups
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return "", "", []string{}
|
|
||||||
}
|
|
||||||
|
|
||||||
func Init(appName string) {
|
func Init(appName string) {
|
||||||
InitDaemon(appName)
|
InitDaemon(appName)
|
||||||
api := &tools.API{}
|
api := &tools.API{}
|
||||||
@ -202,6 +153,49 @@ func SetConfig(mongoUrl string, database string, natsUrl string, lokiUrl string,
|
|||||||
}()
|
}()
|
||||||
logs.CreateLogger("main")
|
logs.CreateLogger("main")
|
||||||
mongo.MONGOService.Init(models.GetModelsNames(), config.GetConfig()) // init the mongo service
|
mongo.MONGOService.Init(models.GetModelsNames(), config.GetConfig()) // init the mongo service
|
||||||
|
/*
|
||||||
|
Here we will check if the resource model is already stored in the database
|
||||||
|
If not we will store it
|
||||||
|
Resource model is the model that will define the structure of the resources
|
||||||
|
*/
|
||||||
|
accessor := (&resource_model.ResourceModel{}).GetAccessor(nil)
|
||||||
|
for _, model := range []string{tools.DATA_RESOURCE.String(), tools.PROCESSING_RESOURCE.String(), tools.STORAGE_RESOURCE.String(), tools.COMPUTE_RESOURCE.String(), tools.WORKFLOW_RESOURCE.String()} {
|
||||||
|
data, code, _ := accessor.Search(nil, model)
|
||||||
|
if code == 404 || len(data) == 0 {
|
||||||
|
refs := map[string]string{}
|
||||||
|
m := map[string]resource_model.Model{}
|
||||||
|
// TODO Specify the model for each resource
|
||||||
|
// for now only processing is specified here (not an elegant way)
|
||||||
|
if model == tools.DATA_RESOURCE.String() || model == tools.STORAGE_RESOURCE.String() {
|
||||||
|
refs["path"] = "string"
|
||||||
|
}
|
||||||
|
if model == tools.PROCESSING_RESOURCE.String() {
|
||||||
|
m["command"] = resource_model.Model{
|
||||||
|
Type: "string",
|
||||||
|
ReadOnly: false,
|
||||||
|
}
|
||||||
|
m["args"] = resource_model.Model{
|
||||||
|
Type: "string",
|
||||||
|
ReadOnly: false,
|
||||||
|
}
|
||||||
|
m["env"] = resource_model.Model{
|
||||||
|
Type: "string",
|
||||||
|
ReadOnly: false,
|
||||||
|
}
|
||||||
|
m["volumes"] = resource_model.Model{
|
||||||
|
Type: "map[string]string",
|
||||||
|
ReadOnly: false,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
accessor.StoreOne(&resource_model.ResourceModel{
|
||||||
|
ResourceType: model,
|
||||||
|
VarRefs: refs,
|
||||||
|
Model: map[string]map[string]resource_model.Model{
|
||||||
|
"container": m,
|
||||||
|
},
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
return cfg
|
return cfg
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -231,64 +225,6 @@ func GetConfLoader() *onion.Onion {
|
|||||||
return config.GetConfLoader()
|
return config.GetConfLoader()
|
||||||
}
|
}
|
||||||
|
|
||||||
type Request struct {
|
|
||||||
collection LibDataEnum
|
|
||||||
user string
|
|
||||||
peerID string
|
|
||||||
groups []string
|
|
||||||
caller *tools.HTTPCaller
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewRequest(collection LibDataEnum, user string, peerID string, groups []string, caller *tools.HTTPCaller) *Request {
|
|
||||||
return &Request{collection: collection, user: user, peerID: peerID, groups: groups, caller: caller}
|
|
||||||
}
|
|
||||||
|
|
||||||
func ToScheduler(m interface{}) (n *workflow_execution.WorkflowSchedule) {
|
|
||||||
defer func() {
|
|
||||||
if r := recover(); r != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
return m.(*workflow_execution.WorkflowSchedule)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *Request) Schedule(wfID string, scheduler *workflow_execution.WorkflowSchedule) (*workflow_execution.WorkflowSchedule, error) {
|
|
||||||
ws, _, _, err := scheduler.Schedules(wfID, &tools.APIRequest{
|
|
||||||
Caller: r.caller,
|
|
||||||
Username: r.user,
|
|
||||||
PeerID: r.peerID,
|
|
||||||
Groups: r.groups,
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
return ws, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *Request) CheckBooking(wfID string, start string, end string, durationInS float64, cron string) bool {
|
|
||||||
ok, _, _, _, _, err := workflow_execution.NewScheduler(start, end, durationInS, cron).GetBuyAndBook(wfID, &tools.APIRequest{
|
|
||||||
Caller: r.caller,
|
|
||||||
Username: r.user,
|
|
||||||
PeerID: r.peerID,
|
|
||||||
Groups: r.groups,
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
fmt.Println(err)
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
return ok
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *Request) PaymentTunnel(o *order.Order, scheduler *workflow_execution.WorkflowSchedule) error {
|
|
||||||
/*return o.Pay(scheduler, &tools.APIRequest{
|
|
||||||
Caller: r.caller,
|
|
||||||
Username: r.user,
|
|
||||||
PeerID: r.peerID,
|
|
||||||
Groups: r.groups,
|
|
||||||
})*/
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* Search will search for the data in the database
|
* Search will search for the data in the database
|
||||||
* @param filters *dbs.Filters
|
* @param filters *dbs.Filters
|
||||||
@ -297,19 +233,18 @@ func (r *Request) PaymentTunnel(o *order.Order, scheduler *workflow_execution.Wo
|
|||||||
* @param c ...*tools.HTTPCaller
|
* @param c ...*tools.HTTPCaller
|
||||||
* @return data LibDataShallow
|
* @return data LibDataShallow
|
||||||
*/
|
*/
|
||||||
func (r *Request) Search(filters *dbs.Filters, word string, isDraft bool) (data LibDataShallow) {
|
func Search(filters *dbs.Filters, word string, collection LibDataEnum, c ...*tools.HTTPCaller) (data LibDataShallow) {
|
||||||
defer func() { // recover the panic
|
defer func() { // recover the panic
|
||||||
if r := recover(); r != nil {
|
if r := recover(); r != nil {
|
||||||
tools.UncatchedError = append(tools.UncatchedError, errors.New("Panic recovered in Search : "+fmt.Sprintf("%v", r)))
|
tools.UncatchedError = append(tools.UncatchedError, errors.New("Panic recovered in Search : "+fmt.Sprintf("%v", r)))
|
||||||
data = LibDataShallow{Data: nil, Code: 500, Err: "Panic recovered in LoadAll : " + fmt.Sprintf("%v", r) + " - " + string(debug.Stack())}
|
data = LibDataShallow{Data: nil, Code: 500, Err: "Panic recovered in LoadAll : " + fmt.Sprintf("%v", r) + " - " + string(debug.Stack())}
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
d, code, err := models.Model(r.collection.EnumIndex()).GetAccessor(&tools.APIRequest{
|
var caller *tools.HTTPCaller // define the caller
|
||||||
Caller: r.caller,
|
if len(c) > 0 {
|
||||||
Username: r.user,
|
caller = c[0]
|
||||||
PeerID: r.peerID,
|
}
|
||||||
Groups: r.groups,
|
d, code, err := models.Model(collection.EnumIndex()).GetAccessor(caller).Search(filters, word)
|
||||||
}).Search(filters, word, isDraft)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
data = LibDataShallow{Data: d, Code: code, Err: err.Error()}
|
data = LibDataShallow{Data: d, Code: code, Err: err.Error()}
|
||||||
return
|
return
|
||||||
@ -324,19 +259,18 @@ func (r *Request) Search(filters *dbs.Filters, word string, isDraft bool) (data
|
|||||||
* @param c ...*tools.HTTPCaller
|
* @param c ...*tools.HTTPCaller
|
||||||
* @return data LibDataShallow
|
* @return data LibDataShallow
|
||||||
*/
|
*/
|
||||||
func (r *Request) LoadAll(isDraft bool) (data LibDataShallow) {
|
func LoadAll(collection LibDataEnum, c ...*tools.HTTPCaller) (data LibDataShallow) {
|
||||||
defer func() { // recover the panic
|
defer func() { // recover the panic
|
||||||
if r := recover(); r != nil {
|
if r := recover(); r != nil {
|
||||||
tools.UncatchedError = append(tools.UncatchedError, errors.New("Panic recovered in LoadAll : "+fmt.Sprintf("%v", r)+" - "+string(debug.Stack())))
|
tools.UncatchedError = append(tools.UncatchedError, errors.New("Panic recovered in LoadAll : "+fmt.Sprintf("%v", r)+" - "+string(debug.Stack())))
|
||||||
data = LibDataShallow{Data: nil, Code: 500, Err: "Panic recovered in LoadAll : " + fmt.Sprintf("%v", r) + " - " + string(debug.Stack())}
|
data = LibDataShallow{Data: nil, Code: 500, Err: "Panic recovered in LoadAll : " + fmt.Sprintf("%v", r) + " - " + string(debug.Stack())}
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
d, code, err := models.Model(r.collection.EnumIndex()).GetAccessor(&tools.APIRequest{
|
var caller *tools.HTTPCaller // define the caller
|
||||||
Caller: r.caller,
|
if len(c) > 0 {
|
||||||
Username: r.user,
|
caller = c[0]
|
||||||
PeerID: r.peerID,
|
}
|
||||||
Groups: r.groups,
|
d, code, err := models.Model(collection.EnumIndex()).GetAccessor(caller).LoadAll()
|
||||||
}).LoadAll(isDraft)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
data = LibDataShallow{Data: d, Code: code, Err: err.Error()}
|
data = LibDataShallow{Data: d, Code: code, Err: err.Error()}
|
||||||
return
|
return
|
||||||
@ -352,19 +286,18 @@ func (r *Request) LoadAll(isDraft bool) (data LibDataShallow) {
|
|||||||
* @param c ...*tools.HTTPCaller
|
* @param c ...*tools.HTTPCaller
|
||||||
* @return data LibData
|
* @return data LibData
|
||||||
*/
|
*/
|
||||||
func (r *Request) LoadOne(id string) (data LibData) {
|
func LoadOne(collection LibDataEnum, id string, c ...*tools.HTTPCaller) (data LibData) {
|
||||||
defer func() { // recover the panic
|
defer func() { // recover the panic
|
||||||
if r := recover(); r != nil {
|
if r := recover(); r != nil {
|
||||||
tools.UncatchedError = append(tools.UncatchedError, errors.New("Panic recovered in LoadOne : "+fmt.Sprintf("%v", r)+" - "+string(debug.Stack())))
|
tools.UncatchedError = append(tools.UncatchedError, errors.New("Panic recovered in LoadOne : "+fmt.Sprintf("%v", r)+" - "+string(debug.Stack())))
|
||||||
data = LibData{Data: nil, Code: 500, Err: "Panic recovered in LoadOne : " + fmt.Sprintf("%v", r) + " - " + string(debug.Stack())}
|
data = LibData{Data: nil, Code: 500, Err: "Panic recovered in LoadOne : " + fmt.Sprintf("%v", r) + " - " + string(debug.Stack())}
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
d, code, err := models.Model(r.collection.EnumIndex()).GetAccessor(&tools.APIRequest{
|
var caller *tools.HTTPCaller // define the caller
|
||||||
Caller: r.caller,
|
if len(c) > 0 {
|
||||||
Username: r.user,
|
caller = c[0]
|
||||||
PeerID: r.peerID,
|
}
|
||||||
Groups: r.groups,
|
d, code, err := models.Model(collection.EnumIndex()).GetAccessor(caller).LoadOne(id)
|
||||||
}).LoadOne(id)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
data = LibData{Data: d, Code: code, Err: err.Error()}
|
data = LibData{Data: d, Code: code, Err: err.Error()}
|
||||||
return
|
return
|
||||||
@ -381,20 +314,19 @@ func (r *Request) LoadOne(id string) (data LibData) {
|
|||||||
* @param c ...*tools.HTTPCaller
|
* @param c ...*tools.HTTPCaller
|
||||||
* @return data LibData
|
* @return data LibData
|
||||||
*/
|
*/
|
||||||
func (r *Request) UpdateOne(set map[string]interface{}, id string) (data LibData) {
|
func UpdateOne(collection LibDataEnum, set map[string]interface{}, id string, c ...*tools.HTTPCaller) (data LibData) {
|
||||||
defer func() { // recover the panic
|
defer func() { // recover the panic
|
||||||
if r := recover(); r != nil {
|
if r := recover(); r != nil {
|
||||||
tools.UncatchedError = append(tools.UncatchedError, errors.New("Panic recovered in UpdateOne : "+fmt.Sprintf("%v", r)+" - "+string(debug.Stack())))
|
tools.UncatchedError = append(tools.UncatchedError, errors.New("Panic recovered in UpdateOne : "+fmt.Sprintf("%v", r)+" - "+string(debug.Stack())))
|
||||||
data = LibData{Data: nil, Code: 500, Err: "Panic recovered in UpdateOne : " + fmt.Sprintf("%v", r) + " - " + string(debug.Stack())}
|
data = LibData{Data: nil, Code: 500, Err: "Panic recovered in UpdateOne : " + fmt.Sprintf("%v", r) + " - " + string(debug.Stack())}
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
model := models.Model(r.collection.EnumIndex())
|
var caller *tools.HTTPCaller // define the caller
|
||||||
d, code, err := model.GetAccessor(&tools.APIRequest{
|
if len(c) > 0 {
|
||||||
Caller: r.caller,
|
caller = c[0]
|
||||||
Username: r.user,
|
}
|
||||||
PeerID: r.peerID,
|
model := models.Model(collection.EnumIndex())
|
||||||
Groups: r.groups,
|
d, code, err := model.GetAccessor(caller).UpdateOne(model.Deserialize(set), id)
|
||||||
}).UpdateOne(model.Deserialize(set, model), id)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
data = LibData{Data: d, Code: code, Err: err.Error()}
|
data = LibData{Data: d, Code: code, Err: err.Error()}
|
||||||
return
|
return
|
||||||
@ -410,19 +342,18 @@ func (r *Request) UpdateOne(set map[string]interface{}, id string) (data LibData
|
|||||||
* @param c ...*tools.HTTPCaller
|
* @param c ...*tools.HTTPCaller
|
||||||
* @return data LibData
|
* @return data LibData
|
||||||
*/
|
*/
|
||||||
func (r *Request) DeleteOne(id string) (data LibData) {
|
func DeleteOne(collection LibDataEnum, id string, c ...*tools.HTTPCaller) (data LibData) {
|
||||||
defer func() { // recover the panic
|
defer func() { // recover the panic
|
||||||
if r := recover(); r != nil {
|
if r := recover(); r != nil {
|
||||||
tools.UncatchedError = append(tools.UncatchedError, errors.New("Panic recovered in DeleteOne : "+fmt.Sprintf("%v", r)+" - "+string(debug.Stack())))
|
tools.UncatchedError = append(tools.UncatchedError, errors.New("Panic recovered in DeleteOne : "+fmt.Sprintf("%v", r)+" - "+string(debug.Stack())))
|
||||||
data = LibData{Data: nil, Code: 500, Err: "Panic recovered in DeleteOne : " + fmt.Sprintf("%v", r) + " - " + string(debug.Stack())}
|
data = LibData{Data: nil, Code: 500, Err: "Panic recovered in DeleteOne : " + fmt.Sprintf("%v", r) + " - " + string(debug.Stack())}
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
d, code, err := models.Model(r.collection.EnumIndex()).GetAccessor(&tools.APIRequest{
|
var caller *tools.HTTPCaller // define the caller
|
||||||
Caller: r.caller,
|
if len(c) > 0 {
|
||||||
Username: r.user,
|
caller = c[0]
|
||||||
PeerID: r.peerID,
|
}
|
||||||
Groups: r.groups,
|
d, code, err := models.Model(collection.EnumIndex()).GetAccessor(caller).DeleteOne(id)
|
||||||
}).DeleteOne(id)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
data = LibData{Data: d, Code: code, Err: err.Error()}
|
data = LibData{Data: d, Code: code, Err: err.Error()}
|
||||||
return
|
return
|
||||||
@ -438,20 +369,19 @@ func (r *Request) DeleteOne(id string) (data LibData) {
|
|||||||
* @param c ...*tools.HTTPCaller
|
* @param c ...*tools.HTTPCaller
|
||||||
* @return data LibData
|
* @return data LibData
|
||||||
*/
|
*/
|
||||||
func (r *Request) StoreOne(object map[string]interface{}) (data LibData) {
|
func StoreOne(collection LibDataEnum, object map[string]interface{}, c ...*tools.HTTPCaller) (data LibData) {
|
||||||
defer func() { // recover the panic
|
defer func() { // recover the panic
|
||||||
if r := recover(); r != nil {
|
if r := recover(); r != nil {
|
||||||
tools.UncatchedError = append(tools.UncatchedError, errors.New("Panic recovered in StoreOne : "+fmt.Sprintf("%v", r)+" - "+string(debug.Stack())))
|
tools.UncatchedError = append(tools.UncatchedError, errors.New("Panic recovered in StoreOne : "+fmt.Sprintf("%v", r)+" - "+string(debug.Stack())))
|
||||||
data = LibData{Data: nil, Code: 500, Err: "Panic recovered in StoreOne : " + fmt.Sprintf("%v", r) + " - " + string(debug.Stack())}
|
data = LibData{Data: nil, Code: 500, Err: "Panic recovered in StoreOne : " + fmt.Sprintf("%v", r) + " - " + string(debug.Stack())}
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
model := models.Model(r.collection.EnumIndex())
|
var caller *tools.HTTPCaller // define the caller
|
||||||
d, code, err := model.GetAccessor(&tools.APIRequest{
|
if len(c) > 0 {
|
||||||
Caller: r.caller,
|
caller = c[0]
|
||||||
Username: r.user,
|
}
|
||||||
PeerID: r.peerID,
|
model := models.Model(collection.EnumIndex())
|
||||||
Groups: r.groups,
|
d, code, err := model.GetAccessor(caller).StoreOne(model.Deserialize(object))
|
||||||
}).StoreOne(model.Deserialize(object, model))
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
data = LibData{Data: d, Code: code, Err: err.Error()}
|
data = LibData{Data: d, Code: code, Err: err.Error()}
|
||||||
return
|
return
|
||||||
@ -467,20 +397,19 @@ func (r *Request) StoreOne(object map[string]interface{}) (data LibData) {
|
|||||||
* @param c ...*tools.HTTPCaller
|
* @param c ...*tools.HTTPCaller
|
||||||
* @return data LibData
|
* @return data LibData
|
||||||
*/
|
*/
|
||||||
func (r *Request) CopyOne(object map[string]interface{}) (data LibData) {
|
func CopyOne(collection LibDataEnum, object map[string]interface{}, c ...*tools.HTTPCaller) (data LibData) {
|
||||||
defer func() { // recover the panic
|
defer func() { // recover the panic
|
||||||
if r := recover(); r != nil {
|
if r := recover(); r != nil {
|
||||||
tools.UncatchedError = append(tools.UncatchedError, errors.New("Panic recovered in CopyOne : "+fmt.Sprintf("%v", r)+" - "+string(debug.Stack())))
|
tools.UncatchedError = append(tools.UncatchedError, errors.New("Panic recovered in CopyOne : "+fmt.Sprintf("%v", r)+" - "+string(debug.Stack())))
|
||||||
data = LibData{Data: nil, Code: 500, Err: "Panic recovered in UpdateOne : " + fmt.Sprintf("%v", r) + " - " + string(debug.Stack())}
|
data = LibData{Data: nil, Code: 500, Err: "Panic recovered in UpdateOne : " + fmt.Sprintf("%v", r) + " - " + string(debug.Stack())}
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
model := models.Model(r.collection.EnumIndex())
|
var caller *tools.HTTPCaller // define the caller
|
||||||
d, code, err := model.GetAccessor(&tools.APIRequest{
|
if len(c) > 0 {
|
||||||
Caller: r.caller,
|
caller = c[0]
|
||||||
Username: r.user,
|
}
|
||||||
PeerID: r.peerID,
|
model := models.Model(collection.EnumIndex())
|
||||||
Groups: r.groups,
|
d, code, err := model.GetAccessor(caller).CopyOne(model.Deserialize(object))
|
||||||
}).CopyOne(model.Deserialize(object, model))
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
data = LibData{Data: d, Code: code, Err: err.Error()}
|
data = LibData{Data: d, Code: code, Err: err.Error()}
|
||||||
return
|
return
|
||||||
@ -491,109 +420,74 @@ func (r *Request) CopyOne(object map[string]interface{}) (data LibData) {
|
|||||||
|
|
||||||
// ================ CAST ========================= //
|
// ================ CAST ========================= //
|
||||||
|
|
||||||
func (l *LibData) ToDataResource() *resources.DataResource {
|
func (l *LibData) ToDataResource() *data.DataResource {
|
||||||
if l.Data.GetAccessor(nil).GetType() == tools.DATA_RESOURCE {
|
if l.Data.GetAccessor(nil).GetType() == tools.DATA_RESOURCE.String() {
|
||||||
return l.Data.(*resources.DataResource)
|
return l.Data.(*data.DataResource)
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (l *LibData) ToComputeResource() *resources.ComputeResource {
|
func (l *LibData) ToComputeResource() *compute.ComputeResource {
|
||||||
if l.Data != nil && l.Data.GetAccessor(nil).GetType() == tools.COMPUTE_RESOURCE {
|
if l.Data != nil && l.Data.GetAccessor(nil).GetType() == tools.COMPUTE_RESOURCE.String() {
|
||||||
return l.Data.(*resources.ComputeResource)
|
return l.Data.(*compute.ComputeResource)
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
func (l *LibData) ToStorageResource() *resources.StorageResource {
|
func (l *LibData) ToStorageResource() *storage.StorageResource {
|
||||||
if l.Data.GetAccessor(nil).GetType() == tools.STORAGE_RESOURCE {
|
if l.Data.GetAccessor(nil).GetType() == tools.STORAGE_RESOURCE.String() {
|
||||||
return l.Data.(*resources.StorageResource)
|
return l.Data.(*storage.StorageResource)
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
func (l *LibData) ToProcessingResource() *resources.ProcessingResource {
|
func (l *LibData) ToProcessingResource() *processing.ProcessingResource {
|
||||||
if l.Data.GetAccessor(nil).GetType() == tools.PROCESSING_RESOURCE {
|
if l.Data.GetAccessor(nil).GetType() == tools.PROCESSING_RESOURCE.String() {
|
||||||
return l.Data.(*resources.ProcessingResource)
|
return l.Data.(*processing.ProcessingResource)
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
func (l *LibData) ToWorkflowResource() *resources.WorkflowResource {
|
func (l *LibData) ToWorkflowResource() *w.WorkflowResource {
|
||||||
if l.Data.GetAccessor(nil).GetType() == tools.WORKFLOW_RESOURCE {
|
if l.Data.GetAccessor(nil).GetType() == tools.WORKFLOW_RESOURCE.String() {
|
||||||
return l.Data.(*resources.WorkflowResource)
|
return l.Data.(*w.WorkflowResource)
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
func (l *LibData) ToPeer() *peer.Peer {
|
func (l *LibData) ToPeer() *peer.Peer {
|
||||||
if l.Data.GetAccessor(nil).GetType() == tools.PEER {
|
if l.Data.GetAccessor(nil).GetType() == tools.PEER.String() {
|
||||||
return l.Data.(*peer.Peer)
|
return l.Data.(*peer.Peer)
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (l *LibData) ToWorkflow() *w2.Workflow {
|
func (l *LibData) ToWorkflow() *w2.Workflow {
|
||||||
if l.Data.GetAccessor(nil).GetType() == tools.WORKFLOW {
|
if l.Data.GetAccessor(nil).GetType() == tools.WORKFLOW.String() {
|
||||||
return l.Data.(*w2.Workflow)
|
return l.Data.(*w2.Workflow)
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
func (l *LibData) ToWorkspace() *workspace.Workspace {
|
func (l *LibData) ToWorkspace() *workspace.Workspace {
|
||||||
if l.Data.GetAccessor(nil).GetType() == tools.WORKSPACE {
|
if l.Data.GetAccessor(nil).GetType() == tools.WORKSPACE.String() {
|
||||||
return l.Data.(*workspace.Workspace)
|
return l.Data.(*workspace.Workspace)
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (l *LibData) ToCollaborativeArea() *collaborative_area.CollaborativeArea {
|
func (l *LibData) ToCollaborativeArea() *collaborative_area.CollaborativeArea {
|
||||||
if l.Data.GetAccessor(nil).GetType() == tools.COLLABORATIVE_AREA {
|
if l.Data.GetAccessor(nil).GetType() == tools.COLLABORATIVE_AREA.String() {
|
||||||
return l.Data.(*collaborative_area.CollaborativeArea)
|
return l.Data.(*collaborative_area.CollaborativeArea)
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (l *LibData) ToRule() *rule.Rule {
|
func (l *LibData) ToRule() *rule.Rule {
|
||||||
if l.Data.GetAccessor(nil).GetType() == tools.COLLABORATIVE_AREA {
|
if l.Data.GetAccessor(nil).GetType() == tools.COLLABORATIVE_AREA.String() {
|
||||||
return l.Data.(*rule.Rule)
|
return l.Data.(*rule.Rule)
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (l *LibData) ToWorkflowExecution() *workflow_execution.WorkflowExecution {
|
func (l *LibData) ToWorkflowExecution() *workflow_execution.WorkflowExecution {
|
||||||
if l.Data.GetAccessor(nil).GetType() == tools.WORKFLOW_EXECUTION {
|
if l.Data.GetAccessor(nil).GetType() == tools.WORKFLOW_EXECUTION.String() {
|
||||||
return l.Data.(*workflow_execution.WorkflowExecution)
|
return l.Data.(*workflow_execution.WorkflowExecution)
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (l *LibData) ToOrder() *order.Order {
|
|
||||||
if l.Data.GetAccessor(nil).GetType() == tools.ORDER {
|
|
||||||
return l.Data.(*order.Order)
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (l *LibData) ToLiveDatacenter() *live.LiveDatacenter {
|
|
||||||
if l.Data.GetAccessor(nil).GetType() == tools.LIVE_DATACENTER {
|
|
||||||
return l.Data.(*live.LiveDatacenter)
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (l *LibData) ToLiveStorage() *live.LiveStorage {
|
|
||||||
if l.Data.GetAccessor(nil).GetType() == tools.LIVE_STORAGE {
|
|
||||||
return l.Data.(*live.LiveStorage)
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (l *LibData) ToBookings() *booking.Booking {
|
|
||||||
if l.Data.GetAccessor(nil).GetType() == tools.BOOKING {
|
|
||||||
return l.Data.(*booking.Booking)
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (l *LibData) ToPurchasedResource() *purchase_resource.PurchaseResource {
|
|
||||||
if l.Data.GetAccessor(nil).GetType() == tools.PURCHASE_RESOURCE {
|
|
||||||
return l.Data.(*purchase_resource.PurchaseResource)
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
5
go.mod
Executable file → Normal file
5
go.mod
Executable file → Normal file
@ -10,13 +10,12 @@ require (
|
|||||||
github.com/nats-io/nats.go v1.37.0
|
github.com/nats-io/nats.go v1.37.0
|
||||||
github.com/robfig/cron/v3 v3.0.1
|
github.com/robfig/cron/v3 v3.0.1
|
||||||
github.com/rs/zerolog v1.33.0
|
github.com/rs/zerolog v1.33.0
|
||||||
github.com/stretchr/testify v1.10.0
|
github.com/stretchr/testify v1.9.0
|
||||||
)
|
)
|
||||||
|
|
||||||
require (
|
require (
|
||||||
github.com/nats-io/nkeys v0.4.7 // indirect
|
github.com/nats-io/nkeys v0.4.7 // indirect
|
||||||
github.com/nats-io/nuid v1.0.1 // indirect
|
github.com/nats-io/nuid v1.0.1 // indirect
|
||||||
github.com/stretchr/objx v0.5.2 // indirect
|
|
||||||
)
|
)
|
||||||
|
|
||||||
require (
|
require (
|
||||||
@ -28,7 +27,6 @@ require (
|
|||||||
|
|
||||||
require (
|
require (
|
||||||
github.com/beorn7/perks v1.0.1 // indirect
|
github.com/beorn7/perks v1.0.1 // indirect
|
||||||
github.com/biter777/countries v1.7.5
|
|
||||||
github.com/cespare/xxhash/v2 v2.2.0 // indirect
|
github.com/cespare/xxhash/v2 v2.2.0 // indirect
|
||||||
github.com/davecgh/go-spew v1.1.1 // indirect
|
github.com/davecgh/go-spew v1.1.1 // indirect
|
||||||
github.com/gabriel-vasile/mimetype v1.4.4 // indirect
|
github.com/gabriel-vasile/mimetype v1.4.4 // indirect
|
||||||
@ -46,7 +44,6 @@ require (
|
|||||||
github.com/prometheus/client_model v0.5.0 // indirect
|
github.com/prometheus/client_model v0.5.0 // indirect
|
||||||
github.com/prometheus/common v0.48.0 // indirect
|
github.com/prometheus/common v0.48.0 // indirect
|
||||||
github.com/prometheus/procfs v0.12.0 // indirect
|
github.com/prometheus/procfs v0.12.0 // indirect
|
||||||
github.com/robfig/cron v1.2.0
|
|
||||||
github.com/shiena/ansicolor v0.0.0-20200904210342-c7312218db18 // indirect
|
github.com/shiena/ansicolor v0.0.0-20200904210342-c7312218db18 // indirect
|
||||||
github.com/xdg-go/pbkdf2 v1.0.0 // indirect
|
github.com/xdg-go/pbkdf2 v1.0.0 // indirect
|
||||||
github.com/xdg-go/scram v1.1.2 // indirect
|
github.com/xdg-go/scram v1.1.2 // indirect
|
||||||
|
8
go.sum
Executable file → Normal file
8
go.sum
Executable file → Normal file
@ -3,8 +3,6 @@ github.com/beego/beego/v2 v2.3.1 h1:7MUKMpJYzOXtCUsTEoXOxsDV/UcHw6CPbaWMlthVNsc=
|
|||||||
github.com/beego/beego/v2 v2.3.1/go.mod h1:5cqHsOHJIxkq44tBpRvtDe59GuVRVv/9/tyVDxd5ce4=
|
github.com/beego/beego/v2 v2.3.1/go.mod h1:5cqHsOHJIxkq44tBpRvtDe59GuVRVv/9/tyVDxd5ce4=
|
||||||
github.com/beorn7/perks v1.0.1 h1:VlbKKnNfV8bJzeqoa4cOKqO6bYr3WgKZxO8Z16+hsOM=
|
github.com/beorn7/perks v1.0.1 h1:VlbKKnNfV8bJzeqoa4cOKqO6bYr3WgKZxO8Z16+hsOM=
|
||||||
github.com/beorn7/perks v1.0.1/go.mod h1:G2ZrVWU2WbWT9wwq4/hrbKbnv/1ERSJQ0ibhJ6rlkpw=
|
github.com/beorn7/perks v1.0.1/go.mod h1:G2ZrVWU2WbWT9wwq4/hrbKbnv/1ERSJQ0ibhJ6rlkpw=
|
||||||
github.com/biter777/countries v1.7.5 h1:MJ+n3+rSxWQdqVJU8eBy9RqcdH6ePPn4PJHocVWUa+Q=
|
|
||||||
github.com/biter777/countries v1.7.5/go.mod h1:1HSpZ526mYqKJcpT5Ti1kcGQ0L0SrXWIaptUWjFfv2E=
|
|
||||||
github.com/cespare/xxhash/v2 v2.2.0 h1:DC2CZ1Ep5Y4k3ZQ899DldepgrayRUGE6BBZ/cd9Cj44=
|
github.com/cespare/xxhash/v2 v2.2.0 h1:DC2CZ1Ep5Y4k3ZQ899DldepgrayRUGE6BBZ/cd9Cj44=
|
||||||
github.com/cespare/xxhash/v2 v2.2.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
|
github.com/cespare/xxhash/v2 v2.2.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
|
||||||
github.com/coreos/etcd v3.3.17+incompatible/go.mod h1:uF7uidLiAD3TWHmW31ZFd/JWoc32PjwdhPthX9715RE=
|
github.com/coreos/etcd v3.3.17+incompatible/go.mod h1:uF7uidLiAD3TWHmW31ZFd/JWoc32PjwdhPthX9715RE=
|
||||||
@ -89,8 +87,6 @@ github.com/prometheus/common v0.48.0 h1:QO8U2CdOzSn1BBsmXJXduaaW+dY/5QLjfB8svtSz
|
|||||||
github.com/prometheus/common v0.48.0/go.mod h1:0/KsvlIEfPQCQ5I2iNSAWKPZziNCvRs5EC6ILDTlAPc=
|
github.com/prometheus/common v0.48.0/go.mod h1:0/KsvlIEfPQCQ5I2iNSAWKPZziNCvRs5EC6ILDTlAPc=
|
||||||
github.com/prometheus/procfs v0.12.0 h1:jluTpSng7V9hY0O2R9DzzJHYb2xULk9VTR1V1R/k6Bo=
|
github.com/prometheus/procfs v0.12.0 h1:jluTpSng7V9hY0O2R9DzzJHYb2xULk9VTR1V1R/k6Bo=
|
||||||
github.com/prometheus/procfs v0.12.0/go.mod h1:pcuDEFsWDnvcgNzo4EEweacyhjeA9Zk3cnaOZAZEfOo=
|
github.com/prometheus/procfs v0.12.0/go.mod h1:pcuDEFsWDnvcgNzo4EEweacyhjeA9Zk3cnaOZAZEfOo=
|
||||||
github.com/robfig/cron v1.2.0 h1:ZjScXvvxeQ63Dbyxy76Fj3AT3Ut0aKsyd2/tl3DTMuQ=
|
|
||||||
github.com/robfig/cron v1.2.0/go.mod h1:JGuDeoQd7Z6yL4zQhZ3OPEVHB7fL6Ka6skscFHfmt2k=
|
|
||||||
github.com/robfig/cron/v3 v3.0.1 h1:WdRxkvbJztn8LMz/QEvLN5sBU+xKpSqwwUO1Pjr4qDs=
|
github.com/robfig/cron/v3 v3.0.1 h1:WdRxkvbJztn8LMz/QEvLN5sBU+xKpSqwwUO1Pjr4qDs=
|
||||||
github.com/robfig/cron/v3 v3.0.1/go.mod h1:eQICP3HwyT7UooqI/z+Ov+PtYAWygg1TEWWzGIFLtro=
|
github.com/robfig/cron/v3 v3.0.1/go.mod h1:eQICP3HwyT7UooqI/z+Ov+PtYAWygg1TEWWzGIFLtro=
|
||||||
github.com/rogpeppe/go-internal v1.10.0 h1:TMyTOH3F/DB16zRVcYyreMH6GnZZrwQVAoYjRBZyWFQ=
|
github.com/rogpeppe/go-internal v1.10.0 h1:TMyTOH3F/DB16zRVcYyreMH6GnZZrwQVAoYjRBZyWFQ=
|
||||||
@ -106,13 +102,9 @@ github.com/smartystreets/assertions v0.0.0-20180927180507-b2de0cb4f26d/go.mod h1
|
|||||||
github.com/smartystreets/goconvey v0.0.0-20190731233626-505e41936337 h1:WN9BUFbdyOsSH/XohnWpXOlq9NBD5sGAB2FciQMUEe8=
|
github.com/smartystreets/goconvey v0.0.0-20190731233626-505e41936337 h1:WN9BUFbdyOsSH/XohnWpXOlq9NBD5sGAB2FciQMUEe8=
|
||||||
github.com/smartystreets/goconvey v0.0.0-20190731233626-505e41936337/go.mod h1:syvi0/a8iFYH4r/RixwvyeAJjdLS9QV7WQ/tjFTllLA=
|
github.com/smartystreets/goconvey v0.0.0-20190731233626-505e41936337/go.mod h1:syvi0/a8iFYH4r/RixwvyeAJjdLS9QV7WQ/tjFTllLA=
|
||||||
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||||
github.com/stretchr/objx v0.5.2 h1:xuMeJ0Sdp5ZMRXx/aWO6RZxdr3beISkG5/G/aIRr3pY=
|
|
||||||
github.com/stretchr/objx v0.5.2/go.mod h1:FRsXN1f5AsAjCGJKqEizvkpNtU+EGNCLh3NxZ/8L+MA=
|
|
||||||
github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
|
github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
|
||||||
github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg=
|
github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg=
|
||||||
github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
|
github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
|
||||||
github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA=
|
|
||||||
github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
|
|
||||||
github.com/xdg-go/pbkdf2 v1.0.0 h1:Su7DPu48wXMwC3bs7MCNG+z4FhcyEuz5dlvchbq0B0c=
|
github.com/xdg-go/pbkdf2 v1.0.0 h1:Su7DPu48wXMwC3bs7MCNG+z4FhcyEuz5dlvchbq0B0c=
|
||||||
github.com/xdg-go/pbkdf2 v1.0.0/go.mod h1:jrpuAogTd400dnrH08LKmI/xc1MbPOebTwRqcT5RDeI=
|
github.com/xdg-go/pbkdf2 v1.0.0/go.mod h1:jrpuAogTd400dnrH08LKmI/xc1MbPOebTwRqcT5RDeI=
|
||||||
github.com/xdg-go/scram v1.1.2 h1:FHX5I5B4i4hKRVRBCFRxq1iQRej7WO3hhBuJf+UUySY=
|
github.com/xdg-go/scram v1.1.2 h1:FHX5I5B4i4hKRVRBCFRxq1iQRej7WO3hhBuJf+UUySY=
|
||||||
|
@ -1,202 +0,0 @@
|
|||||||
package bill
|
|
||||||
|
|
||||||
import (
|
|
||||||
"sync"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/dbs"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/enum"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/pricing"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/order"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/peer"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/resources/purchase_resource"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
)
|
|
||||||
|
|
||||||
/*
|
|
||||||
* Booking is a struct that represents a booking
|
|
||||||
*/
|
|
||||||
|
|
||||||
type Bill struct {
|
|
||||||
utils.AbstractObject
|
|
||||||
OrderID string `json:"order_id" bson:"order_id" validate:"required"`
|
|
||||||
Status enum.CompletionStatus `json:"status" bson:"status" default:"0"`
|
|
||||||
SubOrders map[string]*PeerOrder `json:"sub_orders" bson:"sub_orders"`
|
|
||||||
Total float64 `json:"total" bson:"total" validate:"required"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func GenerateBill(order *order.Order, request *tools.APIRequest) (*Bill, error) {
|
|
||||||
// hhmmm : should get... the loop.
|
|
||||||
return &Bill{
|
|
||||||
AbstractObject: utils.AbstractObject{
|
|
||||||
Name: "bill_" + request.PeerID + "_" + time.Now().UTC().Format("2006-01-02T15:04:05"),
|
|
||||||
IsDraft: false,
|
|
||||||
},
|
|
||||||
OrderID: order.UUID,
|
|
||||||
Status: enum.PENDING,
|
|
||||||
// SubOrders: peerOrders,
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func DraftFirstBill(order *order.Order, request *tools.APIRequest) (*Bill, error) {
|
|
||||||
peers := map[string][]*PeerItemOrder{}
|
|
||||||
for _, p := range order.Purchases {
|
|
||||||
// TODO : if once
|
|
||||||
if _, ok := peers[p.DestPeerID]; !ok {
|
|
||||||
peers[p.DestPeerID] = []*PeerItemOrder{}
|
|
||||||
}
|
|
||||||
peers[p.DestPeerID] = append(peers[p.DestPeerID], &PeerItemOrder{
|
|
||||||
Purchase: p,
|
|
||||||
Item: p.PricedItem,
|
|
||||||
Quantity: 1,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
for _, b := range order.Bookings {
|
|
||||||
// TODO : if once
|
|
||||||
isPurchased := false
|
|
||||||
for _, p := range order.Purchases {
|
|
||||||
if p.ResourceID == b.ResourceID {
|
|
||||||
isPurchased = true
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if isPurchased {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if _, ok := peers[b.DestPeerID]; !ok {
|
|
||||||
peers[b.DestPeerID] = []*PeerItemOrder{}
|
|
||||||
}
|
|
||||||
peers[b.DestPeerID] = append(peers[b.DestPeerID], &PeerItemOrder{
|
|
||||||
Item: b.PricedItem,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
peerOrders := map[string]*PeerOrder{}
|
|
||||||
for peerID, items := range peers {
|
|
||||||
pr, _, err := peer.NewAccessor(request).LoadOne(peerID)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
peerOrders[peerID] = &PeerOrder{
|
|
||||||
PeerID: peerID,
|
|
||||||
BillingAddress: pr.(*peer.Peer).WalletAddress,
|
|
||||||
Items: items,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
bill := &Bill{
|
|
||||||
AbstractObject: utils.AbstractObject{
|
|
||||||
Name: "bill_" + request.PeerID + "_" + time.Now().UTC().Format("2006-01-02T15:04:05"),
|
|
||||||
IsDraft: true,
|
|
||||||
},
|
|
||||||
OrderID: order.UUID,
|
|
||||||
Status: enum.PENDING,
|
|
||||||
SubOrders: peerOrders,
|
|
||||||
}
|
|
||||||
return bill.SumUpBill(request)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *Bill) GetAccessor(request *tools.APIRequest) utils.Accessor {
|
|
||||||
return NewAccessor(request) // Create a new instance of the accessor
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *Bill) StoreDraftDefault() {
|
|
||||||
r.IsDraft = true
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *Bill) CanUpdate(set utils.DBObject) (bool, utils.DBObject) {
|
|
||||||
if !r.IsDraft && r.Status != set.(*Bill).Status {
|
|
||||||
return true, &Bill{Status: set.(*Bill).Status} // only state can be updated
|
|
||||||
}
|
|
||||||
return r.IsDraft, set
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *Bill) CanDelete() bool {
|
|
||||||
return r.IsDraft // only draft order can be deleted
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *Bill) SumUpBill(request *tools.APIRequest) (*Bill, error) {
|
|
||||||
for _, b := range d.SubOrders {
|
|
||||||
err := b.SumUpBill(request)
|
|
||||||
if err != nil {
|
|
||||||
return d, err
|
|
||||||
}
|
|
||||||
d.Total += b.Total
|
|
||||||
}
|
|
||||||
return d, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
type PeerOrder struct {
|
|
||||||
Error string `json:"error,omitempty" bson:"error,omitempty"`
|
|
||||||
PeerID string `json:"peer_id,omitempty" bson:"peer_id,omitempty"`
|
|
||||||
Status enum.CompletionStatus `json:"status" bson:"status" default:"0"`
|
|
||||||
BillingAddress string `json:"billing_address,omitempty" bson:"billing_address,omitempty"`
|
|
||||||
Items []*PeerItemOrder `json:"items,omitempty" bson:"items,omitempty"`
|
|
||||||
Total float64 `json:"total,omitempty" bson:"total,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *PeerOrder) Pay(request *tools.APIRequest, response chan *PeerOrder, wg *sync.WaitGroup) {
|
|
||||||
d.Status = enum.PENDING
|
|
||||||
go func() {
|
|
||||||
// DO SOMETHING TO PAY ON BLOCKCHAIN OR WHATEVER ON RETURN UPDATE STATUS
|
|
||||||
d.Status = enum.PAID // TO REMOVE LATER IT'S A MOCK
|
|
||||||
if d.Status == enum.PAID {
|
|
||||||
for _, b := range d.Items {
|
|
||||||
if !b.Item.IsPurchasable() {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
accessor := purchase_resource.NewAccessor(request)
|
|
||||||
accessor.StoreOne(&purchase_resource.PurchaseResource{
|
|
||||||
ResourceID: b.Item.GetID(),
|
|
||||||
ResourceType: b.Item.GetType(),
|
|
||||||
EndDate: b.Item.GetLocationEnd(),
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if d.Status != enum.PENDING {
|
|
||||||
response <- d
|
|
||||||
}
|
|
||||||
wg.Done()
|
|
||||||
}()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *PeerOrder) SumUpBill(request *tools.APIRequest) error {
|
|
||||||
for _, b := range d.Items {
|
|
||||||
tot, err := b.GetPrice(request) // missing something
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
d.Total += tot
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
type PeerItemOrder struct {
|
|
||||||
Quantity int `json:"quantity,omitempty" bson:"quantity,omitempty"`
|
|
||||||
Purchase *purchase_resource.PurchaseResource `json:"purchase,omitempty" bson:"purchase,omitempty"`
|
|
||||||
Item pricing.PricedItemITF `json:"item,omitempty" bson:"item,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *PeerItemOrder) GetPrice(request *tools.APIRequest) (float64, error) {
|
|
||||||
accessor := purchase_resource.NewAccessor(request)
|
|
||||||
search, code, _ := accessor.Search(&dbs.Filters{
|
|
||||||
And: map[string][]dbs.Filter{
|
|
||||||
"resource_id": {{Operator: dbs.EQUAL.String(), Value: d.Item.GetID()}},
|
|
||||||
},
|
|
||||||
}, "", d.Purchase.IsDraft)
|
|
||||||
if code == 200 && len(search) > 0 {
|
|
||||||
for _, s := range search {
|
|
||||||
if s.(*purchase_resource.PurchaseResource).EndDate == nil || time.Now().UTC().After(*s.(*purchase_resource.PurchaseResource).EndDate) {
|
|
||||||
return 0, nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
p, err := d.Item.GetPrice()
|
|
||||||
if err != nil {
|
|
||||||
return 0, err
|
|
||||||
}
|
|
||||||
return p * float64(d.Quantity), nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// WTF HOW TO SELECT THE RIGHT PRICE ???
|
|
||||||
// SHOULD SET A BUYING STATUS WHEN PAYMENT IS VALIDATED
|
|
@ -1,63 +0,0 @@
|
|||||||
package bill
|
|
||||||
|
|
||||||
import (
|
|
||||||
"cloud.o-forge.io/core/oc-lib/dbs"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/logs"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
)
|
|
||||||
|
|
||||||
type billMongoAccessor struct {
|
|
||||||
utils.AbstractAccessor // AbstractAccessor contains the basic fields of an accessor (model, caller)
|
|
||||||
}
|
|
||||||
|
|
||||||
// New creates a new instance of the billMongoAccessor
|
|
||||||
func NewAccessor(request *tools.APIRequest) *billMongoAccessor {
|
|
||||||
return &billMongoAccessor{
|
|
||||||
AbstractAccessor: utils.AbstractAccessor{
|
|
||||||
Logger: logs.CreateLogger(tools.LIVE_DATACENTER.String()), // Create a logger with the data type
|
|
||||||
Request: request,
|
|
||||||
Type: tools.LIVE_DATACENTER,
|
|
||||||
},
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
|
||||||
* Nothing special here, just the basic CRUD operations
|
|
||||||
*/
|
|
||||||
func (a *billMongoAccessor) DeleteOne(id string) (utils.DBObject, int, error) {
|
|
||||||
return utils.GenericDeleteOne(id, a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *billMongoAccessor) UpdateOne(set utils.DBObject, id string) (utils.DBObject, int, error) {
|
|
||||||
// should verify if a source is existing...
|
|
||||||
return utils.GenericUpdateOne(set, id, a, &Bill{})
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *billMongoAccessor) StoreOne(data utils.DBObject) (utils.DBObject, int, error) {
|
|
||||||
return utils.GenericStoreOne(data.(*Bill), a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *billMongoAccessor) CopyOne(data utils.DBObject) (utils.DBObject, int, error) {
|
|
||||||
return utils.GenericStoreOne(data.(*Bill), a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *billMongoAccessor) LoadOne(id string) (utils.DBObject, int, error) {
|
|
||||||
return utils.GenericLoadOne[*Bill](id, func(d utils.DBObject) (utils.DBObject, int, error) {
|
|
||||||
return d, 200, nil
|
|
||||||
}, a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *billMongoAccessor) LoadAll(isDraft bool) ([]utils.ShallowDBObject, int, error) {
|
|
||||||
return utils.GenericLoadAll[*Bill](a.getExec(), isDraft, a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *billMongoAccessor) Search(filters *dbs.Filters, search string, isDraft bool) ([]utils.ShallowDBObject, int, error) {
|
|
||||||
return utils.GenericSearch[*Bill](filters, search, (&Bill{}).GetObjectFilters(search), a.getExec(), isDraft, a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *billMongoAccessor) getExec() func(utils.DBObject) utils.ShallowDBObject {
|
|
||||||
return func(d utils.DBObject) utils.ShallowDBObject {
|
|
||||||
return d
|
|
||||||
}
|
|
||||||
}
|
|
@ -1,2 +0,0 @@
|
|||||||
# Billing process
|
|
||||||
Scheduler process a drafted order + a first bill corresponding to every once buying.
|
|
@ -1,14 +1,14 @@
|
|||||||
package booking
|
package booking
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"encoding/json"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/dbs"
|
"cloud.o-forge.io/core/oc-lib/dbs"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/enum"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/models"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/pricing"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
"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"
|
"cloud.o-forge.io/core/oc-lib/tools"
|
||||||
|
"github.com/google/uuid"
|
||||||
"go.mongodb.org/mongo-driver/bson/primitive"
|
"go.mongodb.org/mongo-driver/bson/primitive"
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -16,129 +16,76 @@ import (
|
|||||||
* Booking is a struct that represents a booking
|
* Booking is a struct that represents a booking
|
||||||
*/
|
*/
|
||||||
type Booking struct {
|
type Booking struct {
|
||||||
utils.AbstractObject // AbstractObject contains the basic fields of an object (id, name)
|
workflow_execution.WorkflowExecution // WorkflowExecution contains the workflow execution data
|
||||||
PricedItem pricing.PricedItemITF `json:"priced_item,omitempty" bson:"priced_item,omitempty"` // We need to add the validate:"required" tag once the pricing feature is implemented, removed to avoid handling the error
|
ComputeResourceID string `json:"compute_resource_id,omitempty" bson:"compute_resource_id,omitempty" validate:"required"` // ComputeResourceID is the ID of the compute resource specified in the booking
|
||||||
|
|
||||||
ResumeMetrics map[string]map[string]models.MetricResume `json:"resume_metrics,omitempty" bson:"resume_metrics,omitempty"`
|
|
||||||
ExecutionMetrics map[string][]models.MetricsSnapshot `json:"metrics,omitempty" bson:"metrics,omitempty"`
|
|
||||||
|
|
||||||
ExecutionsID string `json:"executions_id,omitempty" bson:"executions_id,omitempty" validate:"required"` // ExecutionsID is the ID of the executions
|
|
||||||
DestPeerID string `json:"dest_peer_id,omitempty"` // DestPeerID is the ID of the destination peer
|
|
||||||
WorkflowID string `json:"workflow_id,omitempty" bson:"workflow_id,omitempty"` // WorkflowID is the ID of the workflow
|
|
||||||
ExecutionID string `json:"execution_id,omitempty" bson:"execution_id,omitempty" validate:"required"`
|
|
||||||
State enum.BookingStatus `json:"state,omitempty" bson:"state,omitempty" validate:"required"` // State is the state of the booking
|
|
||||||
ExpectedStartDate time.Time `json:"expected_start_date,omitempty" bson:"expected_start_date,omitempty" validate:"required"` // ExpectedStartDate is the expected start date of the booking
|
|
||||||
ExpectedEndDate *time.Time `json:"expected_end_date,omitempty" bson:"expected_end_date,omitempty" validate:"required"` // ExpectedEndDate is the expected end date of the booking
|
|
||||||
|
|
||||||
RealStartDate *time.Time `json:"real_start_date,omitempty" bson:"real_start_date,omitempty"` // RealStartDate is the real start date of the booking
|
|
||||||
RealEndDate *time.Time `json:"real_end_date,omitempty" bson:"real_end_date,omitempty"` // RealEndDate is the real end date of the booking
|
|
||||||
|
|
||||||
ResourceType tools.DataType `json:"resource_type,omitempty" bson:"resource_type,omitempty" validate:"required"` // ResourceType is the type of the resource
|
|
||||||
ResourceID string `json:"resource_id,omitempty" bson:"resource_id,omitempty" validate:"required"` // could be a Compute or a Storage
|
|
||||||
}
|
|
||||||
|
|
||||||
func (b *Booking) CalcDeltaOfExecution() map[string]map[string]models.MetricResume {
|
|
||||||
m := map[string]map[string]models.MetricResume{}
|
|
||||||
for instance, snapshot := range b.ExecutionMetrics {
|
|
||||||
m[instance] = map[string]models.MetricResume{}
|
|
||||||
for _, metric := range snapshot {
|
|
||||||
for _, mm := range metric.Metrics {
|
|
||||||
if resume, ok := m[instance][mm.Name]; !ok {
|
|
||||||
m[instance][mm.Name] = models.MetricResume{
|
|
||||||
Delta: 0,
|
|
||||||
LastValue: mm.Value,
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
delta := resume.LastValue - mm.Value
|
|
||||||
if delta == 0 {
|
|
||||||
resume.Delta = delta
|
|
||||||
} else {
|
|
||||||
resume.Delta = (resume.Delta + delta) / 2
|
|
||||||
}
|
|
||||||
resume.LastValue = mm.Value
|
|
||||||
m[instance][mm.Name] = resume
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return m
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// CheckBooking checks if a booking is possible on a specific compute resource
|
// CheckBooking checks if a booking is possible on a specific compute resource
|
||||||
func (wfa *Booking) Check(id string, start time.Time, end *time.Time, parrallelAllowed int) (bool, error) {
|
func (wfa *Booking) CheckBooking(id string, start time.Time, end *time.Time) (bool, error) {
|
||||||
// check if
|
// check if
|
||||||
if end == nil {
|
if end == nil {
|
||||||
// if no end... then Book like a savage
|
// if no end... then Book like a savage
|
||||||
e := start.Add(time.Hour)
|
return true, nil
|
||||||
end = &e
|
|
||||||
}
|
}
|
||||||
accessor := NewAccessor(nil)
|
e := *end
|
||||||
|
accessor := wfa.GetAccessor(nil)
|
||||||
res, code, err := accessor.Search(&dbs.Filters{
|
res, code, err := accessor.Search(&dbs.Filters{
|
||||||
And: map[string][]dbs.Filter{ // check if there is a booking on the same compute resource by filtering on the compute_resource_id, the state and the execution date
|
And: map[string][]dbs.Filter{ // check if there is a booking on the same compute resource by filtering on the compute_resource_id, the state and the execution date
|
||||||
"resource_id": {{Operator: dbs.EQUAL.String(), Value: id}},
|
"compute_resource_id": {{Operator: dbs.EQUAL.String(), Value: id}},
|
||||||
"state": {{Operator: dbs.EQUAL.String(), Value: enum.DRAFT.EnumIndex()}},
|
"workflowexecution.state": {{Operator: dbs.EQUAL.String(), Value: workflow_execution.SCHEDULED.EnumIndex()}},
|
||||||
"expected_start_date": {
|
"workflowexecution.execution_date": {
|
||||||
{Operator: dbs.LTE.String(), Value: primitive.NewDateTimeFromTime(*end)},
|
{Operator: dbs.LTE.String(), Value: primitive.NewDateTimeFromTime(e)},
|
||||||
{Operator: dbs.GTE.String(), Value: primitive.NewDateTimeFromTime(start)},
|
{Operator: dbs.GTE.String(), Value: primitive.NewDateTimeFromTime(start)},
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
}, "", wfa.IsDraft)
|
}, "")
|
||||||
if code != 200 {
|
if code != 200 {
|
||||||
return false, err
|
return false, err
|
||||||
}
|
}
|
||||||
return len(res) <= parrallelAllowed, nil
|
return len(res) == 0, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *Booking) GetDelayForLaunch() time.Duration {
|
// tool to convert the argo status to a state
|
||||||
return d.RealStartDate.Sub(d.ExpectedStartDate)
|
func (wfa *Booking) ArgoStatusToState(status string) *Booking {
|
||||||
|
wfa.WorkflowExecution.ArgoStatusToState(status)
|
||||||
|
return wfa
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *Booking) GetDelayForFinishing() time.Duration {
|
func (ao *Booking) GetID() string {
|
||||||
if d.ExpectedEndDate == nil {
|
return ao.UUID
|
||||||
return time.Duration(0)
|
}
|
||||||
|
|
||||||
|
func (r *Booking) GenerateID() {
|
||||||
|
if r.UUID == "" {
|
||||||
|
r.UUID = uuid.New().String()
|
||||||
}
|
}
|
||||||
return d.RealEndDate.Sub(d.ExpectedStartDate)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *Booking) GetUsualDuration() time.Duration {
|
func (d *Booking) GetName() string {
|
||||||
return d.ExpectedEndDate.Sub(d.ExpectedStartDate)
|
return d.UUID + "_" + d.ExecDate.String()
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *Booking) GetRealDuration() time.Duration {
|
func (d *Booking) GetAccessor(caller *tools.HTTPCaller) utils.Accessor {
|
||||||
if d.RealEndDate == nil || d.RealStartDate == nil {
|
data := New() // Create a new instance of the accessor
|
||||||
return time.Duration(0)
|
data.Init(tools.BOOKING, caller) // Initialize the accessor with the BOOKING model type
|
||||||
|
return data
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dma *Booking) Deserialize(j map[string]interface{}) utils.DBObject {
|
||||||
|
b, err := json.Marshal(j)
|
||||||
|
if err != nil {
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
return d.RealEndDate.Sub(*d.RealStartDate)
|
json.Unmarshal(b, dma)
|
||||||
|
return dma
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *Booking) GetDelayOnDuration() time.Duration {
|
func (dma *Booking) Serialize() map[string]interface{} {
|
||||||
return d.GetRealDuration() - d.GetUsualDuration()
|
var m map[string]interface{}
|
||||||
}
|
b, err := json.Marshal(dma)
|
||||||
|
if err != nil {
|
||||||
func (d *Booking) GetAccessor(request *tools.APIRequest) utils.Accessor {
|
return nil
|
||||||
return NewAccessor(request) // Create a new instance of the accessor
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *Booking) VerifyAuth(request *tools.APIRequest) bool {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *Booking) StoreDraftDefault() {
|
|
||||||
r.IsDraft = false
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *Booking) CanUpdate(set utils.DBObject) (bool, utils.DBObject) {
|
|
||||||
if !r.IsDraft && r.State != set.(*Booking).State || r.RealStartDate != set.(*Booking).RealStartDate || r.RealEndDate != set.(*Booking).RealEndDate {
|
|
||||||
return true, &Booking{
|
|
||||||
State: set.(*Booking).State,
|
|
||||||
RealStartDate: set.(*Booking).RealStartDate,
|
|
||||||
RealEndDate: set.(*Booking).RealEndDate,
|
|
||||||
} // only state can be updated
|
|
||||||
}
|
}
|
||||||
// TODO : HERE WE CAN HANDLE THE CASE WHERE THE BOOKING IS DELAYED OR EXCEEDING OR ending sooner
|
json.Unmarshal(b, &m)
|
||||||
return r.IsDraft, set
|
return m
|
||||||
}
|
|
||||||
|
|
||||||
func (r *Booking) CanDelete() bool {
|
|
||||||
return r.IsDraft // only draft bookings can be deleted
|
|
||||||
}
|
}
|
||||||
|
@ -1,92 +1,103 @@
|
|||||||
package booking
|
package booking
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/dbs"
|
"cloud.o-forge.io/core/oc-lib/dbs"
|
||||||
"cloud.o-forge.io/core/oc-lib/logs"
|
"cloud.o-forge.io/core/oc-lib/dbs/mongo"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/enum"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
"cloud.o-forge.io/core/oc-lib/models/utils"
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
"cloud.o-forge.io/core/oc-lib/models/workflow_execution"
|
||||||
)
|
)
|
||||||
|
|
||||||
type BookingMongoAccessor struct {
|
type bookingMongoAccessor struct {
|
||||||
utils.AbstractAccessor // AbstractAccessor contains the basic fields of an accessor (model, caller)
|
utils.AbstractAccessor // AbstractAccessor contains the basic fields of an accessor (model, caller)
|
||||||
}
|
}
|
||||||
|
|
||||||
// New creates a new instance of the BookingMongoAccessor
|
// New creates a new instance of the bookingMongoAccessor
|
||||||
func NewAccessor(request *tools.APIRequest) *BookingMongoAccessor {
|
func New() *bookingMongoAccessor {
|
||||||
return &BookingMongoAccessor{
|
return &bookingMongoAccessor{}
|
||||||
AbstractAccessor: utils.AbstractAccessor{
|
|
||||||
Logger: logs.CreateLogger(tools.BOOKING.String()), // Create a logger with the data type
|
|
||||||
Request: request,
|
|
||||||
Type: tools.BOOKING,
|
|
||||||
},
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* Nothing special here, just the basic CRUD operations
|
* Nothing special here, just the basic CRUD operations
|
||||||
*/
|
*/
|
||||||
func (a *BookingMongoAccessor) DeleteOne(id string) (utils.DBObject, int, error) {
|
func (wfa *bookingMongoAccessor) DeleteOne(id string) (utils.DBObject, int, error) {
|
||||||
return utils.GenericDeleteOne(id, a)
|
return wfa.GenericDeleteOne(id, wfa)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a *BookingMongoAccessor) UpdateOne(set utils.DBObject, id string) (utils.DBObject, int, error) {
|
func (wfa *bookingMongoAccessor) UpdateOne(set utils.DBObject, id string) (utils.DBObject, int, error) {
|
||||||
if set.(*Booking).State == 0 {
|
return wfa.GenericUpdateOne(set, id, wfa, &Booking{})
|
||||||
return nil, 400, errors.New("state is required")
|
}
|
||||||
|
|
||||||
|
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
|
||||||
}
|
}
|
||||||
realSet := &Booking{State: set.(*Booking).State}
|
res_mongo.Decode(&workflow)
|
||||||
return utils.GenericUpdateOne(realSet, id, a, &Booking{})
|
if workflow.State == workflow_execution.SCHEDULED && time.Now().UTC().After(*workflow.ExecDate) {
|
||||||
}
|
workflow.State = workflow_execution.FORGOTTEN
|
||||||
|
wfa.GenericRawUpdateOne(&workflow, id, wfa)
|
||||||
func (a *BookingMongoAccessor) StoreOne(data utils.DBObject) (utils.DBObject, int, error) {
|
|
||||||
return utils.GenericStoreOne(data, a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *BookingMongoAccessor) CopyOne(data utils.DBObject) (utils.DBObject, int, error) {
|
|
||||||
return utils.GenericStoreOne(data, a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *BookingMongoAccessor) LoadOne(id string) (utils.DBObject, int, error) {
|
|
||||||
return utils.GenericLoadOne[*Booking](id, func(d utils.DBObject) (utils.DBObject, int, error) {
|
|
||||||
now := time.Now()
|
|
||||||
now = now.Add(time.Second * -60)
|
|
||||||
if d.(*Booking).State == enum.DRAFT && now.UTC().After(d.(*Booking).ExpectedStartDate) {
|
|
||||||
return utils.GenericDeleteOne(d.GetID(), a)
|
|
||||||
}
|
|
||||||
if (d.(*Booking).ExpectedEndDate) == nil {
|
|
||||||
d.(*Booking).State = enum.FORGOTTEN
|
|
||||||
utils.GenericRawUpdateOne(d, id, a)
|
|
||||||
} else if d.(*Booking).State == enum.SCHEDULED && now.UTC().After(d.(*Booking).ExpectedStartDate) {
|
|
||||||
d.(*Booking).State = enum.DELAYED
|
|
||||||
utils.GenericRawUpdateOne(d, id, a)
|
|
||||||
}
|
|
||||||
return d, 200, nil
|
|
||||||
}, a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *BookingMongoAccessor) LoadAll(isDraft bool) ([]utils.ShallowDBObject, int, error) {
|
|
||||||
return utils.GenericLoadAll[*Booking](a.getExec(), isDraft, a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *BookingMongoAccessor) Search(filters *dbs.Filters, search string, isDraft bool) ([]utils.ShallowDBObject, int, error) {
|
|
||||||
return utils.GenericSearch[*Booking](filters, search, (&Booking{}).GetObjectFilters(search), a.getExec(), isDraft, a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *BookingMongoAccessor) getExec() func(utils.DBObject) utils.ShallowDBObject {
|
|
||||||
return func(d utils.DBObject) utils.ShallowDBObject {
|
|
||||||
now := time.Now()
|
|
||||||
now = now.Add(time.Second * -60)
|
|
||||||
if d.(*Booking).State == enum.DRAFT && now.UTC().After(d.(*Booking).ExpectedStartDate) {
|
|
||||||
utils.GenericDeleteOne(d.GetID(), a)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
if d.(*Booking).State == enum.SCHEDULED && now.UTC().After(d.(*Booking).ExpectedStartDate) {
|
|
||||||
d.(*Booking).State = enum.DELAYED
|
|
||||||
utils.GenericRawUpdateOne(d, d.GetID(), a)
|
|
||||||
}
|
|
||||||
return d
|
|
||||||
}
|
}
|
||||||
|
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
|
||||||
}
|
}
|
||||||
|
@ -1,87 +0,0 @@
|
|||||||
package booking_test
|
|
||||||
|
|
||||||
import (
|
|
||||||
"testing"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/stretchr/testify/assert"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/booking"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/enum"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestBooking_GetDurations(t *testing.T) {
|
|
||||||
start := time.Now().Add(-2 * time.Hour)
|
|
||||||
end := start.Add(1 * time.Hour)
|
|
||||||
realStart := start.Add(30 * time.Minute)
|
|
||||||
realEnd := realStart.Add(90 * time.Minute)
|
|
||||||
|
|
||||||
b := &booking.Booking{
|
|
||||||
ExpectedStartDate: start,
|
|
||||||
ExpectedEndDate: &end,
|
|
||||||
RealStartDate: &realStart,
|
|
||||||
RealEndDate: &realEnd,
|
|
||||||
}
|
|
||||||
|
|
||||||
assert.Equal(t, 30*time.Minute, b.GetDelayForLaunch())
|
|
||||||
assert.Equal(t, 90*time.Minute, b.GetRealDuration())
|
|
||||||
assert.Equal(t, end.Sub(start), b.GetUsualDuration())
|
|
||||||
assert.Equal(t, b.GetRealDuration()-b.GetUsualDuration(), b.GetDelayOnDuration())
|
|
||||||
assert.Equal(t, realEnd.Sub(start), b.GetDelayForFinishing())
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestBooking_GetAccessor(t *testing.T) {
|
|
||||||
req := &tools.APIRequest{}
|
|
||||||
b := &booking.Booking{}
|
|
||||||
accessor := b.GetAccessor(req)
|
|
||||||
|
|
||||||
assert.NotNil(t, accessor)
|
|
||||||
assert.Equal(t, tools.BOOKING, accessor.(*booking.BookingMongoAccessor).Type)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestBooking_VerifyAuth(t *testing.T) {
|
|
||||||
assert.True(t, (&booking.Booking{}).VerifyAuth(nil))
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestBooking_StoreDraftDefault(t *testing.T) {
|
|
||||||
b := &booking.Booking{}
|
|
||||||
b.StoreDraftDefault()
|
|
||||||
assert.False(t, b.IsDraft)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestBooking_CanUpdate(t *testing.T) {
|
|
||||||
now := time.Now()
|
|
||||||
b := &booking.Booking{
|
|
||||||
State: enum.SCHEDULED,
|
|
||||||
AbstractObject: utils.AbstractObject{IsDraft: false},
|
|
||||||
RealStartDate: &now,
|
|
||||||
}
|
|
||||||
|
|
||||||
set := &booking.Booking{
|
|
||||||
State: enum.DELAYED,
|
|
||||||
RealStartDate: &now,
|
|
||||||
}
|
|
||||||
|
|
||||||
ok, result := b.CanUpdate(set)
|
|
||||||
assert.True(t, ok)
|
|
||||||
assert.Equal(t, enum.DELAYED, result.(*booking.Booking).State)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestBooking_CanDelete(t *testing.T) {
|
|
||||||
b := &booking.Booking{AbstractObject: utils.AbstractObject{IsDraft: true}}
|
|
||||||
assert.True(t, b.CanDelete())
|
|
||||||
|
|
||||||
b.IsDraft = false
|
|
||||||
assert.False(t, b.CanDelete())
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestNewAccessor(t *testing.T) {
|
|
||||||
req := &tools.APIRequest{}
|
|
||||||
accessor := booking.NewAccessor(req)
|
|
||||||
|
|
||||||
assert.NotNil(t, accessor)
|
|
||||||
assert.Equal(t, tools.BOOKING, accessor.Type)
|
|
||||||
assert.Equal(t, req, accessor.Request)
|
|
||||||
}
|
|
@ -1,16 +1,16 @@
|
|||||||
package collaborative_area
|
package collaborative_area
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"slices"
|
"encoding/json"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/config"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/collaborative_area/rules/rule"
|
"cloud.o-forge.io/core/oc-lib/models/collaborative_area/rules/rule"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/peer"
|
"cloud.o-forge.io/core/oc-lib/models/peer"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
"cloud.o-forge.io/core/oc-lib/models/utils"
|
||||||
w "cloud.o-forge.io/core/oc-lib/models/workflow"
|
w "cloud.o-forge.io/core/oc-lib/models/workflow"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/workspace"
|
"cloud.o-forge.io/core/oc-lib/models/workspace"
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
"cloud.o-forge.io/core/oc-lib/tools"
|
||||||
|
"github.com/google/uuid"
|
||||||
)
|
)
|
||||||
|
|
||||||
type CollaborativeAreaRule struct {
|
type CollaborativeAreaRule struct {
|
||||||
@ -27,13 +27,14 @@ type CollaborativeAreaRule struct {
|
|||||||
type CollaborativeArea struct {
|
type CollaborativeArea struct {
|
||||||
utils.AbstractObject // AbstractObject contains the basic fields of an object (id, name)
|
utils.AbstractObject // AbstractObject contains the basic fields of an object (id, name)
|
||||||
IsSent bool `json:"is_sent" bson:"-"` // IsSent is a flag that indicates if the workspace is sent
|
IsSent bool `json:"is_sent" bson:"-"` // IsSent is a flag that indicates if the workspace is sent
|
||||||
|
CreatorID string `json:"peer_id,omitempty" bson:"peer_id,omitempty" validate:"required"` // CreatorID is the ID of the creator
|
||||||
Version string `json:"version,omitempty" bson:"version,omitempty"` // Version is the version of the workspace
|
Version string `json:"version,omitempty" bson:"version,omitempty"` // Version is the version of the workspace
|
||||||
Description string `json:"description,omitempty" bson:"description,omitempty" validate:"required"` // Description is the description of the workspace
|
Description string `json:"description,omitempty" bson:"description,omitempty" validate:"required"` // Description is the description of the workspace
|
||||||
CollaborativeAreaRule *CollaborativeAreaRule `json:"collaborative_area,omitempty" bson:"collaborative_area,omitempty"` // CollaborativeArea is the collaborative area of the workspace
|
CollaborativeAreaRule *CollaborativeAreaRule `json:"collaborative_area,omitempty" bson:"collaborative_area,omitempty"` // CollaborativeArea is the collaborative area of the workspace
|
||||||
Attributes map[string]interface{} `json:"attributes,omitempty" bson:"attributes,omitempty"` // Attributes is the attributes of the workspace (TODO)
|
Attributes map[string]interface{} `json:"attributes,omitempty" bson:"attributes,omitempty"` // Attributes is the attributes of the workspace (TODO)
|
||||||
Workspaces []string `json:"workspaces" bson:"workspaces"` // Workspaces is the workspaces of the workspace
|
Workspaces []string `json:"workspaces" bson:"workspaces"` // Workspaces is the workspaces of the workspace
|
||||||
Workflows []string `json:"workflows" bson:"workflows"` // Workflows is the workflows of the workspace
|
Workflows []string `json:"workflows" bson:"workflows"` // Workflows is the workflows of the workspace
|
||||||
AllowedPeersGroup map[string][]string `json:"allowed_peers_group" bson:"allowed_peers_group"` // AllowedPeersGroup is the group of allowed peers
|
Peers []string `json:"peers" bson:"peers"` // Peers is the peers of the workspace
|
||||||
Rules []string `json:"rules" bson:"rules,omitempty"` // Rules is the rules of the workspace
|
Rules []string `json:"rules" bson:"rules,omitempty"` // Rules is the rules of the workspace
|
||||||
|
|
||||||
SharedRules []*rule.Rule `json:"shared_rules,omitempty" bson:"-"` // SharedRules is the shared rules of the workspace
|
SharedRules []*rule.Rule `json:"shared_rules,omitempty" bson:"-"` // SharedRules is the shared rules of the workspace
|
||||||
@ -42,62 +43,41 @@ type CollaborativeArea struct {
|
|||||||
SharedPeers []*peer.Peer `json:"shared_peers,omitempty" bson:"-"` // SharedPeers is the shared peers of the workspace
|
SharedPeers []*peer.Peer `json:"shared_peers,omitempty" bson:"-"` // SharedPeers is the shared peers of the workspace
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ao *CollaborativeArea) Clear(peerID string) {
|
func (ao *CollaborativeArea) GetID() string {
|
||||||
if ao.AllowedPeersGroup == nil {
|
return ao.UUID
|
||||||
ao.AllowedPeersGroup = map[string][]string{}
|
|
||||||
}
|
|
||||||
ao.CreatorID = peerID
|
|
||||||
if config.GetConfig().Whitelist {
|
|
||||||
ao.AllowedPeersGroup[peerID] = []string{"*"}
|
|
||||||
} else {
|
|
||||||
ao.AllowedPeersGroup[peerID] = []string{}
|
|
||||||
}
|
|
||||||
// then reset the shared fields
|
|
||||||
if ao.Workspaces == nil {
|
|
||||||
ao.Workspaces = []string{}
|
|
||||||
}
|
|
||||||
if ao.Workflows == nil {
|
|
||||||
ao.Workflows = []string{}
|
|
||||||
}
|
|
||||||
if ao.Rules == nil {
|
|
||||||
ao.Rules = []string{}
|
|
||||||
}
|
|
||||||
if ao.CollaborativeAreaRule == nil {
|
|
||||||
ao.CollaborativeAreaRule = &CollaborativeAreaRule{
|
|
||||||
ShareMode: "private",
|
|
||||||
ExploitedBy: "collaborators only",
|
|
||||||
}
|
|
||||||
}
|
|
||||||
ao.CollaborativeAreaRule.CreatedAt = time.Now().UTC()
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ao *CollaborativeArea) VerifyAuth(request *tools.APIRequest) bool {
|
func (r *CollaborativeArea) GenerateID() {
|
||||||
if (ao.AllowedPeersGroup != nil || config.GetConfig().Whitelist) && request != nil {
|
if r.UUID == "" {
|
||||||
if grps, ok := ao.AllowedPeersGroup[request.PeerID]; ok || config.GetConfig().Whitelist {
|
r.UUID = uuid.New().String()
|
||||||
if slices.Contains(grps, "*") || (!ok && config.GetConfig().Whitelist) {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
for _, grp := range grps {
|
|
||||||
if slices.Contains(request.Groups, grp) {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
return ao.AbstractObject.VerifyAuth(request)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *CollaborativeArea) GetAccessor(request *tools.APIRequest) utils.Accessor {
|
func (d *CollaborativeArea) GetName() string {
|
||||||
return NewAccessor(request) // Create a new instance of the accessor
|
return d.Name
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *CollaborativeArea) Trim() *CollaborativeArea {
|
func (d *CollaborativeArea) GetAccessor(caller *tools.HTTPCaller) utils.Accessor {
|
||||||
return d
|
data := New() // Create a new instance of the accessor
|
||||||
|
data.Init(tools.COLLABORATIVE_AREA, caller) // Initialize the accessor with the SHARED_WORKSPACE model type
|
||||||
|
return data
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *CollaborativeArea) StoreDraftDefault() {
|
func (dma *CollaborativeArea) Deserialize(j map[string]interface{}) utils.DBObject {
|
||||||
d.AllowedPeersGroup = map[string][]string{
|
b, err := json.Marshal(j)
|
||||||
d.CreatorID: []string{"*"},
|
if err != nil {
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
d.IsDraft = false
|
json.Unmarshal(b, dma)
|
||||||
|
return dma
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dma *CollaborativeArea) Serialize() map[string]interface{} {
|
||||||
|
var m map[string]interface{}
|
||||||
|
b, err := json.Marshal(dma)
|
||||||
|
if err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
json.Unmarshal(b, &m)
|
||||||
|
return m
|
||||||
}
|
}
|
||||||
|
@ -1,15 +1,16 @@
|
|||||||
package collaborative_area
|
package collaborative_area
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"slices"
|
"slices"
|
||||||
|
"time"
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/dbs"
|
"cloud.o-forge.io/core/oc-lib/dbs"
|
||||||
"cloud.o-forge.io/core/oc-lib/logs"
|
"cloud.o-forge.io/core/oc-lib/dbs/mongo"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/collaborative_area/rules/rule"
|
"cloud.o-forge.io/core/oc-lib/models/collaborative_area/rules/rule"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/peer"
|
"cloud.o-forge.io/core/oc-lib/models/peer"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
"cloud.o-forge.io/core/oc-lib/models/utils"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/workflow"
|
|
||||||
w "cloud.o-forge.io/core/oc-lib/models/workflow"
|
w "cloud.o-forge.io/core/oc-lib/models/workflow"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/workspace"
|
"cloud.o-forge.io/core/oc-lib/models/workspace"
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
"cloud.o-forge.io/core/oc-lib/tools"
|
||||||
@ -18,150 +19,43 @@ import (
|
|||||||
// SharedWorkspace is a struct that represents a collaborative area
|
// SharedWorkspace is a struct that represents a collaborative area
|
||||||
type collaborativeAreaMongoAccessor struct {
|
type collaborativeAreaMongoAccessor struct {
|
||||||
utils.AbstractAccessor // AbstractAccessor contains the basic fields of an accessor (model, caller)
|
utils.AbstractAccessor // AbstractAccessor contains the basic fields of an accessor (model, caller)
|
||||||
|
|
||||||
workspaceAccessor utils.Accessor
|
|
||||||
workflowAccessor utils.Accessor
|
|
||||||
peerAccessor utils.Accessor
|
|
||||||
ruleAccessor utils.Accessor
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewAccessor(request *tools.APIRequest) *collaborativeAreaMongoAccessor {
|
// New creates a new instance of the collaborativeAreaMongoAccessor
|
||||||
return &collaborativeAreaMongoAccessor{
|
func New() *collaborativeAreaMongoAccessor {
|
||||||
AbstractAccessor: utils.AbstractAccessor{
|
return &collaborativeAreaMongoAccessor{}
|
||||||
Logger: logs.CreateLogger(tools.COLLABORATIVE_AREA.String()), // Create a logger with the data type
|
|
||||||
Request: request,
|
|
||||||
Type: tools.COLLABORATIVE_AREA,
|
|
||||||
},
|
|
||||||
workspaceAccessor: (&workspace.Workspace{}).GetAccessor(request),
|
|
||||||
workflowAccessor: (&w.Workflow{}).GetAccessor(request),
|
|
||||||
peerAccessor: (&peer.Peer{}).GetAccessor(request),
|
|
||||||
ruleAccessor: (&rule.Rule{}).GetAccessor(request),
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// DeleteOne deletes a collaborative area from the database, given its ID, it automatically share to peers if the workspace is shared
|
// DeleteOne deletes a collaborative area from the database, given its ID, it automatically share to peers if the workspace is shared
|
||||||
func (a *collaborativeAreaMongoAccessor) DeleteOne(id string) (utils.DBObject, int, error) {
|
func (wfa *collaborativeAreaMongoAccessor) DeleteOne(id string) (utils.DBObject, int, error) {
|
||||||
set, code, err := a.LoadOne(id)
|
set, code, _ := wfa.LoadOne(id)
|
||||||
if code != 200 {
|
if code == 200 { // always delete on peers than recreate
|
||||||
return nil, code, err
|
wfa.deleteToPeer(set.(*CollaborativeArea))
|
||||||
}
|
}
|
||||||
a.deleteToPeer(set.(*CollaborativeArea))
|
wfa.sharedWorkflow(&CollaborativeArea{}, id) // create all shared workflows
|
||||||
a.sharedWorkflow(&CollaborativeArea{}, id) // create all shared workflows
|
wfa.sharedWorkspace(&CollaborativeArea{}, id) // create all collaborative areas
|
||||||
a.sharedWorkspace(&CollaborativeArea{}, id) // create all collaborative areas
|
return wfa.GenericDeleteOne(id, wfa) // then add on yours
|
||||||
return utils.GenericDeleteOne(id, a) // then add on yours
|
|
||||||
}
|
|
||||||
|
|
||||||
// UpdateOne updates a collaborative area in the database, given its ID and the new data, it automatically share to peers if the workspace is shared
|
|
||||||
func (a *collaborativeAreaMongoAccessor) UpdateOne(set utils.DBObject, id string) (utils.DBObject, int, error) {
|
|
||||||
res, code, err := utils.GenericUpdateOne(set.(*CollaborativeArea).Trim(), id, a, &CollaborativeArea{})
|
|
||||||
// a.deleteToPeer(res.(*CollaborativeArea)) // delete the collaborative area on the peer
|
|
||||||
a.sharedWorkflow(res.(*CollaborativeArea), id) // replace all shared workflows
|
|
||||||
a.sharedWorkspace(res.(*CollaborativeArea), id) // replace all collaborative areas (not shared worspace obj but workspace one)
|
|
||||||
// a.sendToPeer(res.(*CollaborativeArea)) // send the collaborative area (collaborative area object) to the peers
|
|
||||||
return res, code, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// StoreOne stores a collaborative area in the database, it automatically share to peers if the workspace is shared
|
|
||||||
func (a *collaborativeAreaMongoAccessor) StoreOne(data utils.DBObject) (utils.DBObject, int, error) {
|
|
||||||
_, id := (&peer.Peer{}).IsMySelf() // get the local peer
|
|
||||||
data.(*CollaborativeArea).Clear(id) // set the creator
|
|
||||||
// retrieve or proper peer
|
|
||||||
if data.(*CollaborativeArea).CollaborativeAreaRule != nil {
|
|
||||||
data.(*CollaborativeArea).CollaborativeAreaRule = &CollaborativeAreaRule{}
|
|
||||||
}
|
|
||||||
data.(*CollaborativeArea).CollaborativeAreaRule.Creator = id
|
|
||||||
d, code, err := utils.GenericStoreOne(data.(*CollaborativeArea).Trim(), a)
|
|
||||||
if code == 200 {
|
|
||||||
a.sharedWorkflow(d.(*CollaborativeArea), d.GetID()) // create all shared workflows
|
|
||||||
a.sharedWorkspace(d.(*CollaborativeArea), d.GetID()) // create all collaborative areas
|
|
||||||
a.sendToPeer(d.(*CollaborativeArea)) // send the collaborative area (collaborative area object) to the peers
|
|
||||||
}
|
|
||||||
return data, code, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// CopyOne copies a CollaborativeArea in the database
|
|
||||||
func (a *collaborativeAreaMongoAccessor) CopyOne(data utils.DBObject) (utils.DBObject, int, error) {
|
|
||||||
return a.StoreOne(data)
|
|
||||||
}
|
|
||||||
|
|
||||||
func filterEnrich[T utils.ShallowDBObject](arr []string, isDrafted bool, a utils.Accessor) []T {
|
|
||||||
var new []T
|
|
||||||
res, code, _ := a.Search(&dbs.Filters{
|
|
||||||
Or: map[string][]dbs.Filter{
|
|
||||||
"abstractobject.id": {{Operator: dbs.IN.String(), Value: arr}},
|
|
||||||
},
|
|
||||||
}, "", isDrafted)
|
|
||||||
fmt.Println(res, arr, isDrafted, a)
|
|
||||||
if code == 200 {
|
|
||||||
for _, r := range res {
|
|
||||||
new = append(new, r.(T))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return new
|
|
||||||
}
|
|
||||||
|
|
||||||
// enrich is a function that enriches the CollaborativeArea with the shared objects
|
|
||||||
func (a *collaborativeAreaMongoAccessor) enrich(sharedWorkspace *CollaborativeArea, isDrafted bool, request *tools.APIRequest) *CollaborativeArea {
|
|
||||||
sharedWorkspace.SharedWorkspaces = filterEnrich[*workspace.Workspace](sharedWorkspace.Workspaces, isDrafted, a.workspaceAccessor)
|
|
||||||
sharedWorkspace.SharedWorkflows = filterEnrich[*workflow.Workflow](sharedWorkspace.Workflows, isDrafted, a.workflowAccessor)
|
|
||||||
peerskey := []string{}
|
|
||||||
fmt.Println("PEERS 1", sharedWorkspace.AllowedPeersGroup)
|
|
||||||
for k, v := range sharedWorkspace.AllowedPeersGroup {
|
|
||||||
canFound := false
|
|
||||||
for _, t := range request.Groups {
|
|
||||||
if slices.Contains(v, t) {
|
|
||||||
canFound = true
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
fmt.Println("PEERS 2", canFound, v)
|
|
||||||
if slices.Contains(v, "*") || canFound {
|
|
||||||
peerskey = append(peerskey, k)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
fmt.Println("PEERS", peerskey)
|
|
||||||
sharedWorkspace.SharedPeers = filterEnrich[*peer.Peer](peerskey, isDrafted, a.peerAccessor)
|
|
||||||
sharedWorkspace.SharedRules = filterEnrich[*rule.Rule](sharedWorkspace.Rules, isDrafted, a.ruleAccessor)
|
|
||||||
return sharedWorkspace
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *collaborativeAreaMongoAccessor) LoadOne(id string) (utils.DBObject, int, error) {
|
|
||||||
return utils.GenericLoadOne[*CollaborativeArea](id, func(d utils.DBObject) (utils.DBObject, int, error) {
|
|
||||||
return a.enrich(d.(*CollaborativeArea), false, a.Request), 200, nil
|
|
||||||
}, a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *collaborativeAreaMongoAccessor) LoadAll(isDrafted bool) ([]utils.ShallowDBObject, int, error) {
|
|
||||||
return utils.GenericLoadAll[*CollaborativeArea](func(d utils.DBObject) utils.ShallowDBObject {
|
|
||||||
return a.enrich(d.(*CollaborativeArea), isDrafted, a.Request)
|
|
||||||
}, isDrafted, a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *collaborativeAreaMongoAccessor) Search(filters *dbs.Filters, search string, isDrafted bool) ([]utils.ShallowDBObject, int, error) {
|
|
||||||
return utils.GenericSearch[*CollaborativeArea](filters, search, (&CollaborativeArea{}).GetObjectFilters(search),
|
|
||||||
func(d utils.DBObject) utils.ShallowDBObject {
|
|
||||||
return a.enrich(d.(*CollaborativeArea), isDrafted, a.Request)
|
|
||||||
}, isDrafted, a)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/*
|
/*
|
||||||
sharedWorkspace is a function that shares the collaborative area to the peers
|
sharedWorkspace is a function that shares the collaborative area to the peers
|
||||||
*/
|
*/
|
||||||
func (a *collaborativeAreaMongoAccessor) sharedWorkspace(shared *CollaborativeArea, id string) {
|
func (wfa *collaborativeAreaMongoAccessor) sharedWorkspace(shared *CollaborativeArea, id string) {
|
||||||
eldest, code, _ := a.LoadOne(id) // get the eldest
|
eldest, code, _ := wfa.LoadOne(id) // get the eldest
|
||||||
|
accessor := (&workspace.Workspace{}).GetAccessor(nil)
|
||||||
if code == 200 {
|
if code == 200 {
|
||||||
eld := eldest.(*CollaborativeArea)
|
eld := eldest.(*CollaborativeArea)
|
||||||
if eld.Workspaces != nil { // update all your workspaces in the eldest by replacing shared ref by an empty string
|
if eld.Workspaces != nil { // update all your workspaces in the eldest by replacing shared ref by an empty string
|
||||||
for _, v := range eld.Workspaces {
|
for _, v := range eld.Workspaces {
|
||||||
a.workspaceAccessor.UpdateOne(&workspace.Workspace{Shared: ""}, v)
|
accessor.UpdateOne(&workspace.Workspace{Shared: ""}, v)
|
||||||
if a.GetCaller() != nil || a.GetCaller().URLS == nil || a.GetCaller().URLS[tools.WORKSPACE] == nil {
|
if wfa.Caller != nil || wfa.Caller.URLS == nil || wfa.Caller.URLS[tools.WORKSPACE] == nil {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
paccess := (&peer.Peer{}) // send to all peers
|
paccess := (&peer.Peer{}) // send to all peers
|
||||||
for k := range shared.AllowedPeersGroup { // delete the collaborative area on the peer
|
for _, p := range shared.Peers { // delete the collaborative area on the peer
|
||||||
b, err := paccess.LaunchPeerExecution(k, v, tools.WORKSPACE, tools.DELETE, nil, a.GetCaller())
|
b, err := paccess.LaunchPeerExecution(p, v, tools.WORKSPACE, tools.DELETE, nil, wfa.Caller)
|
||||||
if err != nil && b == nil {
|
if err != nil && b == nil {
|
||||||
a.Logger.Error().Msg("Could not send to peer " + k + ". Error: " + err.Error())
|
wfa.Logger.Error().Msg("Could not send to peer " + p + ". Error: " + err.Error())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -169,20 +63,20 @@ func (a *collaborativeAreaMongoAccessor) sharedWorkspace(shared *CollaborativeAr
|
|||||||
}
|
}
|
||||||
if shared.Workspaces != nil {
|
if shared.Workspaces != nil {
|
||||||
for _, v := range shared.Workspaces { // update all the collaborative areas
|
for _, v := range shared.Workspaces { // update all the collaborative areas
|
||||||
workspace, code, _ := a.workspaceAccessor.UpdateOne(&workspace.Workspace{Shared: shared.UUID}, v) // add the shared ref to workspace
|
workspace, code, _ := accessor.UpdateOne(&workspace.Workspace{Shared: shared.UUID}, v) // add the shared ref to workspace
|
||||||
if a.GetCaller() != nil || a.GetCaller().URLS == nil || a.GetCaller().URLS[tools.WORKSPACE] == nil {
|
if wfa.Caller != nil || wfa.Caller.URLS == nil || wfa.Caller.URLS[tools.WORKSPACE] == nil {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
for k := range shared.AllowedPeersGroup {
|
for _, p := range shared.Peers {
|
||||||
if code != 200 {
|
if code != 200 {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
paccess := (&peer.Peer{}) // send to all peers, add the collaborative area on the peer
|
paccess := (&peer.Peer{}) // send to all peers, add the collaborative area on the peer
|
||||||
s := workspace.Serialize(workspace)
|
s := workspace.Serialize()
|
||||||
s["name"] = fmt.Sprintf("%v", s["name"]) + "_" + k
|
s["name"] = fmt.Sprintf("%v", s["name"]) + "_" + p
|
||||||
b, err := paccess.LaunchPeerExecution(k, v, tools.WORKSPACE, tools.POST, s, a.GetCaller())
|
b, err := paccess.LaunchPeerExecution(p, v, tools.WORKSPACE, tools.POST, s, wfa.Caller)
|
||||||
if err != nil && b == nil {
|
if err != nil && b == nil {
|
||||||
a.Logger.Error().Msg("Could not send to peer " + k + ". Error: " + err.Error())
|
wfa.Logger.Error().Msg("Could not send to peer " + p + ". Error: " + err.Error())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -192,13 +86,14 @@ func (a *collaborativeAreaMongoAccessor) sharedWorkspace(shared *CollaborativeAr
|
|||||||
}
|
}
|
||||||
|
|
||||||
// sharedWorkflow is a function that shares the shared workflow to the peers
|
// sharedWorkflow is a function that shares the shared workflow to the peers
|
||||||
func (a *collaborativeAreaMongoAccessor) sharedWorkflow(shared *CollaborativeArea, id string) {
|
func (wfa *collaborativeAreaMongoAccessor) sharedWorkflow(shared *CollaborativeArea, id string) {
|
||||||
eldest, code, _ := a.LoadOne(id) // get the eldest
|
accessor := (&w.Workflow{}).GetAccessor(nil)
|
||||||
|
eldest, code, _ := wfa.LoadOne(id) // get the eldest
|
||||||
if code == 200 {
|
if code == 200 {
|
||||||
eld := eldest.(*CollaborativeArea)
|
eld := eldest.(*CollaborativeArea)
|
||||||
if eld.Workflows != nil {
|
if eld.Workflows != nil {
|
||||||
for _, v := range eld.Workflows {
|
for _, v := range eld.Workflows {
|
||||||
data, code, _ := a.workflowAccessor.LoadOne(v)
|
data, code, _ := accessor.LoadOne(v)
|
||||||
if code == 200 {
|
if code == 200 {
|
||||||
s := data.(*w.Workflow)
|
s := data.(*w.Workflow)
|
||||||
new := []string{}
|
new := []string{}
|
||||||
@ -209,15 +104,15 @@ func (a *collaborativeAreaMongoAccessor) sharedWorkflow(shared *CollaborativeAre
|
|||||||
} // kick the shared reference in your old shared workflow
|
} // kick the shared reference in your old shared workflow
|
||||||
n := &w.Workflow{}
|
n := &w.Workflow{}
|
||||||
n.Shared = new
|
n.Shared = new
|
||||||
a.workflowAccessor.UpdateOne(n, v)
|
accessor.UpdateOne(n, v)
|
||||||
if a.GetCaller() != nil || a.GetCaller().URLS == nil || a.GetCaller().URLS[tools.WORKFLOW] == nil {
|
if wfa.Caller != nil || wfa.Caller.URLS == nil || wfa.Caller.URLS[tools.WORKFLOW] == nil {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
paccess := (&peer.Peer{}) // send to all peers
|
paccess := (&peer.Peer{}) // send to all peers
|
||||||
for k := range shared.AllowedPeersGroup { // delete the shared workflow on the peer
|
for _, p := range shared.Peers { // delete the shared workflow on the peer
|
||||||
b, err := paccess.LaunchPeerExecution(k, v, tools.WORKFLOW, tools.DELETE, nil, a.GetCaller())
|
b, err := paccess.LaunchPeerExecution(p, v, tools.WORKFLOW, tools.DELETE, nil, wfa.Caller)
|
||||||
if err != nil && b == nil {
|
if err != nil && b == nil {
|
||||||
a.Logger.Error().Msg("Could not send to peer " + k + ". Error: " + err.Error())
|
wfa.Logger.Error().Msg("Could not send to peer " + p + ". Error: " + err.Error())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -226,23 +121,23 @@ func (a *collaborativeAreaMongoAccessor) sharedWorkflow(shared *CollaborativeAre
|
|||||||
}
|
}
|
||||||
if shared.Workflows != nil { // update all the shared workflows
|
if shared.Workflows != nil { // update all the shared workflows
|
||||||
for _, v := range shared.Workflows {
|
for _, v := range shared.Workflows {
|
||||||
data, code, _ := a.workflowAccessor.LoadOne(v)
|
data, code, _ := accessor.LoadOne(v)
|
||||||
if code == 200 {
|
if code == 200 {
|
||||||
s := data.(*w.Workflow)
|
s := data.(*w.Workflow)
|
||||||
if !slices.Contains(s.Shared, id) {
|
if !slices.Contains(s.Shared, id) {
|
||||||
s.Shared = append(s.Shared, id)
|
s.Shared = append(s.Shared, id)
|
||||||
workflow, code, _ := a.workflowAccessor.UpdateOne(s, v)
|
workflow, code, _ := accessor.UpdateOne(s, v)
|
||||||
if a.GetCaller() != nil || a.GetCaller().URLS == nil || a.GetCaller().URLS[tools.WORKFLOW] == nil {
|
if wfa.Caller != nil || wfa.Caller.URLS == nil || wfa.Caller.URLS[tools.WORKFLOW] == nil {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
paccess := (&peer.Peer{})
|
paccess := (&peer.Peer{})
|
||||||
for k := range shared.AllowedPeersGroup { // send to all peers
|
for _, p := range shared.Peers { // send to all peers
|
||||||
if code == 200 {
|
if code == 200 {
|
||||||
s := workflow.Serialize(workflow) // add the shared workflow on the peer
|
s := workflow.Serialize() // add the shared workflow on the peer
|
||||||
s["name"] = fmt.Sprintf("%v", s["name"]) + "_" + k
|
s["name"] = fmt.Sprintf("%v", s["name"]) + "_" + p
|
||||||
b, err := paccess.LaunchPeerExecution(k, shared.UUID, tools.WORKFLOW, tools.POST, s, a.GetCaller())
|
b, err := paccess.LaunchPeerExecution(p, shared.UUID, tools.WORKFLOW, tools.POST, s, wfa.Caller)
|
||||||
if err != nil && b == nil {
|
if err != nil && b == nil {
|
||||||
a.Logger.Error().Msg("Could not send to peer " + k + ". Error: " + err.Error())
|
wfa.Logger.Error().Msg("Could not send to peer " + p + ". Error: " + err.Error())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -255,29 +150,194 @@ func (a *collaborativeAreaMongoAccessor) sharedWorkflow(shared *CollaborativeAre
|
|||||||
}
|
}
|
||||||
|
|
||||||
// sharedWorkspace is a function that shares the collaborative area to the peers
|
// sharedWorkspace is a function that shares the collaborative area to the peers
|
||||||
func (a *collaborativeAreaMongoAccessor) deleteToPeer(shared *CollaborativeArea) {
|
func (wfa *collaborativeAreaMongoAccessor) deleteToPeer(shared *CollaborativeArea) {
|
||||||
a.contactPeer(shared, tools.POST)
|
if wfa.Caller == nil || wfa.Caller.URLS == nil || wfa.Caller.URLS[tools.COLLABORATIVE_AREA] == nil || wfa.Caller.Disabled {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
paccess := (&peer.Peer{})
|
||||||
|
for _, v := range shared.Peers {
|
||||||
|
if ok, _ := (&peer.Peer{AbstractObject: utils.AbstractObject{UUID: v}}).IsMySelf(); ok {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
b, err := paccess.LaunchPeerExecution(v, shared.UUID, tools.COLLABORATIVE_AREA, tools.DELETE, nil, wfa.Caller)
|
||||||
|
if err != nil && b == nil {
|
||||||
|
wfa.Logger.Error().Msg("Could not send to peer " + v + ". Error: " + err.Error())
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// sharedWorkspace is a function that shares the collaborative area to the peers
|
// sharedWorkspace is a function that shares the collaborative area to the peers
|
||||||
func (a *collaborativeAreaMongoAccessor) sendToPeer(shared *CollaborativeArea) {
|
func (wfa *collaborativeAreaMongoAccessor) sendToPeer(shared *CollaborativeArea) {
|
||||||
a.contactPeer(shared, tools.POST)
|
if wfa.Caller == nil || wfa.Caller.URLS == nil || wfa.Caller.URLS[tools.COLLABORATIVE_AREA] == nil || wfa.Caller.Disabled {
|
||||||
}
|
|
||||||
|
|
||||||
func (a *collaborativeAreaMongoAccessor) contactPeer(shared *CollaborativeArea, meth tools.METHOD) {
|
|
||||||
if a.GetCaller() == nil || a.GetCaller().URLS == nil || a.GetCaller().URLS[tools.COLLABORATIVE_AREA] == nil || a.GetCaller().Disabled {
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
paccess := (&peer.Peer{})
|
paccess := (&peer.Peer{})
|
||||||
for k := range shared.AllowedPeersGroup {
|
for _, v := range shared.Peers {
|
||||||
if ok, _ := (&peer.Peer{AbstractObject: utils.AbstractObject{UUID: k}}).IsMySelf(); ok || (shared.IsSent && meth == tools.POST) || (!shared.IsSent && meth != tools.POST) {
|
if ok, _ := (&peer.Peer{AbstractObject: utils.AbstractObject{UUID: v}}).IsMySelf(); ok || shared.IsSent {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
shared.IsSent = meth == tools.POST
|
shared.IsSent = true
|
||||||
b, err := paccess.LaunchPeerExecution(k, k, tools.COLLABORATIVE_AREA, meth, shared.Serialize(shared), a.GetCaller())
|
b, err := paccess.LaunchPeerExecution(v, v, tools.COLLABORATIVE_AREA, tools.POST, shared.Serialize(), wfa.Caller)
|
||||||
if err != nil && b == nil {
|
if err != nil && b == nil {
|
||||||
a.Logger.Error().Msg("Could not send to peer " + k + ". Error: " + err.Error())
|
wfa.Logger.Error().Msg("Could not send to peer " + v + ". Error: " + err.Error())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// UpdateOne updates a collaborative area in the database, given its ID and the new data, it automatically share to peers if the workspace is shared
|
||||||
|
func (wfa *collaborativeAreaMongoAccessor) UpdateOne(set utils.DBObject, id string) (utils.DBObject, int, error) {
|
||||||
|
res, code, err := wfa.GenericUpdateOne(set.(*CollaborativeArea), id, wfa, &CollaborativeArea{})
|
||||||
|
fmt.Println("UpdateOne", set, res, code, err)
|
||||||
|
// wfa.deleteToPeer(res.(*CollaborativeArea)) // delete the collaborative area on the peer
|
||||||
|
wfa.sharedWorkflow(res.(*CollaborativeArea), id) // replace all shared workflows
|
||||||
|
wfa.sharedWorkspace(res.(*CollaborativeArea), id) // replace all collaborative areas (not shared worspace obj but workspace one)
|
||||||
|
// wfa.sendToPeer(res.(*CollaborativeArea)) // send the collaborative area (collaborative area object) to the peers
|
||||||
|
return res, code, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// StoreOne stores a collaborative area in the database, it automatically share to peers if the workspace is shared
|
||||||
|
func (wfa *collaborativeAreaMongoAccessor) StoreOne(data utils.DBObject) (utils.DBObject, int, error) {
|
||||||
|
_, id := (&peer.Peer{}).IsMySelf() // get the local peer
|
||||||
|
data.(*CollaborativeArea).CreatorID = id // set the creator id
|
||||||
|
data.(*CollaborativeArea).Peers = append(data.(*CollaborativeArea).Peers, id) // add the creator id to the peers
|
||||||
|
// then reset the shared fields
|
||||||
|
if data.(*CollaborativeArea).Workspaces == nil {
|
||||||
|
data.(*CollaborativeArea).Workspaces = []string{}
|
||||||
|
}
|
||||||
|
if data.(*CollaborativeArea).Workflows == nil {
|
||||||
|
data.(*CollaborativeArea).Workflows = []string{}
|
||||||
|
}
|
||||||
|
if data.(*CollaborativeArea).Rules == nil {
|
||||||
|
data.(*CollaborativeArea).Rules = []string{}
|
||||||
|
}
|
||||||
|
if data.(*CollaborativeArea).CollaborativeAreaRule == nil {
|
||||||
|
data.(*CollaborativeArea).CollaborativeAreaRule = &CollaborativeAreaRule{
|
||||||
|
ShareMode: "private",
|
||||||
|
ExploitedBy: "collaborators only",
|
||||||
|
}
|
||||||
|
}
|
||||||
|
data.(*CollaborativeArea).CollaborativeAreaRule.CreatedAt = time.Now().UTC()
|
||||||
|
// retrieve or proper peer
|
||||||
|
dd, code, err := (&peer.Peer{}).GetAccessor(nil).Search(nil, "0")
|
||||||
|
if code != 200 || len(dd) == 0 {
|
||||||
|
return nil, code, errors.New("Could not retrieve the peer" + err.Error())
|
||||||
|
}
|
||||||
|
data.(*CollaborativeArea).CollaborativeAreaRule.Creator = dd[0].GetID()
|
||||||
|
d, code, err := wfa.GenericStoreOne(data.(*CollaborativeArea), wfa)
|
||||||
|
if code == 200 {
|
||||||
|
wfa.sharedWorkflow(d.(*CollaborativeArea), d.GetID()) // create all shared workflows
|
||||||
|
wfa.sharedWorkspace(d.(*CollaborativeArea), d.GetID()) // create all collaborative areas
|
||||||
|
wfa.sendToPeer(d.(*CollaborativeArea)) // send the collaborative area (collaborative area object) to the peers
|
||||||
|
}
|
||||||
|
return data, code, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// CopyOne copies a CollaborativeArea in the database
|
||||||
|
func (wfa *collaborativeAreaMongoAccessor) CopyOne(data utils.DBObject) (utils.DBObject, int, error) {
|
||||||
|
return wfa.StoreOne(data)
|
||||||
|
}
|
||||||
|
|
||||||
|
// enrich is a function that enriches the CollaborativeArea with the shared objects
|
||||||
|
func (wfa *collaborativeAreaMongoAccessor) enrich(sharedWorkspace *CollaborativeArea) *CollaborativeArea {
|
||||||
|
access := (&workspace.Workspace{}).GetAccessor(nil)
|
||||||
|
res, code, _ := access.Search(&dbs.Filters{
|
||||||
|
Or: map[string][]dbs.Filter{
|
||||||
|
"abstractobject.id": {{Operator: dbs.IN.String(), Value: sharedWorkspace.Workspaces}},
|
||||||
|
},
|
||||||
|
}, "")
|
||||||
|
if code == 200 {
|
||||||
|
for _, r := range res {
|
||||||
|
sharedWorkspace.SharedWorkspaces = append(sharedWorkspace.SharedWorkspaces, r.(*workspace.Workspace))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
access = (&w.Workflow{}).GetAccessor(nil)
|
||||||
|
res, code, _ = access.Search(&dbs.Filters{
|
||||||
|
Or: map[string][]dbs.Filter{
|
||||||
|
"abstractobject.id": {{Operator: dbs.IN.String(), Value: sharedWorkspace.Workflows}},
|
||||||
|
},
|
||||||
|
}, "")
|
||||||
|
if code == 200 {
|
||||||
|
for _, r := range res {
|
||||||
|
sharedWorkspace.SharedWorkflows = append(sharedWorkspace.SharedWorkflows, r.(*w.Workflow))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
access = (&peer.Peer{}).GetAccessor(nil)
|
||||||
|
res, code, _ = access.Search(&dbs.Filters{
|
||||||
|
Or: map[string][]dbs.Filter{
|
||||||
|
"abstractobject.id": {{Operator: dbs.IN.String(), Value: sharedWorkspace.Peers}},
|
||||||
|
},
|
||||||
|
}, "")
|
||||||
|
if code == 200 {
|
||||||
|
for _, r := range res {
|
||||||
|
sharedWorkspace.SharedPeers = append(sharedWorkspace.SharedPeers, r.(*peer.Peer))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
access = (&rule.Rule{}).GetAccessor(nil)
|
||||||
|
res, code, _ = access.Search(&dbs.Filters{
|
||||||
|
Or: map[string][]dbs.Filter{
|
||||||
|
"abstractobject.id": {{Operator: dbs.IN.String(), Value: sharedWorkspace.Rules}},
|
||||||
|
},
|
||||||
|
}, "")
|
||||||
|
if code == 200 {
|
||||||
|
for _, r := range res {
|
||||||
|
sharedWorkspace.SharedRules = append(sharedWorkspace.SharedRules, r.(*rule.Rule))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return sharedWorkspace
|
||||||
|
}
|
||||||
|
|
||||||
|
// LoadOne loads a collaborative area from the database, given its ID and enrich it
|
||||||
|
func (wfa *collaborativeAreaMongoAccessor) LoadOne(id string) (utils.DBObject, int, error) {
|
||||||
|
var sharedWorkspace CollaborativeArea
|
||||||
|
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(&sharedWorkspace)
|
||||||
|
return wfa.enrich(&sharedWorkspace), 200, nil // enrich the collaborative area
|
||||||
|
}
|
||||||
|
|
||||||
|
// LoadAll loads all the collaborative areas from the database and enrich them
|
||||||
|
func (wfa collaborativeAreaMongoAccessor) 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 []CollaborativeArea
|
||||||
|
if err = res_mongo.All(mongo.MngoCtx, &results); err != nil {
|
||||||
|
return nil, 404, err
|
||||||
|
}
|
||||||
|
for _, r := range results {
|
||||||
|
objs = append(objs, wfa.enrich(&r)) // enrich the collaborative area
|
||||||
|
}
|
||||||
|
return objs, 200, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Search searches for collaborative areas in the database, given some filters OR a search string
|
||||||
|
func (wfa *collaborativeAreaMongoAccessor) 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{ // search by name only by default can be override
|
||||||
|
"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 []CollaborativeArea
|
||||||
|
if err = res_mongo.All(mongo.MngoCtx, &results); err != nil {
|
||||||
|
return nil, 404, err
|
||||||
|
}
|
||||||
|
for _, r := range results {
|
||||||
|
objs = append(objs, wfa.enrich(&r)) // enrich the collaborative area
|
||||||
|
}
|
||||||
|
return objs, 200, nil
|
||||||
|
}
|
||||||
|
@ -1,6 +1,8 @@
|
|||||||
package rule
|
package rule
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
"cloud.o-forge.io/core/oc-lib/models/utils"
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
"cloud.o-forge.io/core/oc-lib/tools"
|
||||||
"github.com/google/uuid"
|
"github.com/google/uuid"
|
||||||
@ -16,14 +18,39 @@ type Rule struct {
|
|||||||
Actions []string `json:"actions,omitempty" bson:"actions,omitempty"` // NOT DEFINITIVE TO SPECIFICATION
|
Actions []string `json:"actions,omitempty" bson:"actions,omitempty"` // NOT DEFINITIVE TO SPECIFICATION
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (ao *Rule) GetID() string {
|
||||||
|
return ao.UUID
|
||||||
|
}
|
||||||
|
|
||||||
func (r *Rule) GenerateID() {
|
func (r *Rule) GenerateID() {
|
||||||
r.UUID = uuid.New().String()
|
r.UUID = uuid.New().String()
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *Rule) GetAccessor(request *tools.APIRequest) utils.Accessor {
|
func (d *Rule) GetName() string {
|
||||||
return NewAccessor(request)
|
return d.Name
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *Rule) VerifyAuth(request *tools.APIRequest) bool {
|
func (d *Rule) GetAccessor(caller *tools.HTTPCaller) utils.Accessor {
|
||||||
return true
|
data := New()
|
||||||
|
data.Init(tools.RULE, caller)
|
||||||
|
return data
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dma *Rule) Deserialize(j map[string]interface{}) utils.DBObject {
|
||||||
|
b, err := json.Marshal(j)
|
||||||
|
if err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
json.Unmarshal(b, dma)
|
||||||
|
return dma
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dma *Rule) Serialize() map[string]interface{} {
|
||||||
|
var m map[string]interface{}
|
||||||
|
b, err := json.Marshal(dma)
|
||||||
|
if err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
json.Unmarshal(b, &m)
|
||||||
|
return m
|
||||||
}
|
}
|
||||||
|
@ -2,9 +2,8 @@ package rule
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"cloud.o-forge.io/core/oc-lib/dbs"
|
"cloud.o-forge.io/core/oc-lib/dbs"
|
||||||
"cloud.o-forge.io/core/oc-lib/logs"
|
"cloud.o-forge.io/core/oc-lib/dbs/mongo"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
"cloud.o-forge.io/core/oc-lib/models/utils"
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
type ruleMongoAccessor struct {
|
type ruleMongoAccessor struct {
|
||||||
@ -12,51 +11,80 @@ type ruleMongoAccessor struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// New creates a new instance of the ruleMongoAccessor
|
// New creates a new instance of the ruleMongoAccessor
|
||||||
func NewAccessor(request *tools.APIRequest) *ruleMongoAccessor {
|
func New() *ruleMongoAccessor {
|
||||||
return &ruleMongoAccessor{
|
return &ruleMongoAccessor{}
|
||||||
AbstractAccessor: utils.AbstractAccessor{
|
}
|
||||||
Logger: logs.CreateLogger(tools.RULE.String()), // Create a logger with the data type
|
|
||||||
Request: request,
|
// GetType returns the type of the rule
|
||||||
Type: tools.RULE,
|
func (wfa *ruleMongoAccessor) DeleteOne(id string) (utils.DBObject, int, error) {
|
||||||
},
|
return wfa.GenericDeleteOne(id, wfa)
|
||||||
|
}
|
||||||
|
|
||||||
|
// UpdateOne updates a rule in the database
|
||||||
|
func (wfa *ruleMongoAccessor) UpdateOne(set utils.DBObject, id string) (utils.DBObject, int, error) {
|
||||||
|
return wfa.GenericUpdateOne(set.(*Rule), id, wfa, &Rule{})
|
||||||
|
}
|
||||||
|
|
||||||
|
// StoreOne stores a rule in the database
|
||||||
|
func (wfa *ruleMongoAccessor) StoreOne(data utils.DBObject) (utils.DBObject, int, error) {
|
||||||
|
return wfa.GenericStoreOne(data.(*Rule), wfa)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (wfa *ruleMongoAccessor) CopyOne(data utils.DBObject) (utils.DBObject, int, error) {
|
||||||
|
return wfa.GenericStoreOne(data, wfa)
|
||||||
|
}
|
||||||
|
|
||||||
|
// LoadOne loads a rule from the database
|
||||||
|
func (wfa *ruleMongoAccessor) LoadOne(id string) (utils.DBObject, int, error) {
|
||||||
|
var rule Rule
|
||||||
|
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(&rule)
|
||||||
|
return &rule, 200, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
/*
|
// LoadAll loads all rules from the database
|
||||||
* Nothing special here, just the basic CRUD operations
|
func (wfa ruleMongoAccessor) LoadAll() ([]utils.ShallowDBObject, int, error) {
|
||||||
*/
|
objs := []utils.ShallowDBObject{}
|
||||||
func (a *ruleMongoAccessor) DeleteOne(id string) (utils.DBObject, int, error) {
|
res_mongo, code, err := mongo.MONGOService.LoadAll(wfa.GetType())
|
||||||
return utils.GenericDeleteOne(id, a)
|
if err != nil {
|
||||||
}
|
wfa.Logger.Error().Msg("Could not retrieve any from db. Error: " + err.Error())
|
||||||
|
return nil, code, err
|
||||||
func (a *ruleMongoAccessor) UpdateOne(set utils.DBObject, id string) (utils.DBObject, int, error) {
|
|
||||||
return utils.GenericUpdateOne(set, id, a, &Rule{})
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *ruleMongoAccessor) StoreOne(data utils.DBObject) (utils.DBObject, int, error) {
|
|
||||||
return utils.GenericStoreOne(data, a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *ruleMongoAccessor) CopyOne(data utils.DBObject) (utils.DBObject, int, error) {
|
|
||||||
return utils.GenericStoreOne(data, a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *ruleMongoAccessor) LoadOne(id string) (utils.DBObject, int, error) {
|
|
||||||
return utils.GenericLoadOne[*Rule](id, func(d utils.DBObject) (utils.DBObject, int, error) {
|
|
||||||
return d, 200, nil
|
|
||||||
}, a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *ruleMongoAccessor) LoadAll(isDraft bool) ([]utils.ShallowDBObject, int, error) {
|
|
||||||
return utils.GenericLoadAll[*Rule](a.getExec(), isDraft, a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *ruleMongoAccessor) Search(filters *dbs.Filters, search string, isDraft bool) ([]utils.ShallowDBObject, int, error) {
|
|
||||||
return utils.GenericSearch[*Rule](filters, search, (&Rule{}).GetObjectFilters(search), a.getExec(), isDraft, a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *ruleMongoAccessor) getExec() func(utils.DBObject) utils.ShallowDBObject {
|
|
||||||
return func(d utils.DBObject) utils.ShallowDBObject {
|
|
||||||
return d
|
|
||||||
}
|
}
|
||||||
|
var results []Rule
|
||||||
|
if err = res_mongo.All(mongo.MngoCtx, &results); err != nil {
|
||||||
|
return nil, 404, err
|
||||||
|
}
|
||||||
|
for _, r := range results {
|
||||||
|
objs = append(objs, &r)
|
||||||
|
}
|
||||||
|
return objs, 200, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Search searches for rules in the database, given some filters OR a search string
|
||||||
|
func (wfa *ruleMongoAccessor) 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 []Rule
|
||||||
|
if err = res_mongo.All(mongo.MngoCtx, &results); err != nil {
|
||||||
|
return nil, 404, err
|
||||||
|
}
|
||||||
|
for _, r := range results {
|
||||||
|
objs = append(objs, &r)
|
||||||
|
}
|
||||||
|
return objs, 200, nil
|
||||||
}
|
}
|
||||||
|
@ -1,13 +1,17 @@
|
|||||||
package shallow_collaborative_area
|
package shallow_collaborative_area
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
"cloud.o-forge.io/core/oc-lib/models/utils"
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
"cloud.o-forge.io/core/oc-lib/tools"
|
||||||
|
"github.com/google/uuid"
|
||||||
)
|
)
|
||||||
|
|
||||||
type ShallowCollaborativeArea struct {
|
type ShallowCollaborativeArea struct {
|
||||||
utils.AbstractObject
|
utils.AbstractObject
|
||||||
IsSent bool `json:"is_sent" bson:"-"`
|
IsSent bool `json:"is_sent" bson:"-"`
|
||||||
|
CreatorID string `json:"peer_id,omitempty" bson:"peer_id,omitempty" validate:"required"`
|
||||||
Version string `json:"version,omitempty" bson:"version,omitempty"`
|
Version string `json:"version,omitempty" bson:"version,omitempty"`
|
||||||
Description string `json:"description,omitempty" bson:"description,omitempty" validate:"required"`
|
Description string `json:"description,omitempty" bson:"description,omitempty" validate:"required"`
|
||||||
Attributes map[string]interface{} `json:"attributes,omitempty" bson:"attributes,omitempty"`
|
Attributes map[string]interface{} `json:"attributes,omitempty" bson:"attributes,omitempty"`
|
||||||
@ -17,6 +21,41 @@ type ShallowCollaborativeArea struct {
|
|||||||
Rules []string `json:"rules,omitempty" bson:"rules,omitempty"`
|
Rules []string `json:"rules,omitempty" bson:"rules,omitempty"`
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *ShallowCollaborativeArea) GetAccessor(request *tools.APIRequest) utils.Accessor {
|
func (ao *ShallowCollaborativeArea) GetID() string {
|
||||||
return NewAccessor(request)
|
return ao.UUID
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *ShallowCollaborativeArea) GenerateID() {
|
||||||
|
if r.UUID == "" {
|
||||||
|
r.UUID = uuid.New().String()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *ShallowCollaborativeArea) GetName() string {
|
||||||
|
return d.Name
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *ShallowCollaborativeArea) GetAccessor(caller *tools.HTTPCaller) utils.Accessor {
|
||||||
|
data := New()
|
||||||
|
data.Init(tools.COLLABORATIVE_AREA, caller)
|
||||||
|
return data
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dma *ShallowCollaborativeArea) Deserialize(j map[string]interface{}) utils.DBObject {
|
||||||
|
b, err := json.Marshal(j)
|
||||||
|
if err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
json.Unmarshal(b, dma)
|
||||||
|
return dma
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dma *ShallowCollaborativeArea) Serialize() map[string]interface{} {
|
||||||
|
var m map[string]interface{}
|
||||||
|
b, err := json.Marshal(dma)
|
||||||
|
if err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
json.Unmarshal(b, &m)
|
||||||
|
return m
|
||||||
}
|
}
|
||||||
|
@ -2,55 +2,82 @@ package shallow_collaborative_area
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"cloud.o-forge.io/core/oc-lib/dbs"
|
"cloud.o-forge.io/core/oc-lib/dbs"
|
||||||
"cloud.o-forge.io/core/oc-lib/logs"
|
"cloud.o-forge.io/core/oc-lib/dbs/mongo"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
"cloud.o-forge.io/core/oc-lib/models/utils"
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
type shallowSharedWorkspaceMongoAccessor struct {
|
type shallowSharedWorkspaceMongoAccessor struct {
|
||||||
utils.AbstractAccessor
|
utils.AbstractAccessor
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewAccessor(request *tools.APIRequest) *shallowSharedWorkspaceMongoAccessor {
|
func New() *shallowSharedWorkspaceMongoAccessor {
|
||||||
return &shallowSharedWorkspaceMongoAccessor{
|
return &shallowSharedWorkspaceMongoAccessor{}
|
||||||
AbstractAccessor: utils.AbstractAccessor{
|
}
|
||||||
Logger: logs.CreateLogger(tools.COLLABORATIVE_AREA.String()), // Create a logger with the data type
|
|
||||||
Request: request, // Set the caller
|
func (wfa *shallowSharedWorkspaceMongoAccessor) DeleteOne(id string) (utils.DBObject, int, error) {
|
||||||
Type: tools.COLLABORATIVE_AREA,
|
return wfa.GenericDeleteOne(id, wfa)
|
||||||
},
|
}
|
||||||
|
|
||||||
|
func (wfa *shallowSharedWorkspaceMongoAccessor) UpdateOne(set utils.DBObject, id string) (utils.DBObject, int, error) {
|
||||||
|
return wfa.GenericUpdateOne(set.(*ShallowCollaborativeArea), id, wfa, &ShallowCollaborativeArea{})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (wfa *shallowSharedWorkspaceMongoAccessor) StoreOne(data utils.DBObject) (utils.DBObject, int, error) {
|
||||||
|
return wfa.GenericStoreOne(data.(*ShallowCollaborativeArea), wfa)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (wfa *shallowSharedWorkspaceMongoAccessor) CopyOne(data utils.DBObject) (utils.DBObject, int, error) {
|
||||||
|
return wfa.StoreOne(data)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (wfa *shallowSharedWorkspaceMongoAccessor) LoadOne(id string) (utils.DBObject, int, error) {
|
||||||
|
var sharedWorkspace ShallowCollaborativeArea
|
||||||
|
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(&sharedWorkspace)
|
||||||
|
return &sharedWorkspace, 200, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a *shallowSharedWorkspaceMongoAccessor) DeleteOne(id string) (utils.DBObject, int, error) {
|
func (wfa shallowSharedWorkspaceMongoAccessor) LoadAll() ([]utils.ShallowDBObject, int, error) {
|
||||||
return utils.GenericDeleteOne(id, a)
|
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 []ShallowCollaborativeArea
|
||||||
|
if err = res_mongo.All(mongo.MngoCtx, &results); err != nil {
|
||||||
|
return nil, 404, err
|
||||||
|
}
|
||||||
|
for _, r := range results {
|
||||||
|
objs = append(objs, &r)
|
||||||
|
}
|
||||||
|
return objs, 200, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a *shallowSharedWorkspaceMongoAccessor) UpdateOne(set utils.DBObject, id string) (utils.DBObject, int, error) {
|
func (wfa *shallowSharedWorkspaceMongoAccessor) Search(filters *dbs.Filters, search string) ([]utils.ShallowDBObject, int, error) {
|
||||||
return utils.GenericUpdateOne(set.(*ShallowCollaborativeArea), id, a, &ShallowCollaborativeArea{})
|
objs := []utils.ShallowDBObject{}
|
||||||
}
|
if (filters == nil || len(filters.And) == 0 || len(filters.Or) == 0) && search != "" {
|
||||||
|
filters = &dbs.Filters{
|
||||||
func (a *shallowSharedWorkspaceMongoAccessor) StoreOne(data utils.DBObject) (utils.DBObject, int, error) {
|
Or: map[string][]dbs.Filter{
|
||||||
return utils.GenericStoreOne(data.(*ShallowCollaborativeArea), a)
|
"abstractobject.name": {{Operator: dbs.LIKE.String(), Value: search}},
|
||||||
}
|
},
|
||||||
|
}
|
||||||
func (a *shallowSharedWorkspaceMongoAccessor) CopyOne(data utils.DBObject) (utils.DBObject, int, error) {
|
}
|
||||||
return a.StoreOne(data)
|
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())
|
||||||
func (a *shallowSharedWorkspaceMongoAccessor) LoadOne(id string) (utils.DBObject, int, error) {
|
return nil, code, err
|
||||||
return utils.GenericLoadOne[*ShallowCollaborativeArea](id, func(d utils.DBObject) (utils.DBObject, int, error) {
|
}
|
||||||
return d, 200, nil
|
var results []ShallowCollaborativeArea
|
||||||
}, a)
|
if err = res_mongo.All(mongo.MngoCtx, &results); err != nil {
|
||||||
}
|
return nil, 404, err
|
||||||
|
}
|
||||||
func (a *shallowSharedWorkspaceMongoAccessor) LoadAll(isDraft bool) ([]utils.ShallowDBObject, int, error) {
|
for _, r := range results {
|
||||||
return utils.GenericLoadAll[*ShallowCollaborativeArea](func(d utils.DBObject) utils.ShallowDBObject {
|
objs = append(objs, &r)
|
||||||
return d
|
}
|
||||||
}, isDraft, a)
|
return objs, 200, nil
|
||||||
}
|
|
||||||
|
|
||||||
func (a *shallowSharedWorkspaceMongoAccessor) Search(filters *dbs.Filters, search string, isDraft bool) ([]utils.ShallowDBObject, int, error) {
|
|
||||||
return utils.GenericSearch[*ShallowCollaborativeArea](filters, search, (&ShallowCollaborativeArea{}).GetObjectFilters(search), func(d utils.DBObject) utils.ShallowDBObject {
|
|
||||||
return d
|
|
||||||
}, isDraft, a)
|
|
||||||
}
|
}
|
||||||
|
@ -1,20 +0,0 @@
|
|||||||
package enum
|
|
||||||
|
|
||||||
type InfrastructureType int
|
|
||||||
|
|
||||||
const (
|
|
||||||
DOCKER InfrastructureType = iota
|
|
||||||
KUBERNETES
|
|
||||||
SLURM
|
|
||||||
HW
|
|
||||||
CONDOR
|
|
||||||
)
|
|
||||||
|
|
||||||
func (t InfrastructureType) String() string {
|
|
||||||
return [...]string{"DOCKER", "KUBERNETES", "SLURM", "HW", "CONDOR"}[t]
|
|
||||||
}
|
|
||||||
|
|
||||||
// get list of all infrastructure types
|
|
||||||
func InfrastructureList() []InfrastructureType {
|
|
||||||
return []InfrastructureType{DOCKER, KUBERNETES, SLURM, HW, CONDOR}
|
|
||||||
}
|
|
@ -1,56 +0,0 @@
|
|||||||
package enum
|
|
||||||
|
|
||||||
type StorageSize int
|
|
||||||
|
|
||||||
// StorageType - Enum that defines the type of storage
|
|
||||||
const (
|
|
||||||
GB StorageSize = iota
|
|
||||||
MB
|
|
||||||
KB
|
|
||||||
TB
|
|
||||||
)
|
|
||||||
|
|
||||||
var argoType = [...]string{
|
|
||||||
"Gi",
|
|
||||||
"Mi",
|
|
||||||
"Ki",
|
|
||||||
"Ti",
|
|
||||||
}
|
|
||||||
|
|
||||||
// Size to string
|
|
||||||
func (t StorageSize) String() string {
|
|
||||||
return [...]string{"GB", "MB", "KB", "TB"}[t]
|
|
||||||
}
|
|
||||||
|
|
||||||
func SizeList() []StorageSize {
|
|
||||||
return []StorageSize{GB, MB, KB, TB}
|
|
||||||
}
|
|
||||||
|
|
||||||
// New creates a new instance of the StorageResource struct
|
|
||||||
func (dma StorageSize) ToArgo() string {
|
|
||||||
return argoType[dma]
|
|
||||||
}
|
|
||||||
|
|
||||||
// enum of a data type
|
|
||||||
type StorageType int
|
|
||||||
|
|
||||||
const (
|
|
||||||
FILE = iota
|
|
||||||
STREAM
|
|
||||||
API
|
|
||||||
DATABASE
|
|
||||||
S3
|
|
||||||
MEMORY
|
|
||||||
HARDWARE
|
|
||||||
AZURE
|
|
||||||
GCS
|
|
||||||
)
|
|
||||||
|
|
||||||
// String() - Returns the string representation of the storage type
|
|
||||||
func (t StorageType) String() string {
|
|
||||||
return [...]string{"FILE", "STREAM", "API", "DATABASE", "S3", "MEMORY", "HARDWARE", "AZURE", "GCS"}[t]
|
|
||||||
}
|
|
||||||
|
|
||||||
func TypeList() []StorageType {
|
|
||||||
return []StorageType{FILE, STREAM, API, DATABASE, S3, MEMORY, HARDWARE, AZURE, GCS}
|
|
||||||
}
|
|
@ -1,64 +0,0 @@
|
|||||||
package enum
|
|
||||||
|
|
||||||
type CompletionStatus int
|
|
||||||
|
|
||||||
const (
|
|
||||||
DRAFTED CompletionStatus = iota
|
|
||||||
PENDING
|
|
||||||
CANCEL
|
|
||||||
PARTIAL
|
|
||||||
PAID
|
|
||||||
DISPUTED
|
|
||||||
OVERDUE
|
|
||||||
REFUND
|
|
||||||
)
|
|
||||||
|
|
||||||
func (d CompletionStatus) String() string {
|
|
||||||
return [...]string{"drafted", "pending", "cancel", "partial", "paid", "disputed", "overdue", "refund"}[d]
|
|
||||||
}
|
|
||||||
|
|
||||||
func CompletionStatusList() []CompletionStatus {
|
|
||||||
return []CompletionStatus{DRAFTED, PENDING, CANCEL, PARTIAL, PAID, DISPUTED, OVERDUE, REFUND}
|
|
||||||
}
|
|
||||||
|
|
||||||
type BookingStatus int
|
|
||||||
|
|
||||||
const (
|
|
||||||
DRAFT BookingStatus = iota
|
|
||||||
SCHEDULED
|
|
||||||
STARTED
|
|
||||||
FAILURE
|
|
||||||
SUCCESS
|
|
||||||
FORGOTTEN
|
|
||||||
DELAYED
|
|
||||||
CANCELLED
|
|
||||||
)
|
|
||||||
|
|
||||||
var str = [...]string{
|
|
||||||
"draft",
|
|
||||||
"scheduled",
|
|
||||||
"started",
|
|
||||||
"failure",
|
|
||||||
"success",
|
|
||||||
"forgotten",
|
|
||||||
"delayed",
|
|
||||||
"cancelled",
|
|
||||||
}
|
|
||||||
|
|
||||||
func FromInt(i int) string {
|
|
||||||
return str[i]
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d BookingStatus) String() string {
|
|
||||||
return str[d]
|
|
||||||
}
|
|
||||||
|
|
||||||
// EnumIndex - Creating common behavior - give the type a EnumIndex functio
|
|
||||||
func (d BookingStatus) EnumIndex() int {
|
|
||||||
return int(d)
|
|
||||||
}
|
|
||||||
|
|
||||||
// List
|
|
||||||
func StatusList() []BookingStatus {
|
|
||||||
return []BookingStatus{DRAFT, SCHEDULED, STARTED, FAILURE, SUCCESS, FORGOTTEN, DELAYED, CANCELLED}
|
|
||||||
}
|
|
@ -1,17 +0,0 @@
|
|||||||
package models
|
|
||||||
|
|
||||||
type Container struct {
|
|
||||||
Image string `json:"image,omitempty" bson:"image,omitempty"` // Image is the container image TEMPO
|
|
||||||
Command string `json:"command,omitempty" bson:"command,omitempty"` // Command is the container command
|
|
||||||
Args string `json:"args,omitempty" bson:"args,omitempty"` // Args is the container arguments
|
|
||||||
Env map[string]string `json:"env,omitempty" bson:"env,omitempty"` // Env is the container environment variables
|
|
||||||
Volumes map[string]string `json:"volumes,omitempty" bson:"volumes,omitempty"` // Volumes is the container volumes
|
|
||||||
|
|
||||||
Exposes []Expose `bson:"exposes,omitempty" json:"exposes,omitempty"` // Expose is the execution
|
|
||||||
}
|
|
||||||
|
|
||||||
type Expose struct {
|
|
||||||
Port int `json:"port,omitempty" bson:"port,omitempty"` // Port is the port
|
|
||||||
Reverse string `json:"reverse,omitempty" bson:"reverse,omitempty"` // Reverse is the reverse
|
|
||||||
PAT int `json:"pat,omitempty" bson:"pat,omitempty"` // PAT is the PAT
|
|
||||||
}
|
|
@ -1,20 +0,0 @@
|
|||||||
package models
|
|
||||||
|
|
||||||
// CPU is a struct that represents a CPU
|
|
||||||
type CPU struct {
|
|
||||||
Model string `bson:"model,omitempty" json:"model,omitempty"`
|
|
||||||
FrequencyGhz float64 `bson:"frequency,omitempty" json:"frequency,omitempty"`
|
|
||||||
Cores int `bson:"cores,omitempty" json:"cores,omitempty"`
|
|
||||||
Architecture string `bson:"architecture,omitempty" json:"architecture,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type RAM struct {
|
|
||||||
SizeGb float64 `bson:"size,omitempty" json:"size,omitempty" description:"Units in MB"`
|
|
||||||
Ecc bool `bson:"ecc" json:"ecc" default:"true"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type GPU struct {
|
|
||||||
Model string `bson:"model,omitempty" json:"model,omitempty"`
|
|
||||||
MemoryGb float64 `bson:"memory,omitempty" json:"memory,omitempty" description:"Units in MB"`
|
|
||||||
Cores map[string]int `bson:"cores,omitempty" json:"cores,omitempty"`
|
|
||||||
}
|
|
@ -1,21 +0,0 @@
|
|||||||
package models
|
|
||||||
|
|
||||||
type Artifact struct {
|
|
||||||
AttrPath string `json:"attr_path,omitempty" bson:"attr_path,omitempty" validate:"required"`
|
|
||||||
AttrFrom string `json:"from_path,omitempty" bson:"from_path,omitempty"`
|
|
||||||
Readonly bool `json:"readonly" bson:"readonly" default:"true"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type Param struct {
|
|
||||||
Name string `json:"name" bson:"name" validate:"required"`
|
|
||||||
Attr string `json:"attr,omitempty" bson:"attr,omitempty"`
|
|
||||||
Value string `json:"value,omitempty" bson:"value,omitempty"`
|
|
||||||
Origin string `json:"origin,omitempty" bson:"origin,omitempty"`
|
|
||||||
Readonly bool `json:"readonly" bson:"readonly" default:"true"`
|
|
||||||
Optionnal bool `json:"optionnal" bson:"optionnal" default:"true"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type InOutputs struct {
|
|
||||||
Params []Param `json:"parameters" bson:"parameters"`
|
|
||||||
Artifacts []Artifact `json:"artifacts" bson:"artifacts"`
|
|
||||||
}
|
|
@ -1,17 +0,0 @@
|
|||||||
package models
|
|
||||||
|
|
||||||
type MetricsSnapshot struct {
|
|
||||||
From string `json:"origin"`
|
|
||||||
Metrics []Metric `json:"metrics"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type Metric struct {
|
|
||||||
Name string `json:"name"`
|
|
||||||
Value float64 `json:"value"`
|
|
||||||
Error error `json:"error"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type MetricResume struct {
|
|
||||||
Delta float64 `json:"delta"`
|
|
||||||
LastValue float64 `json:"last_value"`
|
|
||||||
}
|
|
@ -1,42 +0,0 @@
|
|||||||
package common
|
|
||||||
|
|
||||||
import (
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/pricing"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
)
|
|
||||||
|
|
||||||
func GetPlannerNearestStart(start time.Time, planned map[tools.DataType]map[string]pricing.PricedItemITF, request *tools.APIRequest) float64 {
|
|
||||||
near := float64(10000000000) // set a high value
|
|
||||||
for _, items := range planned { // loop through the planned items
|
|
||||||
for _, priced := range items { // loop through the priced items
|
|
||||||
if priced.GetLocationStart() == nil { // if the start is nil,
|
|
||||||
continue // skip the iteration
|
|
||||||
}
|
|
||||||
newS := priced.GetLocationStart() // get the start
|
|
||||||
if newS.Sub(start).Seconds() < near { // if the difference between the start and the new start is less than the nearest start
|
|
||||||
near = newS.Sub(start).Seconds()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return near
|
|
||||||
}
|
|
||||||
|
|
||||||
func GetPlannerLongestTime(end *time.Time, planned map[tools.DataType]map[string]pricing.PricedItemITF, request *tools.APIRequest) float64 {
|
|
||||||
if end == nil {
|
|
||||||
return -1
|
|
||||||
}
|
|
||||||
longestTime := float64(0)
|
|
||||||
for _, priced := range planned[tools.PROCESSING_RESOURCE] {
|
|
||||||
if priced.GetLocationEnd() == nil {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
newS := priced.GetLocationEnd()
|
|
||||||
if end == nil && longestTime < newS.Sub(*end).Seconds() {
|
|
||||||
longestTime = newS.Sub(*end).Seconds()
|
|
||||||
}
|
|
||||||
// get the nearest start from start var
|
|
||||||
}
|
|
||||||
return longestTime
|
|
||||||
}
|
|
@ -1,22 +0,0 @@
|
|||||||
package pricing
|
|
||||||
|
|
||||||
import (
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
)
|
|
||||||
|
|
||||||
type PricedItemITF interface {
|
|
||||||
GetID() string
|
|
||||||
GetType() tools.DataType
|
|
||||||
IsPurchasable() bool
|
|
||||||
IsBooked() bool
|
|
||||||
GetCreatorID() string
|
|
||||||
SelectPricing() PricingProfileITF
|
|
||||||
GetLocationStart() *time.Time
|
|
||||||
SetLocationStart(start time.Time)
|
|
||||||
SetLocationEnd(end time.Time)
|
|
||||||
GetLocationEnd() *time.Time
|
|
||||||
GetExplicitDurationInS() float64
|
|
||||||
GetPrice() (float64, error)
|
|
||||||
}
|
|
@ -1,66 +0,0 @@
|
|||||||
package pricing
|
|
||||||
|
|
||||||
import (
|
|
||||||
"time"
|
|
||||||
)
|
|
||||||
|
|
||||||
type PricingProfileITF interface {
|
|
||||||
IsBooked() bool
|
|
||||||
IsPurchasable() bool
|
|
||||||
GetPurchase() BuyingStrategy
|
|
||||||
GetOverrideStrategyValue() int
|
|
||||||
GetPrice(quantity float64, val float64, start time.Time, end time.Time, params ...string) (float64, error)
|
|
||||||
}
|
|
||||||
|
|
||||||
type RefundType int
|
|
||||||
|
|
||||||
const (
|
|
||||||
REFUND_DEAD_END RefundType = iota
|
|
||||||
REFUND_ON_ERROR
|
|
||||||
REFUND_ON_EARLY_END
|
|
||||||
)
|
|
||||||
|
|
||||||
func (t RefundType) String() string {
|
|
||||||
return [...]string{"REFUND ON DEAD END", "REFUND ON ERROR", "REFUND ON EARLY END"}[t]
|
|
||||||
}
|
|
||||||
|
|
||||||
func RefundTypeList() []RefundType {
|
|
||||||
return []RefundType{REFUND_DEAD_END, REFUND_ON_ERROR, REFUND_ON_EARLY_END}
|
|
||||||
}
|
|
||||||
|
|
||||||
type AccessPricingProfile[T Strategy] struct { // only use for acces such as : DATA && PROCESSING
|
|
||||||
Pricing PricingStrategy[T] `json:"pricing,omitempty" bson:"pricing,omitempty"` // Price is the price of the resource
|
|
||||||
DefaultRefund RefundType `json:"default_refund" bson:"default_refund"` // DefaultRefund is the default refund type of the pricing
|
|
||||||
RefundRatio int32 `json:"refund_ratio" bson:"refund_ratio" default:"0"` // RefundRatio is the refund ratio if missing
|
|
||||||
}
|
|
||||||
|
|
||||||
func (b *AccessPricingProfile[T]) GetOverrideStrategyValue() int {
|
|
||||||
return -1
|
|
||||||
}
|
|
||||||
|
|
||||||
type ExploitPrivilegeStrategy int
|
|
||||||
|
|
||||||
const (
|
|
||||||
BASIC ExploitPrivilegeStrategy = iota
|
|
||||||
GARANTED_ON_DELAY
|
|
||||||
GARANTED
|
|
||||||
)
|
|
||||||
|
|
||||||
func ExploitPrivilegeStrategyList() []ExploitPrivilegeStrategy {
|
|
||||||
return []ExploitPrivilegeStrategy{BASIC, GARANTED_ON_DELAY, GARANTED}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (t ExploitPrivilegeStrategy) String() string {
|
|
||||||
return [...]string{"NO GARANTY", "GARANTED ON SPECIFIC DELAY", "GARANTED"}[t]
|
|
||||||
}
|
|
||||||
|
|
||||||
type ExploitPricingProfile[T Strategy] struct { // only use for exploit such as : STORAGE, COMPUTE, WORKFLOW
|
|
||||||
AccessPricingProfile[T]
|
|
||||||
AdditionnalRefundTypes []RefundType `json:"refund_types" bson:"refund_types"` // RefundTypes is the refund types of the pricing
|
|
||||||
|
|
||||||
PrivilegeStrategy ExploitPrivilegeStrategy `json:"privilege_strategy,omitempty" bson:"privilege_strategy,omitempty"` // Strategy is the strategy of the pricing
|
|
||||||
GarantedDelaySecond uint `json:"garanted_delay_second,omitempty" bson:"garanted_delay_second,omitempty"` // GarantedDelaySecond is the garanted delay of the pricing
|
|
||||||
|
|
||||||
Exceeding bool `json:"exceeding" bson:"exceeding"` // Exceeding is the exceeding of the bill
|
|
||||||
ExceedingRatio int32 `json:"exceeding_ratio" bson:"exceeding_ratio" default:"0"` // ExceedingRatio is the exceeding ratio of the bill
|
|
||||||
}
|
|
@ -1,181 +0,0 @@
|
|||||||
package pricing
|
|
||||||
|
|
||||||
import (
|
|
||||||
"errors"
|
|
||||||
"fmt"
|
|
||||||
"strconv"
|
|
||||||
"time"
|
|
||||||
)
|
|
||||||
|
|
||||||
type BillingStrategy int // BAM BAM
|
|
||||||
|
|
||||||
// should except... on
|
|
||||||
const (
|
|
||||||
BILL_ONCE BillingStrategy = iota // is a permanent buying ( predictible )
|
|
||||||
BILL_PER_WEEK
|
|
||||||
BILL_PER_MONTH
|
|
||||||
BILL_PER_YEAR
|
|
||||||
)
|
|
||||||
|
|
||||||
func (t BillingStrategy) IsBillingStrategyAllowed(bs int) (BillingStrategy, bool) {
|
|
||||||
switch t {
|
|
||||||
case BILL_ONCE:
|
|
||||||
return BILL_ONCE, bs == 0
|
|
||||||
case BILL_PER_WEEK:
|
|
||||||
case BILL_PER_MONTH:
|
|
||||||
case BILL_PER_YEAR:
|
|
||||||
return t, bs != 0
|
|
||||||
}
|
|
||||||
return t, false
|
|
||||||
}
|
|
||||||
|
|
||||||
func (t BillingStrategy) String() string {
|
|
||||||
return [...]string{"BILL_ONCE", "BILL_PER_WEEK", "BILL_PER_MONTH", "BILL_PER_YEAR"}[t]
|
|
||||||
}
|
|
||||||
|
|
||||||
func BillingStrategyList() []BillingStrategy {
|
|
||||||
return []BillingStrategy{BILL_ONCE, BILL_PER_WEEK, BILL_PER_MONTH, BILL_PER_YEAR}
|
|
||||||
}
|
|
||||||
|
|
||||||
type BuyingStrategy int
|
|
||||||
|
|
||||||
// should except... on
|
|
||||||
const (
|
|
||||||
PERMANENT BuyingStrategy = iota // is a permanent buying ( predictible )
|
|
||||||
UNDEFINED_SUBSCRIPTION // a endless subscription ( unpredictible )
|
|
||||||
SUBSCRIPTION // a defined subscription ( predictible )
|
|
||||||
// PAY_PER_USE // per request. ( unpredictible )
|
|
||||||
)
|
|
||||||
|
|
||||||
func (t BuyingStrategy) String() string {
|
|
||||||
return [...]string{"PERMANENT", "UNDEFINED_SUBSCRIPTION", "SUBSCRIPTION"}[t]
|
|
||||||
}
|
|
||||||
|
|
||||||
func (t BuyingStrategy) IsBillingStrategyAllowed(bs BillingStrategy) (BillingStrategy, bool) {
|
|
||||||
switch t {
|
|
||||||
case PERMANENT:
|
|
||||||
return BILL_ONCE, bs == BILL_ONCE
|
|
||||||
case UNDEFINED_SUBSCRIPTION:
|
|
||||||
return BILL_PER_MONTH, bs != BILL_ONCE
|
|
||||||
case SUBSCRIPTION:
|
|
||||||
/*case PAY_PER_USE:
|
|
||||||
return bs, true*/
|
|
||||||
}
|
|
||||||
return bs, false
|
|
||||||
}
|
|
||||||
|
|
||||||
func BuyingStrategyList() []BuyingStrategy {
|
|
||||||
return []BuyingStrategy{PERMANENT, UNDEFINED_SUBSCRIPTION, SUBSCRIPTION}
|
|
||||||
}
|
|
||||||
|
|
||||||
type Strategy interface {
|
|
||||||
GetStrategy() string
|
|
||||||
GetStrategyValue() int
|
|
||||||
}
|
|
||||||
|
|
||||||
type TimePricingStrategy int
|
|
||||||
|
|
||||||
const (
|
|
||||||
ONCE TimePricingStrategy = iota
|
|
||||||
PER_SECOND
|
|
||||||
PER_MINUTE
|
|
||||||
PER_HOUR
|
|
||||||
PER_DAY
|
|
||||||
PER_WEEK
|
|
||||||
PER_MONTH
|
|
||||||
)
|
|
||||||
|
|
||||||
func IsTimeStrategy(i int) bool {
|
|
||||||
return len(TimePricingStrategyList()) < i
|
|
||||||
}
|
|
||||||
|
|
||||||
func (t TimePricingStrategy) String() string {
|
|
||||||
return [...]string{"ONCE", "PER SECOND", "PER MINUTE", "PER HOUR", "PER DAY", "PER WEEK", "PER MONTH"}[t]
|
|
||||||
}
|
|
||||||
|
|
||||||
func TimePricingStrategyList() []TimePricingStrategy {
|
|
||||||
return []TimePricingStrategy{ONCE, PER_SECOND, PER_MINUTE, PER_HOUR, PER_DAY, PER_WEEK, PER_MONTH}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (t TimePricingStrategy) GetStrategy() string {
|
|
||||||
return [...]string{"ONCE", "PER_SECOND", "PER_MINUTE", "PER_HOUR", "PER_DAY", "PER_WEEK", "PER_MONTH"}[t]
|
|
||||||
}
|
|
||||||
|
|
||||||
func (t TimePricingStrategy) GetStrategyValue() int {
|
|
||||||
return int(t)
|
|
||||||
}
|
|
||||||
|
|
||||||
func getAverageTimeInSecond(averageTimeInSecond float64, start time.Time, end *time.Time) float64 {
|
|
||||||
now := time.Now()
|
|
||||||
after := now.Add(time.Duration(averageTimeInSecond) * time.Second)
|
|
||||||
|
|
||||||
fromAverageDuration := after.Sub(now).Seconds()
|
|
||||||
var tEnd time.Time
|
|
||||||
if end == nil {
|
|
||||||
tEnd = start.Add(1 * time.Hour)
|
|
||||||
} else {
|
|
||||||
tEnd = *end
|
|
||||||
}
|
|
||||||
fromDateDuration := tEnd.Sub(start).Seconds()
|
|
||||||
|
|
||||||
if fromAverageDuration > fromDateDuration {
|
|
||||||
return fromAverageDuration
|
|
||||||
}
|
|
||||||
return fromDateDuration
|
|
||||||
}
|
|
||||||
|
|
||||||
func BookingEstimation(t TimePricingStrategy, price float64, locationDurationInSecond float64, start time.Time, end *time.Time) (float64, error) {
|
|
||||||
locationDurationInSecond = getAverageTimeInSecond(locationDurationInSecond, start, end)
|
|
||||||
priceStr := fmt.Sprintf("%v", price)
|
|
||||||
p, err := strconv.ParseFloat(priceStr, 64)
|
|
||||||
if err != nil {
|
|
||||||
return 0, err
|
|
||||||
}
|
|
||||||
switch t {
|
|
||||||
case ONCE:
|
|
||||||
return p, nil
|
|
||||||
case PER_HOUR:
|
|
||||||
return p * float64(locationDurationInSecond/3600), nil
|
|
||||||
case PER_MINUTE:
|
|
||||||
return p * float64(locationDurationInSecond/60), nil
|
|
||||||
case PER_SECOND:
|
|
||||||
return p * locationDurationInSecond, nil
|
|
||||||
case PER_DAY:
|
|
||||||
return p * float64(locationDurationInSecond/86400), nil
|
|
||||||
case PER_WEEK:
|
|
||||||
return p * float64(locationDurationInSecond/604800), nil
|
|
||||||
case PER_MONTH:
|
|
||||||
return p * float64(locationDurationInSecond/2592000), nil
|
|
||||||
}
|
|
||||||
return 0, errors.New("pricing strategy not found")
|
|
||||||
}
|
|
||||||
|
|
||||||
// may suppress in pricing strategy -> to set in map
|
|
||||||
type PricingStrategy[T Strategy] struct {
|
|
||||||
Price float64 `json:"price" bson:"price" default:"0"` // Price is the Price of the pricing
|
|
||||||
Currency string `json:"currency" bson:"currency" default:"USD"` // Currency is the currency of the pricing
|
|
||||||
BuyingStrategy BuyingStrategy `json:"buying_strategy" bson:"buying_strategy" default:"0"` // BuyingStrategy is the buying strategy of the pricing
|
|
||||||
TimePricingStrategy TimePricingStrategy `json:"time_pricing_strategy" bson:"time_pricing_strategy" default:"0"` // TimePricingStrategy is the time pricing strategy of the pricing
|
|
||||||
OverrideStrategy T `json:"override_strategy" bson:"override_strategy" default:"-1"` // Modulation is the modulation of the pricing
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p PricingStrategy[T]) GetPrice(amountOfData float64, bookingTimeDuration float64, start time.Time, end *time.Time) (float64, error) {
|
|
||||||
if p.BuyingStrategy == SUBSCRIPTION {
|
|
||||||
return BookingEstimation(p.GetTimePricingStrategy(), p.Price*float64(amountOfData), bookingTimeDuration, start, end)
|
|
||||||
} else if p.BuyingStrategy == PERMANENT {
|
|
||||||
return p.Price, nil
|
|
||||||
}
|
|
||||||
return p.Price * float64(amountOfData), nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p PricingStrategy[T]) GetBuyingStrategy() BuyingStrategy {
|
|
||||||
return p.BuyingStrategy
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p PricingStrategy[T]) GetTimePricingStrategy() TimePricingStrategy {
|
|
||||||
return p.TimePricingStrategy
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p PricingStrategy[T]) GetOverrideStrategy() T {
|
|
||||||
return p.OverrideStrategy
|
|
||||||
}
|
|
@ -1,129 +0,0 @@
|
|||||||
package pricing_test
|
|
||||||
|
|
||||||
import (
|
|
||||||
"testing"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/stretchr/testify/assert"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/pricing"
|
|
||||||
)
|
|
||||||
|
|
||||||
type DummyStrategy int
|
|
||||||
|
|
||||||
func (d DummyStrategy) GetStrategy() string { return "DUMMY" }
|
|
||||||
func (d DummyStrategy) GetStrategyValue() int { return int(d) }
|
|
||||||
|
|
||||||
func TestBuyingStrategy_String(t *testing.T) {
|
|
||||||
assert.Equal(t, "UNLIMITED", pricing.PERMANENT.String())
|
|
||||||
assert.Equal(t, "SUBSCRIPTION", pricing.SUBSCRIPTION.String())
|
|
||||||
//assert.Equal(t, "PAY PER USE", pricing.PAY_PER_USE.String())
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestBuyingStrategyList(t *testing.T) {
|
|
||||||
list := pricing.BuyingStrategyList()
|
|
||||||
assert.Equal(t, 3, len(list))
|
|
||||||
assert.Contains(t, list, pricing.SUBSCRIPTION)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestTimePricingStrategy_String(t *testing.T) {
|
|
||||||
assert.Equal(t, "ONCE", pricing.ONCE.String())
|
|
||||||
assert.Equal(t, "PER SECOND", pricing.PER_SECOND.String())
|
|
||||||
assert.Equal(t, "PER MONTH", pricing.PER_MONTH.String())
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestTimePricingStrategyList(t *testing.T) {
|
|
||||||
list := pricing.TimePricingStrategyList()
|
|
||||||
assert.Equal(t, 7, len(list))
|
|
||||||
assert.Contains(t, list, pricing.PER_DAY)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestTimePricingStrategy_Methods(t *testing.T) {
|
|
||||||
ts := pricing.PER_MINUTE
|
|
||||||
assert.Equal(t, "PER_MINUTE", ts.GetStrategy())
|
|
||||||
assert.Equal(t, 2, ts.GetStrategyValue())
|
|
||||||
}
|
|
||||||
|
|
||||||
func Test_getAverageTimeInSecond_WithEnd(t *testing.T) {
|
|
||||||
start := time.Now()
|
|
||||||
end := start.Add(30 * time.Minute)
|
|
||||||
|
|
||||||
_, err := pricing.BookingEstimation(pricing.PER_MINUTE, 2.0, 1200, start, &end)
|
|
||||||
assert.NoError(t, err)
|
|
||||||
}
|
|
||||||
|
|
||||||
func Test_getAverageTimeInSecond_WithoutEnd(t *testing.T) {
|
|
||||||
start := time.Now()
|
|
||||||
|
|
||||||
// getAverageTimeInSecond is tested via BookingEstimation
|
|
||||||
price, err := pricing.BookingEstimation(pricing.PER_HOUR, 10.0, 100, start, nil)
|
|
||||||
assert.NoError(t, err)
|
|
||||||
assert.True(t, price > 0)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestBookingEstimation(t *testing.T) {
|
|
||||||
start := time.Now()
|
|
||||||
end := start.Add(2 * time.Hour)
|
|
||||||
strategies := map[pricing.TimePricingStrategy]float64{
|
|
||||||
pricing.ONCE: 50,
|
|
||||||
pricing.PER_HOUR: 10,
|
|
||||||
pricing.PER_MINUTE: 1,
|
|
||||||
pricing.PER_SECOND: 0.1,
|
|
||||||
pricing.PER_DAY: 100,
|
|
||||||
pricing.PER_WEEK: 500,
|
|
||||||
pricing.PER_MONTH: 2000,
|
|
||||||
}
|
|
||||||
|
|
||||||
for strategy, price := range strategies {
|
|
||||||
t.Run(strategy.String(), func(t *testing.T) {
|
|
||||||
cost, err := pricing.BookingEstimation(strategy, price, 3600, start, &end)
|
|
||||||
assert.NoError(t, err)
|
|
||||||
assert.True(t, cost >= 0)
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
_, err := pricing.BookingEstimation(999, 10, 3600, start, &end)
|
|
||||||
assert.Error(t, err)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestPricingStrategy_Getters(t *testing.T) {
|
|
||||||
ps := pricing.PricingStrategy[DummyStrategy]{
|
|
||||||
Price: 20,
|
|
||||||
Currency: "USD",
|
|
||||||
BuyingStrategy: pricing.SUBSCRIPTION,
|
|
||||||
TimePricingStrategy: pricing.PER_MINUTE,
|
|
||||||
OverrideStrategy: DummyStrategy(1),
|
|
||||||
}
|
|
||||||
|
|
||||||
assert.Equal(t, pricing.SUBSCRIPTION, ps.GetBuyingStrategy())
|
|
||||||
assert.Equal(t, pricing.PER_MINUTE, ps.GetTimePricingStrategy())
|
|
||||||
assert.Equal(t, DummyStrategy(1), ps.GetOverrideStrategy())
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestPricingStrategy_GetPrice(t *testing.T) {
|
|
||||||
start := time.Now()
|
|
||||||
end := start.Add(1 * time.Hour)
|
|
||||||
|
|
||||||
// SUBSCRIPTION case
|
|
||||||
ps := pricing.PricingStrategy[DummyStrategy]{
|
|
||||||
Price: 5,
|
|
||||||
BuyingStrategy: pricing.SUBSCRIPTION,
|
|
||||||
TimePricingStrategy: pricing.PER_HOUR,
|
|
||||||
}
|
|
||||||
|
|
||||||
p, err := ps.GetPrice(2, 3600, start, &end)
|
|
||||||
assert.NoError(t, err)
|
|
||||||
assert.True(t, p > 0)
|
|
||||||
|
|
||||||
// UNLIMITED case
|
|
||||||
ps.BuyingStrategy = pricing.PERMANENT
|
|
||||||
p, err = ps.GetPrice(10, 0, start, &end)
|
|
||||||
assert.NoError(t, err)
|
|
||||||
assert.Equal(t, 5.0, p)
|
|
||||||
|
|
||||||
// PAY_PER_USE case
|
|
||||||
//ps.BuyingStrategy = pricing.PAY_PER_USE
|
|
||||||
p, err = ps.GetPrice(3, 0, start, &end)
|
|
||||||
assert.NoError(t, err)
|
|
||||||
assert.Equal(t, 15.0, p)
|
|
||||||
}
|
|
@ -1,18 +0,0 @@
|
|||||||
package live
|
|
||||||
|
|
||||||
import (
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/resources"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
)
|
|
||||||
|
|
||||||
type LiveInterface interface {
|
|
||||||
utils.DBObject
|
|
||||||
GetMonitorPath() string
|
|
||||||
GetResourcesID() []string
|
|
||||||
SetResourcesID(string)
|
|
||||||
GetResourceAccessor(request *tools.APIRequest) utils.Accessor
|
|
||||||
GetResource() resources.ResourceInterface
|
|
||||||
GetResourceInstance() resources.ResourceInstanceITF
|
|
||||||
SetResourceInstance(res resources.ResourceInterface, i resources.ResourceInstanceITF) resources.ResourceInterface
|
|
||||||
}
|
|
@ -1,71 +0,0 @@
|
|||||||
package live
|
|
||||||
|
|
||||||
import (
|
|
||||||
"slices"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/resources"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
"github.com/biter777/countries"
|
|
||||||
)
|
|
||||||
|
|
||||||
/*
|
|
||||||
* LiveDatacenter is a struct that represents a compute units in your datacenters
|
|
||||||
*/
|
|
||||||
type Credentials struct {
|
|
||||||
Login string `json:"login,omitempty" bson:"login,omitempty"`
|
|
||||||
Pass string `json:"password,omitempty" bson:"password,omitempty"`
|
|
||||||
Token string `json:"token,omitempty" bson:"token,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type Certs struct {
|
|
||||||
AuthorityCertificate string `json:"authority_certificate,omitempty" bson:"authority_certificate,omitempty"`
|
|
||||||
ClientCertificate string `json:"client_certificate,omitempty" bson:"client_certificate,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type LiveCerts struct {
|
|
||||||
Host string `json:"host,omitempty" bson:"host,omitempty"`
|
|
||||||
Port string `json:"port,omitempty" bson:"port,omitempty"`
|
|
||||||
|
|
||||||
Certificates *Certs `json:"certs,omitempty" bson:"certs,omitempty"`
|
|
||||||
Credentials *Credentials `json:"creds,omitempty" bson:"creds,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// TODO in the future multiple type of certs depending of infra type
|
|
||||||
|
|
||||||
type AbstractLive struct {
|
|
||||||
utils.AbstractObject
|
|
||||||
Certs LiveCerts `json:"certs,omitempty" bson:"certs,omitempty"`
|
|
||||||
|
|
||||||
MonitorPath string `json:"monitor_path,omitempty" bson:"monitor_path,omitempty"`
|
|
||||||
Location resources.GeoPoint `json:"location,omitempty" bson:"location,omitempty"`
|
|
||||||
Country countries.CountryCode `json:"country,omitempty" bson:"country,omitempty"`
|
|
||||||
AccessProtocol string `json:"access_protocol,omitempty" bson:"access_protocol,omitempty"`
|
|
||||||
ResourcesID []string `json:"resources_id" bson:"resources_id"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *AbstractLive) GetMonitorPath() string {
|
|
||||||
return d.MonitorPath
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *AbstractLive) GetResourcesID() []string {
|
|
||||||
return d.ResourcesID
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *AbstractLive) SetResourcesID(id string) {
|
|
||||||
if !slices.Contains(d.ResourcesID, id) {
|
|
||||||
d.ResourcesID = append(d.ResourcesID, id)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *AbstractLive) GetResourceType() tools.DataType {
|
|
||||||
return tools.INVALID
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *AbstractLive) StoreDraftDefault() {
|
|
||||||
r.IsDraft = true
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *AbstractLive) CanDelete() bool {
|
|
||||||
return r.IsDraft // only draft ComputeUnits can be deleted
|
|
||||||
}
|
|
@ -1,50 +0,0 @@
|
|||||||
package live
|
|
||||||
|
|
||||||
import (
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/enum"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/models"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/resources"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
)
|
|
||||||
|
|
||||||
/*
|
|
||||||
* LiveDatacenter is a struct that represents a compute units in your datacenters
|
|
||||||
*/
|
|
||||||
|
|
||||||
type LiveDatacenter struct {
|
|
||||||
AbstractLive
|
|
||||||
|
|
||||||
StorageType enum.StorageType `bson:"storage_type" json:"storage_type" default:"-1"` // Type is the type of the storage
|
|
||||||
Acronym string `bson:"acronym,omitempty" json:"acronym,omitempty"` // Acronym is the acronym of the storage
|
|
||||||
|
|
||||||
Architecture string `json:"architecture,omitempty" bson:"architecture,omitempty"` // Architecture is the architecture
|
|
||||||
Infrastructure enum.InfrastructureType `json:"infrastructure" bson:"infrastructure" default:"-1"` // Infrastructure is the infrastructure
|
|
||||||
Source string `json:"source,omitempty" bson:"source,omitempty"` // Source is the source of the resource
|
|
||||||
SecurityLevel string `json:"security_level,omitempty" bson:"security_level,omitempty"`
|
|
||||||
PowerSources []string `json:"power_sources,omitempty" bson:"power_sources,omitempty"`
|
|
||||||
AnnualCO2Emissions float64 `json:"annual_co2_emissions,omitempty" bson:"co2_emissions,omitempty"`
|
|
||||||
CPUs map[string]*models.CPU `bson:"cpus,omitempty" json:"cpus,omitempty"` // CPUs is the list of CPUs key is model
|
|
||||||
GPUs map[string]*models.GPU `bson:"gpus,omitempty" json:"gpus,omitempty"` // GPUs is the list of GPUs key is model
|
|
||||||
Nodes []*resources.ComputeNode `json:"nodes,omitempty" bson:"nodes,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *LiveDatacenter) GetAccessor(request *tools.APIRequest) utils.Accessor {
|
|
||||||
return NewAccessor[*LiveDatacenter](tools.LIVE_DATACENTER, request) // Create a new instance of the accessor
|
|
||||||
}
|
|
||||||
func (d *LiveDatacenter) GetResourceAccessor(request *tools.APIRequest) utils.Accessor {
|
|
||||||
return resources.NewAccessor[*resources.ComputeResource](tools.COMPUTE_RESOURCE, request, func() utils.DBObject { return &resources.ComputeResource{} })
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *LiveDatacenter) GetResource() resources.ResourceInterface {
|
|
||||||
return &resources.ComputeResource{}
|
|
||||||
}
|
|
||||||
func (d *LiveDatacenter) GetResourceInstance() resources.ResourceInstanceITF {
|
|
||||||
return &resources.ComputeResourceInstance{}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *LiveDatacenter) SetResourceInstance(res resources.ResourceInterface, i resources.ResourceInstanceITF) resources.ResourceInterface {
|
|
||||||
r := res.(*resources.ComputeResource)
|
|
||||||
r.Instances = append(r.Instances, i.(*resources.ComputeResourceInstance))
|
|
||||||
return r
|
|
||||||
}
|
|
@ -1,117 +0,0 @@
|
|||||||
package live
|
|
||||||
|
|
||||||
import (
|
|
||||||
"encoding/json"
|
|
||||||
"errors"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/dbs"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/logs"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
)
|
|
||||||
|
|
||||||
type computeUnitsMongoAccessor[T LiveInterface] struct {
|
|
||||||
utils.AbstractAccessor // AbstractAccessor contains the basic fields of an accessor (model, caller)
|
|
||||||
}
|
|
||||||
|
|
||||||
// New creates a new instance of the computeUnitsMongoAccessor
|
|
||||||
func NewAccessor[T LiveInterface](t tools.DataType, request *tools.APIRequest) *computeUnitsMongoAccessor[T] {
|
|
||||||
return &computeUnitsMongoAccessor[T]{
|
|
||||||
AbstractAccessor: utils.AbstractAccessor{
|
|
||||||
Logger: logs.CreateLogger(t.String()), // Create a logger with the data type
|
|
||||||
Request: request,
|
|
||||||
Type: t,
|
|
||||||
},
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
|
||||||
* Nothing special here, just the basic CRUD operations
|
|
||||||
*/
|
|
||||||
func (a *computeUnitsMongoAccessor[T]) DeleteOne(id string) (utils.DBObject, int, error) {
|
|
||||||
return utils.GenericDeleteOne(id, a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *computeUnitsMongoAccessor[T]) UpdateOne(set utils.DBObject, id string) (utils.DBObject, int, error) {
|
|
||||||
// should verify if a source is existing...
|
|
||||||
return utils.GenericUpdateOne(set, id, a, &LiveDatacenter{})
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *computeUnitsMongoAccessor[T]) StoreOne(data utils.DBObject) (utils.DBObject, int, error) {
|
|
||||||
return utils.GenericStoreOne(data.(*LiveDatacenter), a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *computeUnitsMongoAccessor[T]) CopyOne(data utils.DBObject) (utils.DBObject, int, error) {
|
|
||||||
// is a publisher... that become a resources.
|
|
||||||
if data.IsDrafted() {
|
|
||||||
return nil, 422, errors.New("can't publish a drafted compute units")
|
|
||||||
}
|
|
||||||
live := data.(T)
|
|
||||||
if live.GetMonitorPath() == "" || live.GetID() != "" {
|
|
||||||
return nil, 422, errors.New("publishing is only allowed is it can be monitored and be accessible")
|
|
||||||
}
|
|
||||||
if res, code, err := a.LoadOne(live.GetID()); err != nil {
|
|
||||||
return nil, code, err
|
|
||||||
} else {
|
|
||||||
live = res.(T)
|
|
||||||
}
|
|
||||||
resAccess := live.GetResourceAccessor(a.Request)
|
|
||||||
instance := live.GetResourceInstance()
|
|
||||||
b, _ := json.Marshal(live)
|
|
||||||
json.Unmarshal(b, instance)
|
|
||||||
|
|
||||||
if len(live.GetResourcesID()) > 0 {
|
|
||||||
for _, r := range live.GetResourcesID() {
|
|
||||||
// TODO dependent of a existing resource
|
|
||||||
res, code, err := resAccess.LoadOne(r)
|
|
||||||
if err == nil {
|
|
||||||
return nil, code, err
|
|
||||||
}
|
|
||||||
existingResource := live.GetResource()
|
|
||||||
b, _ := json.Marshal(res)
|
|
||||||
json.Unmarshal(b, existingResource)
|
|
||||||
live.SetResourceInstance(existingResource, instance)
|
|
||||||
resAccess.UpdateOne(existingResource, existingResource.GetID())
|
|
||||||
}
|
|
||||||
if live.GetID() != "" {
|
|
||||||
return a.LoadOne(live.GetID())
|
|
||||||
} else {
|
|
||||||
return a.StoreOne(live)
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
r := live.GetResource()
|
|
||||||
b, _ := json.Marshal(live)
|
|
||||||
json.Unmarshal(b, &r)
|
|
||||||
live.SetResourceInstance(r, instance)
|
|
||||||
res, code, err := resAccess.StoreOne(r)
|
|
||||||
if err != nil {
|
|
||||||
return nil, code, err
|
|
||||||
}
|
|
||||||
live.SetResourcesID(res.GetID())
|
|
||||||
if live.GetID() != "" {
|
|
||||||
return a.UpdateOne(live, live.GetID())
|
|
||||||
} else {
|
|
||||||
return a.StoreOne(live)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *computeUnitsMongoAccessor[T]) LoadOne(id string) (utils.DBObject, int, error) {
|
|
||||||
return utils.GenericLoadOne[T](id, func(d utils.DBObject) (utils.DBObject, int, error) {
|
|
||||||
return d, 200, nil
|
|
||||||
}, a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *computeUnitsMongoAccessor[T]) LoadAll(isDraft bool) ([]utils.ShallowDBObject, int, error) {
|
|
||||||
return utils.GenericLoadAll[T](a.getExec(), isDraft, a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *computeUnitsMongoAccessor[T]) Search(filters *dbs.Filters, search string, isDraft bool) ([]utils.ShallowDBObject, int, error) {
|
|
||||||
return utils.GenericSearch[*LiveDatacenter](filters, search, (&LiveDatacenter{}).GetObjectFilters(search), a.getExec(), isDraft, a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *computeUnitsMongoAccessor[T]) getExec() func(utils.DBObject) utils.ShallowDBObject {
|
|
||||||
return func(d utils.DBObject) utils.ShallowDBObject {
|
|
||||||
return d
|
|
||||||
}
|
|
||||||
}
|
|
@ -1,46 +0,0 @@
|
|||||||
package live
|
|
||||||
|
|
||||||
import (
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/enum"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/resources"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
)
|
|
||||||
|
|
||||||
/*
|
|
||||||
* LiveStorage is a struct that represents a compute units in your datacenters
|
|
||||||
*/
|
|
||||||
|
|
||||||
type LiveStorage struct {
|
|
||||||
AbstractLive
|
|
||||||
|
|
||||||
Source string `bson:"source,omitempty" json:"source,omitempty"` // Source is the source of the storage
|
|
||||||
Path string `bson:"path,omitempty" json:"path,omitempty"` // Path is the store folders in the source
|
|
||||||
Local bool `bson:"local" json:"local"`
|
|
||||||
SecurityLevel string `bson:"security_level,omitempty" json:"security_level,omitempty"`
|
|
||||||
SizeType enum.StorageSize `bson:"size_type" json:"size_type" default:"0"` // SizeType is the type of the storage size
|
|
||||||
SizeGB int64 `bson:"size,omitempty" json:"size,omitempty"` // Size is the size of the storage
|
|
||||||
Encryption bool `bson:"encryption,omitempty" json:"encryption,omitempty"` // Encryption is a flag that indicates if the storage is encrypted
|
|
||||||
Redundancy string `bson:"redundancy,omitempty" json:"redundancy,omitempty"` // Redundancy is the redundancy of the storage
|
|
||||||
Throughput string `bson:"throughput,omitempty" json:"throughput,omitempty"` // Throughput is the throughput of the storage
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *LiveStorage) GetAccessor(request *tools.APIRequest) utils.Accessor {
|
|
||||||
return NewAccessor[*LiveStorage](tools.LIVE_STORAGE, request) // Create a new instance of the accessor
|
|
||||||
}
|
|
||||||
func (d *LiveStorage) GetResourceAccessor(request *tools.APIRequest) utils.Accessor {
|
|
||||||
return resources.NewAccessor[*resources.ComputeResource](tools.STORAGE_RESOURCE, request, func() utils.DBObject { return &resources.StorageResource{} })
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *LiveStorage) GetResource() resources.ResourceInterface {
|
|
||||||
return &resources.StorageResource{}
|
|
||||||
}
|
|
||||||
func (d *LiveStorage) GetResourceInstance() resources.ResourceInstanceITF {
|
|
||||||
return &resources.StorageResourceInstance{}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *LiveStorage) SetResourceInstance(res resources.ResourceInterface, i resources.ResourceInstanceITF) resources.ResourceInterface {
|
|
||||||
r := res.(*resources.StorageResource)
|
|
||||||
r.Instances = append(r.Instances, i.(*resources.StorageResourceInstance))
|
|
||||||
return r
|
|
||||||
}
|
|
@ -2,17 +2,18 @@ package models
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"cloud.o-forge.io/core/oc-lib/logs"
|
"cloud.o-forge.io/core/oc-lib/logs"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/bill"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/live"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/order"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/resources/purchase_resource"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
"cloud.o-forge.io/core/oc-lib/tools"
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/booking"
|
"cloud.o-forge.io/core/oc-lib/models/booking"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/collaborative_area"
|
"cloud.o-forge.io/core/oc-lib/models/collaborative_area"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/collaborative_area/rules/rule"
|
"cloud.o-forge.io/core/oc-lib/models/collaborative_area/rules/rule"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/peer"
|
"cloud.o-forge.io/core/oc-lib/models/peer"
|
||||||
resource "cloud.o-forge.io/core/oc-lib/models/resources"
|
"cloud.o-forge.io/core/oc-lib/models/resource_model"
|
||||||
|
d "cloud.o-forge.io/core/oc-lib/models/resources/data"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/resources/compute"
|
||||||
|
p "cloud.o-forge.io/core/oc-lib/models/resources/processing"
|
||||||
|
s "cloud.o-forge.io/core/oc-lib/models/resources/storage"
|
||||||
|
w "cloud.o-forge.io/core/oc-lib/models/resources/workflow"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
"cloud.o-forge.io/core/oc-lib/models/utils"
|
||||||
w2 "cloud.o-forge.io/core/oc-lib/models/workflow"
|
w2 "cloud.o-forge.io/core/oc-lib/models/workflow"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/workflow_execution"
|
"cloud.o-forge.io/core/oc-lib/models/workflow_execution"
|
||||||
@ -23,33 +24,29 @@ import (
|
|||||||
This package contains the models used in the application
|
This package contains the models used in the application
|
||||||
It's used to create the models dynamically
|
It's used to create the models dynamically
|
||||||
*/
|
*/
|
||||||
var ModelsCatalog = map[string]func() utils.DBObject{
|
var models = map[string]func() utils.DBObject{
|
||||||
tools.WORKFLOW_RESOURCE.String(): func() utils.DBObject { return &resource.WorkflowResource{} },
|
tools.WORKFLOW_RESOURCE.String(): func() utils.DBObject { return &w.WorkflowResource{} },
|
||||||
tools.DATA_RESOURCE.String(): func() utils.DBObject { return &resource.DataResource{} },
|
tools.DATA_RESOURCE.String(): func() utils.DBObject { return &d.DataResource{} },
|
||||||
tools.COMPUTE_RESOURCE.String(): func() utils.DBObject { return &resource.ComputeResource{} },
|
tools.COMPUTE_RESOURCE.String(): func() utils.DBObject { return &compute.ComputeResource{} },
|
||||||
tools.STORAGE_RESOURCE.String(): func() utils.DBObject { return &resource.StorageResource{} },
|
tools.STORAGE_RESOURCE.String(): func() utils.DBObject { return &s.StorageResource{} },
|
||||||
tools.PROCESSING_RESOURCE.String(): func() utils.DBObject { return &resource.ProcessingResource{} },
|
tools.PROCESSING_RESOURCE.String(): func() utils.DBObject { return &p.ProcessingResource{} },
|
||||||
tools.WORKFLOW.String(): func() utils.DBObject { return &w2.Workflow{} },
|
tools.WORKFLOW.String(): func() utils.DBObject { return &w2.Workflow{} },
|
||||||
tools.WORKFLOW_EXECUTION.String(): func() utils.DBObject { return &workflow_execution.WorkflowExecution{} },
|
tools.WORKFLOW_EXECUTION.String(): func() utils.DBObject { return &workflow_execution.WorkflowExecution{} },
|
||||||
tools.WORKSPACE.String(): func() utils.DBObject { return &w3.Workspace{} },
|
tools.WORKSPACE.String(): func() utils.DBObject { return &w3.Workspace{} },
|
||||||
|
tools.RESOURCE_MODEL.String(): func() utils.DBObject { return &resource_model.ResourceModel{} },
|
||||||
tools.PEER.String(): func() utils.DBObject { return &peer.Peer{} },
|
tools.PEER.String(): func() utils.DBObject { return &peer.Peer{} },
|
||||||
tools.COLLABORATIVE_AREA.String(): func() utils.DBObject { return &collaborative_area.CollaborativeArea{} },
|
tools.COLLABORATIVE_AREA.String(): func() utils.DBObject { return &collaborative_area.CollaborativeArea{} },
|
||||||
tools.RULE.String(): func() utils.DBObject { return &rule.Rule{} },
|
tools.RULE.String(): func() utils.DBObject { return &rule.Rule{} },
|
||||||
tools.BOOKING.String(): func() utils.DBObject { return &booking.Booking{} },
|
tools.BOOKING.String(): func() utils.DBObject { return &booking.Booking{} },
|
||||||
tools.WORKFLOW_HISTORY.String(): func() utils.DBObject { return &w2.WorkflowHistory{} },
|
tools.WORKFLOW_HISTORY.String(): func() utils.DBObject { return &w2.WorkflowHistory{} },
|
||||||
tools.WORKSPACE_HISTORY.String(): func() utils.DBObject { return &w3.WorkspaceHistory{} },
|
tools.WORKSPACE_HISTORY.String(): func() utils.DBObject { return &w3.WorkspaceHistory{} },
|
||||||
tools.ORDER.String(): func() utils.DBObject { return &order.Order{} },
|
|
||||||
tools.PURCHASE_RESOURCE.String(): func() utils.DBObject { return &purchase_resource.PurchaseResource{} },
|
|
||||||
tools.LIVE_DATACENTER.String(): func() utils.DBObject { return &live.LiveDatacenter{} },
|
|
||||||
tools.LIVE_STORAGE.String(): func() utils.DBObject { return &live.LiveStorage{} },
|
|
||||||
tools.BILL.String(): func() utils.DBObject { return &bill.Bill{} },
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Model returns the model object based on the model type
|
// Model returns the model object based on the model type
|
||||||
func Model(model int) utils.DBObject {
|
func Model(model int) utils.DBObject {
|
||||||
log := logs.GetLogger()
|
log := logs.GetLogger()
|
||||||
if _, ok := ModelsCatalog[tools.FromInt(model)]; ok {
|
if _, ok := models[tools.FromInt(model)]; ok {
|
||||||
return ModelsCatalog[tools.FromInt(model)]()
|
return models[tools.FromInt(model)]()
|
||||||
}
|
}
|
||||||
log.Error().Msg("Can't find model " + tools.FromInt(model) + ".")
|
log.Error().Msg("Can't find model " + tools.FromInt(model) + ".")
|
||||||
return nil
|
return nil
|
||||||
@ -58,7 +55,7 @@ func Model(model int) utils.DBObject {
|
|||||||
// GetModelsNames returns the names of the models
|
// GetModelsNames returns the names of the models
|
||||||
func GetModelsNames() []string {
|
func GetModelsNames() []string {
|
||||||
names := []string{}
|
names := []string{}
|
||||||
for name := range ModelsCatalog {
|
for name := range models {
|
||||||
names = append(names, name)
|
names = append(names, name)
|
||||||
}
|
}
|
||||||
return names
|
return names
|
||||||
|
@ -1,53 +0,0 @@
|
|||||||
package order
|
|
||||||
|
|
||||||
import (
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/booking"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/enum"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/pricing"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/resources/purchase_resource"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
)
|
|
||||||
|
|
||||||
/*
|
|
||||||
* Booking is a struct that represents a booking
|
|
||||||
*/
|
|
||||||
|
|
||||||
type Order struct {
|
|
||||||
utils.AbstractObject
|
|
||||||
ExecutionsID string `json:"executions_id" bson:"executions_id" validate:"required"`
|
|
||||||
Status enum.CompletionStatus `json:"status" bson:"status" default:"0"`
|
|
||||||
Purchases []*purchase_resource.PurchaseResource `json:"purchases" bson:"purchases"`
|
|
||||||
Bookings []*booking.Booking `json:"bookings" bson:"bookings"`
|
|
||||||
|
|
||||||
Billing map[pricing.BillingStrategy][]*booking.Booking `json:"billing" bson:"billing"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *Order) StoreDraftDefault() {
|
|
||||||
r.IsDraft = true
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *Order) CanUpdate(set utils.DBObject) (bool, utils.DBObject) {
|
|
||||||
if !r.IsDraft && r.Status != set.(*Order).Status {
|
|
||||||
return true, &Order{Status: set.(*Order).Status} // only state can be updated
|
|
||||||
}
|
|
||||||
return r.IsDraft, set
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *Order) CanDelete() bool {
|
|
||||||
return r.IsDraft // only draft order can be deleted
|
|
||||||
}
|
|
||||||
|
|
||||||
func (o *Order) Quantity() int {
|
|
||||||
return len(o.Purchases) + len(o.Purchases)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *Order) SetName() {
|
|
||||||
d.Name = d.UUID + "_order_" + "_" + time.Now().UTC().Format("2006-01-02T15:04:05")
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *Order) GetAccessor(request *tools.APIRequest) utils.Accessor {
|
|
||||||
return NewAccessor(request) // Create a new instance of the accessor
|
|
||||||
}
|
|
@ -1,64 +0,0 @@
|
|||||||
package order
|
|
||||||
|
|
||||||
import (
|
|
||||||
"errors"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/dbs"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/logs"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
)
|
|
||||||
|
|
||||||
type orderMongoAccessor struct {
|
|
||||||
utils.AbstractAccessor // AbstractAccessor contains the basic fields of an accessor (model, caller)
|
|
||||||
}
|
|
||||||
|
|
||||||
// New creates a new instance of the orderMongoAccessor
|
|
||||||
func NewAccessor(request *tools.APIRequest) *orderMongoAccessor {
|
|
||||||
return &orderMongoAccessor{
|
|
||||||
AbstractAccessor: utils.AbstractAccessor{
|
|
||||||
Logger: logs.CreateLogger(tools.ORDER.String()), // Create a logger with the data type
|
|
||||||
Request: request,
|
|
||||||
Type: tools.ORDER,
|
|
||||||
},
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
|
||||||
* Nothing special here, just the basic CRUD operations
|
|
||||||
*/
|
|
||||||
func (a *orderMongoAccessor) DeleteOne(id string) (utils.DBObject, int, error) {
|
|
||||||
return utils.GenericDeleteOne(id, a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *orderMongoAccessor) UpdateOne(set utils.DBObject, id string) (utils.DBObject, int, error) {
|
|
||||||
return utils.GenericUpdateOne(set, id, a, &Order{})
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *orderMongoAccessor) StoreOne(data utils.DBObject) (utils.DBObject, int, error) {
|
|
||||||
return utils.GenericStoreOne(data,a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *orderMongoAccessor) CopyOne(data utils.DBObject) (utils.DBObject, int, error) {
|
|
||||||
return nil, 404, errors.New("Not implemented")
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *orderMongoAccessor) LoadOne(id string) (utils.DBObject, int, error) {
|
|
||||||
return utils.GenericLoadOne[*Order](id, func(d utils.DBObject) (utils.DBObject, int, error) {
|
|
||||||
return d, 200, nil
|
|
||||||
}, a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *orderMongoAccessor) LoadAll(isDraft bool) ([]utils.ShallowDBObject, int, error) {
|
|
||||||
return utils.GenericLoadAll[*Order](a.getExec(), isDraft, a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *orderMongoAccessor) Search(filters *dbs.Filters, search string, isDraft bool) ([]utils.ShallowDBObject, int, error) {
|
|
||||||
return utils.GenericSearch[*Order](filters, search, (&Order{}).GetObjectFilters(search), a.getExec(), isDraft, a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *orderMongoAccessor) getExec() func(utils.DBObject) utils.ShallowDBObject {
|
|
||||||
return func(d utils.DBObject) utils.ShallowDBObject {
|
|
||||||
return d
|
|
||||||
}
|
|
||||||
}
|
|
@ -1 +0,0 @@
|
|||||||
package tests
|
|
@ -1,10 +1,12 @@
|
|||||||
package peer
|
package peer
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"encoding/json"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
"cloud.o-forge.io/core/oc-lib/models/utils"
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
"cloud.o-forge.io/core/oc-lib/tools"
|
||||||
|
"github.com/google/uuid"
|
||||||
)
|
)
|
||||||
|
|
||||||
// now write a go enum for the state partner with self, blacklist, partner
|
// now write a go enum for the state partner with self, blacklist, partner
|
||||||
@ -29,18 +31,13 @@ func (m PeerState) EnumIndex() int {
|
|||||||
// Peer is a struct that represents a peer
|
// Peer is a struct that represents a peer
|
||||||
type Peer struct {
|
type Peer struct {
|
||||||
utils.AbstractObject
|
utils.AbstractObject
|
||||||
Url string `json:"url" bson:"url" validate:"required"` // Url is the URL of the peer (base64url)
|
Url string `json:"url,omitempty" bson:"url,omitempty" validate:"required"` // Url is the URL of the peer (base64url)
|
||||||
WalletAddress string `json:"wallet_address" bson:"wallet_address" validate:"required"` // WalletAddress is the wallet address of the peer
|
PublicKey string `json:"public_key,omitempty" bson:"public_key,omitempty"` // PublicKey is the public key of the peer
|
||||||
PublicKey string `json:"public_key" bson:"public_key" validate:"required"` // PublicKey is the public key of the peer
|
Services map[string]int `json:"services,omitempty" bson:"services,omitempty"`
|
||||||
State PeerState `json:"state" bson:"state" default:"0"`
|
State PeerState `json:"state" bson:"state" default:"0"`
|
||||||
ServicesState map[string]int `json:"services_state,omitempty" bson:"services_state,omitempty"`
|
|
||||||
FailedExecution []PeerExecution `json:"failed_execution" bson:"failed_execution"` // FailedExecution is the list of failed executions, to be retried
|
FailedExecution []PeerExecution `json:"failed_execution" bson:"failed_execution"` // FailedExecution is the list of failed executions, to be retried
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ao *Peer) VerifyAuth(request *tools.APIRequest) bool {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
|
|
||||||
// AddExecution adds an execution to the list of failed executions
|
// AddExecution adds an execution to the list of failed executions
|
||||||
func (ao *Peer) AddExecution(exec PeerExecution) {
|
func (ao *Peer) AddExecution(exec PeerExecution) {
|
||||||
found := false
|
found := false
|
||||||
@ -67,25 +64,56 @@ func (ao *Peer) RemoveExecution(exec PeerExecution) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// IsMySelf checks if the peer is the local peer
|
// IsMySelf checks if the peer is the local peer
|
||||||
func (p *Peer) IsMySelf() (bool, string) {
|
func (ao *Peer) IsMySelf() (bool, string) {
|
||||||
d, code, err := NewAccessor(nil).Search(nil, SELF.String(), p.IsDraft)
|
d, code, err := ao.GetAccessor(nil).Search(nil, SELF.String())
|
||||||
if code != 200 || err != nil || len(d) == 0 {
|
if code != 200 || err != nil || len(d) == 0 {
|
||||||
return false, ""
|
return false, ""
|
||||||
}
|
}
|
||||||
id := d[0].GetID()
|
id := d[0].GetID()
|
||||||
return p.UUID == id, id
|
return ao.UUID == id, id
|
||||||
}
|
}
|
||||||
|
|
||||||
// LaunchPeerExecution launches an execution on a peer
|
// LaunchPeerExecution launches an execution on a peer
|
||||||
func (p *Peer) LaunchPeerExecution(peerID string, dataID string, dt tools.DataType, method tools.METHOD, body interface{}, caller *tools.HTTPCaller) (map[string]interface{}, error) {
|
func (p *Peer) LaunchPeerExecution(peerID string, dataID string, dt tools.DataType, method tools.METHOD, body map[string]interface{}, caller *tools.HTTPCaller) (*PeerExecution, error) {
|
||||||
p.UUID = peerID
|
p.UUID = peerID
|
||||||
return cache.LaunchPeerExecution(peerID, dataID, dt, method, body, caller) // Launch the execution on the peer through the cache
|
return cache.LaunchPeerExecution(peerID, dataID, dt, method, body, caller) // Launch the execution on the peer through the cache
|
||||||
}
|
}
|
||||||
func (d *Peer) GetAccessor(request *tools.APIRequest) utils.Accessor {
|
|
||||||
data := NewAccessor(request) // Create a new instance of the accessor
|
func (ao *Peer) GetID() string {
|
||||||
|
return ao.UUID
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *Peer) GenerateID() {
|
||||||
|
if r.UUID == "" {
|
||||||
|
r.UUID = uuid.New().String()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *Peer) GetName() string {
|
||||||
|
return d.Name
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *Peer) GetAccessor(caller *tools.HTTPCaller) utils.Accessor {
|
||||||
|
data := New() // Create a new instance of the accessor
|
||||||
|
data.Init(tools.PEER, caller) // Initialize the accessor with the PEER model type
|
||||||
return data
|
return data
|
||||||
}
|
}
|
||||||
|
|
||||||
func (r *Peer) CanDelete() bool {
|
func (dma *Peer) Deserialize(j map[string]interface{}) utils.DBObject {
|
||||||
return false // only draft order can be deleted
|
b, err := json.Marshal(j)
|
||||||
|
if err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
json.Unmarshal(b, dma)
|
||||||
|
return dma
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dma *Peer) Serialize() map[string]interface{} {
|
||||||
|
var m map[string]interface{}
|
||||||
|
b, err := json.Marshal(dma)
|
||||||
|
if err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
json.Unmarshal(b, &m)
|
||||||
|
return m
|
||||||
}
|
}
|
||||||
|
@ -4,6 +4,7 @@ import (
|
|||||||
"encoding/json"
|
"encoding/json"
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"regexp"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
"cloud.o-forge.io/core/oc-lib/tools"
|
||||||
@ -14,11 +15,11 @@ import (
|
|||||||
* it defines the execution data
|
* it defines the execution data
|
||||||
*/
|
*/
|
||||||
type PeerExecution struct {
|
type PeerExecution struct {
|
||||||
Method string `json:"method" bson:"method"`
|
Method string `json:"method" bson:"method"`
|
||||||
Url string `json:"url" bson:"url"`
|
Url string `json:"url" bson:"url"`
|
||||||
Body interface{} `json:"body" bson:"body"`
|
Body map[string]interface{} `json:"body" bson:"body"`
|
||||||
DataType int `json:"data_type" bson:"data_type"`
|
DataType int `json:"data_type" bson:"data_type"`
|
||||||
DataID string `json:"data_id" bson:"data_id"`
|
DataID string `json:"data_id" bson:"data_id"`
|
||||||
}
|
}
|
||||||
|
|
||||||
var cache = &PeerCache{} // Singleton instance of the peer cache
|
var cache = &PeerCache{} // Singleton instance of the peer cache
|
||||||
@ -28,69 +29,99 @@ type PeerCache struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// urlFormat formats the URL of the peer with the data type API function
|
// urlFormat formats the URL of the peer with the data type API function
|
||||||
func (p *PeerCache) urlFormat(hostUrl string, dt tools.DataType) string {
|
func (p *PeerCache) urlFormat(url string, dt tools.DataType) string {
|
||||||
return hostUrl + "/" + strings.ReplaceAll(dt.API(), "oc-", "")
|
// localhost is replaced by the local peer URL
|
||||||
|
// because localhost must collide on a web request security protocol
|
||||||
|
localhost := ""
|
||||||
|
if strings.Contains(url, "localhost") {
|
||||||
|
localhost = "localhost"
|
||||||
|
}
|
||||||
|
if strings.Contains(url, "127.0.0.1") {
|
||||||
|
localhost = "127.0.0.1"
|
||||||
|
}
|
||||||
|
if localhost != "" {
|
||||||
|
r := regexp.MustCompile("(" + localhost + ":[0-9]+)")
|
||||||
|
t := r.FindString(url)
|
||||||
|
if t != "" {
|
||||||
|
url = strings.Replace(url, t, dt.API()+":8080/oc", -1)
|
||||||
|
} else {
|
||||||
|
url = strings.ReplaceAll(url, localhost, dt.API()+":8080/oc")
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
url = url + "/" + dt.API()
|
||||||
|
}
|
||||||
|
return url
|
||||||
}
|
}
|
||||||
|
|
||||||
// checkPeerStatus checks the status of a peer
|
// checkPeerStatus checks the status of a peer
|
||||||
func (p *PeerCache) checkPeerStatus(peerID string, appName string) (*Peer, bool) {
|
func (p *PeerCache) checkPeerStatus(peerID string, appName string, caller *tools.HTTPCaller) (*Peer, bool) {
|
||||||
api := tools.API{}
|
api := tools.API{}
|
||||||
access := NewShallowAccessor()
|
access := (&Peer{}).GetAccessor(nil)
|
||||||
res, code, _ := access.LoadOne(peerID) // Load the peer from db
|
res, code, _ := access.LoadOne(peerID) // Load the peer from db
|
||||||
if code != 200 { // no peer no party
|
if code != 200 { // no peer no party
|
||||||
return nil, false
|
return nil, false
|
||||||
}
|
}
|
||||||
url := p.urlFormat(res.(*Peer).Url, tools.PEER) + "/status" // Format the URL
|
methods := caller.URLS[tools.PEER] // Get the methods url of the peer
|
||||||
|
if methods == nil {
|
||||||
|
return res.(*Peer), false
|
||||||
|
}
|
||||||
|
meth := methods[tools.POST] // Get the POST method to check status
|
||||||
|
if meth == "" {
|
||||||
|
return res.(*Peer), false
|
||||||
|
}
|
||||||
|
url := p.urlFormat(res.(*Peer).Url, tools.PEER) + meth // Format the URL
|
||||||
|
fmt.Println("Checking peer status on", url, "...")
|
||||||
state, services := api.CheckRemotePeer(url)
|
state, services := api.CheckRemotePeer(url)
|
||||||
res.(*Peer).ServicesState = services // Update the services states of the peer
|
fmt.Println("Checking peer status on", url, state, services) // Check the status of the peer
|
||||||
|
res.(*Peer).Services = services // Update the services states of the peer
|
||||||
access.UpdateOne(res, peerID) // Update the peer in the db
|
access.UpdateOne(res, peerID) // Update the peer in the db
|
||||||
return res.(*Peer), state != tools.DEAD && services[appName] == 0 // Return the peer and its status
|
return res.(*Peer), state != tools.DEAD && services[appName] == 0 // Return the peer and its status
|
||||||
}
|
}
|
||||||
|
|
||||||
// LaunchPeerExecution launches an execution on a peer
|
// LaunchPeerExecution launches an execution on a peer
|
||||||
// The method contacts the path described by : peer.Url + datatype path (from enums) + replacement of id by dataID
|
|
||||||
func (p *PeerCache) LaunchPeerExecution(peerID string, dataID string,
|
func (p *PeerCache) LaunchPeerExecution(peerID string, dataID string,
|
||||||
dt tools.DataType, method tools.METHOD, body interface{}, caller tools.HTTPCallerITF) (map[string]interface{}, error) {
|
dt tools.DataType, method tools.METHOD, body map[string]interface{}, caller *tools.HTTPCaller) (*PeerExecution, error) {
|
||||||
fmt.Println("Launching peer execution on", caller.GetUrls(), dt, method)
|
fmt.Println("Launching peer execution on", caller.URLS, dt, method)
|
||||||
methods := caller.GetUrls()[dt] // Get the methods url of the data type
|
methods := caller.URLS[dt] // Get the methods url of the data type
|
||||||
if m, ok := methods[method]; !ok || m == "" {
|
if m, ok := methods[method]; !ok || m == "" {
|
||||||
return map[string]interface{}{}, errors.New("Requested method " + method.String() + " not declared in HTTPCaller")
|
return nil, errors.New("no path found")
|
||||||
}
|
}
|
||||||
path := methods[method] // Get the path corresponding to the action we want to execute
|
meth := methods[method] // Get the method url to execute
|
||||||
path = strings.ReplaceAll(path, ":id", dataID) // Replace the id in the path in case of a DELETE / UPDATE method (it's a standard naming in OC)
|
meth = strings.ReplaceAll(meth, ":id", dataID) // Replace the id in the url in case of a DELETE / UPDATE method (it's a standard naming in OC)
|
||||||
url := ""
|
url := ""
|
||||||
|
|
||||||
// Check the status of the peer
|
// Check the status of the peer
|
||||||
if mypeer, ok := p.checkPeerStatus(peerID, dt.API()); !ok && mypeer != nil {
|
if mypeer, ok := p.checkPeerStatus(peerID, dt.API(), caller); !ok && mypeer != nil {
|
||||||
// If the peer is not reachable, add the execution to the failed executions list
|
// If the peer is not reachable, add the execution to the failed executions list
|
||||||
pexec := &PeerExecution{
|
pexec := &PeerExecution{
|
||||||
Method: method.String(),
|
Method: method.String(),
|
||||||
Url: p.urlFormat((mypeer.Url), dt) + path, // the url is constitued of : host URL + resource path + action path (ex : mypeer.com/datacenter/resourcetype/path/to/action)
|
Url: p.urlFormat((mypeer.Url)+meth, dt),
|
||||||
Body: body,
|
Body: body,
|
||||||
DataType: dt.EnumIndex(),
|
DataType: dt.EnumIndex(),
|
||||||
DataID: dataID,
|
DataID: dataID,
|
||||||
}
|
}
|
||||||
mypeer.AddExecution(*pexec)
|
mypeer.AddExecution(*pexec)
|
||||||
NewShallowAccessor().UpdateOne(mypeer, peerID) // Update the peer in the db
|
mypeer.GetAccessor(nil).UpdateOne(mypeer, peerID) // Update the peer in the db
|
||||||
return map[string]interface{}{}, errors.New("peer is " + peerID + " not reachable")
|
return nil, errors.New("peer is not reachable")
|
||||||
} else {
|
} else {
|
||||||
if mypeer == nil {
|
if mypeer == nil {
|
||||||
return map[string]interface{}{}, errors.New("peer " + peerID + " not found")
|
return nil, errors.New("peer not found")
|
||||||
}
|
}
|
||||||
// If the peer is reachable, launch the execution
|
// If the peer is reachable, launch the execution
|
||||||
url = p.urlFormat((mypeer.Url), dt) + path // Format the URL
|
url = p.urlFormat((mypeer.Url)+meth, dt) // Format the URL
|
||||||
tmp := mypeer.FailedExecution // Get the failed executions list
|
tmp := mypeer.FailedExecution // Get the failed executions list
|
||||||
mypeer.FailedExecution = []PeerExecution{} // Reset the failed executions list
|
mypeer.FailedExecution = []PeerExecution{} // Reset the failed executions list
|
||||||
NewShallowAccessor().UpdateOne(mypeer, peerID) // Update the peer in the db
|
mypeer.GetAccessor(nil).UpdateOne(mypeer, peerID) // Update the peer in the db
|
||||||
for _, v := range tmp { // Retry the failed executions
|
for _, v := range tmp { // Retry the failed executions
|
||||||
go p.Exec(v.Url, tools.ToMethod(v.Method), v.Body, caller)
|
go p.exec(v.Url, tools.ToMethod(v.Method), v.Body, caller)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return p.Exec(url, method, body, caller) // Execute the method
|
fmt.Println("URL exec", url)
|
||||||
|
return nil, p.exec(url, method, body, caller) // Execute the method
|
||||||
}
|
}
|
||||||
|
|
||||||
// exec executes the method on the peer
|
// exec executes the method on the peer
|
||||||
func (p *PeerCache) Exec(url string, method tools.METHOD, body interface{}, caller tools.HTTPCallerITF) (map[string]interface{}, error) {
|
func (p *PeerCache) exec(url string, method tools.METHOD, body map[string]interface{}, caller *tools.HTTPCaller) error {
|
||||||
var b []byte
|
var b []byte
|
||||||
var err error
|
var err error
|
||||||
if method == tools.POST { // Execute the POST method if it's a POST method
|
if method == tools.POST { // Execute the POST method if it's a POST method
|
||||||
@ -103,15 +134,12 @@ func (p *PeerCache) Exec(url string, method tools.METHOD, body interface{}, call
|
|||||||
b, err = caller.CallDelete(url, "")
|
b, err = caller.CallDelete(url, "")
|
||||||
}
|
}
|
||||||
var m map[string]interface{}
|
var m map[string]interface{}
|
||||||
|
json.Unmarshal(b, &m)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return m, err
|
return err
|
||||||
}
|
|
||||||
err = json.Unmarshal(b, &m)
|
|
||||||
if err != nil {
|
|
||||||
return m, err
|
|
||||||
}
|
}
|
||||||
if e, ok := m["error"]; ok && e != "<nil>" && e != "" { // Check if there is an error in the response
|
if e, ok := m["error"]; ok && e != "<nil>" && e != "" { // Check if there is an error in the response
|
||||||
return m, errors.New(fmt.Sprintf("%v", m["error"]))
|
return errors.New(fmt.Sprintf("%v", m["error"]))
|
||||||
}
|
}
|
||||||
return m, nil
|
return nil
|
||||||
}
|
}
|
||||||
|
@ -4,40 +4,17 @@ import (
|
|||||||
"strconv"
|
"strconv"
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/dbs"
|
"cloud.o-forge.io/core/oc-lib/dbs"
|
||||||
"cloud.o-forge.io/core/oc-lib/logs"
|
"cloud.o-forge.io/core/oc-lib/dbs/mongo"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
"cloud.o-forge.io/core/oc-lib/models/utils"
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
type peerMongoAccessor struct {
|
type peerMongoAccessor struct {
|
||||||
utils.AbstractAccessor // AbstractAccessor contains the basic fields of an accessor (model, caller)
|
utils.AbstractAccessor // AbstractAccessor contains the basic fields of an accessor (model, caller)
|
||||||
OverrideAuth bool
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// New creates a new instance of the peerMongoAccessor
|
// New creates a new instance of the peerMongoAccessor
|
||||||
func NewShallowAccessor() *peerMongoAccessor {
|
func New() *peerMongoAccessor {
|
||||||
return &peerMongoAccessor{
|
return &peerMongoAccessor{}
|
||||||
OverrideAuth: true,
|
|
||||||
AbstractAccessor: utils.AbstractAccessor{
|
|
||||||
Logger: logs.CreateLogger(tools.PEER.String()), // Create a logger with the data type
|
|
||||||
Type: tools.PEER,
|
|
||||||
},
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewAccessor(request *tools.APIRequest) *peerMongoAccessor {
|
|
||||||
return &peerMongoAccessor{
|
|
||||||
OverrideAuth: false,
|
|
||||||
AbstractAccessor: utils.AbstractAccessor{
|
|
||||||
Logger: logs.CreateLogger(tools.PEER.String()), // Create a logger with the data type
|
|
||||||
Request: request,
|
|
||||||
Type: tools.PEER,
|
|
||||||
},
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (wfa *peerMongoAccessor) ShouldVerifyAuth() bool {
|
|
||||||
return !wfa.OverrideAuth
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/*
|
/*
|
||||||
@ -45,55 +22,80 @@ func (wfa *peerMongoAccessor) ShouldVerifyAuth() bool {
|
|||||||
*/
|
*/
|
||||||
|
|
||||||
func (wfa *peerMongoAccessor) DeleteOne(id string) (utils.DBObject, int, error) {
|
func (wfa *peerMongoAccessor) DeleteOne(id string) (utils.DBObject, int, error) {
|
||||||
return utils.GenericDeleteOne(id, wfa)
|
return wfa.GenericDeleteOne(id, wfa)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (wfa *peerMongoAccessor) UpdateOne(set utils.DBObject, id string) (utils.DBObject, int, error) {
|
func (wfa *peerMongoAccessor) UpdateOne(set utils.DBObject, id string) (utils.DBObject, int, error) {
|
||||||
return utils.GenericUpdateOne(set.(*Peer), id, wfa, &Peer{})
|
return wfa.GenericUpdateOne(set.(*Peer), id, wfa, &Peer{})
|
||||||
}
|
}
|
||||||
|
|
||||||
func (wfa *peerMongoAccessor) StoreOne(data utils.DBObject) (utils.DBObject, int, error) {
|
func (wfa *peerMongoAccessor) StoreOne(data utils.DBObject) (utils.DBObject, int, error) {
|
||||||
return utils.GenericStoreOne(data.(*Peer), wfa)
|
return wfa.GenericStoreOne(data.(*Peer), wfa)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (wfa *peerMongoAccessor) CopyOne(data utils.DBObject) (utils.DBObject, int, error) {
|
func (wfa *peerMongoAccessor) CopyOne(data utils.DBObject) (utils.DBObject, int, error) {
|
||||||
return utils.GenericStoreOne(data, wfa)
|
return wfa.GenericStoreOne(data, wfa)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (dca *peerMongoAccessor) LoadOne(id string) (utils.DBObject, int, error) {
|
func (wfa *peerMongoAccessor) LoadOne(id string) (utils.DBObject, int, error) {
|
||||||
return utils.GenericLoadOne[*Peer](id, func(d utils.DBObject) (utils.DBObject, int, error) {
|
var peer Peer
|
||||||
return d, 200, nil
|
res_mongo, code, err := mongo.MONGOService.LoadOne(id, wfa.GetType())
|
||||||
}, dca)
|
if err != nil {
|
||||||
|
wfa.Logger.Error().Msg("Could not retrieve " + id + " from db. Error: " + err.Error())
|
||||||
|
return nil, code, err
|
||||||
|
}
|
||||||
|
res_mongo.Decode(&peer)
|
||||||
|
|
||||||
|
return &peer, 200, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (wfa *peerMongoAccessor) LoadAll(isDraft bool) ([]utils.ShallowDBObject, int, error) {
|
func (wfa peerMongoAccessor) LoadAll() ([]utils.ShallowDBObject, int, error) {
|
||||||
return utils.GenericLoadAll[*Peer](func(d utils.DBObject) utils.ShallowDBObject {
|
objs := []utils.ShallowDBObject{}
|
||||||
return d
|
res_mongo, code, err := mongo.MONGOService.LoadAll(wfa.GetType())
|
||||||
}, isDraft, wfa)
|
if err != nil {
|
||||||
|
wfa.Logger.Error().Msg("Could not retrieve any from db. Error: " + err.Error())
|
||||||
|
return nil, code, err
|
||||||
|
}
|
||||||
|
var results []Peer
|
||||||
|
if err = res_mongo.All(mongo.MngoCtx, &results); err != nil {
|
||||||
|
return nil, 404, err
|
||||||
|
}
|
||||||
|
for _, r := range results {
|
||||||
|
objs = append(objs, &r)
|
||||||
|
}
|
||||||
|
return objs, 200, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (wfa *peerMongoAccessor) Search(filters *dbs.Filters, search string, isDraft bool) ([]utils.ShallowDBObject, int, error) {
|
func (wfa *peerMongoAccessor) Search(filters *dbs.Filters, search string) ([]utils.ShallowDBObject, int, error) {
|
||||||
return utils.GenericSearch[*Peer](filters, search, wfa.GetDefaultFilter(search),
|
objs := []utils.ShallowDBObject{}
|
||||||
func(d utils.DBObject) utils.ShallowDBObject {
|
if (filters == nil || len(filters.And) == 0 || len(filters.Or) == 0) && search != "" {
|
||||||
return d
|
s, err := strconv.Atoi(search)
|
||||||
}, isDraft, wfa)
|
if err == nil {
|
||||||
}
|
filters = &dbs.Filters{
|
||||||
func (a *peerMongoAccessor) GetDefaultFilter(search string) *dbs.Filters {
|
Or: map[string][]dbs.Filter{ // search by name if no filters are provided
|
||||||
if i, err := strconv.Atoi(search); err == nil {
|
"state": {{Operator: dbs.EQUAL.String(), Value: s}},
|
||||||
return &dbs.Filters{
|
},
|
||||||
Or: map[string][]dbs.Filter{ // search by name if no filters are provided
|
}
|
||||||
"state": {{Operator: dbs.EQUAL.String(), Value: i}},
|
} else {
|
||||||
},
|
filters = &dbs.Filters{
|
||||||
}
|
Or: map[string][]dbs.Filter{ // search by name if no filters are provided
|
||||||
} else {
|
"abstractobject.name": {{Operator: dbs.LIKE.String(), Value: search}},
|
||||||
if search == "*" {
|
"url": {{Operator: dbs.LIKE.String(), Value: search}},
|
||||||
search = ""
|
},
|
||||||
}
|
}
|
||||||
return &dbs.Filters{
|
|
||||||
Or: map[string][]dbs.Filter{ // search by name if no filters are provided
|
|
||||||
"abstractobject.name": {{Operator: dbs.LIKE.String(), Value: search}},
|
|
||||||
"url": {{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 []Peer
|
||||||
|
if err = res_mongo.All(mongo.MngoCtx, &results); err != nil {
|
||||||
|
return nil, 404, err
|
||||||
|
}
|
||||||
|
for _, r := range results {
|
||||||
|
objs = append(objs, &r)
|
||||||
|
}
|
||||||
|
return objs, 200, nil
|
||||||
}
|
}
|
||||||
|
@ -1,100 +0,0 @@
|
|||||||
package peer_test
|
|
||||||
|
|
||||||
import (
|
|
||||||
"encoding/json"
|
|
||||||
"testing"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/peer"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
"github.com/stretchr/testify/assert"
|
|
||||||
"github.com/stretchr/testify/mock"
|
|
||||||
)
|
|
||||||
|
|
||||||
// MockHTTPCaller mocks tools.HTTPCaller
|
|
||||||
type MockHTTPCaller struct {
|
|
||||||
mock.Mock
|
|
||||||
URLS map[tools.DataType]map[tools.METHOD]string
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *MockHTTPCaller) GetUrls() map[tools.DataType]map[tools.METHOD]string {
|
|
||||||
return c.URLS
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *MockHTTPCaller) CallPost(url, token string, body interface{}, types ...string) ([]byte, error) {
|
|
||||||
args := m.Called(url, token, body)
|
|
||||||
return args.Get(0).([]byte), args.Error(1)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *MockHTTPCaller) CallGet(url, token string, types ...string) ([]byte, error) {
|
|
||||||
args := m.Called(url, token)
|
|
||||||
return args.Get(0).([]byte), args.Error(1)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *MockHTTPCaller) CallDelete(url, token string) ([]byte, error) {
|
|
||||||
args := m.Called(url, token)
|
|
||||||
return args.Get(0).([]byte), args.Error(1)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLaunchPeerExecution_PeerNotReachable(t *testing.T) {
|
|
||||||
cache := &peer.PeerCache{}
|
|
||||||
caller := &MockHTTPCaller{
|
|
||||||
URLS: map[tools.DataType]map[tools.METHOD]string{
|
|
||||||
tools.PEER: {
|
|
||||||
tools.POST: "/execute/:id",
|
|
||||||
},
|
|
||||||
},
|
|
||||||
}
|
|
||||||
exec, err := cache.LaunchPeerExecution("peer-id", "data-id", tools.PEER, tools.POST, map[string]string{"a": "b"}, caller)
|
|
||||||
assert.Nil(t, exec)
|
|
||||||
assert.Error(t, err)
|
|
||||||
assert.Contains(t, err.Error(), "not reachable")
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestExecSuccess(t *testing.T) {
|
|
||||||
cache := &peer.PeerCache{}
|
|
||||||
caller := &MockHTTPCaller{}
|
|
||||||
url := "http://mockpeer/resource"
|
|
||||||
response := map[string]interface{}{"result": "ok"}
|
|
||||||
data, _ := json.Marshal(response)
|
|
||||||
|
|
||||||
caller.On("CallPost", url, "", mock.Anything).Return(data, nil)
|
|
||||||
_, err := cache.Exec(url, tools.POST, map[string]string{"key": "value"}, caller)
|
|
||||||
assert.NoError(t, err)
|
|
||||||
caller.AssertExpectations(t)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestExecReturnsErrorField(t *testing.T) {
|
|
||||||
cache := &peer.PeerCache{}
|
|
||||||
caller := &MockHTTPCaller{}
|
|
||||||
url := "http://mockpeer/resource"
|
|
||||||
response := map[string]interface{}{"error": "something failed"}
|
|
||||||
data, _ := json.Marshal(response)
|
|
||||||
|
|
||||||
caller.On("CallPost", url, "", mock.Anything).Return(data, nil)
|
|
||||||
_, err := cache.Exec(url, tools.POST, map[string]string{"key": "value"}, caller)
|
|
||||||
assert.Error(t, err)
|
|
||||||
assert.Equal(t, "something failed", err.Error())
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestExecInvalidJSON(t *testing.T) {
|
|
||||||
cache := &peer.PeerCache{}
|
|
||||||
caller := &MockHTTPCaller{}
|
|
||||||
url := "http://mockpeer/resource"
|
|
||||||
caller.On("CallPost", url, "", mock.Anything).Return([]byte("{invalid json}"), nil)
|
|
||||||
_, err := cache.Exec(url, tools.POST, map[string]string{"key": "value"}, caller)
|
|
||||||
assert.Error(t, err)
|
|
||||||
assert.Contains(t, err.Error(), "invalid character")
|
|
||||||
}
|
|
||||||
|
|
||||||
type mockAccessor struct {
|
|
||||||
loadOne func(string) (interface{}, int, error)
|
|
||||||
updateOne func(interface{}, string) error
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *mockAccessor) LoadOne(id string) (interface{}, int, error) {
|
|
||||||
return m.loadOne(id)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *mockAccessor) UpdateOne(i interface{}, id string) error {
|
|
||||||
return m.updateOne(i, id)
|
|
||||||
}
|
|
@ -1,127 +0,0 @@
|
|||||||
package peer_test
|
|
||||||
|
|
||||||
import (
|
|
||||||
"testing"
|
|
||||||
|
|
||||||
"github.com/stretchr/testify/assert"
|
|
||||||
"github.com/stretchr/testify/mock"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/dbs"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/peer"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
|
||||||
)
|
|
||||||
|
|
||||||
type MockAccessor struct {
|
|
||||||
mock.Mock
|
|
||||||
utils.AbstractAccessor
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *MockAccessor) DeleteOne(id string) (utils.DBObject, int, error) {
|
|
||||||
args := m.Called(id)
|
|
||||||
return args.Get(0).(utils.DBObject), args.Int(1), args.Error(2)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *MockAccessor) UpdateOne(set utils.DBObject, id string) (utils.DBObject, int, error) {
|
|
||||||
args := m.Called(set, id)
|
|
||||||
return args.Get(0).(utils.DBObject), args.Int(1), args.Error(2)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *MockAccessor) StoreOne(data utils.DBObject) (utils.DBObject, int, error) {
|
|
||||||
args := m.Called(data)
|
|
||||||
return args.Get(0).(utils.DBObject), args.Int(1), args.Error(2)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *MockAccessor) LoadOne(id string) (utils.DBObject, int, error) {
|
|
||||||
args := m.Called(id)
|
|
||||||
return args.Get(0).(utils.DBObject), args.Int(1), args.Error(2)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *MockAccessor) LoadAll(isDraft bool) ([]utils.ShallowDBObject, int, error) {
|
|
||||||
args := m.Called(isDraft)
|
|
||||||
return args.Get(0).([]utils.ShallowDBObject), args.Int(1), args.Error(2)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *MockAccessor) Search(filters *dbs.Filters, search string, isDraft bool) ([]utils.ShallowDBObject, int, error) {
|
|
||||||
args := m.Called(filters, search, isDraft)
|
|
||||||
return args.Get(0).([]utils.ShallowDBObject), args.Int(1), args.Error(2)
|
|
||||||
}
|
|
||||||
|
|
||||||
func newTestPeer() *peer.Peer {
|
|
||||||
return &peer.Peer{
|
|
||||||
Url: "http://localhost",
|
|
||||||
WalletAddress: "0x123",
|
|
||||||
PublicKey: "pubkey",
|
|
||||||
State: peer.SELF,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestDeleteOne_UsingMock(t *testing.T) {
|
|
||||||
mockAcc := new(MockAccessor)
|
|
||||||
mockAcc.On("DeleteOne", "id").Return(newTestPeer(), 200, nil)
|
|
||||||
|
|
||||||
obj, code, err := mockAcc.DeleteOne("id")
|
|
||||||
assert.NoError(t, err)
|
|
||||||
assert.Equal(t, 200, code)
|
|
||||||
assert.NotNil(t, obj)
|
|
||||||
mockAcc.AssertExpectations(t)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestUpdateOne_UsingMock(t *testing.T) {
|
|
||||||
mockAcc := new(MockAccessor)
|
|
||||||
peerObj := newTestPeer()
|
|
||||||
mockAcc.On("UpdateOne", peerObj, "id").Return(peerObj, 200, nil)
|
|
||||||
|
|
||||||
obj, code, err := mockAcc.UpdateOne(peerObj, "id")
|
|
||||||
assert.NoError(t, err)
|
|
||||||
assert.Equal(t, 200, code)
|
|
||||||
assert.Equal(t, peerObj, obj)
|
|
||||||
mockAcc.AssertExpectations(t)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestStoreOne_UsingMock(t *testing.T) {
|
|
||||||
mockAcc := new(MockAccessor)
|
|
||||||
peerObj := newTestPeer()
|
|
||||||
mockAcc.On("StoreOne", peerObj).Return(peerObj, 200, nil)
|
|
||||||
|
|
||||||
obj, code, err := mockAcc.StoreOne(peerObj)
|
|
||||||
assert.NoError(t, err)
|
|
||||||
assert.Equal(t, 200, code)
|
|
||||||
assert.Equal(t, peerObj, obj)
|
|
||||||
mockAcc.AssertExpectations(t)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoadOne_UsingMock(t *testing.T) {
|
|
||||||
mockAcc := new(MockAccessor)
|
|
||||||
mockAcc.On("LoadOne", "test-id").Return(newTestPeer(), 200, nil)
|
|
||||||
|
|
||||||
obj, code, err := mockAcc.LoadOne("test-id")
|
|
||||||
assert.NoError(t, err)
|
|
||||||
assert.Equal(t, 200, code)
|
|
||||||
assert.NotNil(t, obj)
|
|
||||||
mockAcc.AssertExpectations(t)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoadAll_UsingMock(t *testing.T) {
|
|
||||||
mockAcc := new(MockAccessor)
|
|
||||||
expected := []utils.ShallowDBObject{newTestPeer()}
|
|
||||||
mockAcc.On("LoadAll", false).Return(expected, 200, nil)
|
|
||||||
|
|
||||||
objs, code, err := mockAcc.LoadAll(false)
|
|
||||||
assert.NoError(t, err)
|
|
||||||
assert.Equal(t, 200, code)
|
|
||||||
assert.Equal(t, expected, objs)
|
|
||||||
mockAcc.AssertExpectations(t)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestSearch_UsingMock(t *testing.T) {
|
|
||||||
mockAcc := new(MockAccessor)
|
|
||||||
filters := &dbs.Filters{}
|
|
||||||
expected := []utils.ShallowDBObject{newTestPeer()}
|
|
||||||
mockAcc.On("Search", filters, "test", false).Return(expected, 200, nil)
|
|
||||||
|
|
||||||
objs, code, err := mockAcc.Search(filters, "test", false)
|
|
||||||
assert.NoError(t, err)
|
|
||||||
assert.Equal(t, 200, code)
|
|
||||||
assert.Equal(t, expected, objs)
|
|
||||||
mockAcc.AssertExpectations(t)
|
|
||||||
}
|
|
125
models/resource_model/resource_model.go
Normal file
125
models/resource_model/resource_model.go
Normal file
@ -0,0 +1,125 @@
|
|||||||
|
package resource_model
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/utils"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/tools"
|
||||||
|
"github.com/google/uuid"
|
||||||
|
)
|
||||||
|
|
||||||
|
type WebResource struct {
|
||||||
|
Protocol string `bson:"protocol,omitempty" json:"protocol,omitempty"` // Protocol is the protocol of the URL
|
||||||
|
Path string `bson:"path,omitempty" json:"path,omitempty"` // Path is the path of the URL
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* AbstractResource is a struct that represents a resource
|
||||||
|
* it defines the resource data
|
||||||
|
*/
|
||||||
|
type AbstractResource struct {
|
||||||
|
utils.AbstractObject // AbstractObject contains the basic fields of an object (id, name)
|
||||||
|
ShortDescription string `json:"short_description,omitempty" bson:"short_description,omitempty" validate:"required"` // ShortDescription is the short description of the resource
|
||||||
|
Description string `json:"description,omitempty" bson:"description,omitempty"` // Description is the description of the resource
|
||||||
|
Logo string `json:"logo,omitempty" bson:"logo,omitempty" validate:"required"` // Logo is the logo of the resource
|
||||||
|
Owner string `json:"owner,omitempty" bson:"owner,omitempty" validate:"required"` // Owner is the owner of the resource
|
||||||
|
OwnerLogo string `json:"owner_logo,omitempty" bson:"owner_logo,omitempty"` // OwnerLogo is the owner logo of the resource
|
||||||
|
SourceUrl string `json:"source_url,omitempty" bson:"source_url,omitempty" validate:"required"` // SourceUrl is the source URL of the resource
|
||||||
|
PeerID string `json:"peer_id,omitempty" bson:"peer_id,omitempty"` // PeerID is the ID of the peer getting this resource
|
||||||
|
Price string `json:"price,omitempty" bson:"price,omitempty"` // Price is the price of access to the resource
|
||||||
|
License string `json:"license,omitempty" bson:"license,omitempty"` // License is the license of the resource
|
||||||
|
ResourceModel *ResourceModel `json:"resource_model,omitempty" bson:"resource_model,omitempty"` // ResourceModel is the model of the resource
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* GetModelType returns the type of the model key
|
||||||
|
*/
|
||||||
|
func (abs *AbstractResource) GetModelType(cat string, key string) interface{} {
|
||||||
|
if abs.ResourceModel == nil || abs.ResourceModel.Model == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
if _, ok := abs.ResourceModel.Model[key]; !ok {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return abs.ResourceModel.Model[cat][key].Type
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* GetModelKeys returns the keys of the model
|
||||||
|
*/
|
||||||
|
func (abs *AbstractResource) GetModelKeys() []string {
|
||||||
|
keys := make([]string, 0)
|
||||||
|
for k := range abs.ResourceModel.Model {
|
||||||
|
keys = append(keys, k)
|
||||||
|
}
|
||||||
|
return keys
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* GetModelReadOnly returns the readonly of the model key
|
||||||
|
*/
|
||||||
|
func (abs *AbstractResource) GetModelReadOnly(cat string, key string) interface{} {
|
||||||
|
if abs.ResourceModel == nil || abs.ResourceModel.Model == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
if _, ok := abs.ResourceModel.Model[key]; !ok {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return abs.ResourceModel.Model[cat][key].ReadOnly
|
||||||
|
}
|
||||||
|
|
||||||
|
type Model struct {
|
||||||
|
Type string `json:"type,omitempty" bson:"type,omitempty"` // Type is the type of the model
|
||||||
|
ReadOnly bool `json:"readonly,omitempty" bson:"readonly,omitempty"` // ReadOnly is the readonly of the model
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* ResourceModel is a struct that represents a resource model
|
||||||
|
* it defines the resource metadata and specificity
|
||||||
|
* Warning: This struct is not user available, it is only used by the system
|
||||||
|
*/
|
||||||
|
type ResourceModel struct {
|
||||||
|
UUID string `json:"id,omitempty" bson:"id,omitempty" validate:"required"`
|
||||||
|
ResourceType string `json:"resource_type,omitempty" bson:"resource_type,omitempty" validate:"required"`
|
||||||
|
VarRefs map[string]string `json:"var_refs,omitempty" bson:"var_refs,omitempty"` // VarRefs is the variable references of the model
|
||||||
|
Model map[string]map[string]Model `json:"model,omitempty" bson:"model,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ao *ResourceModel) GetID() string {
|
||||||
|
return ao.UUID
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ao *ResourceModel) UpToDate() {}
|
||||||
|
|
||||||
|
func (r *ResourceModel) GenerateID() {
|
||||||
|
r.UUID = uuid.New().String()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *ResourceModel) GetName() string {
|
||||||
|
return d.UUID
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *ResourceModel) GetAccessor(caller *tools.HTTPCaller) utils.Accessor {
|
||||||
|
data := &ResourceModelMongoAccessor{}
|
||||||
|
data.Init(tools.RESOURCE_MODEL, caller)
|
||||||
|
return data
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dma *ResourceModel) Deserialize(j map[string]interface{}) utils.DBObject {
|
||||||
|
b, err := json.Marshal(j)
|
||||||
|
if err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
json.Unmarshal(b, dma)
|
||||||
|
return dma
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dma *ResourceModel) Serialize() map[string]interface{} {
|
||||||
|
var m map[string]interface{}
|
||||||
|
b, err := json.Marshal(dma)
|
||||||
|
if err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
json.Unmarshal(b, &m)
|
||||||
|
return m
|
||||||
|
}
|
83
models/resource_model/resource_model_mongo_accessor.go
Normal file
83
models/resource_model/resource_model_mongo_accessor.go
Normal file
@ -0,0 +1,83 @@
|
|||||||
|
package resource_model
|
||||||
|
|
||||||
|
import (
|
||||||
|
"cloud.o-forge.io/core/oc-lib/dbs"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/dbs/mongo"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/utils"
|
||||||
|
)
|
||||||
|
|
||||||
|
type ResourceModelMongoAccessor struct {
|
||||||
|
utils.AbstractAccessor // AbstractAccessor contains the basic fields of an accessor (model, caller)
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Nothing special here, just the basic CRUD operations
|
||||||
|
*/
|
||||||
|
|
||||||
|
func (wfa *ResourceModelMongoAccessor) DeleteOne(id string) (utils.DBObject, int, error) {
|
||||||
|
return wfa.GenericDeleteOne(id, wfa)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (wfa *ResourceModelMongoAccessor) UpdateOne(set utils.DBObject, id string) (utils.DBObject, int, error) {
|
||||||
|
return wfa.GenericUpdateOne(set, id, wfa, &ResourceModel{})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (wfa *ResourceModelMongoAccessor) StoreOne(data utils.DBObject) (utils.DBObject, int, error) {
|
||||||
|
return wfa.GenericStoreOne(data, wfa)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (wfa *ResourceModelMongoAccessor) CopyOne(data utils.DBObject) (utils.DBObject, int, error) {
|
||||||
|
return wfa.GenericStoreOne(data, wfa)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (wfa *ResourceModelMongoAccessor) LoadOne(id string) (utils.DBObject, int, error) {
|
||||||
|
var workflow ResourceModel
|
||||||
|
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)
|
||||||
|
return &workflow, 200, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (wfa ResourceModelMongoAccessor) 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 []ResourceModel
|
||||||
|
if err = res_mongo.All(mongo.MngoCtx, &results); err != nil {
|
||||||
|
return nil, 404, err
|
||||||
|
}
|
||||||
|
for _, r := range results {
|
||||||
|
objs = append(objs, &r)
|
||||||
|
}
|
||||||
|
return objs, 200, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (wfa *ResourceModelMongoAccessor) 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{
|
||||||
|
"resource_type": {{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 []ResourceModel
|
||||||
|
if err = res_mongo.All(mongo.MngoCtx, &results); err != nil {
|
||||||
|
return nil, 404, err
|
||||||
|
}
|
||||||
|
for _, r := range results {
|
||||||
|
objs = append(objs, &r)
|
||||||
|
}
|
||||||
|
return objs, 200, nil
|
||||||
|
}
|
@ -1,203 +0,0 @@
|
|||||||
package resources
|
|
||||||
|
|
||||||
import (
|
|
||||||
"errors"
|
|
||||||
"strings"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/enum"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/models"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/pricing"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
)
|
|
||||||
|
|
||||||
/*
|
|
||||||
* ComputeResource is a struct that represents a compute resource
|
|
||||||
* it defines the resource compute
|
|
||||||
*/
|
|
||||||
type ComputeResource struct {
|
|
||||||
AbstractInstanciatedResource[*ComputeResourceInstance]
|
|
||||||
Architecture string `json:"architecture,omitempty" bson:"architecture,omitempty"` // Architecture is the architecture
|
|
||||||
Infrastructure enum.InfrastructureType `json:"infrastructure" bson:"infrastructure" default:"-1"` // Infrastructure is the infrastructure
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *ComputeResource) GetAccessor(request *tools.APIRequest) utils.Accessor {
|
|
||||||
return NewAccessor[*ComputeResource](tools.COMPUTE_RESOURCE, request, func() utils.DBObject { return &ComputeResource{} })
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *ComputeResource) GetType() string {
|
|
||||||
return tools.COMPUTE_RESOURCE.String()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (abs *ComputeResource) ConvertToPricedResource(t tools.DataType, request *tools.APIRequest) pricing.PricedItemITF {
|
|
||||||
if t != tools.COMPUTE_RESOURCE {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
p := abs.AbstractInstanciatedResource.ConvertToPricedResource(t, request)
|
|
||||||
priced := p.(*PricedResource)
|
|
||||||
return &PricedComputeResource{
|
|
||||||
PricedResource: *priced,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
type ComputeNode struct {
|
|
||||||
Name string `json:"name,omitempty" bson:"name,omitempty"`
|
|
||||||
Quantity int64 `json:"quantity" bson:"quantity" default:"1"`
|
|
||||||
RAM *models.RAM `bson:"ram,omitempty" json:"ram,omitempty"` // RAM is the RAM
|
|
||||||
CPUs map[string]int64 `bson:"cpus,omitempty" json:"cpus,omitempty"` // CPUs is the list of CPUs key is model
|
|
||||||
GPUs map[string]int64 `bson:"gpus,omitempty" json:"gpus,omitempty"` // GPUs is the list of GPUs key is model
|
|
||||||
}
|
|
||||||
|
|
||||||
type ComputeResourceInstance struct {
|
|
||||||
ResourceInstance[*ComputeResourcePartnership]
|
|
||||||
Source string `json:"source,omitempty" bson:"source,omitempty"` // Source is the source of the resource
|
|
||||||
SecurityLevel string `json:"security_level,omitempty" bson:"security_level,omitempty"`
|
|
||||||
PowerSources []string `json:"power_sources,omitempty" bson:"power_sources,omitempty"`
|
|
||||||
AnnualCO2Emissions float64 `json:"annual_co2_emissions,omitempty" bson:"co2_emissions,omitempty"`
|
|
||||||
CPUs map[string]*models.CPU `bson:"cpus,omitempty" json:"cpus,omitempty"` // CPUs is the list of CPUs key is model
|
|
||||||
GPUs map[string]*models.GPU `bson:"gpus,omitempty" json:"gpus,omitempty"` // GPUs is the list of GPUs key is model
|
|
||||||
Nodes []*ComputeNode `json:"nodes,omitempty" bson:"nodes,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type ComputeResourcePartnership struct {
|
|
||||||
ResourcePartnerShip[*ComputeResourcePricingProfile]
|
|
||||||
MinGaranteedCPUsCores map[string]float64 `json:"garanteed_cpus,omitempty" bson:"garanteed_cpus,omitempty"`
|
|
||||||
MinGaranteedGPUsMemoryGB map[string]float64 `json:"garanteed_gpus,omitempty" bson:"garanteed_gpus,omitempty"`
|
|
||||||
MinGaranteedRAMSize float64 `json:"garanteed_ram,omitempty" bson:"garanteed_ram,omitempty"`
|
|
||||||
|
|
||||||
MaxAllowedCPUsCores map[string]float64 `json:"allowed_cpus,omitempty" bson:"allowed_cpus,omitempty"`
|
|
||||||
MaxAllowedGPUsMemoryGB map[string]float64 `json:"allowed_gpus,omitempty" bson:"allowed_gpus,omitempty"`
|
|
||||||
MaxAllowedRAMSize float64 `json:"allowed_ram,omitempty" bson:"allowed_ram,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type ComputeResourcePricingProfile struct {
|
|
||||||
pricing.ExploitPricingProfile[pricing.TimePricingStrategy]
|
|
||||||
// ExploitPricingProfile is the pricing profile of a compute it means that we exploit the resource for an amount of continuous time
|
|
||||||
CPUsPrices map[string]float64 `json:"cpus_prices,omitempty" bson:"cpus_prices,omitempty"` // CPUsPrices is the prices of the CPUs
|
|
||||||
GPUsPrices map[string]float64 `json:"gpus_prices,omitempty" bson:"gpus_prices,omitempty"` // GPUsPrices is the prices of the GPUs
|
|
||||||
RAMPrice float64 `json:"ram_price" bson:"ram_price" default:"-1"` // RAMPrice is the price of the RAM
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *ComputeResourcePricingProfile) IsPurchasable() bool {
|
|
||||||
return p.Pricing.BuyingStrategy != pricing.UNDEFINED_SUBSCRIPTION
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *ComputeResourcePricingProfile) GetPurchase() pricing.BuyingStrategy {
|
|
||||||
return p.Pricing.BuyingStrategy
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *ComputeResourcePricingProfile) IsBooked() bool {
|
|
||||||
if p.Pricing.BuyingStrategy == pricing.PERMANENT {
|
|
||||||
p.Pricing.BuyingStrategy = pricing.SUBSCRIPTION
|
|
||||||
}
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *ComputeResourcePricingProfile) GetOverrideStrategyValue() int {
|
|
||||||
return -1
|
|
||||||
}
|
|
||||||
|
|
||||||
// NOT A PROPER QUANTITY
|
|
||||||
// amountOfData is the number of CPUs, GPUs or RAM dependings on the params
|
|
||||||
func (p *ComputeResourcePricingProfile) GetPrice(amountOfData float64, explicitDuration float64, start time.Time, end time.Time, params ...string) (float64, error) {
|
|
||||||
if len(params) < 1 {
|
|
||||||
return 0, errors.New("params must be set")
|
|
||||||
}
|
|
||||||
pp := float64(0)
|
|
||||||
model := params[1]
|
|
||||||
if strings.Contains(params[0], "cpus") && len(params) > 1 {
|
|
||||||
if _, ok := p.CPUsPrices[model]; ok {
|
|
||||||
p.Pricing.Price = p.CPUsPrices[model]
|
|
||||||
}
|
|
||||||
r, err := p.Pricing.GetPrice(amountOfData, explicitDuration, start, &end)
|
|
||||||
if err != nil {
|
|
||||||
return 0, err
|
|
||||||
}
|
|
||||||
pp += r
|
|
||||||
|
|
||||||
}
|
|
||||||
if strings.Contains(params[0], "gpus") && len(params) > 1 {
|
|
||||||
if _, ok := p.GPUsPrices[model]; ok {
|
|
||||||
p.Pricing.Price = p.GPUsPrices[model]
|
|
||||||
}
|
|
||||||
r, err := p.Pricing.GetPrice(amountOfData, explicitDuration, start, &end)
|
|
||||||
if err != nil {
|
|
||||||
return 0, err
|
|
||||||
}
|
|
||||||
pp += r
|
|
||||||
}
|
|
||||||
if strings.Contains(params[0], "ram") {
|
|
||||||
if p.RAMPrice >= 0 {
|
|
||||||
p.Pricing.Price = p.RAMPrice
|
|
||||||
}
|
|
||||||
r, err := p.Pricing.GetPrice(float64(amountOfData), explicitDuration, start, &end)
|
|
||||||
if err != nil {
|
|
||||||
return 0, err
|
|
||||||
}
|
|
||||||
pp += r
|
|
||||||
}
|
|
||||||
return pp, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
type PricedComputeResource struct {
|
|
||||||
PricedResource
|
|
||||||
|
|
||||||
CPUsLocated map[string]float64 `json:"cpus_in_use" bson:"cpus_in_use"` // CPUsInUse is the list of CPUs in use
|
|
||||||
GPUsLocated map[string]float64 `json:"gpus_in_use" bson:"gpus_in_use"` // GPUsInUse is the list of GPUs in use
|
|
||||||
RAMLocated float64 `json:"ram_in_use" bson:"ram_in_use"` // RAMInUse is the RAM in use
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *PricedComputeResource) GetType() tools.DataType {
|
|
||||||
return tools.COMPUTE_RESOURCE
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *PricedComputeResource) GetPrice() (float64, error) {
|
|
||||||
now := time.Now()
|
|
||||||
if r.UsageStart == nil {
|
|
||||||
r.UsageStart = &now
|
|
||||||
}
|
|
||||||
if r.UsageEnd == nil {
|
|
||||||
add := r.UsageStart.Add(time.Duration(1 * time.Hour))
|
|
||||||
r.UsageEnd = &add
|
|
||||||
}
|
|
||||||
if r.SelectedPricing == nil {
|
|
||||||
return 0, errors.New("pricing profile must be set on Priced Compute" + r.ResourceID)
|
|
||||||
}
|
|
||||||
pricing := r.SelectedPricing
|
|
||||||
price := float64(0)
|
|
||||||
for _, l := range []map[string]float64{r.CPUsLocated, r.GPUsLocated} {
|
|
||||||
for model, amountOfData := range l {
|
|
||||||
cpus, err := pricing.GetPrice(float64(amountOfData), r.ExplicitBookingDurationS, *r.UsageStart, *r.UsageEnd, "cpus", model)
|
|
||||||
if err != nil {
|
|
||||||
return 0, err
|
|
||||||
}
|
|
||||||
price += cpus
|
|
||||||
}
|
|
||||||
}
|
|
||||||
ram, err := pricing.GetPrice(r.RAMLocated, r.ExplicitBookingDurationS, *r.UsageStart, *r.UsageEnd, "ram")
|
|
||||||
if err != nil {
|
|
||||||
return 0, err
|
|
||||||
}
|
|
||||||
price += ram
|
|
||||||
return price, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
|
||||||
* FillWithDefaultProcessingUsage fills the order item with the default processing usage
|
|
||||||
* it depends on the processing usage only if nothing is set, during order
|
|
||||||
*/
|
|
||||||
func (i *PricedComputeResource) FillWithDefaultProcessingUsage(usage *ProcessingUsage) {
|
|
||||||
for _, cpu := range usage.CPUs {
|
|
||||||
if _, ok := i.CPUsLocated[cpu.Model]; !ok {
|
|
||||||
i.CPUsLocated[cpu.Model] = 0
|
|
||||||
}
|
|
||||||
if i.CPUsLocated[cpu.Model] < float64(cpu.Cores) {
|
|
||||||
i.CPUsLocated[cpu.Model] = float64(cpu.Cores)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
for _, cpu := range usage.GPUs {
|
|
||||||
i.GPUsLocated[cpu.Model] = 1
|
|
||||||
}
|
|
||||||
i.RAMLocated = usage.RAM.SizeGb
|
|
||||||
}
|
|
101
models/resources/compute/compute.go
Normal file
101
models/resources/compute/compute.go
Normal file
@ -0,0 +1,101 @@
|
|||||||
|
package compute
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/resource_model"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/utils"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/tools"
|
||||||
|
)
|
||||||
|
|
||||||
|
type TechnologyEnum int
|
||||||
|
|
||||||
|
const (
|
||||||
|
DOCKER TechnologyEnum = iota
|
||||||
|
KUBERNETES
|
||||||
|
SLURM
|
||||||
|
HW
|
||||||
|
CONDOR
|
||||||
|
)
|
||||||
|
|
||||||
|
func (t TechnologyEnum) String() string {
|
||||||
|
return [...]string{"DOCKER", "KUBERNETES", "SLURM", "HW", "CONDOR"}[t]
|
||||||
|
}
|
||||||
|
|
||||||
|
type AccessEnum int
|
||||||
|
|
||||||
|
const (
|
||||||
|
SSH AccessEnum = iota
|
||||||
|
SSH_KUBE_API
|
||||||
|
SSH_SLURM
|
||||||
|
SSH_DOCKER
|
||||||
|
OPENCLOUD
|
||||||
|
VPN
|
||||||
|
)
|
||||||
|
|
||||||
|
func (a AccessEnum) String() string {
|
||||||
|
return [...]string{"SSH", "SSH_KUBE_API", "SSH_SLURM", "SSH_DOCKER", "OPENCLOUD", "VPN"}[a]
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* ComputeResource is a struct that represents a compute resource
|
||||||
|
* it defines the resource compute
|
||||||
|
*/
|
||||||
|
type ComputeResource struct {
|
||||||
|
resource_model.AbstractResource
|
||||||
|
Technology TechnologyEnum `json:"technology" bson:"technology" default:"0"` // Technology is the technology
|
||||||
|
Architecture string `json:"architecture,omitempty" bson:"architecture,omitempty"` // Architecture is the architecture
|
||||||
|
Access AccessEnum `json:"access" bson:"access default:"0"` // Access is the access
|
||||||
|
|
||||||
|
Localisation string `json:"localisation,omitempty" bson:"localisation,omitempty"` // Localisation is the localisation
|
||||||
|
|
||||||
|
CPUs []*CPU `bson:"cpus,omitempty" json:"cpus,omitempty"` // CPUs is the list of CPUs
|
||||||
|
RAM *RAM `bson:"ram,omitempty" json:"ram,omitempty"` // RAM is the RAM
|
||||||
|
GPUs []*GPU `bson:"gpus,omitempty" json:"gpus,omitempty"` // GPUs is the list of GPUs
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dma *ComputeResource) Deserialize(j map[string]interface{}) utils.DBObject {
|
||||||
|
b, err := json.Marshal(j)
|
||||||
|
if err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
json.Unmarshal(b, dma)
|
||||||
|
return dma
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dma *ComputeResource) Serialize() map[string]interface{} {
|
||||||
|
var m map[string]interface{}
|
||||||
|
b, err := json.Marshal(dma)
|
||||||
|
if err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
json.Unmarshal(b, &m)
|
||||||
|
return m
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *ComputeResource) GetAccessor(caller *tools.HTTPCaller) utils.Accessor {
|
||||||
|
data := New()
|
||||||
|
data.Init(tools.COMPUTE_RESOURCE, caller)
|
||||||
|
return data
|
||||||
|
}
|
||||||
|
|
||||||
|
// CPU is a struct that represents a CPU
|
||||||
|
type CPU struct {
|
||||||
|
Cores uint `bson:"cores,omitempty" json:"cores,omitempty"` //TODO: validate
|
||||||
|
Architecture string `bson:"architecture,omitempty" json:"architecture,omitempty"` //TOOD: enum
|
||||||
|
Shared bool `bson:"shared,omitempty" json:"shared,omitempty"`
|
||||||
|
MinimumMemory uint `bson:"minimum_memory,omitempty" json:"minimum_memory,omitempty"`
|
||||||
|
Platform string `bson:"platform,omitempty" json:"platform,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type RAM struct {
|
||||||
|
Size uint `bson:"size,omitempty" json:"size,omitempty" description:"Units in MB"`
|
||||||
|
Ecc bool `bson:"ecc,omitempty" json:"ecc,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type GPU struct {
|
||||||
|
CudaCores uint `bson:"cuda_cores,omitempty" json:"cuda_cores,omitempty"`
|
||||||
|
Model string `bson:"model,omitempty" json:"model,omitempty"`
|
||||||
|
Memory uint `bson:"memory,omitempty" json:"memory,omitempty" description:"Units in MB"`
|
||||||
|
TensorCores uint `bson:"tensor_cores,omitempty" json:"tensor_cores,omitempty"`
|
||||||
|
}
|
112
models/resources/compute/compute_mongo_accessor.go
Normal file
112
models/resources/compute/compute_mongo_accessor.go
Normal file
@ -0,0 +1,112 @@
|
|||||||
|
package compute
|
||||||
|
|
||||||
|
import (
|
||||||
|
"cloud.o-forge.io/core/oc-lib/dbs"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/dbs/mongo"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/resource_model"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/utils"
|
||||||
|
)
|
||||||
|
|
||||||
|
type computeMongoAccessor struct {
|
||||||
|
utils.AbstractAccessor // AbstractAccessor contains the basic fields of an accessor (model, caller)
|
||||||
|
}
|
||||||
|
|
||||||
|
// New creates a new instance of the computeMongoAccessor
|
||||||
|
func New() *computeMongoAccessor {
|
||||||
|
return &computeMongoAccessor{}
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Nothing special here, just the basic CRUD operations
|
||||||
|
*/
|
||||||
|
|
||||||
|
func (dca *computeMongoAccessor) DeleteOne(id string) (utils.DBObject, int, error) {
|
||||||
|
return dca.GenericDeleteOne(id, dca)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dca *computeMongoAccessor) UpdateOne(set utils.DBObject, id string) (utils.DBObject, int, error) {
|
||||||
|
set.(*ComputeResource).ResourceModel = nil
|
||||||
|
return dca.GenericUpdateOne(set, id, dca, &ComputeResource{})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dca *computeMongoAccessor) StoreOne(data utils.DBObject) (utils.DBObject, int, error) {
|
||||||
|
data.(*ComputeResource).ResourceModel = nil
|
||||||
|
return dca.GenericStoreOne(data, dca)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dca *computeMongoAccessor) CopyOne(data utils.DBObject) (utils.DBObject, int, error) {
|
||||||
|
return dca.GenericStoreOne(data, dca)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dca *computeMongoAccessor) LoadOne(id string) (utils.DBObject, int, error) {
|
||||||
|
var compute ComputeResource
|
||||||
|
|
||||||
|
res_mongo, code, err := mongo.MONGOService.LoadOne(id, dca.GetType())
|
||||||
|
if err != nil {
|
||||||
|
dca.Logger.Error().Msg("Could not retrieve " + id + " from db. Error: " + err.Error())
|
||||||
|
return nil, code, err
|
||||||
|
}
|
||||||
|
|
||||||
|
res_mongo.Decode(&compute)
|
||||||
|
accessor := (&resource_model.ResourceModel{}).GetAccessor(nil)
|
||||||
|
resources, _, err := accessor.Search(nil, dca.GetType())
|
||||||
|
if err == nil && len(resources) > 0 {
|
||||||
|
compute.ResourceModel = resources[0].(*resource_model.ResourceModel)
|
||||||
|
}
|
||||||
|
return &compute, 200, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (wfa computeMongoAccessor) 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 []ComputeResource
|
||||||
|
if err = res_mongo.All(mongo.MngoCtx, &results); err != nil {
|
||||||
|
return nil, 404, err
|
||||||
|
}
|
||||||
|
accessor := (&resource_model.ResourceModel{}).GetAccessor(nil)
|
||||||
|
resources, _, err := accessor.Search(nil, wfa.GetType())
|
||||||
|
for _, r := range results {
|
||||||
|
if err == nil && len(resources) > 0 {
|
||||||
|
r.ResourceModel = resources[0].(*resource_model.ResourceModel)
|
||||||
|
}
|
||||||
|
objs = append(objs, &r) // only get the abstract resource !
|
||||||
|
}
|
||||||
|
return objs, 200, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (wfa *computeMongoAccessor) 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 like name, short_description, description, owner, url if no filters are provided
|
||||||
|
"abstractresource.abstractobject.name": {{Operator: dbs.LIKE.String(), Value: search}},
|
||||||
|
"abstractresource.short_description": {{Operator: dbs.LIKE.String(), Value: search}},
|
||||||
|
"abstractresource.description": {{Operator: dbs.LIKE.String(), Value: search}},
|
||||||
|
"abstractresource.owner": {{Operator: dbs.LIKE.String(), Value: search}},
|
||||||
|
"abstractresource.source_url": {{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 []ComputeResource
|
||||||
|
if err = res_mongo.All(mongo.MngoCtx, &results); err != nil {
|
||||||
|
return nil, 404, err
|
||||||
|
}
|
||||||
|
accessor := (&resource_model.ResourceModel{}).GetAccessor(nil)
|
||||||
|
resources, _, err := accessor.Search(nil, wfa.GetType())
|
||||||
|
for _, r := range results {
|
||||||
|
if err == nil && len(resources) > 0 {
|
||||||
|
r.ResourceModel = resources[0].(*resource_model.ResourceModel)
|
||||||
|
}
|
||||||
|
objs = append(objs, &r) // only get the abstract resource !
|
||||||
|
}
|
||||||
|
return objs, 200, nil
|
||||||
|
}
|
46
models/resources/compute/compute_test.go
Normal file
46
models/resources/compute/compute_test.go
Normal file
@ -0,0 +1,46 @@
|
|||||||
|
package compute
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/resource_model"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/utils"
|
||||||
|
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestStoreOneCompute(t *testing.T) {
|
||||||
|
dc := ComputeResource{
|
||||||
|
AbstractResource: resource_model.AbstractResource{
|
||||||
|
AbstractObject: utils.AbstractObject{Name: "testCompute"},
|
||||||
|
Description: "Lorem Ipsum",
|
||||||
|
Logo: "azerty.com",
|
||||||
|
Owner: "toto",
|
||||||
|
OwnerLogo: "totoLogo",
|
||||||
|
SourceUrl: "azerty.fr",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
dcma := New()
|
||||||
|
id, _, _ := dcma.StoreOne(&dc)
|
||||||
|
|
||||||
|
assert.NotEmpty(t, id)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestLoadOneCompute(t *testing.T) {
|
||||||
|
dc := ComputeResource{
|
||||||
|
AbstractResource: resource_model.AbstractResource{
|
||||||
|
AbstractObject: utils.AbstractObject{Name: "testCompute"},
|
||||||
|
Description: "Lorem Ipsum",
|
||||||
|
Logo: "azerty.com",
|
||||||
|
Owner: "toto",
|
||||||
|
OwnerLogo: "totoLogo",
|
||||||
|
SourceUrl: "azerty.fr",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
dcma := New()
|
||||||
|
new_dc, _, _ := dcma.StoreOne(&dc)
|
||||||
|
|
||||||
|
assert.Equal(t, dc, new_dc)
|
||||||
|
}
|
@ -1,184 +0,0 @@
|
|||||||
package resources
|
|
||||||
|
|
||||||
import (
|
|
||||||
"errors"
|
|
||||||
"fmt"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/models"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/pricing"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
)
|
|
||||||
|
|
||||||
/*
|
|
||||||
* DataResource is a struct that represents a data resource
|
|
||||||
* it defines the resource data
|
|
||||||
*/
|
|
||||||
type DataResource struct {
|
|
||||||
AbstractInstanciatedResource[*DataInstance]
|
|
||||||
Type string `bson:"type,omitempty" json:"type,omitempty"`
|
|
||||||
Quality string `bson:"quality,omitempty" json:"quality,omitempty"`
|
|
||||||
OpenData bool `bson:"open_data" json:"open_data" default:"false"` // Type is the type of the storage
|
|
||||||
Static bool `bson:"static" json:"static" default:"false"`
|
|
||||||
UpdatePeriod *time.Time `bson:"update_period,omitempty" json:"update_period,omitempty"`
|
|
||||||
PersonalData bool `bson:"personal_data,omitempty" json:"personal_data,omitempty"`
|
|
||||||
AnonymizedPersonalData bool `bson:"anonymized_personal_data,omitempty" json:"anonymized_personal_data,omitempty"`
|
|
||||||
SizeGB float64 `json:"size,omitempty" bson:"size,omitempty"` // SizeGB is the size of the data License DataLicense `json:"license" bson:"license" description:"license of the data" default:"0"` // License is the license of the data
|
|
||||||
// ? Interest DataLicense `json:"interest" bson:"interest" description:"interest of the data" default:"0"` // Interest is the interest of the data
|
|
||||||
Example string `json:"example,omitempty" bson:"example,omitempty" description:"base64 encoded data"` // Example is an example of the data
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *DataResource) GetAccessor(request *tools.APIRequest) utils.Accessor {
|
|
||||||
return NewAccessor[*DataResource](tools.DATA_RESOURCE, request, func() utils.DBObject { return &DataResource{} }) // Create a new instance of the accessor
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *DataResource) GetType() string {
|
|
||||||
return tools.DATA_RESOURCE.String()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (abs *DataResource) ConvertToPricedResource(t tools.DataType, request *tools.APIRequest) pricing.PricedItemITF {
|
|
||||||
if t != tools.DATA_RESOURCE {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
p := abs.AbstractInstanciatedResource.ConvertToPricedResource(t, request)
|
|
||||||
priced := p.(*PricedResource)
|
|
||||||
return &PricedDataResource{
|
|
||||||
PricedResource: *priced,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
type DataInstance struct {
|
|
||||||
ResourceInstance[*DataResourcePartnership]
|
|
||||||
Source string `json:"source,omitempty" bson:"source,omitempty"` // Source is the source of the data
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ri *DataInstance) StoreDraftDefault() {
|
|
||||||
found := false
|
|
||||||
for _, p := range ri.ResourceInstance.Env {
|
|
||||||
if p.Attr == "source" {
|
|
||||||
found = true
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if !found {
|
|
||||||
ri.ResourceInstance.Env = append(ri.ResourceInstance.Env, models.Param{
|
|
||||||
Attr: "source",
|
|
||||||
Value: ri.Source,
|
|
||||||
Readonly: true,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
ri.ResourceInstance.StoreDraftDefault()
|
|
||||||
}
|
|
||||||
|
|
||||||
type DataResourcePartnership struct {
|
|
||||||
ResourcePartnerShip[*DataResourcePricingProfile]
|
|
||||||
MaxDownloadableGbAllowed float64 `json:"allowed_gb,omitempty" bson:"allowed_gb,omitempty"`
|
|
||||||
PersonalDataAllowed bool `json:"personal_data_allowed,omitempty" bson:"personal_data_allowed,omitempty"`
|
|
||||||
AnonymizedPersonalDataAllowed bool `json:"anonymized_personal_data_allowed,omitempty" bson:"anonymized_personal_data_allowed,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type DataResourcePricingStrategy int
|
|
||||||
|
|
||||||
const (
|
|
||||||
PER_DOWNLOAD DataResourcePricingStrategy = iota + 6
|
|
||||||
PER_TB_DOWNLOADED
|
|
||||||
PER_GB_DOWNLOADED
|
|
||||||
PER_MB_DOWNLOADED
|
|
||||||
PER_KB_DOWNLOADED
|
|
||||||
)
|
|
||||||
|
|
||||||
func (t DataResourcePricingStrategy) String() string {
|
|
||||||
return [...]string{"PER DOWNLOAD", "PER TB DOWNLOADED", "PER GB DOWNLOADED", "PER MB DOWNLOADED", "PER KB DOWNLOADED"}[t]
|
|
||||||
}
|
|
||||||
|
|
||||||
func DataResourcePricingStrategyList() []DataResourcePricingStrategy {
|
|
||||||
return []DataResourcePricingStrategy{PER_DOWNLOAD, PER_TB_DOWNLOADED, PER_GB_DOWNLOADED, PER_MB_DOWNLOADED, PER_KB_DOWNLOADED}
|
|
||||||
}
|
|
||||||
|
|
||||||
func ToDataResourcePricingStrategy(i int) DataResourcePricingStrategy {
|
|
||||||
return DataResourcePricingStrategy(i)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (t DataResourcePricingStrategy) GetStrategy() string {
|
|
||||||
return [...]string{"PER_DOWNLOAD", "PER_GB", "PER_MB", "PER_KB"}[t]
|
|
||||||
}
|
|
||||||
|
|
||||||
func (t DataResourcePricingStrategy) GetStrategyValue() int {
|
|
||||||
return int(t)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (t DataResourcePricingStrategy) GetQuantity(amountOfDataGB float64) (float64, error) {
|
|
||||||
switch t {
|
|
||||||
case PER_DOWNLOAD:
|
|
||||||
return 1, nil
|
|
||||||
case PER_TB_DOWNLOADED:
|
|
||||||
return amountOfDataGB * 1000, nil
|
|
||||||
case PER_GB_DOWNLOADED:
|
|
||||||
return amountOfDataGB, nil
|
|
||||||
case PER_MB_DOWNLOADED:
|
|
||||||
return amountOfDataGB / 1000, nil
|
|
||||||
case PER_KB_DOWNLOADED:
|
|
||||||
return amountOfDataGB / 1000000, nil
|
|
||||||
}
|
|
||||||
return 0, errors.New("pricing strategy not found")
|
|
||||||
}
|
|
||||||
|
|
||||||
type DataResourcePricingProfile struct {
|
|
||||||
pricing.AccessPricingProfile[DataResourcePricingStrategy] // AccessPricingProfile is the pricing profile of a data it means that we can access the data for an amount of time
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *DataResourcePricingProfile) GetOverrideStrategyValue() int {
|
|
||||||
return p.Pricing.OverrideStrategy.GetStrategyValue()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *DataResourcePricingProfile) GetPrice(amountOfData float64, explicitDuration float64, start time.Time, end time.Time, params ...string) (float64, error) {
|
|
||||||
return p.Pricing.GetPrice(amountOfData, explicitDuration, start, &end)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *DataResourcePricingProfile) GetPurchase() pricing.BuyingStrategy {
|
|
||||||
return p.Pricing.BuyingStrategy
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *DataResourcePricingProfile) IsPurchasable() bool {
|
|
||||||
return p.Pricing.BuyingStrategy != pricing.UNDEFINED_SUBSCRIPTION
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *DataResourcePricingProfile) IsBooked() bool {
|
|
||||||
// TODO WHAT ABOUT PAY PER USE... it's a complicate CASE
|
|
||||||
return p.Pricing.BuyingStrategy != pricing.PERMANENT
|
|
||||||
}
|
|
||||||
|
|
||||||
type PricedDataResource struct {
|
|
||||||
PricedResource
|
|
||||||
UsageStorageGB float64 `json:"storage_gb,omitempty" bson:"storage_gb,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *PricedDataResource) GetType() tools.DataType {
|
|
||||||
return tools.DATA_RESOURCE
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *PricedDataResource) GetPrice() (float64, error) {
|
|
||||||
fmt.Println("GetPrice", r.UsageStart, r.UsageEnd)
|
|
||||||
now := time.Now()
|
|
||||||
if r.UsageStart == nil {
|
|
||||||
r.UsageStart = &now
|
|
||||||
}
|
|
||||||
if r.UsageEnd == nil {
|
|
||||||
add := r.UsageStart.Add(time.Duration(1 * time.Hour))
|
|
||||||
r.UsageEnd = &add
|
|
||||||
}
|
|
||||||
if r.SelectedPricing == nil {
|
|
||||||
return 0, errors.New("pricing profile must be set on Priced Data" + r.ResourceID)
|
|
||||||
}
|
|
||||||
pricing := r.SelectedPricing
|
|
||||||
var err error
|
|
||||||
amountOfData := float64(1)
|
|
||||||
if pricing.GetOverrideStrategyValue() >= 0 {
|
|
||||||
amountOfData, err = ToDataResourcePricingStrategy(pricing.GetOverrideStrategyValue()).GetQuantity(r.UsageStorageGB)
|
|
||||||
if err != nil {
|
|
||||||
return 0, err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return pricing.GetPrice(amountOfData, r.ExplicitBookingDurationS, *r.UsageStart, *r.UsageEnd)
|
|
||||||
}
|
|
65
models/resources/data/data.go
Normal file
65
models/resources/data/data.go
Normal file
@ -0,0 +1,65 @@
|
|||||||
|
package data
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/resource_model"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/utils"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/tools"
|
||||||
|
)
|
||||||
|
|
||||||
|
// enum of public private or licenced data
|
||||||
|
type DataLicense int
|
||||||
|
|
||||||
|
const (
|
||||||
|
PUBLIC DataLicense = iota
|
||||||
|
PRIVATE
|
||||||
|
LICENCED
|
||||||
|
)
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Struct of Usage Conditions
|
||||||
|
*/
|
||||||
|
type UsageConditions struct {
|
||||||
|
Usage string `json:"usage,omitempty" bson:"usage,omitempty" description:"usage of the data"` // Usage is the usage of the data
|
||||||
|
Actors []string `json:"actors,omitempty" bson:"actors,omitempty" description:"actors of the data"` // Actors is the actors of the data
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* DataResource is a struct that represents a data resource
|
||||||
|
* it defines the resource data
|
||||||
|
*/
|
||||||
|
type DataResource struct {
|
||||||
|
resource_model.AbstractResource // AbstractResource contains the basic fields of an object (id, name)
|
||||||
|
resource_model.WebResource
|
||||||
|
Type string `bson:"type,omitempty" json:"type,omitempty"` // Type is the type of the storage
|
||||||
|
UsageConditions UsageConditions `json:"usage_conditions,omitempty" bson:"usage_conditions,omitempty" description:"usage conditions of the data"` // UsageConditions is the usage conditions of the data
|
||||||
|
License DataLicense `json:"license" bson:"license" description:"license of the data" default:"0"` // License is the license of the data
|
||||||
|
Interest DataLicense `json:"interest" bson:"interest" description:"interest of the data" default:"0"` // Interest is the interest of the data
|
||||||
|
Example string `json:"example,omitempty" bson:"example,omitempty" description:"base64 encoded data"` // Example is an example of the data
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dma *DataResource) Deserialize(j map[string]interface{}) utils.DBObject {
|
||||||
|
b, err := json.Marshal(j)
|
||||||
|
if err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
json.Unmarshal(b, dma)
|
||||||
|
return dma
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dma *DataResource) Serialize() map[string]interface{} {
|
||||||
|
var m map[string]interface{}
|
||||||
|
b, err := json.Marshal(dma)
|
||||||
|
if err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
json.Unmarshal(b, &m)
|
||||||
|
return m
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *DataResource) GetAccessor(caller *tools.HTTPCaller) utils.Accessor {
|
||||||
|
data := New() // Create a new instance of the accessor
|
||||||
|
data.Init(tools.DATA_RESOURCE, caller) // Initialize the accessor with the DATA_RESOURCE model type
|
||||||
|
return data
|
||||||
|
}
|
110
models/resources/data/data_mongo_accessor.go
Normal file
110
models/resources/data/data_mongo_accessor.go
Normal file
@ -0,0 +1,110 @@
|
|||||||
|
package data
|
||||||
|
|
||||||
|
import (
|
||||||
|
"cloud.o-forge.io/core/oc-lib/dbs"
|
||||||
|
mongo "cloud.o-forge.io/core/oc-lib/dbs/mongo"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/resource_model"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/utils"
|
||||||
|
)
|
||||||
|
|
||||||
|
type dataMongoAccessor struct {
|
||||||
|
utils.AbstractAccessor // AbstractAccessor contains the basic fields of an accessor (model, caller)
|
||||||
|
}
|
||||||
|
|
||||||
|
// New creates a new instance of the dataMongoAccessor
|
||||||
|
func New() *dataMongoAccessor {
|
||||||
|
return &dataMongoAccessor{}
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Nothing special here, just the basic CRUD operations
|
||||||
|
*/
|
||||||
|
|
||||||
|
func (dma *dataMongoAccessor) DeleteOne(id string) (utils.DBObject, int, error) {
|
||||||
|
return dma.GenericDeleteOne(id, dma)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dma *dataMongoAccessor) UpdateOne(set utils.DBObject, id string) (utils.DBObject, int, error) {
|
||||||
|
set.(*DataResource).ResourceModel = nil
|
||||||
|
return dma.GenericUpdateOne(set, id, dma, &DataResource{})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dma *dataMongoAccessor) StoreOne(data utils.DBObject) (utils.DBObject, int, error) {
|
||||||
|
data.(*DataResource).ResourceModel = nil
|
||||||
|
return dma.GenericStoreOne(data, dma)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dma *dataMongoAccessor) CopyOne(data utils.DBObject) (utils.DBObject, int, error) {
|
||||||
|
return dma.GenericStoreOne(data, dma)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dma *dataMongoAccessor) LoadOne(id string) (utils.DBObject, int, error) {
|
||||||
|
var data DataResource
|
||||||
|
res_mongo, code, err := mongo.MONGOService.LoadOne(id, dma.GetType())
|
||||||
|
if err != nil {
|
||||||
|
dma.Logger.Error().Msg("Could not retrieve " + id + " from db. Error: " + err.Error())
|
||||||
|
return nil, code, err
|
||||||
|
}
|
||||||
|
res_mongo.Decode(&data)
|
||||||
|
accessor := (&resource_model.ResourceModel{}).GetAccessor(nil)
|
||||||
|
resources, _, err := accessor.Search(nil, dma.GetType())
|
||||||
|
if err == nil && len(resources) > 0 {
|
||||||
|
data.ResourceModel = resources[0].(*resource_model.ResourceModel)
|
||||||
|
}
|
||||||
|
return &data, 200, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (wfa dataMongoAccessor) 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 []DataResource
|
||||||
|
if err = res_mongo.All(mongo.MngoCtx, &results); err != nil {
|
||||||
|
return nil, 404, err
|
||||||
|
}
|
||||||
|
accessor := (&resource_model.ResourceModel{}).GetAccessor(nil)
|
||||||
|
resources, _, err := accessor.Search(nil, wfa.GetType())
|
||||||
|
for _, r := range results {
|
||||||
|
if err == nil && len(resources) > 0 {
|
||||||
|
r.ResourceModel = resources[0].(*resource_model.ResourceModel)
|
||||||
|
}
|
||||||
|
objs = append(objs, &r) // only get the abstract resource !
|
||||||
|
}
|
||||||
|
return objs, 200, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (wfa *dataMongoAccessor) 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 like name, short_description, description, owner, url if no filters are provided
|
||||||
|
"abstractresource.abstractobject.name": {{Operator: dbs.LIKE.String(), Value: search}},
|
||||||
|
"abstractresource.short_description": {{Operator: dbs.LIKE.String(), Value: search}},
|
||||||
|
"abstractresource.description": {{Operator: dbs.LIKE.String(), Value: search}},
|
||||||
|
"abstractresource.owner": {{Operator: dbs.LIKE.String(), Value: search}},
|
||||||
|
"abstractresource.source_url": {{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 []DataResource
|
||||||
|
if err = res_mongo.All(mongo.MngoCtx, &results); err != nil {
|
||||||
|
return nil, 404, err
|
||||||
|
}
|
||||||
|
accessor := (&resource_model.ResourceModel{}).GetAccessor(nil)
|
||||||
|
resources, _, err := accessor.Search(nil, wfa.GetType())
|
||||||
|
for _, r := range results {
|
||||||
|
if err == nil && len(resources) > 0 {
|
||||||
|
r.ResourceModel = resources[0].(*resource_model.ResourceModel)
|
||||||
|
}
|
||||||
|
objs = append(objs, &r) // only get the abstract resource !
|
||||||
|
}
|
||||||
|
return objs, 200, nil
|
||||||
|
}
|
53
models/resources/data/data_test.go
Normal file
53
models/resources/data/data_test.go
Normal file
@ -0,0 +1,53 @@
|
|||||||
|
package data
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/resource_model"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/utils"
|
||||||
|
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestStoreOneData(t *testing.T) {
|
||||||
|
d := DataResource{
|
||||||
|
WebResource: resource_model.WebResource{
|
||||||
|
Protocol: "http", Path: "azerty.fr",
|
||||||
|
},
|
||||||
|
Example: "123456",
|
||||||
|
AbstractResource: resource_model.AbstractResource{
|
||||||
|
AbstractObject: utils.AbstractObject{Name: "testData"},
|
||||||
|
Description: "Lorem Ipsum",
|
||||||
|
Logo: "azerty.com",
|
||||||
|
Owner: "toto",
|
||||||
|
OwnerLogo: "totoLogo",
|
||||||
|
SourceUrl: "azerty.fr",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
dma := New()
|
||||||
|
id, _, _ := dma.StoreOne(&d)
|
||||||
|
|
||||||
|
assert.NotEmpty(t, id)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestLoadOneDate(t *testing.T) {
|
||||||
|
d := DataResource{
|
||||||
|
WebResource: resource_model.WebResource{
|
||||||
|
Protocol: "http", Path: "azerty.fr",
|
||||||
|
},
|
||||||
|
Example: "123456",
|
||||||
|
AbstractResource: resource_model.AbstractResource{
|
||||||
|
AbstractObject: utils.AbstractObject{Name: "testData"},
|
||||||
|
Description: "Lorem Ipsum",
|
||||||
|
Logo: "azerty.com",
|
||||||
|
Owner: "toto",
|
||||||
|
OwnerLogo: "totoLogo",
|
||||||
|
SourceUrl: "azerty.fr",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
dma := New()
|
||||||
|
new_d, _, _ := dma.StoreOne(&d)
|
||||||
|
assert.Equal(t, d, new_d)
|
||||||
|
}
|
@ -1,38 +0,0 @@
|
|||||||
package resources
|
|
||||||
|
|
||||||
import (
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/pricing"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
)
|
|
||||||
|
|
||||||
type ResourceInterface interface {
|
|
||||||
utils.DBObject
|
|
||||||
Trim()
|
|
||||||
ConvertToPricedResource(t tools.DataType, request *tools.APIRequest) pricing.PricedItemITF
|
|
||||||
GetType() string
|
|
||||||
GetSelectedInstance() ResourceInstanceITF
|
|
||||||
ClearEnv() utils.DBObject
|
|
||||||
SetAllowedInstances(request *tools.APIRequest)
|
|
||||||
}
|
|
||||||
|
|
||||||
type ResourceInstanceITF interface {
|
|
||||||
utils.DBObject
|
|
||||||
GetID() string
|
|
||||||
GetName() string
|
|
||||||
StoreDraftDefault()
|
|
||||||
ClearEnv()
|
|
||||||
GetProfile() pricing.PricingProfileITF
|
|
||||||
GetPricingsProfiles(peerID string, groups []string) []pricing.PricingProfileITF
|
|
||||||
GetPeerGroups() ([]ResourcePartnerITF, []map[string][]string)
|
|
||||||
ClearPeerGroups()
|
|
||||||
GetSelectedPartnership(peerID string, groups []string) ResourcePartnerITF
|
|
||||||
GetPartnerships(peerID string, groups []string) []ResourcePartnerITF
|
|
||||||
}
|
|
||||||
|
|
||||||
type ResourcePartnerITF interface {
|
|
||||||
GetPricingsProfiles(peerID string, groups []string) []pricing.PricingProfileITF
|
|
||||||
GetPeerGroups() map[string][]string
|
|
||||||
ClearPeerGroups()
|
|
||||||
GetProfile(buying int, strategy int) pricing.PricingProfileITF
|
|
||||||
}
|
|
@ -1,65 +0,0 @@
|
|||||||
package resources
|
|
||||||
|
|
||||||
import (
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
)
|
|
||||||
|
|
||||||
type ResourceSet struct {
|
|
||||||
Datas []string `bson:"datas,omitempty" json:"datas,omitempty"`
|
|
||||||
Storages []string `bson:"storages,omitempty" json:"storages,omitempty"`
|
|
||||||
Processings []string `bson:"processings,omitempty" json:"processings,omitempty"`
|
|
||||||
Computes []string `bson:"computes,omitempty" json:"computes,omitempty"`
|
|
||||||
Workflows []string `bson:"workflows,omitempty" json:"workflows,omitempty"`
|
|
||||||
|
|
||||||
DataResources []*DataResource `bson:"-" json:"data_resources,omitempty"`
|
|
||||||
StorageResources []*StorageResource `bson:"-" json:"storage_resources,omitempty"`
|
|
||||||
ProcessingResources []*ProcessingResource `bson:"-" json:"processing_resources,omitempty"`
|
|
||||||
ComputeResources []*ComputeResource `bson:"-" json:"compute_resources,omitempty"`
|
|
||||||
WorkflowResources []*WorkflowResource `bson:"-" json:"workflow_resources,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *ResourceSet) Clear() {
|
|
||||||
r.DataResources = nil
|
|
||||||
r.StorageResources = nil
|
|
||||||
r.ProcessingResources = nil
|
|
||||||
r.ComputeResources = nil
|
|
||||||
r.WorkflowResources = nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *ResourceSet) Fill(request *tools.APIRequest) {
|
|
||||||
r.Clear()
|
|
||||||
for k, v := range map[utils.DBObject][]string{
|
|
||||||
(&DataResource{}): r.Datas,
|
|
||||||
(&ComputeResource{}): r.Computes,
|
|
||||||
(&StorageResource{}): r.Storages,
|
|
||||||
(&ProcessingResource{}): r.Processings,
|
|
||||||
(&WorkflowResource{}): r.Workflows,
|
|
||||||
} {
|
|
||||||
for _, id := range v {
|
|
||||||
d, _, e := k.GetAccessor(request).LoadOne(id)
|
|
||||||
if e == nil {
|
|
||||||
switch k.(type) {
|
|
||||||
case *DataResource:
|
|
||||||
r.DataResources = append(r.DataResources, d.(*DataResource))
|
|
||||||
case *ComputeResource:
|
|
||||||
r.ComputeResources = append(r.ComputeResources, d.(*ComputeResource))
|
|
||||||
case *StorageResource:
|
|
||||||
r.StorageResources = append(r.StorageResources, d.(*StorageResource))
|
|
||||||
case *ProcessingResource:
|
|
||||||
r.ProcessingResources = append(r.ProcessingResources, d.(*ProcessingResource))
|
|
||||||
case *WorkflowResource:
|
|
||||||
r.WorkflowResources = append(r.WorkflowResources, d.(*WorkflowResource))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
type ItemResource struct {
|
|
||||||
Data *DataResource `bson:"data,omitempty" json:"data,omitempty"`
|
|
||||||
Processing *ProcessingResource `bson:"processing,omitempty" json:"processing,omitempty"`
|
|
||||||
Storage *StorageResource `bson:"storage,omitempty" json:"storage,omitempty"`
|
|
||||||
Compute *ComputeResource `bson:"compute,omitempty" json:"compute,omitempty"`
|
|
||||||
Workflow *WorkflowResource `bson:"workflow,omitempty" json:"workflow,omitempty"`
|
|
||||||
}
|
|
@ -1,100 +0,0 @@
|
|||||||
package resources
|
|
||||||
|
|
||||||
import (
|
|
||||||
"errors"
|
|
||||||
"fmt"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/pricing"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
)
|
|
||||||
|
|
||||||
type PricedResource struct {
|
|
||||||
Name string `json:"name,omitempty" bson:"name,omitempty"`
|
|
||||||
Logo string `json:"logo,omitempty" bson:"logo,omitempty"`
|
|
||||||
InstancesRefs map[string]string `json:"instances_refs,omitempty" bson:"instances_refs,omitempty"`
|
|
||||||
SelectedPricing pricing.PricingProfileITF `json:"selected_pricing,omitempty" bson:"selected_pricing,omitempty"`
|
|
||||||
ExplicitBookingDurationS float64 `json:"explicit_location_duration_s,omitempty" bson:"explicit_location_duration_s,omitempty"`
|
|
||||||
UsageStart *time.Time `json:"start,omitempty" bson:"start,omitempty"`
|
|
||||||
UsageEnd *time.Time `json:"end,omitempty" bson:"end,omitempty"`
|
|
||||||
CreatorID string `json:"peer_id,omitempty" bson:"peer_id,omitempty"`
|
|
||||||
ResourceID string `json:"resource_id,omitempty" bson:"resource_id,omitempty"`
|
|
||||||
ResourceType tools.DataType `json:"resource_type,omitempty" bson:"resource_type,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func (abs *PricedResource) SelectPricing() pricing.PricingProfileITF {
|
|
||||||
return abs.SelectedPricing
|
|
||||||
}
|
|
||||||
|
|
||||||
func (abs *PricedResource) GetID() string {
|
|
||||||
return abs.ResourceID
|
|
||||||
}
|
|
||||||
|
|
||||||
func (abs *PricedResource) GetType() tools.DataType {
|
|
||||||
return abs.ResourceType
|
|
||||||
}
|
|
||||||
|
|
||||||
func (abs *PricedResource) GetCreatorID() string {
|
|
||||||
return abs.CreatorID
|
|
||||||
}
|
|
||||||
|
|
||||||
func (abs *PricedResource) IsPurchasable() bool {
|
|
||||||
if abs.SelectedPricing == nil {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
return (abs.SelectedPricing).IsPurchasable()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (abs *PricedResource) IsBooked() bool {
|
|
||||||
if abs.SelectedPricing == nil {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
return (abs.SelectedPricing).IsBooked()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (abs *PricedResource) GetLocationEnd() *time.Time {
|
|
||||||
return abs.UsageEnd
|
|
||||||
}
|
|
||||||
|
|
||||||
func (abs *PricedResource) GetLocationStart() *time.Time {
|
|
||||||
return abs.UsageStart
|
|
||||||
}
|
|
||||||
|
|
||||||
func (abs *PricedResource) SetLocationStart(start time.Time) {
|
|
||||||
abs.UsageStart = &start
|
|
||||||
}
|
|
||||||
|
|
||||||
func (abs *PricedResource) SetLocationEnd(end time.Time) {
|
|
||||||
abs.UsageEnd = &end
|
|
||||||
}
|
|
||||||
|
|
||||||
func (abs *PricedResource) GetExplicitDurationInS() float64 {
|
|
||||||
if abs.ExplicitBookingDurationS == 0 {
|
|
||||||
if abs.UsageEnd == nil && abs.UsageStart == nil {
|
|
||||||
return time.Duration(1 * time.Hour).Seconds()
|
|
||||||
}
|
|
||||||
if abs.UsageEnd == nil {
|
|
||||||
add := abs.UsageStart.Add(time.Duration(1 * time.Hour))
|
|
||||||
abs.UsageEnd = &add
|
|
||||||
}
|
|
||||||
return abs.UsageEnd.Sub(*abs.UsageStart).Seconds()
|
|
||||||
}
|
|
||||||
return abs.ExplicitBookingDurationS
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *PricedResource) GetPrice() (float64, error) {
|
|
||||||
fmt.Println("GetPrice", r.UsageStart, r.UsageEnd)
|
|
||||||
now := time.Now()
|
|
||||||
if r.UsageStart == nil {
|
|
||||||
r.UsageStart = &now
|
|
||||||
}
|
|
||||||
if r.UsageEnd == nil {
|
|
||||||
add := r.UsageStart.Add(time.Duration(1 * time.Hour))
|
|
||||||
r.UsageEnd = &add
|
|
||||||
}
|
|
||||||
if r.SelectedPricing == nil {
|
|
||||||
return 0, errors.New("pricing profile must be set on Priced Resource " + r.ResourceID)
|
|
||||||
}
|
|
||||||
pricing := r.SelectedPricing
|
|
||||||
return pricing.GetPrice(1, 0, *r.UsageStart, *r.UsageEnd)
|
|
||||||
}
|
|
@ -1,94 +0,0 @@
|
|||||||
package resources
|
|
||||||
|
|
||||||
import (
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/enum"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/models"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/pricing"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
)
|
|
||||||
|
|
||||||
type ProcessingUsage struct {
|
|
||||||
CPUs map[string]*models.CPU `bson:"cpus,omitempty" json:"cpus,omitempty"` // CPUs is the list of CPUs key is model
|
|
||||||
GPUs map[string]*models.GPU `bson:"gpus,omitempty" json:"gpus,omitempty"` // GPUs is the list of GPUs key is model
|
|
||||||
RAM *models.RAM `bson:"ram,omitempty" json:"ram,omitempty"` // RAM is the RAM
|
|
||||||
|
|
||||||
StorageGb float64 `bson:"storage,omitempty" json:"storage,omitempty"` // Storage is the storage
|
|
||||||
Hypothesis string `bson:"hypothesis,omitempty" json:"hypothesis,omitempty"`
|
|
||||||
ScalingModel string `bson:"scaling_model,omitempty" json:"scaling_model,omitempty"` // ScalingModel is the scaling model
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
|
||||||
* ProcessingResource is a struct that represents a processing resource
|
|
||||||
* it defines the resource processing
|
|
||||||
*/
|
|
||||||
type ProcessingResource struct {
|
|
||||||
AbstractInstanciatedResource[*ProcessingInstance]
|
|
||||||
Infrastructure enum.InfrastructureType `json:"infrastructure" bson:"infrastructure" default:"-1"` // Infrastructure is the infrastructure
|
|
||||||
IsService bool `json:"is_service,omitempty" bson:"is_service,omitempty"` // IsService is a flag that indicates if the processing is a service
|
|
||||||
Usage *ProcessingUsage `bson:"usage,omitempty" json:"usage,omitempty"` // Usage is the usage of the processing
|
|
||||||
OpenSource bool `json:"open_source" bson:"open_source" default:"false"`
|
|
||||||
License string `json:"license,omitempty" bson:"license,omitempty"`
|
|
||||||
Maturity string `json:"maturity,omitempty" bson:"maturity,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *ProcessingResource) GetType() string {
|
|
||||||
return tools.PROCESSING_RESOURCE.String()
|
|
||||||
}
|
|
||||||
|
|
||||||
type ProcessingResourceAccess struct {
|
|
||||||
Container *models.Container `json:"container,omitempty" bson:"container,omitempty"` // Container is the container
|
|
||||||
}
|
|
||||||
|
|
||||||
type ProcessingInstance struct {
|
|
||||||
ResourceInstance[*ResourcePartnerShip[*ProcessingResourcePricingProfile]]
|
|
||||||
Access *ProcessingResourceAccess `json:"access,omitempty" bson:"access,omitempty"` // Access is the access
|
|
||||||
}
|
|
||||||
|
|
||||||
type PricedProcessingResource struct {
|
|
||||||
PricedResource
|
|
||||||
IsService bool
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *PricedProcessingResource) GetType() tools.DataType {
|
|
||||||
return tools.PROCESSING_RESOURCE
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *PricedProcessingResource) GetExplicitDurationInS() float64 {
|
|
||||||
if a.ExplicitBookingDurationS == 0 {
|
|
||||||
if a.IsService || a.UsageStart == nil {
|
|
||||||
if a.IsService {
|
|
||||||
return -1
|
|
||||||
}
|
|
||||||
return time.Duration(1 * time.Hour).Seconds()
|
|
||||||
}
|
|
||||||
return a.UsageEnd.Sub(*a.UsageStart).Seconds()
|
|
||||||
}
|
|
||||||
return a.ExplicitBookingDurationS
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *ProcessingResource) GetAccessor(request *tools.APIRequest) utils.Accessor {
|
|
||||||
return NewAccessor[*ProcessingResource](tools.PROCESSING_RESOURCE, request, func() utils.DBObject { return &ProcessingResource{} }) // Create a new instance of the accessor
|
|
||||||
}
|
|
||||||
|
|
||||||
type ProcessingResourcePricingProfile struct {
|
|
||||||
pricing.AccessPricingProfile[pricing.TimePricingStrategy] // AccessPricingProfile is the pricing profile of a data it means that we can access the data for an amount of time
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *ProcessingResourcePricingProfile) IsPurchasable() bool {
|
|
||||||
return p.Pricing.BuyingStrategy != pricing.UNDEFINED_SUBSCRIPTION
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *ProcessingResourcePricingProfile) IsBooked() bool {
|
|
||||||
return p.Pricing.BuyingStrategy != pricing.PERMANENT
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *ProcessingResourcePricingProfile) GetPurchase() pricing.BuyingStrategy {
|
|
||||||
return p.Pricing.BuyingStrategy
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *ProcessingResourcePricingProfile) GetPrice(amountOfData float64, val float64, start time.Time, end time.Time, params ...string) (float64, error) {
|
|
||||||
return p.Pricing.GetPrice(amountOfData, val, start, &end)
|
|
||||||
}
|
|
67
models/resources/processing/processing.go
Normal file
67
models/resources/processing/processing.go
Normal file
@ -0,0 +1,67 @@
|
|||||||
|
package processing
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/resource_model"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/resources/compute"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/utils"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/tools"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Container struct {
|
||||||
|
Image string `json:"image,omitempty" bson:"image,omitempty"` // Image is the container image
|
||||||
|
Command string `json:"command,omitempty" bson:"command,omitempty"` // Command is the container command
|
||||||
|
Args string `json:"args,omitempty" bson:"args,omitempty"` // Args is the container arguments
|
||||||
|
Env map[string]string `json:"env,omitempty" bson:"env,omitempty"` // Env is the container environment variables
|
||||||
|
Volumes map[string]string `json:"volumes,omitempty" bson:"volumes,omitempty"` // Volumes is the container volumes
|
||||||
|
}
|
||||||
|
|
||||||
|
type Expose struct {
|
||||||
|
Port int `json:"port,omitempty" bson:"port,omitempty"` // Port is the port
|
||||||
|
Reverse string `json:"reverse,omitempty" bson:"reverse,omitempty"` // Reverse is the reverse
|
||||||
|
PAT int `json:"pat,omitempty" bson:"pat,omitempty"` // PAT is the PAT
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* ProcessingResource is a struct that represents a processing resource
|
||||||
|
* it defines the resource processing
|
||||||
|
*/
|
||||||
|
type ProcessingResource struct {
|
||||||
|
resource_model.AbstractResource
|
||||||
|
IsService bool `json:"is_service,omitempty" bson:"is_service,omitempty"` // IsService is a flag that indicates if the processing is a service
|
||||||
|
CPUs []*compute.CPU `bson:"cpus,omitempty" json:"cp_us,omitempty"` // CPUs is the list of CPUs
|
||||||
|
GPUs []*compute.GPU `bson:"gpus,omitempty" json:"gp_us,omitempty"` // GPUs is the list of GPUs
|
||||||
|
RAM *compute.RAM `bson:"ram,omitempty" json:"ram,omitempty"` // RAM is the RAM
|
||||||
|
Storage uint `bson:"storage,omitempty" json:"storage,omitempty"` // Storage is the storage
|
||||||
|
Parallel bool `bson:"parallel,omitempty" json:"parallel,omitempty"` // Parallel is a flag that indicates if the processing is parallel
|
||||||
|
ScalingModel uint `bson:"scaling_model,omitempty" json:"scaling_model,omitempty"` // ScalingModel is the scaling model
|
||||||
|
DiskIO string `bson:"disk_io,omitempty" json:"disk_io,omitempty"` // DiskIO is the disk IO
|
||||||
|
Container *Container `bson:"container,omitempty" json:"container,omitempty"` // Container is the container
|
||||||
|
Expose []Expose `bson:"expose,omitempty" json:"expose,omitempty"` // Expose is the execution
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dma *ProcessingResource) Deserialize(j map[string]interface{}) utils.DBObject {
|
||||||
|
b, err := json.Marshal(j)
|
||||||
|
if err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
json.Unmarshal(b, dma)
|
||||||
|
return dma
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dma *ProcessingResource) Serialize() map[string]interface{} {
|
||||||
|
var m map[string]interface{}
|
||||||
|
b, err := json.Marshal(dma)
|
||||||
|
if err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
json.Unmarshal(b, &m)
|
||||||
|
return m
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *ProcessingResource) GetAccessor(caller *tools.HTTPCaller) utils.Accessor {
|
||||||
|
data := New() // Create a new instance of the accessor
|
||||||
|
data.Init(tools.PROCESSING_RESOURCE, caller) // Initialize the accessor with the PROCESSING_RESOURCE model type
|
||||||
|
return data
|
||||||
|
}
|
114
models/resources/processing/processing_mongo_accessor.go
Normal file
114
models/resources/processing/processing_mongo_accessor.go
Normal file
@ -0,0 +1,114 @@
|
|||||||
|
package processing
|
||||||
|
|
||||||
|
import (
|
||||||
|
"cloud.o-forge.io/core/oc-lib/dbs"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/dbs/mongo"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/resource_model"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/utils"
|
||||||
|
)
|
||||||
|
|
||||||
|
type processingMongoAccessor struct {
|
||||||
|
utils.AbstractAccessor // AbstractAccessor contains the basic fields of an accessor (model, caller)
|
||||||
|
}
|
||||||
|
|
||||||
|
// New creates a new instance of the processingMongoAccessor
|
||||||
|
func New() *processingMongoAccessor {
|
||||||
|
return &processingMongoAccessor{}
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Nothing special here, just the basic CRUD operations
|
||||||
|
*/
|
||||||
|
|
||||||
|
func (pma *processingMongoAccessor) DeleteOne(id string) (utils.DBObject, int, error) {
|
||||||
|
return pma.GenericDeleteOne(id, pma)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (pma *processingMongoAccessor) UpdateOne(set utils.DBObject, id string) (utils.DBObject, int, error) {
|
||||||
|
set.(*ProcessingResource).ResourceModel = nil
|
||||||
|
return pma.GenericUpdateOne(set, id, pma, &ProcessingResource{})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (pma *processingMongoAccessor) StoreOne(data utils.DBObject) (utils.DBObject, int, error) {
|
||||||
|
data.(*ProcessingResource).ResourceModel = nil
|
||||||
|
return pma.GenericStoreOne(data, pma)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (pma *processingMongoAccessor) CopyOne(data utils.DBObject) (utils.DBObject, int, error) {
|
||||||
|
return pma.GenericStoreOne(data, pma)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (pma *processingMongoAccessor) LoadOne(id string) (utils.DBObject, int, error) {
|
||||||
|
|
||||||
|
var processing ProcessingResource
|
||||||
|
|
||||||
|
res_mongo, code, err := mongo.MONGOService.LoadOne(id, pma.GetType())
|
||||||
|
if err != nil {
|
||||||
|
pma.Logger.Error().Msg("Could not retrieve " + id + " from db. Error: " + err.Error())
|
||||||
|
return nil, code, err
|
||||||
|
}
|
||||||
|
|
||||||
|
res_mongo.Decode(&processing)
|
||||||
|
accessor := (&resource_model.ResourceModel{}).GetAccessor(nil)
|
||||||
|
resources, _, err := accessor.Search(nil, pma.GetType())
|
||||||
|
if err == nil && len(resources) > 0 {
|
||||||
|
processing.ResourceModel = resources[0].(*resource_model.ResourceModel)
|
||||||
|
}
|
||||||
|
return &processing, 200, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (wfa processingMongoAccessor) 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 []ProcessingResource
|
||||||
|
if err = res_mongo.All(mongo.MngoCtx, &results); err != nil {
|
||||||
|
return nil, 404, err
|
||||||
|
}
|
||||||
|
accessor := (&resource_model.ResourceModel{}).GetAccessor(nil)
|
||||||
|
resources, _, err := accessor.Search(nil, wfa.GetType())
|
||||||
|
for _, r := range results {
|
||||||
|
if err == nil && len(resources) > 0 {
|
||||||
|
r.ResourceModel = resources[0].(*resource_model.ResourceModel)
|
||||||
|
}
|
||||||
|
objs = append(objs, &r) // only get the abstract resource !
|
||||||
|
}
|
||||||
|
return objs, 200, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Search searches for processing resources in the database, given some filters OR a search string
|
||||||
|
func (wfa *processingMongoAccessor) 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 like name, short_description, description, owner, url if no filters are provided
|
||||||
|
"abstractresource.abstractobject.name": {{Operator: dbs.LIKE.String(), Value: search}},
|
||||||
|
"abstractresource.short_description": {{Operator: dbs.LIKE.String(), Value: search}},
|
||||||
|
"abstractresource.description": {{Operator: dbs.LIKE.String(), Value: search}},
|
||||||
|
"abstractresource.owner": {{Operator: dbs.LIKE.String(), Value: search}},
|
||||||
|
"abstractresource.source_url": {{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 []ProcessingResource
|
||||||
|
if err = res_mongo.All(mongo.MngoCtx, &results); err != nil {
|
||||||
|
return nil, 404, err
|
||||||
|
}
|
||||||
|
accessor := (&resource_model.ResourceModel{}).GetAccessor(nil)
|
||||||
|
resources, _, err := accessor.Search(nil, wfa.GetType())
|
||||||
|
for _, r := range results {
|
||||||
|
if err == nil && len(resources) > 0 {
|
||||||
|
r.ResourceModel = resources[0].(*resource_model.ResourceModel)
|
||||||
|
}
|
||||||
|
objs = append(objs, &r) // only get the abstract resource !
|
||||||
|
}
|
||||||
|
return objs, 200, nil
|
||||||
|
}
|
38
models/resources/processing/processing_test.go
Normal file
38
models/resources/processing/processing_test.go
Normal file
@ -0,0 +1,38 @@
|
|||||||
|
package processing
|
||||||
|
|
||||||
|
/*
|
||||||
|
func TestStoreOneProcessing(t *testing.T) {
|
||||||
|
p := ProcessingResource{Container: "totoCont",
|
||||||
|
AbstractResource: resources.AbstractResource{
|
||||||
|
AbstractObject: utils.AbstractObject{Name: "testData"},
|
||||||
|
Description: "Lorem Ipsum",
|
||||||
|
Logo: "azerty.com",
|
||||||
|
Owner: "toto",
|
||||||
|
OwnerLogo: "totoLogo",
|
||||||
|
SourceUrl: "azerty.fr",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
sma := ProcessingMongoAccessor{}
|
||||||
|
id, _, _ := sma.StoreOne(&p)
|
||||||
|
|
||||||
|
assert.NotEmpty(t, id)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestLoadOneProcessing(t *testing.T) {
|
||||||
|
p := ProcessingResource{Container: "totoCont",
|
||||||
|
AbstractResource: resources.AbstractResource{
|
||||||
|
AbstractObject: utils.AbstractObject{Name: "testData"},
|
||||||
|
Description: "Lorem Ipsum",
|
||||||
|
Logo: "azerty.com",
|
||||||
|
Owner: "toto",
|
||||||
|
OwnerLogo: "totoLogo",
|
||||||
|
SourceUrl: "azerty.fr",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
sma := ProcessingMongoAccessor{}
|
||||||
|
new_s, _, _ := sma.StoreOne(&p)
|
||||||
|
assert.Equal(t, p, new_s)
|
||||||
|
}
|
||||||
|
*/
|
@ -1,34 +0,0 @@
|
|||||||
package purchase_resource
|
|
||||||
|
|
||||||
import (
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/pricing"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
)
|
|
||||||
|
|
||||||
type PurchaseResource struct {
|
|
||||||
utils.AbstractObject
|
|
||||||
DestPeerID string
|
|
||||||
PricedItem pricing.PricedItemITF `json:"priced_item,omitempty" bson:"priced_item,omitempty" validate:"required"`
|
|
||||||
ExecutionsID string `json:"executions_id,omitempty" bson:"executions_id,omitempty" validate:"required"` // ExecutionsID is the ID of the executions
|
|
||||||
EndDate *time.Time `json:"end_buying_date,omitempty" bson:"end_buying_date,omitempty"`
|
|
||||||
ResourceID string `json:"resource_id" bson:"resource_id" validate:"required"`
|
|
||||||
ResourceType tools.DataType `json:"resource_type" bson:"resource_type" validate:"required"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *PurchaseResource) GetAccessor(request *tools.APIRequest) utils.Accessor {
|
|
||||||
return NewAccessor(request) // Create a new instance of the accessor
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *PurchaseResource) CanUpdate(set utils.DBObject) (bool, utils.DBObject) {
|
|
||||||
return r.IsDraft, set // only draft buying can be updated
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *PurchaseResource) CanDelete() bool { // ENDBuyingDate is passed
|
|
||||||
if r.EndDate != nil {
|
|
||||||
return time.Now().UTC().After(*r.EndDate)
|
|
||||||
}
|
|
||||||
return false // only draft bookings can be deleted
|
|
||||||
}
|
|
@ -1,72 +0,0 @@
|
|||||||
package purchase_resource
|
|
||||||
|
|
||||||
import (
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/dbs"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/logs"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
)
|
|
||||||
|
|
||||||
type PurchaseResourceMongoAccessor struct {
|
|
||||||
utils.AbstractAccessor // AbstractAccessor contains the basic fields of an accessor (model, caller)
|
|
||||||
}
|
|
||||||
|
|
||||||
// New creates a new instance of the bookingMongoAccessor
|
|
||||||
func NewAccessor(request *tools.APIRequest) *PurchaseResourceMongoAccessor {
|
|
||||||
return &PurchaseResourceMongoAccessor{
|
|
||||||
AbstractAccessor: utils.AbstractAccessor{
|
|
||||||
Logger: logs.CreateLogger(tools.PURCHASE_RESOURCE.String()), // Create a logger with the data type
|
|
||||||
Request: request,
|
|
||||||
Type: tools.PURCHASE_RESOURCE,
|
|
||||||
},
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
|
||||||
* Nothing special here, just the basic CRUD operations
|
|
||||||
*/
|
|
||||||
func (a *PurchaseResourceMongoAccessor) DeleteOne(id string) (utils.DBObject, int, error) {
|
|
||||||
return utils.GenericDeleteOne(id, a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *PurchaseResourceMongoAccessor) UpdateOne(set utils.DBObject, id string) (utils.DBObject, int, error) {
|
|
||||||
return utils.GenericUpdateOne(set, id, a, &PurchaseResource{})
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *PurchaseResourceMongoAccessor) StoreOne(data utils.DBObject) (utils.DBObject, int, error) {
|
|
||||||
return utils.GenericStoreOne(data, a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *PurchaseResourceMongoAccessor) CopyOne(data utils.DBObject) (utils.DBObject, int, error) {
|
|
||||||
return utils.GenericStoreOne(data, a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *PurchaseResourceMongoAccessor) LoadOne(id string) (utils.DBObject, int, error) {
|
|
||||||
return utils.GenericLoadOne[*PurchaseResource](id, func(d utils.DBObject) (utils.DBObject, int, error) {
|
|
||||||
if d.(*PurchaseResource).EndDate != nil && time.Now().UTC().After(*d.(*PurchaseResource).EndDate) {
|
|
||||||
utils.GenericDeleteOne(id, a)
|
|
||||||
return nil, 404, nil
|
|
||||||
}
|
|
||||||
return d, 200, nil
|
|
||||||
}, a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *PurchaseResourceMongoAccessor) LoadAll(isDraft bool) ([]utils.ShallowDBObject, int, error) {
|
|
||||||
return utils.GenericLoadAll[*PurchaseResource](a.getExec(), isDraft, a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *PurchaseResourceMongoAccessor) Search(filters *dbs.Filters, search string, isDraft bool) ([]utils.ShallowDBObject, int, error) {
|
|
||||||
return utils.GenericSearch[*PurchaseResource](filters, search, (&PurchaseResource{}).GetObjectFilters(search), a.getExec(), isDraft, a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *PurchaseResourceMongoAccessor) getExec() func(utils.DBObject) utils.ShallowDBObject {
|
|
||||||
return func(d utils.DBObject) utils.ShallowDBObject {
|
|
||||||
if d.(*PurchaseResource).EndDate != nil && time.Now().UTC().After(*d.(*PurchaseResource).EndDate) {
|
|
||||||
utils.GenericDeleteOne(d.GetID(), a)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
return d
|
|
||||||
}
|
|
||||||
}
|
|
@ -1,56 +0,0 @@
|
|||||||
package purchase_resource_test
|
|
||||||
|
|
||||||
import (
|
|
||||||
"testing"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/stretchr/testify/assert"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/resources/purchase_resource"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestGetAccessor(t *testing.T) {
|
|
||||||
req := &tools.APIRequest{}
|
|
||||||
res := &purchase_resource.PurchaseResource{}
|
|
||||||
accessor := res.GetAccessor(req)
|
|
||||||
|
|
||||||
assert.NotNil(t, accessor)
|
|
||||||
assert.Equal(t, tools.PURCHASE_RESOURCE, accessor.(*purchase_resource.PurchaseResourceMongoAccessor).Type)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestCanUpdate(t *testing.T) {
|
|
||||||
set := &purchase_resource.PurchaseResource{ResourceID: "id"}
|
|
||||||
r := &purchase_resource.PurchaseResource{
|
|
||||||
AbstractObject: utils.AbstractObject{IsDraft: true},
|
|
||||||
}
|
|
||||||
can, updated := r.CanUpdate(set)
|
|
||||||
assert.True(t, can)
|
|
||||||
assert.Equal(t, set, updated)
|
|
||||||
|
|
||||||
r.IsDraft = false
|
|
||||||
can, _ = r.CanUpdate(set)
|
|
||||||
assert.False(t, can)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestCanDelete(t *testing.T) {
|
|
||||||
now := time.Now().UTC()
|
|
||||||
past := now.Add(-1 * time.Hour)
|
|
||||||
future := now.Add(1 * time.Hour)
|
|
||||||
|
|
||||||
t.Run("nil EndDate", func(t *testing.T) {
|
|
||||||
r := &purchase_resource.PurchaseResource{}
|
|
||||||
assert.False(t, r.CanDelete())
|
|
||||||
})
|
|
||||||
|
|
||||||
t.Run("EndDate in past", func(t *testing.T) {
|
|
||||||
r := &purchase_resource.PurchaseResource{EndDate: &past}
|
|
||||||
assert.True(t, r.CanDelete())
|
|
||||||
})
|
|
||||||
|
|
||||||
t.Run("EndDate in future", func(t *testing.T) {
|
|
||||||
r := &purchase_resource.PurchaseResource{EndDate: &future}
|
|
||||||
assert.False(t, r.CanDelete())
|
|
||||||
})
|
|
||||||
}
|
|
291
models/resources/resource.go
Executable file → Normal file
291
models/resources/resource.go
Executable file → Normal file
@ -1,266 +1,57 @@
|
|||||||
package resources
|
package resources
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"slices"
|
"cloud.o-forge.io/core/oc-lib/models/resource_model"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/resources/data"
|
||||||
"cloud.o-forge.io/core/oc-lib/config"
|
"cloud.o-forge.io/core/oc-lib/models/resources/compute"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/models"
|
"cloud.o-forge.io/core/oc-lib/models/resources/processing"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/pricing"
|
"cloud.o-forge.io/core/oc-lib/models/resources/storage"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/peer"
|
w "cloud.o-forge.io/core/oc-lib/models/resources/workflow"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
"github.com/biter777/countries"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// AbstractResource is the struct containing all of the attributes commons to all ressources
|
// AbstractResource is the struct containing all of the attributes commons to all ressources
|
||||||
type AbstractResource struct {
|
|
||||||
utils.AbstractObject // AbstractObject contains the basic fields of an object (id, name)
|
// Resource is the interface to be implemented by all classes inheriting from Resource to have the same behavior
|
||||||
Type string `json:"type,omitempty" bson:"type,omitempty"` // Type is the type of the resource
|
|
||||||
Logo string `json:"logo,omitempty" bson:"logo,omitempty" validate:"required"` // Logo is the logo of the resource
|
// http://www.inanzzz.com/index.php/post/wqbs/a-basic-usage-of-int-and-string-enum-types-in-golang
|
||||||
Description string `json:"description,omitempty" bson:"description,omitempty"` // Description is the description of the resource
|
type ResourceSet struct {
|
||||||
ShortDescription string `json:"short_description,omitempty" bson:"short_description,omitempty" validate:"required"` // ShortDescription is the short description of the resource
|
Datas []string `bson:"datas,omitempty" json:"datas,omitempty"`
|
||||||
Owners []utils.Owner `json:"owners,omitempty" bson:"owners,omitempty"` // Owners is the list of owners of the resource
|
Storages []string `bson:"storages,omitempty" json:"storages,omitempty"`
|
||||||
UsageRestrictions string `bson:"usage_restrictions,omitempty" json:"usage_restrictions,omitempty"`
|
Processings []string `bson:"processings,omitempty" json:"processings,omitempty"`
|
||||||
SelectedInstanceIndex *int `json:"selected_instance_index,omitempty" bson:"selected_instance_index,omitempty"` // SelectedInstance is the selected instance
|
Computes []string `bson:"computes,omitempty" json:"computes,omitempty"`
|
||||||
|
Workflows []string `bson:"workflows,omitempty" json:"workflows,omitempty"`
|
||||||
|
|
||||||
|
DataResources []*data.DataResource `bson:"-" json:"data_resources,omitempty"`
|
||||||
|
StorageResources []*storage.StorageResource `bson:"-" json:"storage_resources,omitempty"`
|
||||||
|
ProcessingResources []*processing.ProcessingResource `bson:"-" json:"processing_resources,omitempty"`
|
||||||
|
ComputeResources []*compute.ComputeResource `bson:"-" json:"compute_resources,omitempty"`
|
||||||
|
WorkflowResources []*w.WorkflowResource `bson:"-" json:"workflow_resources,omitempty"`
|
||||||
}
|
}
|
||||||
|
|
||||||
func (r *AbstractResource) GetSelectedInstance() ResourceInstanceITF {
|
type ItemResource struct {
|
||||||
return nil
|
Data *data.DataResource `bson:"data,omitempty" json:"data,omitempty"`
|
||||||
|
Processing *processing.ProcessingResource `bson:"processing,omitempty" json:"processing,omitempty"`
|
||||||
|
Storage *storage.StorageResource `bson:"storage,omitempty" json:"storage,omitempty"`
|
||||||
|
Compute *compute.ComputeResource `bson:"compute,omitempty" json:"compute,omitempty"`
|
||||||
|
Workflow *w.WorkflowResource `bson:"workflow,omitempty" json:"workflow,omitempty"`
|
||||||
}
|
}
|
||||||
|
|
||||||
func (r *AbstractResource) GetType() string {
|
func (i *ItemResource) GetAbstractRessource() *resource_model.AbstractResource {
|
||||||
return tools.INVALID.String()
|
|
||||||
}
|
if(i.Data != nil){
|
||||||
|
return &i.Data.AbstractResource
|
||||||
func (r *AbstractResource) StoreDraftDefault() {
|
|
||||||
r.IsDraft = true
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *AbstractResource) CanUpdate(set utils.DBObject) (bool, utils.DBObject) {
|
|
||||||
if r.IsDraft != set.IsDrafted() && set.IsDrafted() {
|
|
||||||
return true, set // only state can be updated
|
|
||||||
}
|
}
|
||||||
return r.IsDraft != set.IsDrafted() && set.IsDrafted(), set
|
if(i.Processing != nil){
|
||||||
}
|
return &i.Processing.AbstractResource
|
||||||
|
|
||||||
func (r *AbstractResource) CanDelete() bool {
|
|
||||||
return r.IsDraft // only draft bookings can be deleted
|
|
||||||
}
|
|
||||||
|
|
||||||
type AbstractInstanciatedResource[T ResourceInstanceITF] struct {
|
|
||||||
AbstractResource // AbstractResource contains the basic fields of an object (id, name)
|
|
||||||
Instances []T `json:"instances,omitempty" bson:"instances,omitempty"` // Bill is the bill of the resource // Bill is the bill of the resource
|
|
||||||
}
|
|
||||||
|
|
||||||
func (abs *AbstractInstanciatedResource[T]) ConvertToPricedResource(t tools.DataType, request *tools.APIRequest) pricing.PricedItemITF {
|
|
||||||
instances := map[string]string{}
|
|
||||||
profiles := []pricing.PricingProfileITF{}
|
|
||||||
for _, instance := range abs.Instances {
|
|
||||||
instances[instance.GetID()] = instance.GetName()
|
|
||||||
profiles = instance.GetPricingsProfiles(request.PeerID, request.Groups)
|
|
||||||
}
|
}
|
||||||
var profile pricing.PricingProfileITF
|
if(i.Storage != nil){
|
||||||
if t := abs.GetSelectedInstance(); t != nil {
|
return &i.Storage.AbstractResource
|
||||||
profile = t.GetProfile()
|
|
||||||
}
|
}
|
||||||
if profile == nil && len(profiles) > 0 {
|
if(i.Compute != nil){
|
||||||
profile = profiles[0]
|
return &i.Compute.AbstractResource
|
||||||
}
|
}
|
||||||
return &PricedResource{
|
if(i.Workflow != nil){
|
||||||
Name: abs.Name,
|
return &i.Workflow.AbstractResource
|
||||||
Logo: abs.Logo,
|
|
||||||
ResourceID: abs.UUID,
|
|
||||||
ResourceType: t,
|
|
||||||
InstancesRefs: instances,
|
|
||||||
SelectedPricing: profile,
|
|
||||||
CreatorID: abs.CreatorID,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (abs *AbstractInstanciatedResource[T]) ClearEnv() utils.DBObject {
|
|
||||||
for _, instance := range abs.Instances {
|
|
||||||
instance.ClearEnv()
|
|
||||||
}
|
|
||||||
return abs
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *AbstractInstanciatedResource[T]) GetSelectedInstance() ResourceInstanceITF {
|
|
||||||
if r.SelectedInstanceIndex != nil && len(r.Instances) > *r.SelectedInstanceIndex {
|
|
||||||
return r.Instances[*r.SelectedInstanceIndex]
|
|
||||||
}
|
|
||||||
if len(r.Instances) > 0 {
|
|
||||||
return r.Instances[0]
|
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (abs *AbstractInstanciatedResource[T]) SetAllowedInstances(request *tools.APIRequest) {
|
|
||||||
if request != nil && request.PeerID == abs.CreatorID && request.PeerID != "" {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
abs.Instances = VerifyAuthAction[T](abs.Instances, request)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *AbstractInstanciatedResource[T]) Trim() {
|
|
||||||
d.Type = d.GetType()
|
|
||||||
if ok, _ := (&peer.Peer{AbstractObject: utils.AbstractObject{UUID: d.CreatorID}}).IsMySelf(); !ok {
|
|
||||||
for _, instance := range d.Instances {
|
|
||||||
instance.ClearPeerGroups()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (abs *AbstractInstanciatedResource[T]) VerifyAuth(request *tools.APIRequest) bool {
|
|
||||||
return len(VerifyAuthAction[T](abs.Instances, request)) > 0 || abs.AbstractObject.VerifyAuth(request)
|
|
||||||
}
|
|
||||||
|
|
||||||
func VerifyAuthAction[T ResourceInstanceITF](baseInstance []T, request *tools.APIRequest) []T {
|
|
||||||
instances := []T{}
|
|
||||||
for _, instance := range baseInstance {
|
|
||||||
_, peerGroups := instance.GetPeerGroups()
|
|
||||||
for _, peers := range peerGroups {
|
|
||||||
if request == nil {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if grps, ok := peers[request.PeerID]; ok || config.GetConfig().Whitelist {
|
|
||||||
if (ok && slices.Contains(grps, "*")) || (!ok && config.GetConfig().Whitelist) {
|
|
||||||
instances = append(instances, instance)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
for _, grp := range grps {
|
|
||||||
if slices.Contains(request.Groups, grp) {
|
|
||||||
instances = append(instances, instance)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return instances
|
|
||||||
}
|
|
||||||
|
|
||||||
type GeoPoint struct {
|
|
||||||
Latitude float64 `json:"latitude,omitempty" bson:"latitude,omitempty"`
|
|
||||||
Longitude float64 `json:"longitude,omitempty" bson:"longitude,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type ResourceInstance[T ResourcePartnerITF] struct {
|
|
||||||
utils.AbstractObject
|
|
||||||
Location GeoPoint `json:"location,omitempty" bson:"location,omitempty"`
|
|
||||||
Country countries.CountryCode `json:"country,omitempty" bson:"country,omitempty"`
|
|
||||||
AccessProtocol string `json:"access_protocol,omitempty" bson:"access_protocol,omitempty"`
|
|
||||||
Env []models.Param `json:"env,omitempty" bson:"env,omitempty"`
|
|
||||||
Inputs []models.Param `json:"inputs,omitempty" bson:"inputs,omitempty"`
|
|
||||||
Outputs []models.Param `json:"outputs,omitempty" bson:"outputs,omitempty"`
|
|
||||||
SelectedPartnershipIndex int `json:"selected_partnership_index,omitempty" bson:"selected_partnership_index,omitempty"`
|
|
||||||
SelectedBuyingStrategy int `json:"selected_buying_strategy,omitempty" bson:"selected_buying_strategy,omitempty"`
|
|
||||||
SelectedStrategy int `json:"selected_strategy,omitempty" bson:"selected_strategy,omitempty"`
|
|
||||||
Partnerships []T `json:"partnerships,omitempty" bson:"partnerships,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ri *ResourceInstance[T]) ClearEnv() {
|
|
||||||
ri.Env = []models.Param{}
|
|
||||||
ri.Inputs = []models.Param{}
|
|
||||||
ri.Outputs = []models.Param{}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ri *ResourceInstance[T]) GetProfile() pricing.PricingProfileITF {
|
|
||||||
if len(ri.Partnerships) > ri.SelectedPartnershipIndex {
|
|
||||||
prts := ri.Partnerships[ri.SelectedPartnershipIndex]
|
|
||||||
return prts.GetProfile(ri.SelectedBuyingStrategy, ri.SelectedBuyingStrategy)
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ri *ResourceInstance[T]) GetSelectedPartnership(peerID string, groups []string) ResourcePartnerITF {
|
|
||||||
if len(ri.Partnerships) > ri.SelectedPartnershipIndex {
|
|
||||||
return ri.Partnerships[ri.SelectedPartnershipIndex]
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ri *ResourceInstance[T]) GetPartnerships(peerID string, groups []string) []ResourcePartnerITF {
|
|
||||||
partners := []ResourcePartnerITF{}
|
|
||||||
for _, p := range ri.Partnerships {
|
|
||||||
if p.GetPeerGroups()[peerID] != nil {
|
|
||||||
for _, g := range p.GetPeerGroups()[peerID] {
|
|
||||||
if slices.Contains(groups, g) {
|
|
||||||
partners = append(partners, p)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return partners
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ri *ResourceInstance[T]) GetPricingsProfiles(peerID string, groups []string) []pricing.PricingProfileITF {
|
|
||||||
pricings := []pricing.PricingProfileITF{}
|
|
||||||
for _, p := range ri.Partnerships {
|
|
||||||
pricings = append(pricings, p.GetPricingsProfiles(peerID, groups)...)
|
|
||||||
}
|
|
||||||
return pricings
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ri *ResourceInstance[T]) GetPeerGroups() ([]ResourcePartnerITF, []map[string][]string) {
|
|
||||||
groups := []map[string][]string{}
|
|
||||||
partners := []ResourcePartnerITF{}
|
|
||||||
for _, p := range ri.Partnerships {
|
|
||||||
partners = append(partners, p)
|
|
||||||
groups = append(groups, p.GetPeerGroups())
|
|
||||||
}
|
|
||||||
return partners, groups
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ri *ResourceInstance[T]) ClearPeerGroups() {
|
|
||||||
for _, p := range ri.Partnerships {
|
|
||||||
p.ClearPeerGroups()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
type ResourcePartnerShip[T pricing.PricingProfileITF] struct {
|
|
||||||
Namespace string `json:"namespace" bson:"namespace" default:"default-namespace"`
|
|
||||||
PeerGroups map[string][]string `json:"peer_groups,omitempty" bson:"peer_groups,omitempty"`
|
|
||||||
PricingProfiles map[int]map[int]T `json:"pricing_profiles,omitempty" bson:"pricing_profiles,omitempty"`
|
|
||||||
// to upgrade pricing profiles. to be a map BuyingStrategy, map of Strategy
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ri *ResourcePartnerShip[T]) GetProfile(buying int, strategy int) pricing.PricingProfileITF {
|
|
||||||
if strat, ok := ri.PricingProfiles[buying]; ok {
|
|
||||||
if profile, ok := strat[strategy]; ok {
|
|
||||||
return profile
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
|
||||||
Le pricing doit être selectionné lors d'un scheduling...
|
|
||||||
le type de paiement défini le type de stratégie de paiement
|
|
||||||
note : il faut rajouté - une notion de facturation
|
|
||||||
Une order est l'ensemble de la commande... un booking une réservation, une purchase un acte d'achat.
|
|
||||||
Une bill (facture) représente alors... l'emission d'une facture à un instant T en but d'être honorée, envoyée ... etc.
|
|
||||||
*/
|
|
||||||
func (ri *ResourcePartnerShip[T]) GetPricingsProfiles(peerID string, groups []string) []pricing.PricingProfileITF {
|
|
||||||
profiles := []pricing.PricingProfileITF{}
|
|
||||||
if ri.PeerGroups[peerID] == nil {
|
|
||||||
return profiles
|
|
||||||
}
|
|
||||||
for _, p := range ri.PeerGroups[peerID] {
|
|
||||||
if slices.Contains(groups, p) || slices.Contains(groups, "*") {
|
|
||||||
for _, ri := range ri.PricingProfiles {
|
|
||||||
for _, i := range ri {
|
|
||||||
profiles = append(profiles, i)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return profiles
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return profiles
|
|
||||||
}
|
|
||||||
|
|
||||||
func (rp *ResourcePartnerShip[T]) GetPeerGroups() map[string][]string {
|
|
||||||
return rp.PeerGroups
|
|
||||||
}
|
|
||||||
|
|
||||||
func (rp *ResourcePartnerShip[T]) ClearPeerGroups() {
|
|
||||||
rp.PeerGroups = map[string][]string{}
|
|
||||||
}
|
|
@ -1,106 +0,0 @@
|
|||||||
package resources
|
|
||||||
|
|
||||||
import (
|
|
||||||
"errors"
|
|
||||||
"slices"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/dbs"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/logs"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
)
|
|
||||||
|
|
||||||
type ResourceMongoAccessor[T ResourceInterface] struct {
|
|
||||||
utils.AbstractAccessor // AbstractAccessor contains the basic fields of an accessor (model, caller)
|
|
||||||
generateData func() utils.DBObject
|
|
||||||
}
|
|
||||||
|
|
||||||
// New creates a new instance of the computeMongoAccessor
|
|
||||||
func NewAccessor[T ResourceInterface](t tools.DataType, request *tools.APIRequest, g func() utils.DBObject) *ResourceMongoAccessor[T] {
|
|
||||||
if !slices.Contains([]tools.DataType{
|
|
||||||
tools.COMPUTE_RESOURCE, tools.STORAGE_RESOURCE,
|
|
||||||
tools.PROCESSING_RESOURCE, tools.WORKFLOW_RESOURCE,
|
|
||||||
tools.DATA_RESOURCE,
|
|
||||||
}, t) {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
return &ResourceMongoAccessor[T]{
|
|
||||||
AbstractAccessor: utils.AbstractAccessor{
|
|
||||||
Logger: logs.CreateLogger(t.String()), // Create a logger with the data type
|
|
||||||
Request: request,
|
|
||||||
Type: t,
|
|
||||||
},
|
|
||||||
generateData: g,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
|
||||||
* Nothing special here, just the basic CRUD operations
|
|
||||||
*/
|
|
||||||
func (dca *ResourceMongoAccessor[T]) DeleteOne(id string) (utils.DBObject, int, error) {
|
|
||||||
return utils.GenericDeleteOne(id, dca)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (dca *ResourceMongoAccessor[T]) UpdateOne(set utils.DBObject, id string) (utils.DBObject, int, error) {
|
|
||||||
if dca.GetType() == tools.COMPUTE_RESOURCE {
|
|
||||||
return nil, 404, errors.New("can't update a non existing computing units resource not reported onto compute units catalog")
|
|
||||||
}
|
|
||||||
set.(T).Trim()
|
|
||||||
return utils.GenericUpdateOne(set, id, dca, dca.generateData())
|
|
||||||
}
|
|
||||||
|
|
||||||
func (dca *ResourceMongoAccessor[T]) StoreOne(data utils.DBObject) (utils.DBObject, int, error) {
|
|
||||||
if dca.GetType() == tools.COMPUTE_RESOURCE {
|
|
||||||
return nil, 404, errors.New("can't create a non existing computing units resource not reported onto compute units catalog")
|
|
||||||
}
|
|
||||||
data.(T).Trim()
|
|
||||||
return utils.GenericStoreOne(data, dca)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (dca *ResourceMongoAccessor[T]) CopyOne(data utils.DBObject) (utils.DBObject, int, error) {
|
|
||||||
if dca.GetType() == tools.COMPUTE_RESOURCE {
|
|
||||||
return nil, 404, errors.New("can't copy/publish a non existing computing units resource not reported onto compute units catalog")
|
|
||||||
}
|
|
||||||
return dca.StoreOne(data)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (dca *ResourceMongoAccessor[T]) LoadOne(id string) (utils.DBObject, int, error) {
|
|
||||||
return utils.GenericLoadOne[T](id, func(d utils.DBObject) (utils.DBObject, int, error) {
|
|
||||||
d.(T).SetAllowedInstances(dca.Request)
|
|
||||||
return d, 200, nil
|
|
||||||
}, dca)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (wfa *ResourceMongoAccessor[T]) LoadAll(isDraft bool) ([]utils.ShallowDBObject, int, error) {
|
|
||||||
return utils.GenericLoadAll[T](func(d utils.DBObject) utils.ShallowDBObject {
|
|
||||||
d.(T).SetAllowedInstances(wfa.Request)
|
|
||||||
return d
|
|
||||||
}, isDraft, wfa)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (wfa *ResourceMongoAccessor[T]) Search(filters *dbs.Filters, search string, isDraft bool) ([]utils.ShallowDBObject, int, error) {
|
|
||||||
if filters == nil && search == "*" {
|
|
||||||
return utils.GenericLoadAll[T](func(d utils.DBObject) utils.ShallowDBObject {
|
|
||||||
d.(T).SetAllowedInstances(wfa.Request)
|
|
||||||
return d
|
|
||||||
}, isDraft, wfa)
|
|
||||||
}
|
|
||||||
return utils.GenericSearch[T](filters, search, wfa.getResourceFilter(search),
|
|
||||||
func(d utils.DBObject) utils.ShallowDBObject {
|
|
||||||
d.(T).SetAllowedInstances(wfa.Request)
|
|
||||||
return d
|
|
||||||
}, isDraft, wfa)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (abs *ResourceMongoAccessor[T]) getResourceFilter(search string) *dbs.Filters {
|
|
||||||
return &dbs.Filters{
|
|
||||||
Or: map[string][]dbs.Filter{ // filter by like name, short_description, description, owner, url if no filters are provided
|
|
||||||
"abstractintanciatedresource.abstractresource.abstractobject.name": {{Operator: dbs.LIKE.String(), Value: search}},
|
|
||||||
"abstractintanciatedresource.abstractresource.type": {{Operator: dbs.LIKE.String(), Value: search}},
|
|
||||||
"abstractintanciatedresource.abstractresource.short_description": {{Operator: dbs.LIKE.String(), Value: search}},
|
|
||||||
"abstractintanciatedresource.abstractresource.description": {{Operator: dbs.LIKE.String(), Value: search}},
|
|
||||||
"abstractintanciatedresource.abstractresource.owners.name": {{Operator: dbs.LIKE.String(), Value: search}},
|
|
||||||
"abstractintanciatedresource.abstractresource.abstractobject.creator_id": {{Operator: dbs.EQUAL.String(), Value: search}},
|
|
||||||
},
|
|
||||||
}
|
|
||||||
}
|
|
@ -1,204 +0,0 @@
|
|||||||
package resources
|
|
||||||
|
|
||||||
import (
|
|
||||||
"errors"
|
|
||||||
"fmt"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/enum"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/models"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/pricing"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
)
|
|
||||||
|
|
||||||
/*
|
|
||||||
* StorageResource is a struct that represents a storage resource
|
|
||||||
* it defines the resource storage
|
|
||||||
*/
|
|
||||||
type StorageResource struct {
|
|
||||||
AbstractInstanciatedResource[*StorageResourceInstance] // AbstractResource contains the basic fields of an object (id, name)
|
|
||||||
StorageType enum.StorageType `bson:"storage_type" json:"storage_type" default:"-1"` // Type is the type of the storage
|
|
||||||
Acronym string `bson:"acronym,omitempty" json:"acronym,omitempty"` // Acronym is the acronym of the storage
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *StorageResource) GetAccessor(request *tools.APIRequest) utils.Accessor {
|
|
||||||
return NewAccessor[*StorageResource](tools.STORAGE_RESOURCE, request, func() utils.DBObject { return &StorageResource{} }) // Create a new instance of the accessor
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *StorageResource) GetType() string {
|
|
||||||
return tools.STORAGE_RESOURCE.String()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (abs *StorageResource) ConvertToPricedResource(t tools.DataType, request *tools.APIRequest) pricing.PricedItemITF {
|
|
||||||
if t != tools.STORAGE_RESOURCE {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
p := abs.AbstractInstanciatedResource.ConvertToPricedResource(t, request)
|
|
||||||
priced := p.(*PricedResource)
|
|
||||||
return &PricedStorageResource{
|
|
||||||
PricedResource: *priced,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
type StorageResourceInstance struct {
|
|
||||||
ResourceInstance[*StorageResourcePartnership]
|
|
||||||
Source string `bson:"source,omitempty" json:"source,omitempty"` // Source is the source of the storage
|
|
||||||
Path string `bson:"path,omitempty" json:"path,omitempty"` // Path is the store folders in the source
|
|
||||||
Local bool `bson:"local" json:"local"`
|
|
||||||
SecurityLevel string `bson:"security_level,omitempty" json:"security_level,omitempty"`
|
|
||||||
SizeType enum.StorageSize `bson:"size_type" json:"size_type" default:"0"` // SizeType is the type of the storage size
|
|
||||||
SizeGB int64 `bson:"size,omitempty" json:"size,omitempty"` // Size is the size of the storage
|
|
||||||
Encryption bool `bson:"encryption,omitempty" json:"encryption,omitempty"` // Encryption is a flag that indicates if the storage is encrypted
|
|
||||||
Redundancy string `bson:"redundancy,omitempty" json:"redundancy,omitempty"` // Redundancy is the redundancy of the storage
|
|
||||||
Throughput string `bson:"throughput,omitempty" json:"throughput,omitempty"` // Throughput is the throughput of the storage
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ri *StorageResourceInstance) ClearEnv() {
|
|
||||||
ri.Env = []models.Param{}
|
|
||||||
ri.Inputs = []models.Param{}
|
|
||||||
ri.Outputs = []models.Param{}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ri *StorageResourceInstance) StoreDraftDefault() {
|
|
||||||
found := false
|
|
||||||
for _, p := range ri.ResourceInstance.Env {
|
|
||||||
if p.Attr == "source" {
|
|
||||||
found = true
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if !found {
|
|
||||||
ri.ResourceInstance.Env = append(ri.ResourceInstance.Env, models.Param{
|
|
||||||
Attr: "source",
|
|
||||||
Value: ri.Source,
|
|
||||||
Readonly: true,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
ri.ResourceInstance.StoreDraftDefault()
|
|
||||||
}
|
|
||||||
|
|
||||||
type StorageResourcePartnership struct {
|
|
||||||
ResourcePartnerShip[*StorageResourcePricingProfile]
|
|
||||||
MaxSizeGBAllowed float64 `json:"allowed_gb,omitempty" bson:"allowed_gb,omitempty"`
|
|
||||||
OnlyEncryptedAllowed bool `json:"personal_data_allowed,omitempty" bson:"personal_data_allowed,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type PrivilegeStoragePricingStrategy int
|
|
||||||
|
|
||||||
const (
|
|
||||||
BASIC_STORAGE PrivilegeStoragePricingStrategy = iota
|
|
||||||
GARANTED_ON_DELAY_STORAGE
|
|
||||||
GARANTED_STORAGE
|
|
||||||
)
|
|
||||||
|
|
||||||
func PrivilegeStoragePricingStrategyList() []PrivilegeStoragePricingStrategy {
|
|
||||||
return []PrivilegeStoragePricingStrategy{BASIC_STORAGE, GARANTED_ON_DELAY_STORAGE, GARANTED_STORAGE}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (t PrivilegeStoragePricingStrategy) String() string {
|
|
||||||
return [...]string{"NO MEMORY HOLDING", "KEEPED ON MEMORY GARANTED DURING DELAY", "KEEPED ON MEMORY GARANTED"}[t]
|
|
||||||
}
|
|
||||||
|
|
||||||
type StorageResourcePricingStrategy int
|
|
||||||
|
|
||||||
const (
|
|
||||||
PER_DATA_STORED StorageResourcePricingStrategy = iota + 6
|
|
||||||
PER_TB_STORED
|
|
||||||
PER_GB_STORED
|
|
||||||
PER_MB_STORED
|
|
||||||
PER_KB_STORED
|
|
||||||
)
|
|
||||||
|
|
||||||
func StorageResourcePricingStrategyList() []StorageResourcePricingStrategy {
|
|
||||||
return []StorageResourcePricingStrategy{PER_DATA_STORED, PER_TB_STORED, PER_GB_STORED, PER_MB_STORED, PER_KB_STORED}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (t StorageResourcePricingStrategy) String() string {
|
|
||||||
return [...]string{"PER DATA STORED", "PER TB STORED", "PER GB STORED", "PER MB STORED", "PER KB STORED"}[t]
|
|
||||||
}
|
|
||||||
|
|
||||||
func (t StorageResourcePricingStrategy) GetStrategy() string {
|
|
||||||
return [...]string{"PER_DATA_STORED", "PER_GB_STORED", "PER_MB_STORED", "PER_KB_STORED"}[t]
|
|
||||||
}
|
|
||||||
|
|
||||||
func (t StorageResourcePricingStrategy) GetStrategyValue() int {
|
|
||||||
return int(t)
|
|
||||||
}
|
|
||||||
|
|
||||||
func ToStorageResourcePricingStrategy(i int) StorageResourcePricingStrategy {
|
|
||||||
return StorageResourcePricingStrategy(i)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (t StorageResourcePricingStrategy) GetQuantity(amountOfDataGB float64) (float64, error) {
|
|
||||||
switch t {
|
|
||||||
case PER_DATA_STORED:
|
|
||||||
return amountOfDataGB, nil
|
|
||||||
case PER_TB_STORED:
|
|
||||||
return amountOfDataGB * 1000, nil
|
|
||||||
case PER_GB_STORED:
|
|
||||||
return amountOfDataGB, nil
|
|
||||||
case PER_MB_STORED:
|
|
||||||
return (amountOfDataGB * 1000), nil
|
|
||||||
case PER_KB_STORED:
|
|
||||||
return amountOfDataGB * 1000000, nil
|
|
||||||
}
|
|
||||||
return 0, errors.New("pricing strategy not found")
|
|
||||||
}
|
|
||||||
|
|
||||||
type StorageResourcePricingProfile struct {
|
|
||||||
pricing.ExploitPricingProfile[StorageResourcePricingStrategy] // ExploitPricingProfile is the pricing profile of a storage it means that we exploit the resource for an amount of continuous time
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *StorageResourcePricingProfile) GetPurchase() pricing.BuyingStrategy {
|
|
||||||
return p.Pricing.BuyingStrategy
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *StorageResourcePricingProfile) IsPurchasable() bool {
|
|
||||||
return p.Pricing.BuyingStrategy != pricing.UNDEFINED_SUBSCRIPTION
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *StorageResourcePricingProfile) IsBooked() bool {
|
|
||||||
if p.Pricing.BuyingStrategy == pricing.PERMANENT {
|
|
||||||
p.Pricing.BuyingStrategy = pricing.SUBSCRIPTION
|
|
||||||
}
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *StorageResourcePricingProfile) GetPrice(amountOfData float64, val float64, start time.Time, end time.Time, params ...string) (float64, error) {
|
|
||||||
return p.Pricing.GetPrice(amountOfData, val, start, &end)
|
|
||||||
}
|
|
||||||
|
|
||||||
type PricedStorageResource struct {
|
|
||||||
PricedResource
|
|
||||||
UsageStorageGB float64 `json:"storage_gb,omitempty" bson:"storage_gb,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *PricedStorageResource) GetType() tools.DataType {
|
|
||||||
return tools.STORAGE_RESOURCE
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *PricedStorageResource) GetPrice() (float64, error) {
|
|
||||||
fmt.Println("GetPrice", r.UsageStart, r.UsageEnd)
|
|
||||||
now := time.Now()
|
|
||||||
if r.UsageStart == nil {
|
|
||||||
r.UsageStart = &now
|
|
||||||
}
|
|
||||||
if r.UsageEnd == nil {
|
|
||||||
add := r.UsageStart.Add(time.Duration(1 * time.Hour))
|
|
||||||
r.UsageEnd = &add
|
|
||||||
}
|
|
||||||
if r.SelectedPricing == nil {
|
|
||||||
return 0, errors.New("pricing profile must be set on Priced Storage" + r.ResourceID)
|
|
||||||
}
|
|
||||||
pricing := r.SelectedPricing
|
|
||||||
var err error
|
|
||||||
amountOfData := float64(1)
|
|
||||||
if pricing.GetOverrideStrategyValue() >= 0 {
|
|
||||||
amountOfData, err = ToStorageResourcePricingStrategy(pricing.GetOverrideStrategyValue()).GetQuantity(r.UsageStorageGB)
|
|
||||||
if err != nil {
|
|
||||||
return 0, err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return pricing.GetPrice(amountOfData, r.ExplicitBookingDurationS, *r.UsageStart, *r.UsageEnd)
|
|
||||||
}
|
|
84
models/resources/storage/storage.go
Normal file
84
models/resources/storage/storage.go
Normal file
@ -0,0 +1,84 @@
|
|||||||
|
package storage
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/resource_model"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/utils"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/tools"
|
||||||
|
)
|
||||||
|
|
||||||
|
type StorageSize int
|
||||||
|
|
||||||
|
// StorageType - Enum that defines the type of storage
|
||||||
|
const (
|
||||||
|
GB StorageSize = iota
|
||||||
|
MB
|
||||||
|
KB
|
||||||
|
)
|
||||||
|
|
||||||
|
var argoType = [...]string{
|
||||||
|
"Gi",
|
||||||
|
"Mi",
|
||||||
|
"Ki",
|
||||||
|
}
|
||||||
|
|
||||||
|
// New creates a new instance of the StorageResource struct
|
||||||
|
func (dma StorageSize) ToArgo() string {
|
||||||
|
return argoType[dma]
|
||||||
|
}
|
||||||
|
|
||||||
|
// enum of a data type
|
||||||
|
type StorageType int
|
||||||
|
|
||||||
|
const (
|
||||||
|
FILE = iota
|
||||||
|
STREAM
|
||||||
|
API
|
||||||
|
DATABASE
|
||||||
|
S3
|
||||||
|
MEMORY
|
||||||
|
HARDWARE
|
||||||
|
)
|
||||||
|
|
||||||
|
/*
|
||||||
|
* StorageResource is a struct that represents a storage resource
|
||||||
|
* it defines the resource storage
|
||||||
|
*/
|
||||||
|
type StorageResource struct {
|
||||||
|
resource_model.AbstractResource // AbstractResource contains the basic fields of an object (id, name)
|
||||||
|
resource_model.WebResource
|
||||||
|
Type StorageType `bson:"type,omitempty" json:"type,omitempty"` // Type is the type of the storage
|
||||||
|
Acronym string `bson:"acronym,omitempty" json:"acronym,omitempty"` // Acronym is the acronym of the storage
|
||||||
|
SizeType StorageSize `bson:"size_type" json:"size_type" default:"0"` // SizeType is the type of the storage size
|
||||||
|
Size uint `bson:"size,omitempty" json:"size,omitempty"` // Size is the size of the storage
|
||||||
|
Local bool `bson:"local" json:"local"` // Local is a flag that indicates if the storage is local
|
||||||
|
Encryption bool `bson:"encryption,omitempty" json:"encryption,omitempty"` // Encryption is a flag that indicates if the storage is encrypted
|
||||||
|
Redundancy string `bson:"redundancy,omitempty" json:"redundancy,omitempty"` // Redundancy is the redundancy of the storage
|
||||||
|
Throughput string `bson:"throughput,omitempty" json:"throughput,omitempty"` // Throughput is the throughput of the storage
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dma *StorageResource) Deserialize(j map[string]interface{}) utils.DBObject {
|
||||||
|
b, err := json.Marshal(j)
|
||||||
|
if err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
json.Unmarshal(b, dma)
|
||||||
|
return dma
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dma *StorageResource) Serialize() map[string]interface{} {
|
||||||
|
var m map[string]interface{}
|
||||||
|
b, err := json.Marshal(dma)
|
||||||
|
if err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
json.Unmarshal(b, &m)
|
||||||
|
return m
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *StorageResource) GetAccessor(caller *tools.HTTPCaller) utils.Accessor {
|
||||||
|
data := New() // Create a new instance of the accessor
|
||||||
|
data.Init(tools.STORAGE_RESOURCE, caller) // Initialize the accessor with the STORAGE_RESOURCE model type
|
||||||
|
return data
|
||||||
|
}
|
114
models/resources/storage/storage_mongo_accessor.go
Normal file
114
models/resources/storage/storage_mongo_accessor.go
Normal file
@ -0,0 +1,114 @@
|
|||||||
|
package storage
|
||||||
|
|
||||||
|
import (
|
||||||
|
"cloud.o-forge.io/core/oc-lib/dbs"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/dbs/mongo"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/resource_model"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/utils"
|
||||||
|
)
|
||||||
|
|
||||||
|
type storageMongoAccessor struct {
|
||||||
|
utils.AbstractAccessor // AbstractAccessor contains the basic fields of an accessor (model, caller)
|
||||||
|
}
|
||||||
|
|
||||||
|
// New creates a new instance of the storageMongoAccessor
|
||||||
|
func New() *storageMongoAccessor {
|
||||||
|
return &storageMongoAccessor{}
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Nothing special here, just the basic CRUD operations
|
||||||
|
*/
|
||||||
|
|
||||||
|
func (sma *storageMongoAccessor) DeleteOne(id string) (utils.DBObject, int, error) {
|
||||||
|
return sma.GenericDeleteOne(id, sma)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (sma *storageMongoAccessor) UpdateOne(set utils.DBObject, id string) (utils.DBObject, int, error) {
|
||||||
|
set.(*StorageResource).ResourceModel = nil
|
||||||
|
return sma.GenericUpdateOne(set, id, sma, &StorageResource{})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (sma *storageMongoAccessor) StoreOne(data utils.DBObject) (utils.DBObject, int, error) {
|
||||||
|
data.(*StorageResource).ResourceModel = nil
|
||||||
|
return sma.GenericStoreOne(data, sma)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (sma *storageMongoAccessor) CopyOne(data utils.DBObject) (utils.DBObject, int, error) {
|
||||||
|
return sma.GenericStoreOne(data, sma)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (sma *storageMongoAccessor) LoadOne(id string) (utils.DBObject, int, error) {
|
||||||
|
|
||||||
|
var storage StorageResource
|
||||||
|
|
||||||
|
res_mongo, code, err := mongo.MONGOService.LoadOne(id, sma.GetType())
|
||||||
|
if err != nil {
|
||||||
|
sma.Logger.Error().Msg("Could not retrieve " + id + " from db. Error: " + err.Error())
|
||||||
|
return nil, code, err
|
||||||
|
}
|
||||||
|
|
||||||
|
res_mongo.Decode(&storage)
|
||||||
|
accessor := (&resource_model.ResourceModel{}).GetAccessor(nil)
|
||||||
|
resources, _, err := accessor.Search(nil, sma.GetType())
|
||||||
|
if err == nil && len(resources) > 0 {
|
||||||
|
storage.ResourceModel = resources[0].(*resource_model.ResourceModel)
|
||||||
|
}
|
||||||
|
return &storage, 200, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (wfa storageMongoAccessor) 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 []StorageResource
|
||||||
|
if err = res_mongo.All(mongo.MngoCtx, &results); err != nil {
|
||||||
|
return nil, 404, err
|
||||||
|
}
|
||||||
|
accessor := (&resource_model.ResourceModel{}).GetAccessor(nil)
|
||||||
|
resources, _, err := accessor.Search(nil, wfa.GetType())
|
||||||
|
for _, r := range results {
|
||||||
|
if err == nil && len(resources) > 0 {
|
||||||
|
r.ResourceModel = resources[0].(*resource_model.ResourceModel)
|
||||||
|
}
|
||||||
|
objs = append(objs, &r) // only get the abstract resource !
|
||||||
|
}
|
||||||
|
return objs, 200, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Search searches for storage resources in the database, given some filters OR a search string
|
||||||
|
func (wfa *storageMongoAccessor) 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 like name, short_description, description, owner, url if no filters are provided
|
||||||
|
"abstractresource.abstractobject.name": {{Operator: dbs.LIKE.String(), Value: search}},
|
||||||
|
"abstractresource.short_description": {{Operator: dbs.LIKE.String(), Value: search}},
|
||||||
|
"abstractresource.description": {{Operator: dbs.LIKE.String(), Value: search}},
|
||||||
|
"abstractresource.owner": {{Operator: dbs.LIKE.String(), Value: search}},
|
||||||
|
"abstractresource.source_url": {{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 []StorageResource
|
||||||
|
if err = res_mongo.All(mongo.MngoCtx, &results); err != nil {
|
||||||
|
return nil, 404, err
|
||||||
|
}
|
||||||
|
accessor := (&resource_model.ResourceModel{}).GetAccessor(nil)
|
||||||
|
resources, _, err := accessor.Search(nil, wfa.GetType())
|
||||||
|
for _, r := range results {
|
||||||
|
if err == nil && len(resources) > 0 {
|
||||||
|
r.ResourceModel = resources[0].(*resource_model.ResourceModel)
|
||||||
|
}
|
||||||
|
objs = append(objs, &r) // only get the abstract resource !
|
||||||
|
}
|
||||||
|
return objs, 200, nil
|
||||||
|
}
|
46
models/resources/storage/storage_test.go
Normal file
46
models/resources/storage/storage_test.go
Normal file
@ -0,0 +1,46 @@
|
|||||||
|
package storage
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/resource_model"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/utils"
|
||||||
|
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestStoreOneStorage(t *testing.T) {
|
||||||
|
s := StorageResource{Size: 123, WebResource: resource_model.WebResource{Protocol: "http", Path: "azerty.fr"},
|
||||||
|
AbstractResource: resource_model.AbstractResource{
|
||||||
|
AbstractObject: utils.AbstractObject{Name: "testData"},
|
||||||
|
Description: "Lorem Ipsum",
|
||||||
|
Logo: "azerty.com",
|
||||||
|
Owner: "toto",
|
||||||
|
OwnerLogo: "totoLogo",
|
||||||
|
SourceUrl: "azerty.fr",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
sma := New()
|
||||||
|
id, _, _ := sma.StoreOne(&s)
|
||||||
|
|
||||||
|
assert.NotEmpty(t, id)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestLoadOneStorage(t *testing.T) {
|
||||||
|
s := StorageResource{Size: 123, WebResource: resource_model.WebResource{Protocol: "http", Path: "azerty.fr"},
|
||||||
|
AbstractResource: resource_model.AbstractResource{
|
||||||
|
AbstractObject: utils.AbstractObject{Name: "testData"},
|
||||||
|
Description: "Lorem Ipsum",
|
||||||
|
Logo: "azerty.com",
|
||||||
|
Owner: "toto",
|
||||||
|
OwnerLogo: "totoLogo",
|
||||||
|
SourceUrl: "azerty.fr",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
sma := New()
|
||||||
|
new_s, _, _ := sma.StoreOne(&s)
|
||||||
|
|
||||||
|
assert.Equal(t, s, new_s)
|
||||||
|
}
|
@ -1,108 +0,0 @@
|
|||||||
package resources_test
|
|
||||||
|
|
||||||
import (
|
|
||||||
"testing"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/models"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/pricing"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/resources"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
|
|
||||||
"github.com/stretchr/testify/assert"
|
|
||||||
"github.com/stretchr/testify/require"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestComputeResource_GetType(t *testing.T) {
|
|
||||||
r := &resources.ComputeResource{}
|
|
||||||
assert.Equal(t, tools.COMPUTE_RESOURCE.String(), r.GetType())
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestComputeResource_GetAccessor(t *testing.T) {
|
|
||||||
req := &tools.APIRequest{}
|
|
||||||
cr := &resources.ComputeResource{}
|
|
||||||
accessor := cr.GetAccessor(req)
|
|
||||||
assert.NotNil(t, accessor)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestComputeResource_ConvertToPricedResource(t *testing.T) {
|
|
||||||
req := &tools.APIRequest{}
|
|
||||||
cr := &resources.ComputeResource{}
|
|
||||||
cr.UUID = "comp123"
|
|
||||||
cr.AbstractInstanciatedResource.UUID = cr.UUID
|
|
||||||
result := cr.ConvertToPricedResource(tools.COMPUTE_RESOURCE, req)
|
|
||||||
assert.NotNil(t, result)
|
|
||||||
assert.IsType(t, &resources.PricedComputeResource{}, result)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestComputeResourcePricingProfile_GetPrice_CPUs(t *testing.T) {
|
|
||||||
start := time.Now()
|
|
||||||
end := start.Add(1 * time.Hour)
|
|
||||||
profile := resources.ComputeResourcePricingProfile{
|
|
||||||
CPUsPrices: map[string]float64{"Xeon": 2.0},
|
|
||||||
ExploitPricingProfile: pricing.ExploitPricingProfile[pricing.TimePricingStrategy]{
|
|
||||||
AccessPricingProfile: pricing.AccessPricingProfile[pricing.TimePricingStrategy]{
|
|
||||||
Pricing: pricing.PricingStrategy[pricing.TimePricingStrategy]{Price: 1.0},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
price, err := profile.GetPrice(2, 3600, start, end, "cpus", "Xeon")
|
|
||||||
require.NoError(t, err)
|
|
||||||
assert.Greater(t, price, float64(0))
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestComputeResourcePricingProfile_GetPrice_InvalidParams(t *testing.T) {
|
|
||||||
profile := resources.ComputeResourcePricingProfile{}
|
|
||||||
_, err := profile.GetPrice(1, 3600, time.Now(), time.Now())
|
|
||||||
assert.Error(t, err)
|
|
||||||
assert.Equal(t, "params must be set", err.Error())
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestPricedComputeResource_GetPrice(t *testing.T) {
|
|
||||||
start := time.Now()
|
|
||||||
end := start.Add(1 * time.Hour)
|
|
||||||
r := resources.PricedComputeResource{
|
|
||||||
PricedResource: resources.PricedResource{
|
|
||||||
ResourceID: "comp456",
|
|
||||||
UsageStart: &start,
|
|
||||||
UsageEnd: &end,
|
|
||||||
ExplicitBookingDurationS: 3600,
|
|
||||||
},
|
|
||||||
CPUsLocated: map[string]float64{"Xeon": 2},
|
|
||||||
GPUsLocated: map[string]float64{"Tesla": 1},
|
|
||||||
RAMLocated: 4,
|
|
||||||
}
|
|
||||||
|
|
||||||
price, err := r.GetPrice()
|
|
||||||
require.NoError(t, err)
|
|
||||||
assert.Greater(t, price, float64(0))
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestPricedComputeResource_GetPrice_MissingProfile(t *testing.T) {
|
|
||||||
r := resources.PricedComputeResource{
|
|
||||||
PricedResource: resources.PricedResource{
|
|
||||||
ResourceID: "comp789",
|
|
||||||
},
|
|
||||||
}
|
|
||||||
_, err := r.GetPrice()
|
|
||||||
require.Error(t, err)
|
|
||||||
assert.Contains(t, err.Error(), "pricing profile must be set")
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestPricedComputeResource_FillWithDefaultProcessingUsage(t *testing.T) {
|
|
||||||
usage := &resources.ProcessingUsage{
|
|
||||||
CPUs: map[string]*models.CPU{"t": {Model: "Xeon", Cores: 4}},
|
|
||||||
GPUs: map[string]*models.GPU{"t1": {Model: "Tesla"}},
|
|
||||||
RAM: &models.RAM{SizeGb: 16},
|
|
||||||
}
|
|
||||||
r := &resources.PricedComputeResource{
|
|
||||||
CPUsLocated: make(map[string]float64),
|
|
||||||
GPUsLocated: make(map[string]float64),
|
|
||||||
RAMLocated: 0,
|
|
||||||
}
|
|
||||||
r.FillWithDefaultProcessingUsage(usage)
|
|
||||||
assert.Equal(t, float64(4), r.CPUsLocated["Xeon"])
|
|
||||||
assert.Equal(t, float64(1), r.GPUsLocated["Tesla"])
|
|
||||||
assert.Equal(t, float64(16), r.RAMLocated)
|
|
||||||
}
|
|
@ -1,119 +0,0 @@
|
|||||||
package resources_test
|
|
||||||
|
|
||||||
import (
|
|
||||||
"testing"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/models"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/pricing"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/resources"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
|
|
||||||
"github.com/stretchr/testify/assert"
|
|
||||||
"github.com/stretchr/testify/require"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestDataResource_GetType(t *testing.T) {
|
|
||||||
d := &resources.DataResource{}
|
|
||||||
assert.Equal(t, tools.DATA_RESOURCE.String(), d.GetType())
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestDataResource_GetAccessor(t *testing.T) {
|
|
||||||
req := &tools.APIRequest{}
|
|
||||||
acc := (&resources.DataResource{}).GetAccessor(req)
|
|
||||||
assert.NotNil(t, acc)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestDataResource_ConvertToPricedResource(t *testing.T) {
|
|
||||||
d := &resources.DataResource{}
|
|
||||||
d.UUID = "123"
|
|
||||||
res := d.ConvertToPricedResource(tools.DATA_RESOURCE, &tools.APIRequest{})
|
|
||||||
assert.IsType(t, &resources.PricedDataResource{}, res)
|
|
||||||
|
|
||||||
nilRes := d.ConvertToPricedResource(tools.PROCESSING_RESOURCE, &tools.APIRequest{})
|
|
||||||
assert.Nil(t, nilRes)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestDataInstance_StoreDraftDefault(t *testing.T) {
|
|
||||||
di := &resources.DataInstance{
|
|
||||||
Source: "test-src",
|
|
||||||
ResourceInstance: resources.ResourceInstance[*resources.DataResourcePartnership]{
|
|
||||||
Env: []models.Param{},
|
|
||||||
},
|
|
||||||
}
|
|
||||||
di.StoreDraftDefault()
|
|
||||||
assert.Len(t, di.ResourceInstance.Env, 1)
|
|
||||||
assert.Equal(t, "source", di.ResourceInstance.Env[0].Attr)
|
|
||||||
assert.Equal(t, "test-src", di.ResourceInstance.Env[0].Value)
|
|
||||||
|
|
||||||
// Call again, should not duplicate
|
|
||||||
di.StoreDraftDefault()
|
|
||||||
assert.Len(t, di.ResourceInstance.Env, 1)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestDataResourcePricingStrategy_GetQuantity(t *testing.T) {
|
|
||||||
tests := []struct {
|
|
||||||
strategy resources.DataResourcePricingStrategy
|
|
||||||
input float64
|
|
||||||
expected float64
|
|
||||||
}{
|
|
||||||
{resources.PER_DOWNLOAD, 1, 1},
|
|
||||||
{resources.PER_TB_DOWNLOADED, 1, 1000},
|
|
||||||
{resources.PER_GB_DOWNLOADED, 2.5, 2.5},
|
|
||||||
{resources.PER_MB_DOWNLOADED, 1, 0.001},
|
|
||||||
{resources.PER_KB_DOWNLOADED, 1, 0.000001},
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, tt := range tests {
|
|
||||||
q, err := tt.strategy.GetQuantity(tt.input)
|
|
||||||
require.NoError(t, err)
|
|
||||||
assert.InDelta(t, tt.expected, q, 1e-9)
|
|
||||||
}
|
|
||||||
|
|
||||||
_, err := resources.DataResourcePricingStrategy(999).GetQuantity(1)
|
|
||||||
assert.Error(t, err)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestDataResourcePricingProfile_IsPurchased(t *testing.T) {
|
|
||||||
profile := &resources.DataResourcePricingProfile{}
|
|
||||||
profile.Pricing.BuyingStrategy = pricing.SUBSCRIPTION
|
|
||||||
assert.True(t, profile.IsPurchasable())
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestPricedDataResource_GetPrice(t *testing.T) {
|
|
||||||
now := time.Now()
|
|
||||||
later := now.Add(1 * time.Hour)
|
|
||||||
mockPrice := 42.0
|
|
||||||
|
|
||||||
pricingProfile := &resources.DataResourcePricingProfile{AccessPricingProfile: pricing.AccessPricingProfile[resources.DataResourcePricingStrategy]{
|
|
||||||
Pricing: pricing.PricingStrategy[resources.DataResourcePricingStrategy]{Price: 42.0}},
|
|
||||||
}
|
|
||||||
pricingProfile.Pricing.OverrideStrategy = resources.PER_GB_DOWNLOADED
|
|
||||||
|
|
||||||
r := &resources.PricedDataResource{
|
|
||||||
PricedResource: resources.PricedResource{
|
|
||||||
UsageStart: &now,
|
|
||||||
UsageEnd: &later,
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
price, err := r.GetPrice()
|
|
||||||
require.NoError(t, err)
|
|
||||||
assert.Equal(t, mockPrice, price)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestPricedDataResource_GetPrice_NoProfiles(t *testing.T) {
|
|
||||||
r := &resources.PricedDataResource{
|
|
||||||
PricedResource: resources.PricedResource{
|
|
||||||
ResourceID: "test-resource",
|
|
||||||
},
|
|
||||||
}
|
|
||||||
_, err := r.GetPrice()
|
|
||||||
assert.Error(t, err)
|
|
||||||
assert.Contains(t, err.Error(), "pricing profile must be set")
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestPricedDataResource_GetType(t *testing.T) {
|
|
||||||
r := &resources.PricedDataResource{}
|
|
||||||
assert.Equal(t, tools.DATA_RESOURCE, r.GetType())
|
|
||||||
}
|
|
@ -1,140 +0,0 @@
|
|||||||
package resources_test
|
|
||||||
|
|
||||||
import (
|
|
||||||
"errors"
|
|
||||||
"testing"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/stretchr/testify/assert"
|
|
||||||
"github.com/stretchr/testify/require"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/pricing"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/resources"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
)
|
|
||||||
|
|
||||||
// ---- Mock PricingProfile ----
|
|
||||||
|
|
||||||
type MockPricingProfile struct {
|
|
||||||
pricing.PricingProfileITF
|
|
||||||
Purchased bool
|
|
||||||
ReturnErr bool
|
|
||||||
ReturnCost float64
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *MockPricingProfile) IsPurchasable() bool {
|
|
||||||
return m.Purchased
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *MockPricingProfile) GetPrice(amount float64, explicitDuration float64, start time.Time, end time.Time, _ ...string) (float64, error) {
|
|
||||||
if m.ReturnErr {
|
|
||||||
return 0, errors.New("mock error")
|
|
||||||
}
|
|
||||||
return m.ReturnCost, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// ---- Tests ----
|
|
||||||
|
|
||||||
func TestGetIDAndCreatorAndType(t *testing.T) {
|
|
||||||
r := resources.PricedResource{
|
|
||||||
ResourceID: "res-123",
|
|
||||||
CreatorID: "user-abc",
|
|
||||||
ResourceType: tools.DATA_RESOURCE,
|
|
||||||
}
|
|
||||||
assert.Equal(t, "res-123", r.GetID())
|
|
||||||
assert.Equal(t, "user-abc", r.GetCreatorID())
|
|
||||||
assert.Equal(t, tools.DATA_RESOURCE, r.GetType())
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestIsPurchased(t *testing.T) {
|
|
||||||
t.Run("nil selected pricing returns false", func(t *testing.T) {
|
|
||||||
r := &resources.PricedResource{}
|
|
||||||
assert.False(t, r.IsPurchasable())
|
|
||||||
})
|
|
||||||
|
|
||||||
t.Run("returns true if pricing profile is purchased", func(t *testing.T) {
|
|
||||||
mock := &MockPricingProfile{Purchased: true}
|
|
||||||
r := &resources.PricedResource{SelectedPricing: mock}
|
|
||||||
assert.True(t, r.IsPurchasable())
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestGetAndSetLocationStartEnd(t *testing.T) {
|
|
||||||
r := &resources.PricedResource{}
|
|
||||||
|
|
||||||
now := time.Now()
|
|
||||||
r.SetLocationStart(now)
|
|
||||||
r.SetLocationEnd(now.Add(2 * time.Hour))
|
|
||||||
|
|
||||||
assert.Equal(t, now, *r.GetLocationStart())
|
|
||||||
assert.Equal(t, now.Add(2*time.Hour), *r.GetLocationEnd())
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestGetExplicitDurationInS(t *testing.T) {
|
|
||||||
t.Run("uses explicit duration if set", func(t *testing.T) {
|
|
||||||
r := &resources.PricedResource{ExplicitBookingDurationS: 3600}
|
|
||||||
assert.Equal(t, 3600.0, r.GetExplicitDurationInS())
|
|
||||||
})
|
|
||||||
|
|
||||||
t.Run("computes duration from start and end", func(t *testing.T) {
|
|
||||||
start := time.Now()
|
|
||||||
end := start.Add(2 * time.Hour)
|
|
||||||
r := &resources.PricedResource{UsageStart: &start, UsageEnd: &end}
|
|
||||||
assert.InDelta(t, 7200.0, r.GetExplicitDurationInS(), 0.1)
|
|
||||||
})
|
|
||||||
|
|
||||||
t.Run("defaults to 1 hour when times not set", func(t *testing.T) {
|
|
||||||
r := &resources.PricedResource{}
|
|
||||||
assert.InDelta(t, 3600.0, r.GetExplicitDurationInS(), 0.1)
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestGetPrice(t *testing.T) {
|
|
||||||
t.Run("returns error if no pricing profile", func(t *testing.T) {
|
|
||||||
r := &resources.PricedResource{ResourceID: "no-profile"}
|
|
||||||
price, err := r.GetPrice()
|
|
||||||
require.Error(t, err)
|
|
||||||
assert.Contains(t, err.Error(), "pricing profile must be set")
|
|
||||||
assert.Equal(t, 0.0, price)
|
|
||||||
})
|
|
||||||
|
|
||||||
t.Run("uses first profile if selected is nil", func(t *testing.T) {
|
|
||||||
start := time.Now()
|
|
||||||
end := start.Add(30 * time.Minute)
|
|
||||||
r := &resources.PricedResource{
|
|
||||||
UsageStart: &start,
|
|
||||||
UsageEnd: &end,
|
|
||||||
}
|
|
||||||
price, err := r.GetPrice()
|
|
||||||
require.NoError(t, err)
|
|
||||||
assert.Equal(t, 42.0, price)
|
|
||||||
})
|
|
||||||
|
|
||||||
t.Run("returns error if profile GetPrice fails", func(t *testing.T) {
|
|
||||||
start := time.Now()
|
|
||||||
end := start.Add(1 * time.Hour)
|
|
||||||
mock := &MockPricingProfile{ReturnErr: true}
|
|
||||||
r := &resources.PricedResource{
|
|
||||||
SelectedPricing: mock,
|
|
||||||
UsageStart: &start,
|
|
||||||
UsageEnd: &end,
|
|
||||||
}
|
|
||||||
price, err := r.GetPrice()
|
|
||||||
require.Error(t, err)
|
|
||||||
assert.Equal(t, 0.0, price)
|
|
||||||
})
|
|
||||||
|
|
||||||
t.Run("uses SelectedPricing if set", func(t *testing.T) {
|
|
||||||
start := time.Now()
|
|
||||||
end := start.Add(1 * time.Hour)
|
|
||||||
mock := &MockPricingProfile{ReturnCost: 10.0}
|
|
||||||
r := &resources.PricedResource{
|
|
||||||
SelectedPricing: mock,
|
|
||||||
UsageStart: &start,
|
|
||||||
UsageEnd: &end,
|
|
||||||
}
|
|
||||||
price, err := r.GetPrice()
|
|
||||||
require.NoError(t, err)
|
|
||||||
assert.Equal(t, 10.0, price)
|
|
||||||
})
|
|
||||||
}
|
|
@ -1,106 +0,0 @@
|
|||||||
package resources_test
|
|
||||||
|
|
||||||
import (
|
|
||||||
"testing"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/pricing"
|
|
||||||
. "cloud.o-forge.io/core/oc-lib/models/resources"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
"github.com/stretchr/testify/assert"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestProcessingResource_GetType(t *testing.T) {
|
|
||||||
r := &ProcessingResource{}
|
|
||||||
assert.Equal(t, tools.PROCESSING_RESOURCE.String(), r.GetType())
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestPricedProcessingResource_GetType(t *testing.T) {
|
|
||||||
r := &PricedProcessingResource{}
|
|
||||||
assert.Equal(t, tools.PROCESSING_RESOURCE, r.GetType())
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestPricedProcessingResource_GetExplicitDurationInS(t *testing.T) {
|
|
||||||
now := time.Now()
|
|
||||||
after := now.Add(2 * time.Hour)
|
|
||||||
|
|
||||||
tests := []struct {
|
|
||||||
name string
|
|
||||||
input PricedProcessingResource
|
|
||||||
expected float64
|
|
||||||
}{
|
|
||||||
{
|
|
||||||
name: "Service without explicit duration",
|
|
||||||
input: PricedProcessingResource{
|
|
||||||
IsService: true,
|
|
||||||
},
|
|
||||||
expected: -1,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "Nil start time, non-service",
|
|
||||||
input: PricedProcessingResource{
|
|
||||||
PricedResource: PricedResource{
|
|
||||||
UsageStart: nil,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
expected: float64((1 * time.Hour).Seconds()),
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "Duration computed from start and end",
|
|
||||||
input: PricedProcessingResource{
|
|
||||||
PricedResource: PricedResource{
|
|
||||||
UsageStart: &now,
|
|
||||||
UsageEnd: &after,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
expected: float64((2 * time.Hour).Seconds()),
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "Explicit duration takes precedence",
|
|
||||||
input: PricedProcessingResource{
|
|
||||||
PricedResource: PricedResource{
|
|
||||||
ExplicitBookingDurationS: 1337,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
expected: 1337,
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, test := range tests {
|
|
||||||
t.Run(test.name, func(t *testing.T) {
|
|
||||||
assert.Equal(t, test.expected, test.input.GetExplicitDurationInS())
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestProcessingResource_GetAccessor(t *testing.T) {
|
|
||||||
request := &tools.APIRequest{}
|
|
||||||
r := &ProcessingResource{}
|
|
||||||
acc := r.GetAccessor(request)
|
|
||||||
assert.NotNil(t, acc)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestProcessingResourcePricingProfile_GetPrice(t *testing.T) {
|
|
||||||
start := time.Now()
|
|
||||||
end := start.Add(2 * time.Hour)
|
|
||||||
mockPricing := pricing.AccessPricingProfile[pricing.TimePricingStrategy]{
|
|
||||||
Pricing: pricing.PricingStrategy[pricing.TimePricingStrategy]{
|
|
||||||
Price: 100.0,
|
|
||||||
},
|
|
||||||
}
|
|
||||||
profile := &ProcessingResourcePricingProfile{mockPricing}
|
|
||||||
price, err := profile.GetPrice(0, 0, start, end)
|
|
||||||
assert.NoError(t, err)
|
|
||||||
assert.Equal(t, 100.0, price)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestProcessingResourcePricingProfile_IsPurchased(t *testing.T) {
|
|
||||||
purchased := &ProcessingResourcePricingProfile{
|
|
||||||
AccessPricingProfile: pricing.AccessPricingProfile[pricing.TimePricingStrategy]{
|
|
||||||
Pricing: pricing.PricingStrategy[pricing.TimePricingStrategy]{
|
|
||||||
BuyingStrategy: pricing.PERMANENT,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
}
|
|
||||||
assert.True(t, purchased.IsPurchasable())
|
|
||||||
}
|
|
@ -1,115 +0,0 @@
|
|||||||
package resources_test
|
|
||||||
|
|
||||||
import (
|
|
||||||
"testing"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/pricing"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/resources"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
"github.com/stretchr/testify/assert"
|
|
||||||
)
|
|
||||||
|
|
||||||
type MockInstance struct {
|
|
||||||
ID string
|
|
||||||
Name string
|
|
||||||
resources.ResourceInstance[*MockPartner]
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *MockInstance) GetID() string { return m.ID }
|
|
||||||
func (m *MockInstance) GetName() string { return m.Name }
|
|
||||||
func (m *MockInstance) ClearEnv() {}
|
|
||||||
func (m *MockInstance) ClearPeerGroups() {}
|
|
||||||
func (m *MockInstance) GetProfile() pricing.PricingProfileITF {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
func (m *MockInstance) GetPricingsProfiles(peerID string, groups []string) []pricing.PricingProfileITF {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
func (m *MockInstance) GetPeerGroups() ([]resources.ResourcePartnerITF, []map[string][]string) {
|
|
||||||
return nil, []map[string][]string{
|
|
||||||
{"peer1": {"group1"}},
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
type MockPartner struct {
|
|
||||||
groups map[string][]string
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *MockPartner) GetProfile(buying int, strategy int) pricing.PricingProfileITF {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *MockPartner) GetPeerGroups() map[string][]string {
|
|
||||||
return m.groups
|
|
||||||
}
|
|
||||||
func (m *MockPartner) ClearPeerGroups() {}
|
|
||||||
func (m *MockPartner) GetPricingsProfiles(string, []string) []pricing.PricingProfileITF {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
type MockDBObject struct {
|
|
||||||
utils.AbstractObject
|
|
||||||
isDraft bool
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *MockDBObject) IsDrafted() bool {
|
|
||||||
return m.isDraft
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestGetSelectedInstance_WithValidIndex(t *testing.T) {
|
|
||||||
index := 1
|
|
||||||
inst1 := &MockInstance{ID: "1"}
|
|
||||||
inst2 := &MockInstance{ID: "2"}
|
|
||||||
resource := &resources.AbstractInstanciatedResource[*MockInstance]{
|
|
||||||
AbstractResource: resources.AbstractResource{SelectedInstanceIndex: &index},
|
|
||||||
Instances: []*MockInstance{inst1, inst2},
|
|
||||||
}
|
|
||||||
result := resource.GetSelectedInstance()
|
|
||||||
assert.Equal(t, inst2, result)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestGetSelectedInstance_NoIndex(t *testing.T) {
|
|
||||||
inst := &MockInstance{ID: "1"}
|
|
||||||
resource := &resources.AbstractInstanciatedResource[*MockInstance]{
|
|
||||||
Instances: []*MockInstance{inst},
|
|
||||||
}
|
|
||||||
result := resource.GetSelectedInstance()
|
|
||||||
assert.Equal(t, inst, result)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestCanUpdate_WhenOnlyStateDiffers(t *testing.T) {
|
|
||||||
resource := &resources.AbstractResource{AbstractObject: utils.AbstractObject{IsDraft: false}}
|
|
||||||
set := &MockDBObject{isDraft: true}
|
|
||||||
canUpdate, updated := resource.CanUpdate(set)
|
|
||||||
assert.True(t, canUpdate)
|
|
||||||
assert.Equal(t, set, updated)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestVerifyAuthAction_WithMatchingGroup(t *testing.T) {
|
|
||||||
inst := &MockInstance{
|
|
||||||
ResourceInstance: resources.ResourceInstance[*MockPartner]{
|
|
||||||
Partnerships: []*MockPartner{
|
|
||||||
{groups: map[string][]string{"peer1": {"group1"}}},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
}
|
|
||||||
req := &tools.APIRequest{PeerID: "peer1", Groups: []string{"group1"}}
|
|
||||||
result := resources.VerifyAuthAction([]*MockInstance{inst}, req)
|
|
||||||
assert.Len(t, result, 1)
|
|
||||||
}
|
|
||||||
|
|
||||||
type FakeResource struct {
|
|
||||||
resources.AbstractInstanciatedResource[*MockInstance]
|
|
||||||
}
|
|
||||||
|
|
||||||
func (f *FakeResource) Trim() {}
|
|
||||||
func (f *FakeResource) SetAllowedInstances(*tools.APIRequest) {}
|
|
||||||
func (f *FakeResource) VerifyAuth(*tools.APIRequest) bool { return true }
|
|
||||||
|
|
||||||
func TestNewAccessor_ReturnsValid(t *testing.T) {
|
|
||||||
acc := resources.NewAccessor[*FakeResource](tools.COMPUTE_RESOURCE, &tools.APIRequest{}, func() utils.DBObject {
|
|
||||||
return &FakeResource{}
|
|
||||||
})
|
|
||||||
assert.NotNil(t, acc)
|
|
||||||
}
|
|
@ -1,105 +0,0 @@
|
|||||||
package resources_test
|
|
||||||
|
|
||||||
import (
|
|
||||||
"testing"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/models"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
"github.com/stretchr/testify/assert"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/resources"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestStorageResource_GetType(t *testing.T) {
|
|
||||||
res := &resources.StorageResource{}
|
|
||||||
assert.Equal(t, tools.STORAGE_RESOURCE.String(), res.GetType())
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestStorageResource_GetAccessor(t *testing.T) {
|
|
||||||
res := &resources.StorageResource{}
|
|
||||||
req := &tools.APIRequest{}
|
|
||||||
accessor := res.GetAccessor(req)
|
|
||||||
assert.NotNil(t, accessor)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestStorageResource_ConvertToPricedResource_ValidType(t *testing.T) {
|
|
||||||
res := &resources.StorageResource{}
|
|
||||||
res.AbstractInstanciatedResource.CreatorID = "creator"
|
|
||||||
res.AbstractInstanciatedResource.UUID = "res-id"
|
|
||||||
priced := res.ConvertToPricedResource(tools.STORAGE_RESOURCE, &tools.APIRequest{})
|
|
||||||
assert.NotNil(t, priced)
|
|
||||||
assert.IsType(t, &resources.PricedStorageResource{}, priced)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestStorageResource_ConvertToPricedResource_InvalidType(t *testing.T) {
|
|
||||||
res := &resources.StorageResource{}
|
|
||||||
priced := res.ConvertToPricedResource(tools.COMPUTE_RESOURCE, &tools.APIRequest{})
|
|
||||||
assert.Nil(t, priced)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestStorageResourceInstance_ClearEnv(t *testing.T) {
|
|
||||||
inst := &resources.StorageResourceInstance{
|
|
||||||
ResourceInstance: resources.ResourceInstance[*resources.StorageResourcePartnership]{
|
|
||||||
Env: []models.Param{{Attr: "A"}},
|
|
||||||
Inputs: []models.Param{{Attr: "B"}},
|
|
||||||
Outputs: []models.Param{{Attr: "C"}},
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
inst.ClearEnv()
|
|
||||||
assert.Empty(t, inst.Env)
|
|
||||||
assert.Empty(t, inst.Inputs)
|
|
||||||
assert.Empty(t, inst.Outputs)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestStorageResourceInstance_StoreDraftDefault(t *testing.T) {
|
|
||||||
inst := &resources.StorageResourceInstance{
|
|
||||||
Source: "my-source",
|
|
||||||
ResourceInstance: resources.ResourceInstance[*resources.StorageResourcePartnership]{
|
|
||||||
Env: []models.Param{},
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
inst.StoreDraftDefault()
|
|
||||||
assert.Len(t, inst.Env, 1)
|
|
||||||
assert.Equal(t, "source", inst.Env[0].Attr)
|
|
||||||
assert.Equal(t, "my-source", inst.Env[0].Value)
|
|
||||||
assert.True(t, inst.Env[0].Readonly)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestStorageResourcePricingStrategy_GetQuantity(t *testing.T) {
|
|
||||||
tests := []struct {
|
|
||||||
strategy resources.StorageResourcePricingStrategy
|
|
||||||
dataGB float64
|
|
||||||
expect float64
|
|
||||||
}{
|
|
||||||
{resources.PER_DATA_STORED, 1.2, 1.2},
|
|
||||||
{resources.PER_TB_STORED, 1.2, 1200},
|
|
||||||
{resources.PER_GB_STORED, 2.5, 2.5},
|
|
||||||
{resources.PER_MB_STORED, 1.0, 1000},
|
|
||||||
{resources.PER_KB_STORED, 0.1, 100000},
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, tt := range tests {
|
|
||||||
q, err := tt.strategy.GetQuantity(tt.dataGB)
|
|
||||||
assert.NoError(t, err)
|
|
||||||
assert.Equal(t, tt.expect, q)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestStorageResourcePricingStrategy_GetQuantity_Invalid(t *testing.T) {
|
|
||||||
invalid := resources.StorageResourcePricingStrategy(99)
|
|
||||||
q, err := invalid.GetQuantity(1.0)
|
|
||||||
assert.Error(t, err)
|
|
||||||
assert.Equal(t, 0.0, q)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestPricedStorageResource_GetPrice_NoProfiles(t *testing.T) {
|
|
||||||
res := &resources.PricedStorageResource{
|
|
||||||
PricedResource: resources.PricedResource{
|
|
||||||
ResourceID: "res-id",
|
|
||||||
},
|
|
||||||
}
|
|
||||||
_, err := res.GetPrice()
|
|
||||||
assert.Error(t, err)
|
|
||||||
}
|
|
@ -1,62 +0,0 @@
|
|||||||
package resources_test
|
|
||||||
|
|
||||||
import (
|
|
||||||
"testing"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
"github.com/stretchr/testify/assert"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/resources"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestWorkflowResource_GetType(t *testing.T) {
|
|
||||||
w := &resources.WorkflowResource{}
|
|
||||||
assert.Equal(t, tools.WORKFLOW_RESOURCE.String(), w.GetType())
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestWorkflowResource_ConvertToPricedResource(t *testing.T) {
|
|
||||||
w := &resources.WorkflowResource{
|
|
||||||
AbstractResource: resources.AbstractResource{
|
|
||||||
AbstractObject: utils.AbstractObject{
|
|
||||||
Name: "Test Workflow",
|
|
||||||
UUID: "workflow-uuid",
|
|
||||||
CreatorID: "creator-id",
|
|
||||||
},
|
|
||||||
Logo: "logo.png",
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
req := &tools.APIRequest{
|
|
||||||
PeerID: "peer-1",
|
|
||||||
Groups: []string{"group1"},
|
|
||||||
}
|
|
||||||
|
|
||||||
pr := w.ConvertToPricedResource(tools.WORKFLOW_RESOURCE, req)
|
|
||||||
assert.Equal(t, "creator-id", pr.GetCreatorID())
|
|
||||||
assert.Equal(t, tools.WORKFLOW_RESOURCE, pr.GetType())
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestWorkflowResource_ClearEnv(t *testing.T) {
|
|
||||||
w := &resources.WorkflowResource{}
|
|
||||||
assert.Equal(t, w, w.ClearEnv())
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestWorkflowResource_Trim(t *testing.T) {
|
|
||||||
w := &resources.WorkflowResource{}
|
|
||||||
w.Trim()
|
|
||||||
// nothing to assert; just test that it doesn't panic
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestWorkflowResource_SetAllowedInstances(t *testing.T) {
|
|
||||||
w := &resources.WorkflowResource{}
|
|
||||||
w.SetAllowedInstances(&tools.APIRequest{})
|
|
||||||
// no-op; just confirm no crash
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestWorkflowResource_GetAccessor(t *testing.T) {
|
|
||||||
w := &resources.WorkflowResource{}
|
|
||||||
request := &tools.APIRequest{}
|
|
||||||
accessor := w.GetAccessor(request)
|
|
||||||
assert.NotNil(t, accessor)
|
|
||||||
}
|
|
@ -1,45 +0,0 @@
|
|||||||
package resources
|
|
||||||
|
|
||||||
import (
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/pricing"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
)
|
|
||||||
|
|
||||||
type WorkflowResourcePricingProfile struct{}
|
|
||||||
|
|
||||||
// WorkflowResource is a struct that represents a workflow resource
|
|
||||||
// it defines the resource workflow
|
|
||||||
type WorkflowResource struct {
|
|
||||||
AbstractResource
|
|
||||||
WorkflowID string `bson:"workflow_id,omitempty" json:"workflow_id,omitempty"` // WorkflowID is the ID of the native workflow
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *WorkflowResource) GetAccessor(request *tools.APIRequest) utils.Accessor {
|
|
||||||
return NewAccessor[*WorkflowResource](tools.WORKFLOW_RESOURCE, request, func() utils.DBObject { return &WorkflowResource{} })
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *WorkflowResource) GetType() string {
|
|
||||||
return tools.WORKFLOW_RESOURCE.String()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *WorkflowResource) ClearEnv() utils.DBObject {
|
|
||||||
return d
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *WorkflowResource) Trim() {
|
|
||||||
/* EMPTY */
|
|
||||||
}
|
|
||||||
func (w *WorkflowResource) SetAllowedInstances(request *tools.APIRequest) {
|
|
||||||
/* EMPTY */
|
|
||||||
}
|
|
||||||
|
|
||||||
func (w *WorkflowResource) ConvertToPricedResource(t tools.DataType, request *tools.APIRequest) pricing.PricedItemITF {
|
|
||||||
return &PricedResource{
|
|
||||||
Name: w.Name,
|
|
||||||
Logo: w.Logo,
|
|
||||||
ResourceID: w.UUID,
|
|
||||||
ResourceType: t,
|
|
||||||
CreatorID: w.CreatorID,
|
|
||||||
}
|
|
||||||
}
|
|
70
models/resources/workflow/graph/graph.go
Normal file
70
models/resources/workflow/graph/graph.go
Normal file
@ -0,0 +1,70 @@
|
|||||||
|
package graph
|
||||||
|
|
||||||
|
import (
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/resources"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/utils"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/tools"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Graph is a struct that represents a graph
|
||||||
|
type Graph struct {
|
||||||
|
Zoom float64 `bson:"zoom" json:"zoom" default:"1"` // Zoom is the graphical zoom of the graph
|
||||||
|
Items map[string]GraphItem `bson:"items" json:"items" default:"{}" validate:"required"` // Items is the list of elements in the graph
|
||||||
|
Links []GraphLink `bson:"links" json:"links" default:"{}" validate:"required"` // Links is the list of links between elements in the graph
|
||||||
|
}
|
||||||
|
|
||||||
|
func (g *Graph) GetResource(id string) (string, utils.DBObject) {
|
||||||
|
if item, ok := g.Items[id]; ok {
|
||||||
|
if item.Data != nil {
|
||||||
|
return tools.DATA_RESOURCE.String(), item.Data
|
||||||
|
} else if item.Compute != nil {
|
||||||
|
return tools.COMPUTE_RESOURCE.String(), item.Compute
|
||||||
|
} else if item.Workflow != nil {
|
||||||
|
return tools.WORKFLOW_RESOURCE.String(), item.Workflow
|
||||||
|
} else if item.Processing != nil {
|
||||||
|
return tools.PROCESSING_RESOURCE.String(), item.Processing
|
||||||
|
} else if item.Storage != nil {
|
||||||
|
return tools.STORAGE_RESOURCE.String(), item.Storage
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return "", nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// GraphItem is a struct that represents an item in a graph
|
||||||
|
type GraphItem struct {
|
||||||
|
ID string `bson:"id" json:"id" validate:"required"` // ID is the unique identifier of the item
|
||||||
|
Width float64 `bson:"width" json:"width" validate:"required"` // Width is the graphical width of the item
|
||||||
|
Height float64 `bson:"height" json:"height" validate:"required"` // Height is the graphical height of the item
|
||||||
|
Position Position `bson:"position" json:"position" validate:"required"` // Position is the graphical position of the item
|
||||||
|
*resources.ItemResource // ItemResource is the resource of the item affected to the item
|
||||||
|
}
|
||||||
|
|
||||||
|
// GraphLink is a struct that represents a link between two items in a graph
|
||||||
|
type GraphLink struct {
|
||||||
|
Source Position `bson:"source" json:"source" validate:"required"` // Source is the source graphical position of the link
|
||||||
|
Destination Position `bson:"destination" json:"destination" validate:"required"` // Destination is the destination graphical position of the link
|
||||||
|
Style *GraphLinkStyle `bson:"style,omitempty" json:"style,omitempty"` // Style is the graphical style of the link
|
||||||
|
}
|
||||||
|
|
||||||
|
// GraphLinkStyle is a struct that represents the style of a link in a graph
|
||||||
|
type GraphLinkStyle struct {
|
||||||
|
Color int64 `bson:"color" json:"color"` // Color is the graphical color of the link (int description of a color, can be transpose as hex)
|
||||||
|
Stroke float64 `bson:"stroke" json:"stroke"` // Stroke is the graphical stroke of the link
|
||||||
|
Tension float64 `bson:"tension" json:"tension"` // Tension is the graphical tension of the link
|
||||||
|
HeadRadius float64 `bson:"head_radius" json:"head_radius"` // graphical pin radius
|
||||||
|
DashWidth float64 `bson:"dash_width" json:"dash_width"` // DashWidth is the graphical dash width of the link
|
||||||
|
DashSpace float64 `bson:"dash_space" json:"dash_space"` // DashSpace is the graphical dash space of the link
|
||||||
|
EndArrow Position `bson:"end_arrow" json:"end_arrow"` // EndArrow is the graphical end arrow of the link
|
||||||
|
StartArrow Position `bson:"start_arrow" json:"start_arrow"` // StartArrow is the graphical start arrow of the link
|
||||||
|
ArrowStyle int64 `bson:"arrow_style" json:"arrow_style"` // ArrowStyle is the graphical arrow style of the link (enum foundable in UI)
|
||||||
|
ArrowDirection int64 `bson:"arrow_direction" json:"arrow_direction"` // ArrowDirection is the graphical arrow direction of the link (enum foundable in UI)
|
||||||
|
StartArrowWidth float64 `bson:"start_arrow_width" json:"start_arrow_width"` // StartArrowWidth is the graphical start arrow width of the link
|
||||||
|
EndArrowWidth float64 `bson:"end_arrow_width" json:"end_arrow_width"` // EndArrowWidth is the graphical end arrow width of the link
|
||||||
|
}
|
||||||
|
|
||||||
|
// Position is a struct that represents a graphical position
|
||||||
|
type Position struct {
|
||||||
|
ID string `json:"id" bson:"id"` // ID reprents ItemID (optionnal), TODO: rename to ItemID
|
||||||
|
X float64 `json:"x" bson:"x" validate:"required"` // X is the graphical x position
|
||||||
|
Y float64 `json:"y" bson:"y" validate:"required"` // Y is the graphical y position
|
||||||
|
}
|
41
models/resources/workflow/workflow.go
Normal file
41
models/resources/workflow/workflow.go
Normal file
@ -0,0 +1,41 @@
|
|||||||
|
package oclib
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/resource_model"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/utils"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/tools"
|
||||||
|
)
|
||||||
|
|
||||||
|
// WorkflowResource is a struct that represents a workflow resource
|
||||||
|
// it defines the resource workflow
|
||||||
|
type WorkflowResource struct {
|
||||||
|
resource_model.AbstractResource
|
||||||
|
WorkflowID string `bson:"workflow_id,omitempty" json:"workflow_id,omitempty"` // WorkflowID is the ID of the native workflow
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *WorkflowResource) GetAccessor(caller *tools.HTTPCaller) utils.Accessor {
|
||||||
|
data := New() // Create a new instance of the accessor
|
||||||
|
data.Init(tools.WORKFLOW_RESOURCE, caller) // Initialize the accessor with the WORKFLOW_RESOURCE model type
|
||||||
|
return data
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dma *WorkflowResource) Deserialize(j map[string]interface{}) utils.DBObject {
|
||||||
|
b, err := json.Marshal(j)
|
||||||
|
if err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
json.Unmarshal(b, dma)
|
||||||
|
return dma
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dma *WorkflowResource) Serialize() map[string]interface{} {
|
||||||
|
var m map[string]interface{}
|
||||||
|
b, err := json.Marshal(dma)
|
||||||
|
if err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
json.Unmarshal(b, &m)
|
||||||
|
return m
|
||||||
|
}
|
113
models/resources/workflow/workflow_mongo_accessor.go
Normal file
113
models/resources/workflow/workflow_mongo_accessor.go
Normal file
@ -0,0 +1,113 @@
|
|||||||
|
package oclib
|
||||||
|
|
||||||
|
import (
|
||||||
|
"cloud.o-forge.io/core/oc-lib/dbs"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/dbs/mongo"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/resource_model"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/utils"
|
||||||
|
)
|
||||||
|
|
||||||
|
type workflowResourceMongoAccessor struct {
|
||||||
|
utils.AbstractAccessor // AbstractAccessor contains the basic fields of an accessor (model, caller)
|
||||||
|
}
|
||||||
|
|
||||||
|
func New() *workflowResourceMongoAccessor {
|
||||||
|
return &workflowResourceMongoAccessor{}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (wfa *workflowResourceMongoAccessor) DeleteOne(id string) (utils.DBObject, int, error) {
|
||||||
|
return wfa.GenericDeleteOne(id, wfa)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (wfa *workflowResourceMongoAccessor) UpdateOne(set utils.DBObject, id string) (utils.DBObject, int, error) {
|
||||||
|
set.(*WorkflowResource).ResourceModel = nil
|
||||||
|
return wfa.GenericUpdateOne(set, id, wfa, &WorkflowResource{})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (wfa *workflowResourceMongoAccessor) StoreOne(data utils.DBObject) (utils.DBObject, int, error) {
|
||||||
|
data.(*WorkflowResource).ResourceModel = nil
|
||||||
|
return wfa.GenericStoreOne(data, wfa)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (wfa *workflowResourceMongoAccessor) CopyOne(data utils.DBObject) (utils.DBObject, int, error) {
|
||||||
|
res, _, _ := wfa.LoadOne(data.GetID())
|
||||||
|
data.(*WorkflowResource).WorkflowID = data.GetID()
|
||||||
|
if res == nil {
|
||||||
|
return wfa.GenericStoreOne(data, wfa)
|
||||||
|
} else {
|
||||||
|
data.(*WorkflowResource).UUID = res.GetID()
|
||||||
|
return wfa.GenericUpdateOne(data, res.GetID(), wfa, &WorkflowResource{})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (wfa *workflowResourceMongoAccessor) LoadOne(id string) (utils.DBObject, int, error) {
|
||||||
|
var workflow WorkflowResource
|
||||||
|
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)
|
||||||
|
accessor := (&resource_model.ResourceModel{}).GetAccessor(nil)
|
||||||
|
resources, _, err := accessor.Search(nil, wfa.GetType())
|
||||||
|
if err == nil && len(resources) > 0 {
|
||||||
|
workflow.ResourceModel = resources[0].(*resource_model.ResourceModel)
|
||||||
|
}
|
||||||
|
return &workflow, 200, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (wfa workflowResourceMongoAccessor) 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 []WorkflowResource
|
||||||
|
if err = res_mongo.All(mongo.MngoCtx, &results); err != nil {
|
||||||
|
return nil, 404, err
|
||||||
|
}
|
||||||
|
accessor := (&resource_model.ResourceModel{}).GetAccessor(nil)
|
||||||
|
resources, _, err := accessor.Search(nil, wfa.GetType())
|
||||||
|
for _, r := range results {
|
||||||
|
if err == nil && len(resources) > 0 {
|
||||||
|
r.ResourceModel = resources[0].(*resource_model.ResourceModel)
|
||||||
|
}
|
||||||
|
objs = append(objs, &r)
|
||||||
|
}
|
||||||
|
return objs, 200, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Search searches for workflow resources in the database, given some filters OR a search string
|
||||||
|
func (wfa *workflowResourceMongoAccessor) 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 like name, short_description, description, owner, url if no filters are provided
|
||||||
|
"abstractresource.abstractobject.name": {{Operator: dbs.LIKE.String(), Value: search}},
|
||||||
|
"abstractresource.short_description": {{Operator: dbs.LIKE.String(), Value: search}},
|
||||||
|
"abstractresource.description": {{Operator: dbs.LIKE.String(), Value: search}},
|
||||||
|
"abstractresource.owner": {{Operator: dbs.LIKE.String(), Value: search}},
|
||||||
|
"abstractresource.source_url": {{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 []WorkflowResource
|
||||||
|
if err = res_mongo.All(mongo.MngoCtx, &results); err != nil {
|
||||||
|
return nil, 404, err
|
||||||
|
}
|
||||||
|
accessor := (&resource_model.ResourceModel{}).GetAccessor(nil)
|
||||||
|
resources, _, err := accessor.Search(nil, wfa.GetType())
|
||||||
|
for _, r := range results {
|
||||||
|
if err == nil && len(resources) > 0 {
|
||||||
|
r.ResourceModel = resources[0].(*resource_model.ResourceModel)
|
||||||
|
}
|
||||||
|
objs = append(objs, &r)
|
||||||
|
}
|
||||||
|
return objs, 200, nil
|
||||||
|
}
|
43
models/resources/workflow/workflow_test.go
Normal file
43
models/resources/workflow/workflow_test.go
Normal file
@ -0,0 +1,43 @@
|
|||||||
|
package oclib
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/resource_model"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/utils"
|
||||||
|
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestStoreOneWorkflow(t *testing.T) {
|
||||||
|
w := WorkflowResource{AbstractResource: resource_model.AbstractResource{
|
||||||
|
AbstractObject: utils.AbstractObject{Name: "testWorkflow"},
|
||||||
|
Description: "Lorem Ipsum",
|
||||||
|
Logo: "azerty.com",
|
||||||
|
Owner: "toto",
|
||||||
|
OwnerLogo: "totoLogo",
|
||||||
|
SourceUrl: "azerty.fr",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
wma := New()
|
||||||
|
id, _, _ := wma.StoreOne(&w)
|
||||||
|
|
||||||
|
assert.NotEmpty(t, id)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestLoadOneWorkflow(t *testing.T) {
|
||||||
|
w := WorkflowResource{AbstractResource: resource_model.AbstractResource{
|
||||||
|
AbstractObject: utils.AbstractObject{Name: "testWorkflow"},
|
||||||
|
Description: "Lorem Ipsum",
|
||||||
|
Logo: "azerty.com",
|
||||||
|
Owner: "toto",
|
||||||
|
OwnerLogo: "totoLogo",
|
||||||
|
SourceUrl: "azerty.fr",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
wma := New()
|
||||||
|
new_w, _, _ := wma.StoreOne(&w)
|
||||||
|
assert.Equal(t, w, new_w)
|
||||||
|
}
|
@ -1,38 +0,0 @@
|
|||||||
package models
|
|
||||||
|
|
||||||
import (
|
|
||||||
"strconv"
|
|
||||||
"testing"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models"
|
|
||||||
"github.com/stretchr/testify/assert"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestModel_ReturnsValidInstances(t *testing.T) {
|
|
||||||
for name, _ := range models.ModelsCatalog {
|
|
||||||
t.Run(name, func(t *testing.T) {
|
|
||||||
modelInt, _ := strconv.Atoi(name)
|
|
||||||
obj := models.Model(modelInt)
|
|
||||||
assert.NotNil(t, obj, "Model() returned nil for valid model name %s", name)
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestModel_UnknownModelReturnsNil(t *testing.T) {
|
|
||||||
invalidModelInt := -9999 // unlikely to be valid
|
|
||||||
obj := models.Model(invalidModelInt)
|
|
||||||
assert.Nil(t, obj)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestGetModelsNames_ReturnsAllKeys(t *testing.T) {
|
|
||||||
names := models.GetModelsNames()
|
|
||||||
assert.Len(t, names, len(models.ModelsCatalog))
|
|
||||||
|
|
||||||
seen := make(map[string]bool)
|
|
||||||
for _, name := range names {
|
|
||||||
seen[name] = true
|
|
||||||
}
|
|
||||||
for key := range models.ModelsCatalog {
|
|
||||||
assert.Contains(t, seen, key)
|
|
||||||
}
|
|
||||||
}
|
|
263
models/utils/abstracts.go
Executable file → Normal file
263
models/utils/abstracts.go
Executable file → Normal file
@ -2,9 +2,12 @@ package utils
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/dbs"
|
"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/tools"
|
"cloud.o-forge.io/core/oc-lib/tools"
|
||||||
"github.com/go-playground/validator/v10"
|
"github.com/go-playground/validator/v10"
|
||||||
"github.com/google/uuid"
|
"github.com/google/uuid"
|
||||||
@ -14,37 +17,55 @@ import (
|
|||||||
// single instance of the validator used in every model Struct to validate the fields
|
// single instance of the validator used in every model Struct to validate the fields
|
||||||
var validate = validator.New(validator.WithRequiredStructEnabled())
|
var validate = validator.New(validator.WithRequiredStructEnabled())
|
||||||
|
|
||||||
type AccessMode int
|
|
||||||
|
|
||||||
const (
|
|
||||||
Private AccessMode = iota
|
|
||||||
Public
|
|
||||||
)
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* AbstractObject is a struct that represents the basic fields of an object
|
* AbstractObject is a struct that represents the basic fields of an object
|
||||||
* it defines the object id and name
|
* it defines the object id and name
|
||||||
* every data in base root model should inherit from this struct (only exception is the ResourceModel)
|
* every data in base root model should inherit from this struct (only exception is the ResourceModel)
|
||||||
*/
|
*/
|
||||||
type AbstractObject struct {
|
type AbstractObject struct {
|
||||||
UUID string `json:"id,omitempty" bson:"id,omitempty" validate:"required"`
|
UUID string `json:"id,omitempty" bson:"id,omitempty" validate:"required"`
|
||||||
Name string `json:"name,omitempty" bson:"name,omitempty" validate:"required"`
|
Name string `json:"name,omitempty" bson:"name,omitempty" validate:"required"`
|
||||||
IsDraft bool `json:"is_draft" bson:"is_draft" default:"false"`
|
UpdateDate time.Time `json:"update_date" bson:"update_date"`
|
||||||
CreatorID string `json:"creator_id,omitempty" bson:"creator_id,omitempty"`
|
LastPeerWriter string `json:"last_peer_writer" bson:"last_peer_writer"`
|
||||||
UserCreatorID string `json:"user_creator_id,omitempty" bson:"user_creator_id,omitempty"`
|
|
||||||
CreationDate time.Time `json:"creation_date,omitempty" bson:"creation_date,omitempty"`
|
|
||||||
UpdateDate time.Time `json:"update_date,omitempty" bson:"update_date,omitempty"`
|
|
||||||
UpdaterID string `json:"updater_id,omitempty" bson:"updater_id,omitempty"`
|
|
||||||
UserUpdaterID string `json:"user_updater_id,omitempty" bson:"user_updater_id,omitempty"`
|
|
||||||
AccessMode AccessMode `json:"access_mode" bson:"access_mode" default:"0"`
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ri *AbstractObject) GetAccessor(request *tools.APIRequest) Accessor {
|
// GetID returns the id of the object (abstract)
|
||||||
|
func (ao *AbstractObject) GetID() string {
|
||||||
|
return ao.UUID
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetName returns the name of the object (abstract)
|
||||||
|
func (ao *AbstractObject) GetName() string {
|
||||||
|
return ao.Name
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ao *AbstractObject) UpToDate() {
|
||||||
|
ao.UpdateDate = time.Now()
|
||||||
|
// ao.LastPeerWriter, _ = static.GetMyLocalJsonPeer()
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetAccessor returns the accessor of the object (abstract)
|
||||||
|
func (dma *AbstractObject) GetAccessor(caller *tools.HTTPCaller) Accessor {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (r *AbstractObject) SetID(id string) {
|
func (dma *AbstractObject) Deserialize(j map[string]interface{}) DBObject {
|
||||||
r.UUID = id
|
b, err := json.Marshal(j)
|
||||||
|
if err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
json.Unmarshal(b, dma)
|
||||||
|
return dma
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dma *AbstractObject) Serialize() map[string]interface{} {
|
||||||
|
var m map[string]interface{}
|
||||||
|
b, err := json.Marshal(dma)
|
||||||
|
if err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
json.Unmarshal(b, &m)
|
||||||
|
return m
|
||||||
}
|
}
|
||||||
|
|
||||||
func (r *AbstractObject) GenerateID() {
|
func (r *AbstractObject) GenerateID() {
|
||||||
@ -53,124 +74,100 @@ func (r *AbstractObject) GenerateID() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (r *AbstractObject) StoreDraftDefault() {
|
|
||||||
r.IsDraft = false
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *AbstractObject) CanUpdate(set DBObject) (bool, DBObject) {
|
|
||||||
return true, set
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *AbstractObject) CanDelete() bool {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *AbstractObject) IsDrafted() bool {
|
|
||||||
return r.IsDraft
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetID implements ShallowDBObject.
|
|
||||||
func (ao AbstractObject) GetID() string {
|
|
||||||
return ao.UUID
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetName implements ShallowDBObject.
|
|
||||||
func (ao AbstractObject) GetName() string {
|
|
||||||
return ao.Name
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ao *AbstractObject) GetCreatorID() string {
|
|
||||||
return ao.CreatorID
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ao *AbstractObject) UpToDate(user string, peer string, create bool) {
|
|
||||||
ao.UpdateDate = time.Now()
|
|
||||||
ao.UpdaterID = peer
|
|
||||||
ao.UserUpdaterID = user
|
|
||||||
if create {
|
|
||||||
ao.CreationDate = time.Now()
|
|
||||||
ao.CreatorID = peer
|
|
||||||
ao.UserCreatorID = user
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ao *AbstractObject) VerifyAuth(request *tools.APIRequest) bool {
|
|
||||||
return ao.AccessMode == Public || (request != nil && ao.CreatorID == request.PeerID && request.PeerID != "")
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ao *AbstractObject) GetObjectFilters(search string) *dbs.Filters {
|
|
||||||
if search == "*" {
|
|
||||||
search = ""
|
|
||||||
}
|
|
||||||
return &dbs.Filters{
|
|
||||||
Or: map[string][]dbs.Filter{ // filter by name if no filters are provided
|
|
||||||
"abstractobject.name": {{Operator: dbs.LIKE.String(), Value: search}},
|
|
||||||
}}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (dma *AbstractObject) Deserialize(j map[string]interface{}, obj DBObject) DBObject {
|
|
||||||
b, err := json.Marshal(j)
|
|
||||||
if err != nil {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
json.Unmarshal(b, obj)
|
|
||||||
return obj
|
|
||||||
}
|
|
||||||
|
|
||||||
func (dma *AbstractObject) Serialize(obj DBObject) map[string]interface{} {
|
|
||||||
var m map[string]interface{}
|
|
||||||
b, err := json.Marshal(obj)
|
|
||||||
if err != nil {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
json.Unmarshal(b, &m)
|
|
||||||
return m
|
|
||||||
}
|
|
||||||
|
|
||||||
type AbstractAccessor struct {
|
type AbstractAccessor struct {
|
||||||
Logger zerolog.Logger // Logger is the logger of the accessor, it's a specilized logger for the accessor
|
Logger zerolog.Logger // Logger is the logger of the accessor, it's a specilized logger for the accessor
|
||||||
Type tools.DataType // Type is the data type of the accessor
|
Type string // Type is the data type of the accessor
|
||||||
Request *tools.APIRequest // Caller is the http caller of the accessor (optionnal) only need in a peer connection
|
Caller *tools.HTTPCaller // Caller is the http caller of the accessor (optionnal) only need in a peer connection
|
||||||
ResourceModelAccessor Accessor
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (r *AbstractAccessor) ShouldVerifyAuth() bool {
|
func (dma *AbstractAccessor) GetType() string {
|
||||||
return true
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *AbstractAccessor) GetRequest() *tools.APIRequest {
|
|
||||||
return r.Request
|
|
||||||
}
|
|
||||||
|
|
||||||
func (dma *AbstractAccessor) GetUser() string {
|
|
||||||
if dma.Request == nil {
|
|
||||||
return ""
|
|
||||||
}
|
|
||||||
return dma.Request.Username
|
|
||||||
}
|
|
||||||
|
|
||||||
func (dma *AbstractAccessor) GetPeerID() string {
|
|
||||||
if dma.Request == nil {
|
|
||||||
return ""
|
|
||||||
}
|
|
||||||
return dma.Request.PeerID
|
|
||||||
}
|
|
||||||
func (dma *AbstractAccessor) GetGroups() []string {
|
|
||||||
if dma.Request == nil {
|
|
||||||
return []string{}
|
|
||||||
}
|
|
||||||
return dma.Request.Groups
|
|
||||||
}
|
|
||||||
func (dma *AbstractAccessor) GetLogger() *zerolog.Logger {
|
|
||||||
return &dma.Logger
|
|
||||||
}
|
|
||||||
func (dma *AbstractAccessor) GetType() tools.DataType {
|
|
||||||
return dma.Type
|
return dma.Type
|
||||||
}
|
}
|
||||||
|
|
||||||
func (dma *AbstractAccessor) GetCaller() *tools.HTTPCaller {
|
func (dma *AbstractAccessor) GetCaller() *tools.HTTPCaller {
|
||||||
if dma.Request == nil {
|
return dma.Caller
|
||||||
return nil
|
}
|
||||||
}
|
|
||||||
return dma.Request.Caller
|
// Init initializes the accessor with the data type and the http caller
|
||||||
|
func (dma *AbstractAccessor) Init(t tools.DataType, caller *tools.HTTPCaller) {
|
||||||
|
dma.Logger = logs.CreateLogger(t.String()) // Create a logger with the data type
|
||||||
|
dma.Caller = caller // Set the caller
|
||||||
|
dma.Type = t.String() // Set the data type
|
||||||
|
}
|
||||||
|
|
||||||
|
// GenericLoadOne loads one object from the database (generic)
|
||||||
|
func (wfa *AbstractAccessor) GenericStoreOne(data DBObject, accessor Accessor) (DBObject, int, error) {
|
||||||
|
data.GenerateID()
|
||||||
|
f := dbs.Filters{
|
||||||
|
Or: map[string][]dbs.Filter{
|
||||||
|
"abstractresource.abstractobject.name": {{
|
||||||
|
Operator: dbs.LIKE.String(),
|
||||||
|
Value: data.GetName(),
|
||||||
|
}},
|
||||||
|
"abstractobject.name": {{
|
||||||
|
Operator: dbs.LIKE.String(),
|
||||||
|
Value: data.GetName(),
|
||||||
|
}},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
if cursor, _, _ := accessor.Search(&f, ""); len(cursor) > 0 {
|
||||||
|
return nil, 409, errors.New(accessor.GetType() + " with name " + data.GetName() + " already exists")
|
||||||
|
}
|
||||||
|
err := validate.Struct(data)
|
||||||
|
if err != nil {
|
||||||
|
return nil, 422, err
|
||||||
|
}
|
||||||
|
id, code, err := mongo.MONGOService.StoreOne(data, data.GetID(), wfa.GetType())
|
||||||
|
if err != nil {
|
||||||
|
wfa.Logger.Error().Msg("Could not store " + data.GetName() + " to db. Error: " + err.Error())
|
||||||
|
return nil, code, err
|
||||||
|
}
|
||||||
|
return accessor.LoadOne(id)
|
||||||
|
}
|
||||||
|
|
||||||
|
// GenericLoadOne loads one object from the database (generic)
|
||||||
|
func (dma *AbstractAccessor) GenericDeleteOne(id string, accessor Accessor) (DBObject, int, error) {
|
||||||
|
res, code, err := accessor.LoadOne(id)
|
||||||
|
if err != nil {
|
||||||
|
dma.Logger.Error().Msg("Could not retrieve " + id + " to db. Error: " + err.Error())
|
||||||
|
return nil, code, err
|
||||||
|
}
|
||||||
|
_, code, err = mongo.MONGOService.DeleteOne(id, accessor.GetType())
|
||||||
|
if err != nil {
|
||||||
|
dma.Logger.Error().Msg("Could not delete " + id + " to db. Error: " + err.Error())
|
||||||
|
return nil, code, err
|
||||||
|
}
|
||||||
|
return res, 200, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// GenericLoadOne loads one object from the database (generic)
|
||||||
|
// json expected in entry is a flatted object no need to respect the inheritance hierarchy
|
||||||
|
func (dma *AbstractAccessor) GenericUpdateOne(set DBObject, id string, accessor Accessor, new DBObject) (DBObject, int, error) {
|
||||||
|
r, c, err := accessor.LoadOne(id)
|
||||||
|
if err != nil {
|
||||||
|
return nil, c, err
|
||||||
|
}
|
||||||
|
change := set.Serialize() // get the changes
|
||||||
|
loaded := r.Serialize() // get the loaded object
|
||||||
|
|
||||||
|
for k, v := range change { // apply the changes, with a flatten method
|
||||||
|
loaded[k] = v
|
||||||
|
}
|
||||||
|
id, code, err := mongo.MONGOService.UpdateOne(new.Deserialize(loaded), id, accessor.GetType())
|
||||||
|
if err != nil {
|
||||||
|
dma.Logger.Error().Msg("Could not update " + id + " to db. Error: " + err.Error())
|
||||||
|
return nil, code, err
|
||||||
|
}
|
||||||
|
return accessor.LoadOne(id)
|
||||||
|
}
|
||||||
|
|
||||||
|
// GenericLoadOne loads one object from the database (generic)
|
||||||
|
// json expected in entry is a flatted object no need to respect the inheritance hierarchy
|
||||||
|
func (dma *AbstractAccessor) GenericRawUpdateOne(set DBObject, id string, accessor Accessor) (DBObject, int, error) {
|
||||||
|
id, code, err := mongo.MONGOService.UpdateOne(set, id, accessor.GetType())
|
||||||
|
if err != nil {
|
||||||
|
dma.Logger.Error().Msg("Could not update " + id + " to db. Error: " + err.Error())
|
||||||
|
return nil, code, err
|
||||||
|
}
|
||||||
|
return accessor.LoadOne(id)
|
||||||
}
|
}
|
||||||
|
168
models/utils/common.go
Executable file → Normal file
168
models/utils/common.go
Executable file → Normal file
@ -1,166 +1,8 @@
|
|||||||
package utils
|
package utils
|
||||||
|
|
||||||
import (
|
/*
|
||||||
"errors"
|
type Price struct {
|
||||||
|
Price float64 `json:"price,omitempty" bson:"price,omitempty"`
|
||||||
"cloud.o-forge.io/core/oc-lib/dbs"
|
Currency string `json:"currency,omitempty" bson:"currency,omitempty"`
|
||||||
"cloud.o-forge.io/core/oc-lib/dbs/mongo"
|
|
||||||
mgb "go.mongodb.org/mongo-driver/mongo"
|
|
||||||
)
|
|
||||||
|
|
||||||
type Owner struct {
|
|
||||||
Name string `json:"name,omitempty" bson:"name,omitempty"`
|
|
||||||
Logo string `json:"logo,omitempty" bson:"logo,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func VerifyAccess(a Accessor, id string) error {
|
|
||||||
data, _, err := a.LoadOne(id)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
if a.ShouldVerifyAuth() && !data.VerifyAuth(a.GetRequest()) {
|
|
||||||
return errors.New("you are not allowed to access :" + a.GetType().String())
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// GenericLoadOne loads one object from the database (generic)
|
|
||||||
func GenericStoreOne(data DBObject, a Accessor) (DBObject, int, error) {
|
|
||||||
data.GenerateID()
|
|
||||||
data.StoreDraftDefault()
|
|
||||||
data.UpToDate(a.GetUser(), a.GetPeerID(), true)
|
|
||||||
f := dbs.Filters{
|
|
||||||
Or: map[string][]dbs.Filter{
|
|
||||||
"abstractresource.abstractobject.name": {{
|
|
||||||
Operator: dbs.LIKE.String(),
|
|
||||||
Value: data.GetName(),
|
|
||||||
}},
|
|
||||||
"abstractobject.name": {{
|
|
||||||
Operator: dbs.LIKE.String(),
|
|
||||||
Value: data.GetName(),
|
|
||||||
}},
|
|
||||||
},
|
|
||||||
}
|
|
||||||
if a.ShouldVerifyAuth() && !data.VerifyAuth(a.GetRequest()) {
|
|
||||||
return nil, 403, errors.New("you are not allowed to access : " + a.GetType().String())
|
|
||||||
}
|
|
||||||
if cursor, _, _ := a.Search(&f, "", data.IsDrafted()); len(cursor) > 0 {
|
|
||||||
return nil, 409, errors.New(a.GetType().String() + " with name " + data.GetName() + " already exists")
|
|
||||||
}
|
|
||||||
err := validate.Struct(data)
|
|
||||||
if err != nil {
|
|
||||||
return nil, 422, err
|
|
||||||
}
|
|
||||||
id, code, err := mongo.MONGOService.StoreOne(data, data.GetID(), a.GetType().String())
|
|
||||||
if err != nil {
|
|
||||||
a.GetLogger().Error().Msg("Could not store " + data.GetName() + " to db. Error: " + err.Error())
|
|
||||||
return nil, code, err
|
|
||||||
}
|
|
||||||
return a.LoadOne(id)
|
|
||||||
}
|
|
||||||
|
|
||||||
// GenericLoadOne loads one object from the database (generic)
|
|
||||||
func GenericDeleteOne(id string, a Accessor) (DBObject, int, error) {
|
|
||||||
res, code, err := a.LoadOne(id)
|
|
||||||
if !res.CanDelete() {
|
|
||||||
return nil, 403, errors.New("you are not allowed to delete :" + a.GetType().String())
|
|
||||||
}
|
|
||||||
if err != nil {
|
|
||||||
return nil, code, err
|
|
||||||
}
|
|
||||||
if a.ShouldVerifyAuth() && !res.VerifyAuth(a.GetRequest()) {
|
|
||||||
return nil, 403, errors.New("you are not allowed to access " + a.GetType().String())
|
|
||||||
}
|
|
||||||
_, code, err = mongo.MONGOService.DeleteOne(id, a.GetType().String())
|
|
||||||
if err != nil {
|
|
||||||
a.GetLogger().Error().Msg("Could not delete " + id + " to db. Error: " + err.Error())
|
|
||||||
return nil, code, err
|
|
||||||
}
|
|
||||||
return res, 200, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// GenericLoadOne loads one object from the database (generic)
|
|
||||||
// json expected in entry is a flatted object no need to respect the inheritance hierarchy
|
|
||||||
func GenericUpdateOne(set DBObject, id string, a Accessor, new DBObject) (DBObject, int, error) {
|
|
||||||
r, c, err := a.LoadOne(id)
|
|
||||||
if err != nil {
|
|
||||||
return nil, c, err
|
|
||||||
}
|
|
||||||
ok, newSet := r.CanUpdate(set)
|
|
||||||
if !ok {
|
|
||||||
return nil, 403, errors.New("you are not allowed to delete :" + a.GetType().String())
|
|
||||||
}
|
|
||||||
set = newSet
|
|
||||||
r.UpToDate(a.GetUser(), a.GetPeerID(), false)
|
|
||||||
if a.ShouldVerifyAuth() && !r.VerifyAuth(a.GetRequest()) {
|
|
||||||
return nil, 403, errors.New("you are not allowed to access :" + a.GetType().String())
|
|
||||||
}
|
|
||||||
change := set.Serialize(set) // get the changes
|
|
||||||
loaded := r.Serialize(r) // get the loaded object
|
|
||||||
|
|
||||||
for k, v := range change { // apply the changes, with a flatten method
|
|
||||||
loaded[k] = v
|
|
||||||
}
|
|
||||||
id, code, err := mongo.MONGOService.UpdateOne(new.Deserialize(loaded, new), id, a.GetType().String())
|
|
||||||
if err != nil {
|
|
||||||
a.GetLogger().Error().Msg("Could not update " + id + " to db. Error: " + err.Error())
|
|
||||||
return nil, code, err
|
|
||||||
}
|
|
||||||
return a.LoadOne(id)
|
|
||||||
}
|
|
||||||
|
|
||||||
func GenericLoadOne[T DBObject](id string, f func(DBObject) (DBObject, int, error), a Accessor) (DBObject, int, error) {
|
|
||||||
var data T
|
|
||||||
res_mongo, code, err := mongo.MONGOService.LoadOne(id, a.GetType().String())
|
|
||||||
if err != nil {
|
|
||||||
return nil, code, err
|
|
||||||
}
|
|
||||||
res_mongo.Decode(&data)
|
|
||||||
if a.ShouldVerifyAuth() && !data.VerifyAuth(a.GetRequest()) {
|
|
||||||
return nil, 403, errors.New("you are not allowed to access :" + a.GetType().String())
|
|
||||||
}
|
|
||||||
return f(data)
|
|
||||||
}
|
|
||||||
|
|
||||||
func genericLoadAll[T DBObject](res *mgb.Cursor, code int, err error, onlyDraft bool, f func(DBObject) ShallowDBObject, a Accessor) ([]ShallowDBObject, int, error) {
|
|
||||||
objs := []ShallowDBObject{}
|
|
||||||
var results []T
|
|
||||||
if err != nil {
|
|
||||||
return nil, code, err
|
|
||||||
}
|
|
||||||
if err = res.All(mongo.MngoCtx, &results); err != nil {
|
|
||||||
return nil, 404, err
|
|
||||||
}
|
|
||||||
for _, r := range results {
|
|
||||||
if (a.ShouldVerifyAuth() && !r.VerifyAuth(a.GetRequest())) || f(r) == nil || (onlyDraft && !r.IsDrafted()) || (!onlyDraft && r.IsDrafted()) {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
objs = append(objs, f(r))
|
|
||||||
}
|
|
||||||
return objs, 200, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func GenericLoadAll[T DBObject](f func(DBObject) ShallowDBObject, onlyDraft bool, wfa Accessor) ([]ShallowDBObject, int, error) {
|
|
||||||
res_mongo, code, err := mongo.MONGOService.LoadAll(wfa.GetType().String())
|
|
||||||
return genericLoadAll[T](res_mongo, code, err, onlyDraft, f, wfa)
|
|
||||||
}
|
|
||||||
|
|
||||||
func GenericSearch[T DBObject](filters *dbs.Filters, search string, defaultFilters *dbs.Filters,
|
|
||||||
f func(DBObject) ShallowDBObject, onlyDraft bool, wfa Accessor) ([]ShallowDBObject, int, error) {
|
|
||||||
if filters == nil && search != "" {
|
|
||||||
filters = defaultFilters
|
|
||||||
}
|
|
||||||
res_mongo, code, err := mongo.MONGOService.Search(filters, wfa.GetType().String())
|
|
||||||
return genericLoadAll[T](res_mongo, code, err, onlyDraft, f, wfa)
|
|
||||||
}
|
|
||||||
|
|
||||||
// GenericLoadOne loads one object from the database (generic)
|
|
||||||
// json expected in entry is a flatted object no need to respect the inheritance hierarchy
|
|
||||||
func GenericRawUpdateOne(set DBObject, id string, a Accessor) (DBObject, int, error) {
|
|
||||||
id, code, err := mongo.MONGOService.UpdateOne(set, id, a.GetType().String())
|
|
||||||
if err != nil {
|
|
||||||
a.GetLogger().Error().Msg("Could not update " + id + " to db. Error: " + err.Error())
|
|
||||||
return nil, code, err
|
|
||||||
}
|
|
||||||
return a.LoadOne(id)
|
|
||||||
}
|
}
|
||||||
|
*/
|
||||||
|
33
models/utils/interfaces.go
Executable file → Normal file
33
models/utils/interfaces.go
Executable file → Normal file
@ -3,7 +3,6 @@ package utils
|
|||||||
import (
|
import (
|
||||||
"cloud.o-forge.io/core/oc-lib/dbs"
|
"cloud.o-forge.io/core/oc-lib/dbs"
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
"cloud.o-forge.io/core/oc-lib/tools"
|
||||||
"github.com/rs/zerolog"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// ShallowDBObject is an interface that defines the basic methods shallowed version of a DBObject
|
// ShallowDBObject is an interface that defines the basic methods shallowed version of a DBObject
|
||||||
@ -11,43 +10,31 @@ type ShallowDBObject interface {
|
|||||||
GenerateID()
|
GenerateID()
|
||||||
GetID() string
|
GetID() string
|
||||||
GetName() string
|
GetName() string
|
||||||
Serialize(obj DBObject) map[string]interface{}
|
Deserialize(j map[string]interface{}) DBObject
|
||||||
Deserialize(j map[string]interface{}, obj DBObject) DBObject
|
Serialize() map[string]interface{}
|
||||||
}
|
}
|
||||||
|
|
||||||
// DBObject is an interface that defines the basic methods for a DBObject
|
// DBObject is an interface that defines the basic methods for a DBObject
|
||||||
type DBObject interface {
|
type DBObject interface {
|
||||||
GenerateID()
|
GenerateID()
|
||||||
SetID(id string)
|
|
||||||
GetID() string
|
GetID() string
|
||||||
GetName() string
|
GetName() string
|
||||||
IsDrafted() bool
|
UpToDate()
|
||||||
CanDelete() bool
|
Deserialize(j map[string]interface{}) DBObject
|
||||||
StoreDraftDefault()
|
Serialize() map[string]interface{}
|
||||||
GetCreatorID() string
|
GetAccessor(caller *tools.HTTPCaller) Accessor
|
||||||
UpToDate(user string, peer string, create bool)
|
|
||||||
CanUpdate(set DBObject) (bool, DBObject)
|
|
||||||
VerifyAuth(request *tools.APIRequest) bool
|
|
||||||
Serialize(obj DBObject) map[string]interface{}
|
|
||||||
GetAccessor(request *tools.APIRequest) Accessor
|
|
||||||
Deserialize(j map[string]interface{}, obj DBObject) DBObject
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Accessor is an interface that defines the basic methods for an Accessor
|
// Accessor is an interface that defines the basic methods for an Accessor
|
||||||
type Accessor interface {
|
type Accessor interface {
|
||||||
GetUser() string
|
Init(t tools.DataType, caller *tools.HTTPCaller)
|
||||||
GetPeerID() string
|
GetType() string
|
||||||
GetGroups() []string
|
|
||||||
ShouldVerifyAuth() bool
|
|
||||||
GetType() tools.DataType
|
|
||||||
GetLogger() *zerolog.Logger
|
|
||||||
GetCaller() *tools.HTTPCaller
|
GetCaller() *tools.HTTPCaller
|
||||||
GetRequest() *tools.APIRequest
|
Search(filters *dbs.Filters, search string) ([]ShallowDBObject, int, error)
|
||||||
|
LoadAll() ([]ShallowDBObject, int, error)
|
||||||
LoadOne(id string) (DBObject, int, error)
|
LoadOne(id string) (DBObject, int, error)
|
||||||
DeleteOne(id string) (DBObject, int, error)
|
DeleteOne(id string) (DBObject, int, error)
|
||||||
CopyOne(data DBObject) (DBObject, int, error)
|
CopyOne(data DBObject) (DBObject, int, error)
|
||||||
StoreOne(data DBObject) (DBObject, int, error)
|
StoreOne(data DBObject) (DBObject, int, error)
|
||||||
LoadAll(isDraft bool) ([]ShallowDBObject, int, error)
|
|
||||||
UpdateOne(set DBObject, id string) (DBObject, int, error)
|
UpdateOne(set DBObject, id string) (DBObject, int, error)
|
||||||
Search(filters *dbs.Filters, search string, isDraft bool) ([]ShallowDBObject, int, error)
|
|
||||||
}
|
}
|
||||||
|
@ -1,128 +0,0 @@
|
|||||||
package models_test
|
|
||||||
|
|
||||||
import (
|
|
||||||
"testing"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/dbs"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
"github.com/google/uuid"
|
|
||||||
"github.com/stretchr/testify/assert"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestGenerateID(t *testing.T) {
|
|
||||||
ao := &utils.AbstractObject{}
|
|
||||||
ao.GenerateID()
|
|
||||||
assert.NotEmpty(t, ao.UUID)
|
|
||||||
_, err := uuid.Parse(ao.UUID)
|
|
||||||
assert.NoError(t, err)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestStoreDraftDefault(t *testing.T) {
|
|
||||||
ao := &utils.AbstractObject{IsDraft: true}
|
|
||||||
ao.StoreDraftDefault()
|
|
||||||
assert.False(t, ao.IsDraft)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestCanUpdate(t *testing.T) {
|
|
||||||
ao := &utils.AbstractObject{}
|
|
||||||
res, set := ao.CanUpdate(nil)
|
|
||||||
assert.True(t, res)
|
|
||||||
assert.Nil(t, set)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestCanDelete(t *testing.T) {
|
|
||||||
ao := &utils.AbstractObject{}
|
|
||||||
assert.True(t, ao.CanDelete())
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestIsDrafted(t *testing.T) {
|
|
||||||
ao := &utils.AbstractObject{IsDraft: true}
|
|
||||||
assert.True(t, ao.IsDrafted())
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestGetID(t *testing.T) {
|
|
||||||
u := uuid.New().String()
|
|
||||||
ao := &utils.AbstractObject{UUID: u}
|
|
||||||
assert.Equal(t, u, ao.GetID())
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestGetName(t *testing.T) {
|
|
||||||
name := "MyObject"
|
|
||||||
ao := &utils.AbstractObject{Name: name}
|
|
||||||
assert.Equal(t, name, ao.GetName())
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestGetCreatorID(t *testing.T) {
|
|
||||||
id := "creator-123"
|
|
||||||
ao := &utils.AbstractObject{CreatorID: id}
|
|
||||||
assert.Equal(t, id, ao.GetCreatorID())
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestUpToDate_CreateFalse(t *testing.T) {
|
|
||||||
ao := &utils.AbstractObject{}
|
|
||||||
now := time.Now()
|
|
||||||
time.Sleep(time.Millisecond) // ensure time difference
|
|
||||||
ao.UpToDate("user123", "peer456", false)
|
|
||||||
assert.WithinDuration(t, now, ao.UpdateDate, time.Second)
|
|
||||||
assert.Equal(t, "peer456", ao.UpdaterID)
|
|
||||||
assert.Equal(t, "user123", ao.UserUpdaterID)
|
|
||||||
assert.True(t, ao.CreationDate.IsZero())
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestUpToDate_CreateTrue(t *testing.T) {
|
|
||||||
ao := &utils.AbstractObject{}
|
|
||||||
now := time.Now()
|
|
||||||
time.Sleep(time.Millisecond)
|
|
||||||
ao.UpToDate("user123", "peer456", true)
|
|
||||||
assert.WithinDuration(t, now, ao.UpdateDate, time.Second)
|
|
||||||
assert.WithinDuration(t, now, ao.CreationDate, time.Second)
|
|
||||||
assert.Equal(t, "peer456", ao.UpdaterID)
|
|
||||||
assert.Equal(t, "peer456", ao.CreatorID)
|
|
||||||
assert.Equal(t, "user123", ao.UserUpdaterID)
|
|
||||||
assert.Equal(t, "user123", ao.UserCreatorID)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestVerifyAuth(t *testing.T) {
|
|
||||||
request := &tools.APIRequest{PeerID: "peer123"}
|
|
||||||
ao := &utils.AbstractObject{CreatorID: "peer123"}
|
|
||||||
assert.True(t, ao.VerifyAuth(request))
|
|
||||||
|
|
||||||
ao = &utils.AbstractObject{AccessMode: utils.Public}
|
|
||||||
assert.True(t, ao.VerifyAuth(nil))
|
|
||||||
|
|
||||||
ao = &utils.AbstractObject{AccessMode: utils.Private, CreatorID: "peer123"}
|
|
||||||
request = &tools.APIRequest{PeerID: "wrong"}
|
|
||||||
assert.False(t, ao.VerifyAuth(request))
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestGetObjectFilters(t *testing.T) {
|
|
||||||
ao := &utils.AbstractObject{}
|
|
||||||
f := ao.GetObjectFilters("*")
|
|
||||||
assert.NotNil(t, f)
|
|
||||||
assert.Contains(t, f.Or, "abstractobject.name")
|
|
||||||
assert.Equal(t, dbs.LIKE.String(), f.Or["abstractobject.name"][0].Operator)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestDeserialize(t *testing.T) {
|
|
||||||
ao := &utils.AbstractObject{}
|
|
||||||
input := map[string]interface{}{"name": "test", "id": uuid.New().String()}
|
|
||||||
res := ao.Deserialize(input, &utils.AbstractObject{})
|
|
||||||
assert.NotNil(t, res)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestSerialize(t *testing.T) {
|
|
||||||
ao := &utils.AbstractObject{Name: "test", UUID: uuid.New().String()}
|
|
||||||
m := ao.Serialize(ao)
|
|
||||||
assert.Equal(t, "test", m["name"])
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestAbstractAccessorMethods(t *testing.T) {
|
|
||||||
r := &utils.AbstractAccessor{Request: &tools.APIRequest{Username: "alice", PeerID: "peer1", Groups: []string{"dev"}}}
|
|
||||||
assert.True(t, r.ShouldVerifyAuth())
|
|
||||||
assert.Equal(t, "alice", r.GetUser())
|
|
||||||
assert.Equal(t, "peer1", r.GetPeerID())
|
|
||||||
assert.Equal(t, []string{"dev"}, r.GetGroups())
|
|
||||||
assert.Equal(t, r.Request.Caller, r.GetCaller())
|
|
||||||
}
|
|
@ -1,168 +0,0 @@
|
|||||||
package models_test
|
|
||||||
|
|
||||||
import (
|
|
||||||
"errors"
|
|
||||||
"testing"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/dbs"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
"github.com/rs/zerolog"
|
|
||||||
"github.com/stretchr/testify/assert"
|
|
||||||
"github.com/stretchr/testify/mock"
|
|
||||||
)
|
|
||||||
|
|
||||||
// --- Mock Definitions ---
|
|
||||||
|
|
||||||
type MockDBObject struct {
|
|
||||||
mock.Mock
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *MockAccessor) GetLogger() *zerolog.Logger {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
func (m *MockAccessor) GetGroups() []string {
|
|
||||||
return []string{}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *MockAccessor) GetCaller() *tools.HTTPCaller {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *MockDBObject) GenerateID() { m.Called() }
|
|
||||||
func (m *MockDBObject) StoreDraftDefault() { m.Called() }
|
|
||||||
func (m *MockDBObject) UpToDate(user, peer string, create bool) {
|
|
||||||
m.Called(user, peer, create)
|
|
||||||
}
|
|
||||||
func (m *MockDBObject) VerifyAuth(req *tools.APIRequest) bool {
|
|
||||||
args := m.Called(req)
|
|
||||||
return args.Bool(0)
|
|
||||||
}
|
|
||||||
func (m *MockDBObject) CanDelete() bool {
|
|
||||||
args := m.Called()
|
|
||||||
return args.Bool(0)
|
|
||||||
}
|
|
||||||
func (m *MockDBObject) CanUpdate(set utils.DBObject) (bool, utils.DBObject) {
|
|
||||||
args := m.Called(set)
|
|
||||||
return args.Bool(0), args.Get(1).(utils.DBObject)
|
|
||||||
}
|
|
||||||
func (m *MockDBObject) IsDrafted() bool {
|
|
||||||
args := m.Called()
|
|
||||||
return args.Bool(0)
|
|
||||||
}
|
|
||||||
func (m *MockDBObject) Serialize(obj utils.DBObject) map[string]interface{} {
|
|
||||||
args := m.Called(obj)
|
|
||||||
return args.Get(0).(map[string]interface{})
|
|
||||||
}
|
|
||||||
func (m *MockDBObject) Deserialize(mdata map[string]interface{}, obj utils.DBObject) utils.DBObject {
|
|
||||||
args := m.Called(mdata, obj)
|
|
||||||
return args.Get(0).(utils.DBObject)
|
|
||||||
}
|
|
||||||
func (m *MockDBObject) GetID() string {
|
|
||||||
args := m.Called()
|
|
||||||
return args.String(0)
|
|
||||||
}
|
|
||||||
func (m *MockDBObject) GetName() string {
|
|
||||||
args := m.Called()
|
|
||||||
return args.String(0)
|
|
||||||
}
|
|
||||||
|
|
||||||
type MockAccessor struct {
|
|
||||||
mock.Mock
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *MockAccessor) DeleteOne(id string) (utils.DBObject, int, error) {
|
|
||||||
args := m.Called(id)
|
|
||||||
return args.Get(0).(utils.DBObject), args.Int(1), args.Error(2)
|
|
||||||
}
|
|
||||||
func (m *MockAccessor) LoadOne(id string) (utils.DBObject, int, error) {
|
|
||||||
args := m.Called(id)
|
|
||||||
return args.Get(0).(utils.DBObject), args.Int(1), args.Error(2)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *MockAccessor) LoadAll(isDraft bool) ([]utils.ShallowDBObject, int, error) {
|
|
||||||
args := m.Called(isDraft)
|
|
||||||
return args.Get(0).([]utils.ShallowDBObject), args.Int(1), args.Error(2)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *MockAccessor) UpdateOne(set utils.DBObject, id string) (utils.DBObject, int, error) {
|
|
||||||
args := m.Called(set, id)
|
|
||||||
return args.Get(0).(utils.DBObject), args.Int(1), args.Error(2)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *MockAccessor) CopyOne(data utils.DBObject) (utils.DBObject, int, error) {
|
|
||||||
args := m.Called(data)
|
|
||||||
return args.Get(0).(utils.DBObject), args.Int(1), args.Error(2)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *MockAccessor) StoreOne(data utils.DBObject) (utils.DBObject, int, error) {
|
|
||||||
args := m.Called(data)
|
|
||||||
return args.Get(0).(utils.DBObject), args.Int(1), args.Error(2)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *MockAccessor) ShouldVerifyAuth() bool {
|
|
||||||
args := m.Called()
|
|
||||||
return args.Bool(0)
|
|
||||||
}
|
|
||||||
func (m *MockAccessor) GetRequest() *tools.APIRequest {
|
|
||||||
args := m.Called()
|
|
||||||
return args.Get(0).(*tools.APIRequest)
|
|
||||||
}
|
|
||||||
func (m *MockAccessor) GetType() tools.DataType {
|
|
||||||
args := m.Called()
|
|
||||||
return args.Get(0).(tools.DataType)
|
|
||||||
}
|
|
||||||
func (m *MockAccessor) Search(filters *dbs.Filters, s string, d bool) ([]utils.ShallowDBObject, int, error) {
|
|
||||||
args := m.Called(filters, s, d)
|
|
||||||
return args.Get(0).([]utils.ShallowDBObject), args.Int(1), args.Error(2)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *MockAccessor) GetUser() string {
|
|
||||||
args := m.Called()
|
|
||||||
return args.String(0)
|
|
||||||
}
|
|
||||||
func (m *MockAccessor) GetPeerID() string {
|
|
||||||
args := m.Called()
|
|
||||||
return args.String(0)
|
|
||||||
}
|
|
||||||
|
|
||||||
// --- Test Cases ---
|
|
||||||
|
|
||||||
func TestVerifyAccess_Authorized(t *testing.T) {
|
|
||||||
mockObj := new(MockDBObject)
|
|
||||||
mockAcc := new(MockAccessor)
|
|
||||||
|
|
||||||
req := &tools.APIRequest{PeerID: "peer"}
|
|
||||||
mockAcc.On("LoadOne", "123").Return(mockObj, 200, nil)
|
|
||||||
mockAcc.On("ShouldVerifyAuth").Return(true)
|
|
||||||
mockObj.On("VerifyAuth", req).Return(true)
|
|
||||||
mockAcc.On("GetRequest").Return(req)
|
|
||||||
|
|
||||||
err := utils.VerifyAccess(mockAcc, "123")
|
|
||||||
assert.NoError(t, err)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestVerifyAccess_Unauthorized(t *testing.T) {
|
|
||||||
mockObj := new(MockDBObject)
|
|
||||||
mockAcc := new(MockAccessor)
|
|
||||||
|
|
||||||
req := &tools.APIRequest{PeerID: "peer"}
|
|
||||||
mockAcc.On("LoadOne", "123").Return(mockObj, 200, nil)
|
|
||||||
mockAcc.On("ShouldVerifyAuth").Return(true)
|
|
||||||
mockObj.On("VerifyAuth", req).Return(false)
|
|
||||||
mockAcc.On("GetRequest").Return(req)
|
|
||||||
|
|
||||||
err := utils.VerifyAccess(mockAcc, "123")
|
|
||||||
assert.Error(t, err)
|
|
||||||
assert.Contains(t, err.Error(), "not allowed")
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestVerifyAccess_LoadError(t *testing.T) {
|
|
||||||
mockAcc := new(MockAccessor)
|
|
||||||
|
|
||||||
mockAcc.On("LoadOne", "bad-id").Return(nil, 404, errors.New("not found"))
|
|
||||||
|
|
||||||
err := utils.VerifyAccess(mockAcc, "bad-id")
|
|
||||||
assert.Error(t, err)
|
|
||||||
assert.Equal(t, "not found", err.Error())
|
|
||||||
}
|
|
@ -1,147 +0,0 @@
|
|||||||
package graph
|
|
||||||
|
|
||||||
import (
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/resources"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Graph is a struct that represents a graph
|
|
||||||
type Graph struct {
|
|
||||||
Partial bool `json:"partial" default:"false"` // Partial is a flag that indicates if the graph is partial
|
|
||||||
Zoom float64 `bson:"zoom" json:"zoom" default:"1"` // Zoom is the graphical zoom of the graph
|
|
||||||
Items map[string]GraphItem `bson:"items" json:"items" default:"{}" validate:"required"` // Items is the list of elements in the graph
|
|
||||||
Links []GraphLink `bson:"links" json:"links" default:"{}" validate:"required"` // Links is the list of links between elements in the graph
|
|
||||||
}
|
|
||||||
|
|
||||||
func (g *Graph) Clear(id string) {
|
|
||||||
realItems := map[string]GraphItem{}
|
|
||||||
for k, it := range g.Items {
|
|
||||||
if k == id {
|
|
||||||
realinks := []GraphLink{}
|
|
||||||
for _, link := range g.Links {
|
|
||||||
if link.Source.ID != id && link.Destination.ID != id {
|
|
||||||
realinks = append(realinks, link)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
g.Links = realinks
|
|
||||||
g.Partial = true
|
|
||||||
} else {
|
|
||||||
realItems[k] = it
|
|
||||||
}
|
|
||||||
}
|
|
||||||
g.Items = realItems
|
|
||||||
}
|
|
||||||
|
|
||||||
func (wf *Graph) IsProcessing(item GraphItem) bool {
|
|
||||||
return item.Processing != nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (wf *Graph) IsCompute(item GraphItem) bool {
|
|
||||||
return item.Compute != nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (wf *Graph) IsData(item GraphItem) bool {
|
|
||||||
return item.Data != nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (wf *Graph) IsStorage(item GraphItem) bool {
|
|
||||||
return item.Storage != nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (wf *Graph) IsWorkflow(item GraphItem) bool {
|
|
||||||
return item.Workflow != nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (g *Graph) GetAverageTimeRelatedToProcessingActivity(start time.Time, processings []*resources.ProcessingResource, resource resources.ResourceInterface,
|
|
||||||
f func(GraphItem) resources.ResourceInterface, request *tools.APIRequest) (float64, float64) {
|
|
||||||
nearestStart := float64(10000000000)
|
|
||||||
oneIsInfinite := false
|
|
||||||
longestDuration := float64(0)
|
|
||||||
for _, link := range g.Links {
|
|
||||||
for _, processing := range processings {
|
|
||||||
var source string // source is the source of the link
|
|
||||||
if link.Destination.ID == processing.GetID() && f(g.Items[link.Source.ID]) != nil && f(g.Items[link.Source.ID]).GetID() == resource.GetID() { // if the destination is the processing and the source is not a compute
|
|
||||||
source = link.Source.ID
|
|
||||||
} else if link.Source.ID == processing.GetID() && f(g.Items[link.Source.ID]) != nil && f(g.Items[link.Source.ID]).GetID() == resource.GetID() { // if the source is the processing and the destination is not a compute
|
|
||||||
source = link.Destination.ID
|
|
||||||
}
|
|
||||||
priced := processing.ConvertToPricedResource(tools.PROCESSING_RESOURCE, request)
|
|
||||||
if source != "" {
|
|
||||||
if priced.GetLocationStart() != nil {
|
|
||||||
near := float64(priced.GetLocationStart().Sub(start).Seconds())
|
|
||||||
if near < nearestStart {
|
|
||||||
nearestStart = near
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
if priced.GetLocationEnd() != nil {
|
|
||||||
duration := float64(priced.GetLocationEnd().Sub(*priced.GetLocationStart()).Seconds())
|
|
||||||
if longestDuration < duration {
|
|
||||||
longestDuration = duration
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
oneIsInfinite = true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if oneIsInfinite {
|
|
||||||
return nearestStart, -1
|
|
||||||
}
|
|
||||||
return nearestStart, longestDuration
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
|
||||||
* GetAverageTimeBeforeStart is a function that returns the average time before the start of a processing
|
|
||||||
*/
|
|
||||||
func (g *Graph) GetAverageTimeProcessingBeforeStart(average float64, processingID string, request *tools.APIRequest) float64 {
|
|
||||||
currents := []float64{} // list of current time
|
|
||||||
for _, link := range g.Links { // for each link
|
|
||||||
var source string // source is the source of the link
|
|
||||||
if link.Destination.ID == processingID && g.Items[link.Source.ID].Processing == nil { // if the destination is the processing and the source is not a compute
|
|
||||||
source = link.Source.ID
|
|
||||||
} else if link.Source.ID == processingID && g.Items[link.Source.ID].Processing == nil { // if the source is the processing and the destination is not a compute
|
|
||||||
source = link.Destination.ID
|
|
||||||
}
|
|
||||||
if source == "" { // if source is empty, continue
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
dt, r := g.GetResource(source) // get the resource of the source
|
|
||||||
if r == nil { // if item is nil, continue
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
priced := r.ConvertToPricedResource(dt, request)
|
|
||||||
current := priced.GetExplicitDurationInS() // get the explicit duration of the item
|
|
||||||
if current < 0 { // if current is negative, its means that duration of a before could be infinite continue
|
|
||||||
return current
|
|
||||||
}
|
|
||||||
current += g.GetAverageTimeProcessingBeforeStart(current, source, request) // get the average time before start of the source
|
|
||||||
currents = append(currents, current) // append the current to the currents
|
|
||||||
}
|
|
||||||
var max float64 // get the max time to wait dependancies to finish
|
|
||||||
for _, current := range currents {
|
|
||||||
if current > max {
|
|
||||||
max = current
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return max
|
|
||||||
}
|
|
||||||
|
|
||||||
func (g *Graph) GetResource(id string) (tools.DataType, resources.ResourceInterface) {
|
|
||||||
if item, ok := g.Items[id]; ok {
|
|
||||||
if item.Data != nil {
|
|
||||||
return tools.DATA_RESOURCE, item.Data
|
|
||||||
} else if item.Compute != nil {
|
|
||||||
return tools.COMPUTE_RESOURCE, item.Compute
|
|
||||||
} else if item.Workflow != nil {
|
|
||||||
return tools.WORKFLOW_RESOURCE, item.Workflow
|
|
||||||
} else if item.Processing != nil {
|
|
||||||
return tools.PROCESSING_RESOURCE, item.Processing
|
|
||||||
} else if item.Storage != nil {
|
|
||||||
return tools.STORAGE_RESOURCE, item.Storage
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return tools.INVALID, nil
|
|
||||||
}
|
|
@ -1,38 +0,0 @@
|
|||||||
package graph
|
|
||||||
|
|
||||||
import (
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/resources"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
|
||||||
)
|
|
||||||
|
|
||||||
// GraphItem is a struct that represents an item in a graph
|
|
||||||
type GraphItem struct {
|
|
||||||
ID string `bson:"id" json:"id" validate:"required"` // ID is the unique identifier of the item
|
|
||||||
Width float64 `bson:"width" json:"width" validate:"required"` // Width is the graphical width of the item
|
|
||||||
Height float64 `bson:"height" json:"height" validate:"required"` // Height is the graphical height of the item
|
|
||||||
Position Position `bson:"position" json:"position" validate:"required"` // Position is the graphical position of the item
|
|
||||||
*resources.ItemResource // ItemResource is the resource of the item affected to the item
|
|
||||||
}
|
|
||||||
|
|
||||||
func (g *GraphItem) GetResource() (tools.DataType, resources.ResourceInterface) {
|
|
||||||
if g.Data != nil {
|
|
||||||
return tools.DATA_RESOURCE, g.Data
|
|
||||||
} else if g.Compute != nil {
|
|
||||||
return tools.COMPUTE_RESOURCE, g.Compute
|
|
||||||
} else if g.Workflow != nil {
|
|
||||||
return tools.WORKFLOW_RESOURCE, g.Workflow
|
|
||||||
} else if g.Processing != nil {
|
|
||||||
return tools.PROCESSING_RESOURCE, g.Processing
|
|
||||||
} else if g.Storage != nil {
|
|
||||||
return tools.STORAGE_RESOURCE, g.Storage
|
|
||||||
}
|
|
||||||
return tools.INVALID, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (g *GraphItem) Clear() {
|
|
||||||
g.Data = nil
|
|
||||||
g.Compute = nil
|
|
||||||
g.Workflow = nil
|
|
||||||
g.Processing = nil
|
|
||||||
g.Storage = nil
|
|
||||||
}
|
|
@ -1,56 +0,0 @@
|
|||||||
package graph
|
|
||||||
|
|
||||||
import "cloud.o-forge.io/core/oc-lib/models/common/models"
|
|
||||||
|
|
||||||
type StorageProcessingGraphLink struct {
|
|
||||||
Write bool `json:"write" bson:"write"`
|
|
||||||
Source string `json:"source" bson:"source"`
|
|
||||||
Destination string `json:"destination" bson:"destination"`
|
|
||||||
FileName string `json:"filename" bson:"filename"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// GraphLink is a struct that represents a link between two items in a graph
|
|
||||||
type GraphLink struct {
|
|
||||||
Source Position `bson:"source" json:"source" validate:"required"` // Source is the source graphical position of the link
|
|
||||||
Destination Position `bson:"destination" json:"destination" validate:"required"` // Destination is the destination graphical position of the link
|
|
||||||
Style *GraphLinkStyle `bson:"style,omitempty" json:"style,omitempty"` // Style is the graphical style of the link
|
|
||||||
StorageLinkInfos []StorageProcessingGraphLink `bson:"storage_link_infos,omitempty" json:"storage_link_infos,omitempty"` // StorageLinkInfo is the storage link info
|
|
||||||
Env []models.Param `json:"env" bson:"env"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// tool function to check if a link is a link between a compute and a resource
|
|
||||||
func (l *GraphLink) IsComputeLink(g Graph) (bool, string) {
|
|
||||||
if g.Items == nil {
|
|
||||||
return false, ""
|
|
||||||
}
|
|
||||||
if d, ok := g.Items[l.Source.ID]; ok && d.Compute != nil {
|
|
||||||
return true, d.Compute.UUID
|
|
||||||
}
|
|
||||||
if d, ok := g.Items[l.Destination.ID]; ok && d.Compute != nil {
|
|
||||||
return true, d.Compute.UUID
|
|
||||||
}
|
|
||||||
return false, ""
|
|
||||||
}
|
|
||||||
|
|
||||||
// GraphLinkStyle is a struct that represents the style of a link in a graph
|
|
||||||
type GraphLinkStyle struct {
|
|
||||||
Color int64 `bson:"color" json:"color"` // Color is the graphical color of the link (int description of a color, can be transpose as hex)
|
|
||||||
Stroke float64 `bson:"stroke" json:"stroke"` // Stroke is the graphical stroke of the link
|
|
||||||
Tension float64 `bson:"tension" json:"tension"` // Tension is the graphical tension of the link
|
|
||||||
HeadRadius float64 `bson:"head_radius" json:"head_radius"` // graphical pin radius
|
|
||||||
DashWidth float64 `bson:"dash_width" json:"dash_width"` // DashWidth is the graphical dash width of the link
|
|
||||||
DashSpace float64 `bson:"dash_space" json:"dash_space"` // DashSpace is the graphical dash space of the link
|
|
||||||
EndArrow Position `bson:"end_arrow" json:"end_arrow"` // EndArrow is the graphical end arrow of the link
|
|
||||||
StartArrow Position `bson:"start_arrow" json:"start_arrow"` // StartArrow is the graphical start arrow of the link
|
|
||||||
ArrowStyle int64 `bson:"arrow_style" json:"arrow_style"` // ArrowStyle is the graphical arrow style of the link (enum foundable in UI)
|
|
||||||
ArrowDirection int64 `bson:"arrow_direction" json:"arrow_direction"` // ArrowDirection is the graphical arrow direction of the link (enum foundable in UI)
|
|
||||||
StartArrowWidth float64 `bson:"start_arrow_width" json:"start_arrow_width"` // StartArrowWidth is the graphical start arrow width of the link
|
|
||||||
EndArrowWidth float64 `bson:"end_arrow_width" json:"end_arrow_width"` // EndArrowWidth is the graphical end arrow width of the link
|
|
||||||
}
|
|
||||||
|
|
||||||
// Position is a struct that represents a graphical position
|
|
||||||
type Position struct {
|
|
||||||
ID string `json:"id" bson:"id"` // ID reprents ItemID (optionnal)
|
|
||||||
X float64 `json:"x" bson:"x" validate:"required"` // X is the graphical x position
|
|
||||||
Y float64 `json:"y" bson:"y" validate:"required"` // Y is the graphical y position
|
|
||||||
}
|
|
@ -1,140 +1,103 @@
|
|||||||
package workflow
|
package workflow
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"encoding/json"
|
||||||
"errors"
|
"errors"
|
||||||
"time"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/collaborative_area/shallow_collaborative_area"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/pricing"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/peer"
|
"cloud.o-forge.io/core/oc-lib/models/peer"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/resources"
|
"cloud.o-forge.io/core/oc-lib/models/resources"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/resources/compute"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/resources/storage"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/resources/workflow/graph"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
"cloud.o-forge.io/core/oc-lib/models/utils"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/workflow/graph"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
"cloud.o-forge.io/core/oc-lib/tools"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
/*
|
||||||
|
* AbstractWorkflow is a struct that represents a workflow for resource or native workflow
|
||||||
|
* Warning: there is 2 types of workflows, the resource workflow and the native workflow
|
||||||
|
* native workflow is the one that you create to schedule an execution
|
||||||
|
* resource workflow is the one that is created to set our native workflow in catalog
|
||||||
|
*/
|
||||||
|
type AbstractWorkflow struct {
|
||||||
|
resources.ResourceSet
|
||||||
|
Graph *graph.Graph `bson:"graph,omitempty" json:"graph,omitempty"` // Graph UI & logic representation of the workflow
|
||||||
|
ScheduleActive bool `json:"schedule_active" bson:"schedule_active"` // ScheduleActive is a flag that indicates if the schedule is active, if not the workflow is not scheduled and no execution or booking will be set
|
||||||
|
Schedule *WorkflowSchedule `bson:"schedule,omitempty" json:"schedule,omitempty"` // Schedule is the schedule of the workflow
|
||||||
|
Shared []string `json:"shared,omitempty" bson:"shared,omitempty"` // Shared is the ID of the shared workflow
|
||||||
|
}
|
||||||
|
|
||||||
|
func (w *AbstractWorkflow) GetWorkflows() (list_computings []graph.GraphItem) {
|
||||||
|
for _, item := range w.Graph.Items {
|
||||||
|
if item.Workflow != nil {
|
||||||
|
list_computings = append(list_computings, item)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (w *AbstractWorkflow) GetComputeByRelatedProcessing(processingID string) []*compute.ComputeResource {
|
||||||
|
storages := []*compute.ComputeResource{}
|
||||||
|
for _, link := range w.Graph.Links {
|
||||||
|
nodeID := link.Destination.ID // we considers that the processing is the destination
|
||||||
|
node := w.Graph.Items[link.Source.ID].Compute // we are looking for the storage as source
|
||||||
|
if node == nil { // if the source is not a storage, we consider that the destination is the storage
|
||||||
|
nodeID = link.Source.ID // and the processing is the source
|
||||||
|
node = w.Graph.Items[link.Destination.ID].Compute // we are looking for the storage as destination
|
||||||
|
}
|
||||||
|
if processingID == nodeID && node != nil { // if the storage is linked to the processing
|
||||||
|
storages = append(storages, node)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return storages
|
||||||
|
}
|
||||||
|
|
||||||
|
func (w *AbstractWorkflow) GetStoragesByRelatedProcessing(processingID string) []*storage.StorageResource {
|
||||||
|
storages := []*storage.StorageResource{}
|
||||||
|
for _, link := range w.Graph.Links {
|
||||||
|
nodeID := link.Destination.ID // we considers that the processing is the destination
|
||||||
|
node := w.Graph.Items[link.Source.ID].Storage // we are looking for the storage as source
|
||||||
|
if node == nil { // if the source is not a storage, we consider that the destination is the storage
|
||||||
|
nodeID = link.Source.ID // and the processing is the source
|
||||||
|
node = w.Graph.Items[link.Destination.ID].Storage // we are looking for the storage as destination
|
||||||
|
}
|
||||||
|
if processingID == nodeID && node != nil { // if the storage is linked to the processing
|
||||||
|
storages = append(storages, node)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return storages
|
||||||
|
}
|
||||||
|
|
||||||
|
func (w *AbstractWorkflow) GetProcessings() (list_computings []graph.GraphItem) {
|
||||||
|
for _, item := range w.Graph.Items {
|
||||||
|
if item.Processing != nil {
|
||||||
|
list_computings = append(list_computings, item)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// tool function to check if a link is a link between a compute and a resource
|
||||||
|
func (w *AbstractWorkflow) isDCLink(link graph.GraphLink) (bool, string) {
|
||||||
|
if w.Graph == nil || w.Graph.Items == nil {
|
||||||
|
return false, ""
|
||||||
|
}
|
||||||
|
if d, ok := w.Graph.Items[link.Source.ID]; ok && d.Compute != nil {
|
||||||
|
return true, d.Compute.UUID
|
||||||
|
}
|
||||||
|
if d, ok := w.Graph.Items[link.Destination.ID]; ok && d.Compute != nil {
|
||||||
|
return true, d.Compute.UUID
|
||||||
|
}
|
||||||
|
return false, ""
|
||||||
|
}
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* Workflow is a struct that represents a workflow
|
* Workflow is a struct that represents a workflow
|
||||||
* it defines the native workflow
|
* it defines the native workflow
|
||||||
*/
|
*/
|
||||||
type Workflow struct {
|
type Workflow struct {
|
||||||
utils.AbstractObject // AbstractObject contains the basic fields of an object (id, name)
|
utils.AbstractObject // AbstractObject contains the basic fields of an object (id, name)
|
||||||
resources.ResourceSet
|
AbstractWorkflow // AbstractWorkflow contains the basic fields of a workflow
|
||||||
Graph *graph.Graph `bson:"graph,omitempty" json:"graph,omitempty"` // Graph UI & logic representation of the workflow
|
|
||||||
ScheduleActive bool `json:"schedule_active" bson:"schedule_active"` // ScheduleActive is a flag that indicates if the schedule is active, if not the workflow is not scheduled and no execution or booking will be set
|
|
||||||
// Schedule *WorkflowSchedule `bson:"schedule,omitempty" json:"schedule,omitempty"` // Schedule is the schedule of the workflow
|
|
||||||
Shared []string `json:"shared,omitempty" bson:"shared,omitempty"` // Shared is the ID of the shared workflow // AbstractWorkflow contains the basic fields of a workflow
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *Workflow) GetAccessor(request *tools.APIRequest) utils.Accessor {
|
|
||||||
return NewAccessor(request) // Create a new instance of the accessor
|
|
||||||
}
|
|
||||||
|
|
||||||
type Deps struct {
|
|
||||||
Source string
|
|
||||||
Dest string
|
|
||||||
}
|
|
||||||
|
|
||||||
func (w *Workflow) IsDependancy(id string) []Deps {
|
|
||||||
dependancyOfIDs := []Deps{}
|
|
||||||
for _, link := range w.Graph.Links {
|
|
||||||
if _, ok := w.Graph.Items[link.Destination.ID]; !ok {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
source := w.Graph.Items[link.Destination.ID].Processing
|
|
||||||
if id == link.Source.ID && source != nil {
|
|
||||||
dependancyOfIDs = append(dependancyOfIDs, Deps{Source: source.GetName(), Dest: link.Destination.ID})
|
|
||||||
}
|
|
||||||
sourceWF := w.Graph.Items[link.Destination.ID].Workflow
|
|
||||||
if id == link.Source.ID && sourceWF != nil {
|
|
||||||
dependancyOfIDs = append(dependancyOfIDs, Deps{Source: sourceWF.GetName(), Dest: link.Destination.ID})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return dependancyOfIDs
|
|
||||||
}
|
|
||||||
|
|
||||||
func (w *Workflow) GetDependencies(id string) (dependencies []Deps) {
|
|
||||||
for _, link := range w.Graph.Links {
|
|
||||||
if _, ok := w.Graph.Items[link.Source.ID]; !ok {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
source := w.Graph.Items[link.Source.ID].Processing
|
|
||||||
if id == link.Destination.ID && source != nil {
|
|
||||||
dependencies = append(dependencies, Deps{Source: source.GetName(), Dest: link.Source.ID})
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
func (w *Workflow) GetGraphItems(f func(item graph.GraphItem) bool) (list_datas []graph.GraphItem) {
|
|
||||||
for _, item := range w.Graph.Items {
|
|
||||||
if f(item) {
|
|
||||||
list_datas = append(list_datas, item)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
func (w *Workflow) GetPricedItem(
|
|
||||||
f func(item graph.GraphItem) bool, request *tools.APIRequest, buyingStrategy int, pricingStrategy int) map[string]pricing.PricedItemITF {
|
|
||||||
list_datas := map[string]pricing.PricedItemITF{}
|
|
||||||
for _, item := range w.Graph.Items {
|
|
||||||
if f(item) {
|
|
||||||
dt, res := item.GetResource()
|
|
||||||
ord := res.ConvertToPricedResource(dt, request)
|
|
||||||
list_datas[res.GetID()] = ord
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return list_datas
|
|
||||||
}
|
|
||||||
|
|
||||||
type Related struct {
|
|
||||||
Node resources.ResourceInterface
|
|
||||||
Links []graph.GraphLink
|
|
||||||
}
|
|
||||||
|
|
||||||
func (w *Workflow) GetByRelatedProcessing(processingID string, g func(item graph.GraphItem) bool) map[string]Related {
|
|
||||||
related := map[string]Related{}
|
|
||||||
for _, link := range w.Graph.Links {
|
|
||||||
nodeID := link.Destination.ID
|
|
||||||
var node resources.ResourceInterface
|
|
||||||
if g(w.Graph.Items[link.Source.ID]) {
|
|
||||||
item := w.Graph.Items[link.Source.ID]
|
|
||||||
_, node = item.GetResource()
|
|
||||||
}
|
|
||||||
if node == nil && g(w.Graph.Items[link.Destination.ID]) { // if the source is not a storage, we consider that the destination is the storage
|
|
||||||
nodeID = link.Source.ID
|
|
||||||
item := w.Graph.Items[link.Destination.ID] // and the processing is the source
|
|
||||||
_, node = item.GetResource() // we are looking for the storage as destination
|
|
||||||
}
|
|
||||||
if processingID == nodeID && node != nil { // if the storage is linked to the processing
|
|
||||||
if _, ok := related[processingID]; !ok {
|
|
||||||
related[processingID] = Related{}
|
|
||||||
}
|
|
||||||
rel := related[node.GetID()]
|
|
||||||
rel.Node = node
|
|
||||||
rel.Links = append(rel.Links, link)
|
|
||||||
related[processingID] = rel
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return related
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ao *Workflow) VerifyAuth(request *tools.APIRequest) bool {
|
|
||||||
isAuthorized := false
|
|
||||||
if len(ao.Shared) > 0 {
|
|
||||||
for _, shared := range ao.Shared {
|
|
||||||
shared, code, _ := shallow_collaborative_area.NewAccessor(request).LoadOne(shared)
|
|
||||||
if code != 200 || shared == nil {
|
|
||||||
isAuthorized = false
|
|
||||||
} else {
|
|
||||||
isAuthorized = shared.VerifyAuth(request)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return ao.AbstractObject.VerifyAuth(request) || isAuthorized
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/*
|
/*
|
||||||
@ -145,19 +108,19 @@ func (wfa *Workflow) CheckBooking(caller *tools.HTTPCaller) (bool, error) {
|
|||||||
if wfa.Graph == nil { // no graph no booking
|
if wfa.Graph == nil { // no graph no booking
|
||||||
return false, nil
|
return false, nil
|
||||||
}
|
}
|
||||||
accessor := (&resources.ComputeResource{}).GetAccessor(&tools.APIRequest{Caller: caller})
|
accessor := (&compute.ComputeResource{}).GetAccessor(nil)
|
||||||
for _, link := range wfa.Graph.Links {
|
for _, link := range wfa.Graph.Links {
|
||||||
if ok, compute_id := link.IsComputeLink(*wfa.Graph); ok { // check if the link is a link between a compute and a resource
|
if ok, dc_id := wfa.isDCLink(link); ok { // check if the link is a link between a compute and a resource
|
||||||
compute, code, _ := accessor.LoadOne(compute_id)
|
dc, code, _ := accessor.LoadOne(dc_id)
|
||||||
if code != 200 {
|
if code != 200 {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
// CHECK BOOKING ON PEER, compute could be a remote one
|
// CHECK BOOKING ON PEER, compute could be a remote one
|
||||||
peerID := compute.(*resources.ComputeResource).CreatorID
|
peerID := dc.(*compute.ComputeResource).PeerID
|
||||||
if peerID == "" {
|
if peerID == "" {
|
||||||
return false, errors.New("no peer id")
|
return false, errors.New("no peer id")
|
||||||
} // no peer id no booking, we need to know where to book
|
} // no peer id no booking, we need to know where to book
|
||||||
_, err := (&peer.Peer{}).LaunchPeerExecution(peerID, compute_id, tools.BOOKING, tools.GET, nil, caller)
|
_, err := (&peer.Peer{}).LaunchPeerExecution(peerID, dc_id, tools.BOOKING, tools.GET, nil, caller)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return false, err
|
return false, err
|
||||||
}
|
}
|
||||||
@ -166,96 +129,31 @@ func (wfa *Workflow) CheckBooking(caller *tools.HTTPCaller) (bool, error) {
|
|||||||
return true, nil
|
return true, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (wf *Workflow) Planify(start time.Time, end *time.Time, request *tools.APIRequest) (float64, map[tools.DataType]map[string]pricing.PricedItemITF, *Workflow, error) {
|
func (d *Workflow) GetName() string {
|
||||||
priceds := map[tools.DataType]map[string]pricing.PricedItemITF{}
|
return d.Name
|
||||||
ps, priceds, err := plan[*resources.ProcessingResource](tools.PROCESSING_RESOURCE, wf, priceds, request, wf.Graph.IsProcessing,
|
|
||||||
func(res resources.ResourceInterface, priced pricing.PricedItemITF) (time.Time, float64) {
|
|
||||||
return start.Add(time.Duration(wf.Graph.GetAverageTimeProcessingBeforeStart(0, res.GetID(), request)) * time.Second), priced.GetExplicitDurationInS()
|
|
||||||
}, func(started time.Time, duration float64) *time.Time {
|
|
||||||
s := started.Add(time.Duration(duration))
|
|
||||||
return &s
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
return 0, priceds, nil, err
|
|
||||||
}
|
|
||||||
if _, priceds, err = plan[resources.ResourceInterface](tools.DATA_RESOURCE, wf, priceds, request,
|
|
||||||
wf.Graph.IsData, func(res resources.ResourceInterface, priced pricing.PricedItemITF) (time.Time, float64) {
|
|
||||||
return start, 0
|
|
||||||
}, func(started time.Time, duration float64) *time.Time {
|
|
||||||
return end
|
|
||||||
}); err != nil {
|
|
||||||
return 0, priceds, nil, err
|
|
||||||
}
|
|
||||||
for k, f := range map[tools.DataType]func(graph.GraphItem) bool{tools.STORAGE_RESOURCE: wf.Graph.IsStorage, tools.COMPUTE_RESOURCE: wf.Graph.IsCompute} {
|
|
||||||
if _, priceds, err = plan[resources.ResourceInterface](k, wf, priceds, request,
|
|
||||||
f, func(res resources.ResourceInterface, priced pricing.PricedItemITF) (time.Time, float64) {
|
|
||||||
nearestStart, longestDuration := wf.Graph.GetAverageTimeRelatedToProcessingActivity(start, ps, res, func(i graph.GraphItem) (r resources.ResourceInterface) {
|
|
||||||
if f(i) {
|
|
||||||
_, r = i.GetResource()
|
|
||||||
}
|
|
||||||
return r
|
|
||||||
}, request)
|
|
||||||
return start.Add(time.Duration(nearestStart) * time.Second), longestDuration
|
|
||||||
}, func(started time.Time, duration float64) *time.Time {
|
|
||||||
s := started.Add(time.Duration(duration))
|
|
||||||
return &s
|
|
||||||
}); err != nil {
|
|
||||||
return 0, priceds, nil, err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
longest := common.GetPlannerLongestTime(end, priceds, request)
|
|
||||||
if _, priceds, err = plan[resources.ResourceInterface](tools.WORKFLOW_RESOURCE, wf, priceds, request, wf.Graph.IsWorkflow,
|
|
||||||
func(res resources.ResourceInterface, priced pricing.PricedItemITF) (time.Time, float64) {
|
|
||||||
start := start.Add(time.Duration(common.GetPlannerNearestStart(start, priceds, request)) * time.Second)
|
|
||||||
longest := float64(-1)
|
|
||||||
r, code, err := res.GetAccessor(request).LoadOne(res.GetID())
|
|
||||||
if code != 200 || err != nil {
|
|
||||||
return start, longest
|
|
||||||
}
|
|
||||||
if neoLongest, _, _, err := r.(*Workflow).Planify(start, end, request); err != nil {
|
|
||||||
return start, longest
|
|
||||||
} else if neoLongest > longest {
|
|
||||||
longest = neoLongest
|
|
||||||
}
|
|
||||||
return start.Add(time.Duration(common.GetPlannerNearestStart(start, priceds, request)) * time.Second), longest
|
|
||||||
}, func(start time.Time, longest float64) *time.Time {
|
|
||||||
s := start.Add(time.Duration(longest) * time.Second)
|
|
||||||
return &s
|
|
||||||
}); err != nil {
|
|
||||||
return 0, priceds, nil, err
|
|
||||||
}
|
|
||||||
return longest, priceds, wf, nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func plan[T resources.ResourceInterface](
|
func (d *Workflow) GetAccessor(caller *tools.HTTPCaller) utils.Accessor {
|
||||||
dt tools.DataType, wf *Workflow, priceds map[tools.DataType]map[string]pricing.PricedItemITF, request *tools.APIRequest,
|
data := New() // Create a new instance of the accessor
|
||||||
f func(graph.GraphItem) bool, start func(resources.ResourceInterface, pricing.PricedItemITF) (time.Time, float64), end func(time.Time, float64) *time.Time) ([]T, map[tools.DataType]map[string]pricing.PricedItemITF, error) {
|
data.Init(tools.WORKFLOW, caller) // Initialize the accessor with the WORKFLOW model type
|
||||||
resources := []T{}
|
return data
|
||||||
for _, item := range wf.GetGraphItems(f) {
|
}
|
||||||
if priceds[dt] == nil {
|
|
||||||
priceds[dt] = map[string]pricing.PricedItemITF{}
|
func (dma *Workflow) Deserialize(j map[string]interface{}) utils.DBObject {
|
||||||
}
|
b, err := json.Marshal(j)
|
||||||
dt, realItem := item.GetResource()
|
if err != nil {
|
||||||
if realItem == nil {
|
return nil
|
||||||
return resources, priceds, errors.New("could not load the processing resource")
|
}
|
||||||
}
|
json.Unmarshal(b, dma)
|
||||||
priced := realItem.ConvertToPricedResource(dt, request)
|
return dma
|
||||||
// Should be commented once the Pricing selection feature has been implemented, related to the commit d35ad440fa77763ec7f49ab34a85e47e75581b61
|
}
|
||||||
// if priced.SelectPricing() == nil {
|
|
||||||
// return resources, priceds, errors.New("no pricings are selected... can't proceed")
|
func (dma *Workflow) Serialize() map[string]interface{} {
|
||||||
// }
|
var m map[string]interface{}
|
||||||
started, duration := start(realItem, priced)
|
b, err := json.Marshal(dma)
|
||||||
priced.SetLocationStart(started)
|
if err != nil {
|
||||||
if duration >= 0 {
|
return nil
|
||||||
if e := end(started, duration); e != nil {
|
}
|
||||||
priced.SetLocationEnd(*e)
|
json.Unmarshal(b, &m)
|
||||||
}
|
return m
|
||||||
}
|
|
||||||
if e := end(started, priced.GetExplicitDurationInS()); e != nil {
|
|
||||||
priced.SetLocationEnd(*e)
|
|
||||||
}
|
|
||||||
resources = append(resources, realItem.(T))
|
|
||||||
priceds[dt][item.ID] = priced
|
|
||||||
}
|
|
||||||
return resources, priceds, nil
|
|
||||||
}
|
}
|
||||||
|
@ -8,8 +8,10 @@ import (
|
|||||||
|
|
||||||
type WorkflowHistory struct{ Workflow }
|
type WorkflowHistory struct{ Workflow }
|
||||||
|
|
||||||
func (d *WorkflowHistory) GetAccessor(request *tools.APIRequest) utils.Accessor {
|
func (d *WorkflowHistory) GetAccessor(caller *tools.HTTPCaller) utils.Accessor {
|
||||||
return NewAccessorHistory(request) // Create a new instance of the accessor
|
data := New() // Create a new instance of the accessor
|
||||||
|
data.Init(tools.WORKSPACE_HISTORY, caller) // Initialize the accessor with the WORKSPACE model type
|
||||||
|
return data
|
||||||
}
|
}
|
||||||
func (r *WorkflowHistory) GenerateID() {
|
func (r *WorkflowHistory) GenerateID() {
|
||||||
r.UUID = uuid.New().String()
|
r.UUID = uuid.New().String()
|
||||||
|
@ -2,65 +2,175 @@ package workflow
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"slices"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/dbs"
|
"cloud.o-forge.io/core/oc-lib/dbs"
|
||||||
"cloud.o-forge.io/core/oc-lib/logs"
|
"cloud.o-forge.io/core/oc-lib/dbs/mongo"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/collaborative_area/shallow_collaborative_area"
|
"cloud.o-forge.io/core/oc-lib/models/collaborative_area/shallow_collaborative_area"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/peer"
|
"cloud.o-forge.io/core/oc-lib/models/peer"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/resources"
|
"cloud.o-forge.io/core/oc-lib/models/resources"
|
||||||
|
"cloud.o-forge.io/core/oc-lib/models/resources/compute"
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
"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/models/workspace"
|
"cloud.o-forge.io/core/oc-lib/models/workspace"
|
||||||
"cloud.o-forge.io/core/oc-lib/tools"
|
"cloud.o-forge.io/core/oc-lib/tools"
|
||||||
|
cron "github.com/robfig/cron/v3"
|
||||||
)
|
)
|
||||||
|
|
||||||
type workflowMongoAccessor struct {
|
type workflowMongoAccessor struct {
|
||||||
utils.AbstractAccessor // AbstractAccessor contains the basic fields of an accessor (model, caller)
|
utils.AbstractAccessor // AbstractAccessor contains the basic fields of an accessor (model, caller)
|
||||||
computeResourceAccessor utils.Accessor
|
|
||||||
collaborativeAreaAccessor utils.Accessor
|
|
||||||
workspaceAccessor utils.Accessor
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewAccessorHistory(request *tools.APIRequest) *workflowMongoAccessor {
|
|
||||||
return new(tools.WORKFLOW_HISTORY, request)
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewAccessor(request *tools.APIRequest) *workflowMongoAccessor {
|
|
||||||
return new(tools.WORKFLOW, request)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// New creates a new instance of the workflowMongoAccessor
|
// New creates a new instance of the workflowMongoAccessor
|
||||||
func new(t tools.DataType, request *tools.APIRequest) *workflowMongoAccessor {
|
func New() *workflowMongoAccessor {
|
||||||
return &workflowMongoAccessor{
|
return &workflowMongoAccessor{}
|
||||||
computeResourceAccessor: (&resources.ComputeResource{}).GetAccessor(request),
|
}
|
||||||
collaborativeAreaAccessor: (&shallow_collaborative_area.ShallowCollaborativeArea{}).GetAccessor(request),
|
|
||||||
workspaceAccessor: (&workspace.Workspace{}).GetAccessor(request),
|
/*
|
||||||
AbstractAccessor: utils.AbstractAccessor{
|
* THERE IS A LOT IN THIS FILE SHOULD BE AWARE OF THE COMMENTS
|
||||||
Logger: logs.CreateLogger(t.String()), // Create a logger with the data type
|
*/
|
||||||
Request: request,
|
|
||||||
Type: t,
|
/*
|
||||||
},
|
* getExecutions is a function that returns the executions of a workflow
|
||||||
|
* it returns an array of workflow_execution.WorkflowExecution
|
||||||
|
*/
|
||||||
|
func (wfa *workflowMongoAccessor) getExecutions(id string, data *Workflow) ([]*workflow_execution.WorkflowExecution, error) {
|
||||||
|
workflows_execution := []*workflow_execution.WorkflowExecution{}
|
||||||
|
if data.Schedule != nil { // only set execution on a scheduled workflow
|
||||||
|
if data.Schedule.Start == nil { // if no start date, return an error
|
||||||
|
return workflows_execution, errors.New("should get a start date on the scheduler.")
|
||||||
|
}
|
||||||
|
if data.Schedule.End != nil && data.Schedule.End.IsZero() { // if end date is zero, set it to nil
|
||||||
|
data.Schedule.End = nil
|
||||||
|
}
|
||||||
|
if len(data.Schedule.Cron) > 0 { // if cron is set then end date should be set
|
||||||
|
if data.Schedule.End == nil {
|
||||||
|
return workflows_execution, errors.New("a cron task should have an end date.")
|
||||||
|
}
|
||||||
|
cronStr := strings.Split(data.Schedule.Cron, " ") // split the cron string to treat it
|
||||||
|
if len(cronStr) < 6 { // if the cron string is less than 6 fields, return an error because format is : ss mm hh dd MM dw (6 fields)
|
||||||
|
return nil, errors.New("Bad cron message: " + data.Schedule.Cron + ". Should be at least ss mm hh dd MM dw")
|
||||||
|
}
|
||||||
|
subCron := strings.Join(cronStr[:6], " ")
|
||||||
|
// cron should be parsed as ss mm hh dd MM dw t (min 6 fields)
|
||||||
|
specParser := cron.NewParser(cron.Second | cron.Minute | cron.Hour | cron.Dom | cron.Month | cron.Dow) // create a new cron parser
|
||||||
|
sched, err := specParser.Parse(subCron) // parse the cron string
|
||||||
|
if err != nil {
|
||||||
|
return workflows_execution, errors.New("Bad cron message: " + err.Error())
|
||||||
|
}
|
||||||
|
// loop through the cron schedule to set the executions
|
||||||
|
for s := sched.Next(*data.Schedule.Start); !s.IsZero() && s.Before(*data.Schedule.End); s = sched.Next(s) {
|
||||||
|
obj := &workflow_execution.WorkflowExecution{
|
||||||
|
AbstractObject: utils.AbstractObject{
|
||||||
|
Name: data.Schedule.Name, // set the name of the execution
|
||||||
|
},
|
||||||
|
ExecDate: &s, // set the execution date
|
||||||
|
EndDate: data.Schedule.End, // set the end date
|
||||||
|
State: 1, // set the state to 1 (scheduled)
|
||||||
|
WorkflowID: id, // set the workflow id dependancy of the execution
|
||||||
|
}
|
||||||
|
workflows_execution = append(workflows_execution, obj) // append the execution to the array
|
||||||
|
}
|
||||||
|
|
||||||
|
} else { // if no cron, set the execution to the start date
|
||||||
|
obj := &workflow_execution.WorkflowExecution{ // create a new execution
|
||||||
|
AbstractObject: utils.AbstractObject{
|
||||||
|
Name: data.Schedule.Name,
|
||||||
|
},
|
||||||
|
ExecDate: data.Schedule.Start,
|
||||||
|
EndDate: data.Schedule.End,
|
||||||
|
State: 1,
|
||||||
|
WorkflowID: id,
|
||||||
|
}
|
||||||
|
workflows_execution = append(workflows_execution, obj) // append the execution to the array
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
return workflows_execution, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// DeleteOne deletes a workflow from the database, delete depending executions and bookings
|
// DeleteOne deletes a workflow from the database, delete depending executions and bookings
|
||||||
func (a *workflowMongoAccessor) DeleteOne(id string) (utils.DBObject, int, error) {
|
func (wfa *workflowMongoAccessor) DeleteOne(id string) (utils.DBObject, int, error) {
|
||||||
res, code, err := utils.GenericDeleteOne(id, a)
|
wfa.execution(id, &Workflow{
|
||||||
|
AbstractWorkflow: AbstractWorkflow{ScheduleActive: false},
|
||||||
|
}, true) // delete the executions
|
||||||
|
res, code, err := wfa.GenericDeleteOne(id, wfa)
|
||||||
if res != nil && code == 200 {
|
if res != nil && code == 200 {
|
||||||
a.execute(res.(*Workflow), true, false) // up to date the workspace for the workflow
|
wfa.execute(res.(*Workflow), true, false) // up to date the workspace for the workflow
|
||||||
a.share(res.(*Workflow), true, a.GetCaller())
|
wfa.share(res.(*Workflow), true, wfa.Caller)
|
||||||
}
|
}
|
||||||
return a.verifyResource(res), code, err
|
return res, code, err
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* book is a function that books a workflow on the peers
|
||||||
|
* it takes the workflow id, the real data and the executions
|
||||||
|
* it returns an error if the booking fails
|
||||||
|
*/
|
||||||
|
func (wfa *workflowMongoAccessor) book(id string, realData *Workflow, execs []*workflow_execution.WorkflowExecution) error {
|
||||||
|
if wfa.Caller == nil || wfa.Caller.URLS == nil || wfa.Caller.URLS[tools.BOOKING] == nil {
|
||||||
|
return errors.New("no caller defined")
|
||||||
|
}
|
||||||
|
methods := wfa.Caller.URLS[tools.BOOKING]
|
||||||
|
if _, ok := methods[tools.POST]; !ok {
|
||||||
|
return errors.New("no path found")
|
||||||
|
}
|
||||||
|
res, code, _ := wfa.LoadOne(id)
|
||||||
|
if code != 200 {
|
||||||
|
return errors.New("could not load workflow")
|
||||||
|
}
|
||||||
|
r := res.(*Workflow)
|
||||||
|
g := r.Graph
|
||||||
|
if realData.Graph != nil { // if the graph is set, set it to the real data
|
||||||
|
g = realData.Graph
|
||||||
|
}
|
||||||
|
if g != nil && g.Links != nil && len(g.Links) > 0 { // if the graph is set and has links then book the workflow (even on ourselves)
|
||||||
|
accessor := (&compute.ComputeResource{}).GetAccessor(nil)
|
||||||
|
isDCFound := []string{}
|
||||||
|
for _, link := range g.Links {
|
||||||
|
if ok, dc_id := realData.isDCLink(link); ok { // check if the link is a link between a compute and a resource booking is only on compute
|
||||||
|
if slices.Contains(isDCFound, dc_id) {
|
||||||
|
continue
|
||||||
|
} // if the compute is already found, skip it
|
||||||
|
isDCFound = append(isDCFound, dc_id)
|
||||||
|
dc, code, _ := accessor.LoadOne(dc_id)
|
||||||
|
if code != 200 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
// CHECK BOOKING
|
||||||
|
peerID := dc.(*compute.ComputeResource).PeerID
|
||||||
|
if peerID == "" { // no peer id no booking
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
// BOOKING ON PEER
|
||||||
|
_, err := (&peer.Peer{}).LaunchPeerExecution(peerID, "", tools.BOOKING, tools.POST,
|
||||||
|
(&workflow_execution.WorkflowExecutions{ // it's the standard model for booking see OC-PEER
|
||||||
|
WorkflowID: id, // set the workflow id "WHO"
|
||||||
|
ResourceID: dc_id, // set the compute id "WHERE"
|
||||||
|
Executions: execs, // set the executions to book "WHAT"
|
||||||
|
}).Serialize(), wfa.Caller)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("BOOKING", err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* share is a function that shares a workflow to the peers if the workflow is shared
|
* share is a function that shares a workflow to the peers if the workflow is shared
|
||||||
*/
|
*/
|
||||||
func (a *workflowMongoAccessor) share(realData *Workflow, delete bool, caller *tools.HTTPCaller) {
|
func (wfa *workflowMongoAccessor) share(realData *Workflow, delete bool, caller *tools.HTTPCaller) {
|
||||||
if realData == nil || realData.Shared == nil || len(realData.Shared) == 0 || caller == nil || caller.Disabled { // no shared no sharing
|
if realData == nil || realData.Shared == nil || len(realData.Shared) == 0 || caller == nil || caller.Disabled { // no shared no sharing
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
for _, sharedID := range realData.Shared { // loop through the shared ids
|
for _, sharedID := range realData.Shared { // loop through the shared ids
|
||||||
res, code, _ := a.collaborativeAreaAccessor.LoadOne(sharedID)
|
access := (&shallow_collaborative_area.ShallowCollaborativeArea{}).GetAccessor(nil)
|
||||||
|
res, code, _ := access.LoadOne(sharedID)
|
||||||
if code != 200 {
|
if code != 200 {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
@ -71,87 +181,141 @@ func (a *workflowMongoAccessor) share(realData *Workflow, delete bool, caller *t
|
|||||||
if ok, _ := paccess.IsMySelf(); ok { // if the peer is the current peer, never share because it will create a loop
|
if ok, _ := paccess.IsMySelf(); ok { // if the peer is the current peer, never share because it will create a loop
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
if delete { // if the workflow is deleted, share the deletion orderResourceAccessor utils.Accessor
|
if delete { // if the workflow is deleted, share the deletion
|
||||||
|
|
||||||
history := NewHistory()
|
history := NewHistory()
|
||||||
history.StoreOne(history.MapFromWorkflow(res.(*Workflow)))
|
history.StoreOne(history.MapFromWorkflow(res.(*Workflow)))
|
||||||
_, err = paccess.LaunchPeerExecution(p, res.GetID(), tools.WORKFLOW, tools.DELETE,
|
_, err = paccess.LaunchPeerExecution(p, res.GetID(), tools.WORKFLOW, tools.DELETE, map[string]interface{}{}, caller)
|
||||||
map[string]interface{}{}, caller)
|
|
||||||
} else { // if the workflow is updated, share the update
|
} else { // if the workflow is updated, share the update
|
||||||
_, err = paccess.LaunchPeerExecution(p, res.GetID(), tools.WORKFLOW, tools.PUT,
|
_, err = paccess.LaunchPeerExecution(p, res.GetID(), tools.WORKFLOW, tools.PUT, res.Serialize(), caller)
|
||||||
res.Serialize(res), caller)
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if err != nil {
|
if err != nil {
|
||||||
a.Logger.Error().Msg(err.Error())
|
wfa.Logger.Error().Msg(err.Error())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// UpdateOne updates a workflow in the database
|
/*
|
||||||
func (a *workflowMongoAccessor) UpdateOne(set utils.DBObject, id string) (utils.DBObject, int, error) {
|
* execution is a create or delete function for the workflow executions depending on the schedule of the workflow
|
||||||
// avoid the update if the schedule is the same
|
*/
|
||||||
set = a.verifyResource(set)
|
func (wfa *workflowMongoAccessor) execution(id string, realData *Workflow, delete bool) (int, error) {
|
||||||
if set.(*Workflow).Graph != nil && set.(*Workflow).Graph.Partial {
|
nats := tools.NewNATSCaller() // create a new nats caller because executions are sent to the nats for daemons
|
||||||
return nil, 403, errors.New("you are not allowed to update a partial workflow")
|
mongo.MONGOService.DeleteMultiple(map[string]interface{}{
|
||||||
|
"state": 1, // only delete the scheduled executions only scheduled if executions are in progress or ended, they should not be deleted for registration
|
||||||
|
"workflow_id": id,
|
||||||
|
}, tools.WORKFLOW_EXECUTION.String())
|
||||||
|
err := wfa.book(id, realData, []*workflow_execution.WorkflowExecution{}) // delete the booking of the workflow on the peers
|
||||||
|
fmt.Println("DELETE BOOKING", err)
|
||||||
|
nats.SetNATSPub(tools.WORKFLOW.String(), tools.REMOVE, realData) // send the deletion to the nats
|
||||||
|
if err != nil {
|
||||||
|
return 409, err
|
||||||
}
|
}
|
||||||
res, code, err := utils.GenericUpdateOne(set, id, a, &Workflow{})
|
|
||||||
|
accessor := (&workflow_execution.WorkflowExecution{}).GetAccessor(nil)
|
||||||
|
execs, err := wfa.getExecutions(id, realData) // get the executions of the workflow
|
||||||
|
if err != nil {
|
||||||
|
return 422, err
|
||||||
|
}
|
||||||
|
if !realData.ScheduleActive || delete { // if the schedule is not active, delete the executions
|
||||||
|
execs = []*workflow_execution.WorkflowExecution{}
|
||||||
|
}
|
||||||
|
err = wfa.book(id, realData, execs) // book the workflow on the peers
|
||||||
|
fmt.Println("BOOKING", err)
|
||||||
|
if err != nil {
|
||||||
|
return 409, err // if the booking fails, return an error for integrity between peers
|
||||||
|
}
|
||||||
|
fmt.Println("BOOKING", delete)
|
||||||
|
for _, obj := range execs {
|
||||||
|
_, code, err := accessor.StoreOne(obj)
|
||||||
|
fmt.Println("EXEC", code, err)
|
||||||
|
if code != 200 {
|
||||||
|
return code, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
nats.SetNATSPub(tools.WORKFLOW.String(), tools.CREATE, realData) // send the creation to the nats
|
||||||
|
return 200, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// UpdateOne updates a workflow in the database
|
||||||
|
func (wfa *workflowMongoAccessor) UpdateOne(set utils.DBObject, id string) (utils.DBObject, int, error) {
|
||||||
|
res, code, err := wfa.LoadOne(id)
|
||||||
|
if code != 200 {
|
||||||
|
return nil, 409, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// avoid the update if the schedule is the same
|
||||||
|
avoid := set.(*Workflow).Schedule == nil || (res.(*Workflow).Schedule != nil && res.(*Workflow).ScheduleActive == set.(*Workflow).ScheduleActive && res.(*Workflow).Schedule.Start == set.(*Workflow).Schedule.Start && res.(*Workflow).Schedule.End == set.(*Workflow).Schedule.End && res.(*Workflow).Schedule.Cron == set.(*Workflow).Schedule.Cron)
|
||||||
|
res, code, err = wfa.GenericUpdateOne(set, id, wfa, &Workflow{})
|
||||||
if code != 200 {
|
if code != 200 {
|
||||||
return nil, code, err
|
return nil, code, err
|
||||||
}
|
}
|
||||||
workflow := res.(*Workflow)
|
workflow := res.(*Workflow)
|
||||||
a.execute(workflow, false, true) // update the workspace for the workflow
|
if !avoid { // if the schedule is not avoided, update the executions
|
||||||
a.share(workflow, false, a.GetCaller()) // share the update to the peers
|
if code, err := wfa.execution(id, workflow, false); code != 200 {
|
||||||
|
return nil, code, errors.New("could not update the executions : " + err.Error())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
fmt.Println("UPDATE", workflow.ScheduleActive, workflow.Schedule)
|
||||||
|
if workflow.ScheduleActive && workflow.Schedule != nil { // if the workflow is scheduled, update the executions
|
||||||
|
now := time.Now().UTC()
|
||||||
|
if (workflow.Schedule.End != nil && now.After(*workflow.Schedule.End)) || (workflow.Schedule.End == nil && workflow.Schedule.Start != nil && now.After(*workflow.Schedule.Start)) { // if the start date is passed, then you can book
|
||||||
|
workflow.ScheduleActive = false
|
||||||
|
wfa.GenericRawUpdateOne(workflow, id, wfa)
|
||||||
|
} // if the start date is passed, update the executions
|
||||||
|
}
|
||||||
|
wfa.execute(workflow, false, false) // update the workspace for the workflow
|
||||||
|
wfa.share(workflow, false, wfa.Caller) // share the update to the peers
|
||||||
return res, code, nil
|
return res, code, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// StoreOne stores a workflow in the database
|
// StoreOne stores a workflow in the database
|
||||||
func (a *workflowMongoAccessor) StoreOne(data utils.DBObject) (utils.DBObject, int, error) {
|
func (wfa *workflowMongoAccessor) StoreOne(data utils.DBObject) (utils.DBObject, int, error) {
|
||||||
data = a.verifyResource(data)
|
|
||||||
d := data.(*Workflow)
|
d := data.(*Workflow)
|
||||||
if d.Graph != nil && d.Graph.Partial {
|
if d.ScheduleActive && d.Schedule != nil { // if the workflow is scheduled, update the executions
|
||||||
return nil, 403, errors.New("you are not allowed to update a partial workflow")
|
now := time.Now().UTC()
|
||||||
|
if (d.Schedule.End != nil && now.After(*d.Schedule.End)) || (d.Schedule.End == nil && d.Schedule.Start != nil && now.After(*d.Schedule.Start)) { // if the start date is passed, then you can book
|
||||||
|
d.ScheduleActive = false
|
||||||
|
} // if the start date is passed, update the executions
|
||||||
}
|
}
|
||||||
res, code, err := utils.GenericStoreOne(d, a)
|
res, code, err := wfa.GenericStoreOne(d, wfa)
|
||||||
if err != nil || code != 200 {
|
if err != nil || code != 200 {
|
||||||
return nil, code, err
|
return nil, code, err
|
||||||
}
|
}
|
||||||
workflow := res.(*Workflow)
|
workflow := res.(*Workflow)
|
||||||
|
|
||||||
a.share(workflow, false, a.GetCaller()) // share the creation to the peers
|
wfa.share(workflow, false, wfa.Caller) // share the creation to the peers
|
||||||
a.execute(workflow, false, true) // store the workspace for the workflow
|
//store the executions
|
||||||
|
if code, err := wfa.execution(res.GetID(), workflow, false); err != nil {
|
||||||
|
return nil, code, err
|
||||||
|
}
|
||||||
|
wfa.execute(workflow, false, false) // store the workspace for the workflow
|
||||||
return res, code, nil
|
return res, code, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// CopyOne copies a workflow in the database
|
// CopyOne copies a workflow in the database
|
||||||
func (a *workflowMongoAccessor) CopyOne(data utils.DBObject) (utils.DBObject, int, error) {
|
func (wfa *workflowMongoAccessor) CopyOne(data utils.DBObject) (utils.DBObject, int, error) {
|
||||||
wf := data.(*Workflow)
|
return wfa.GenericStoreOne(data, wfa)
|
||||||
for _, item := range wf.Graph.Items {
|
|
||||||
_, obj := item.GetResource()
|
|
||||||
if obj != nil {
|
|
||||||
obj.ClearEnv()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return utils.GenericStoreOne(data, a)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// execute is a function that executes a workflow
|
// execute is a function that executes a workflow
|
||||||
// it stores the workflow resources in a specific workspace to never have a conflict in UI and logic
|
// it stores the workflow resources in a specific workspace to never have a conflict in UI and logic
|
||||||
func (a *workflowMongoAccessor) execute(workflow *Workflow, delete bool, active bool) {
|
func (wfa *workflowMongoAccessor) execute(workflow *Workflow, delete bool, active bool) {
|
||||||
|
|
||||||
|
accessor := (&workspace.Workspace{}).GetAccessor(nil)
|
||||||
filters := &dbs.Filters{
|
filters := &dbs.Filters{
|
||||||
Or: map[string][]dbs.Filter{ // filter by standard workspace name attached to a workflow
|
Or: map[string][]dbs.Filter{ // filter by standard workspace name attached to a workflow
|
||||||
"abstractobject.name": {{Operator: dbs.LIKE.String(), Value: workflow.Name + "_workspace"}},
|
"abstractobject.name": {{dbs.LIKE.String(), workflow.Name + "_workspace"}},
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
resource, _, err := a.workspaceAccessor.Search(filters, "", workflow.IsDraft)
|
resource, _, err := accessor.Search(filters, "")
|
||||||
if delete { // if delete is set to true, delete the workspace
|
if delete { // if delete is set to true, delete the workspace
|
||||||
for _, r := range resource {
|
for _, r := range resource {
|
||||||
a.workspaceAccessor.DeleteOne(r.GetID())
|
accessor.DeleteOne(r.GetID())
|
||||||
}
|
}
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
if err == nil && len(resource) > 0 { // if the workspace already exists, update it
|
if err == nil && len(resource) > 0 { // if the workspace already exists, update it
|
||||||
a.workspaceAccessor.UpdateOne(&workspace.Workspace{
|
accessor.UpdateOne(&workspace.Workspace{
|
||||||
Active: active,
|
Active: active,
|
||||||
ResourceSet: resources.ResourceSet{
|
ResourceSet: resources.ResourceSet{
|
||||||
Datas: workflow.Datas,
|
Datas: workflow.Datas,
|
||||||
@ -162,7 +326,7 @@ func (a *workflowMongoAccessor) execute(workflow *Workflow, delete bool, active
|
|||||||
},
|
},
|
||||||
}, resource[0].GetID())
|
}, resource[0].GetID())
|
||||||
} else { // if the workspace does not exist, create it
|
} else { // if the workspace does not exist, create it
|
||||||
a.workspaceAccessor.StoreOne(&workspace.Workspace{
|
accessor.StoreOne(&workspace.Workspace{
|
||||||
Active: active,
|
Active: active,
|
||||||
AbstractObject: utils.AbstractObject{Name: workflow.Name + "_workspace"},
|
AbstractObject: utils.AbstractObject{Name: workflow.Name + "_workspace"},
|
||||||
ResourceSet: resources.ResourceSet{
|
ResourceSet: resources.ResourceSet{
|
||||||
@ -176,49 +340,65 @@ func (a *workflowMongoAccessor) execute(workflow *Workflow, delete bool, active
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a *workflowMongoAccessor) LoadOne(id string) (utils.DBObject, int, error) {
|
// LoadOne loads a workflow from the database
|
||||||
return utils.GenericLoadOne[*Workflow](id, func(d utils.DBObject) (utils.DBObject, int, error) {
|
func (wfa *workflowMongoAccessor) LoadOne(id string) (utils.DBObject, int, error) {
|
||||||
w := d.(*Workflow)
|
var workflow Workflow
|
||||||
a.execute(w, false, true) // if no workspace is attached to the workflow, create it
|
res_mongo, code, err := mongo.MONGOService.LoadOne(id, wfa.GetType())
|
||||||
return d, 200, nil
|
if err != nil {
|
||||||
}, a)
|
wfa.Logger.Error().Msg("Could not retrieve " + id + " from db. Error: " + err.Error())
|
||||||
}
|
return nil, code, err
|
||||||
|
|
||||||
func (a *workflowMongoAccessor) LoadAll(isDraft bool) ([]utils.ShallowDBObject, int, error) {
|
|
||||||
return utils.GenericLoadAll[*Workflow](func(d utils.DBObject) utils.ShallowDBObject { return &d.(*Workflow).AbstractObject }, isDraft, a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *workflowMongoAccessor) Search(filters *dbs.Filters, search string, isDraft bool) ([]utils.ShallowDBObject, int, error) {
|
|
||||||
return utils.GenericSearch[*Workflow](filters, search, (&Workflow{}).GetObjectFilters(search), func(d utils.DBObject) utils.ShallowDBObject { return a.verifyResource(d) }, isDraft, a)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *workflowMongoAccessor) verifyResource(obj utils.DBObject) utils.DBObject {
|
|
||||||
wf := obj.(*Workflow)
|
|
||||||
if wf.Graph == nil {
|
|
||||||
return wf
|
|
||||||
}
|
}
|
||||||
for _, item := range wf.Graph.Items {
|
res_mongo.Decode(&workflow)
|
||||||
t, resource := item.GetResource()
|
if workflow.ScheduleActive && workflow.Schedule != nil { // if the workflow is scheduled, update the executions
|
||||||
if resource == nil {
|
now := time.Now().UTC()
|
||||||
continue
|
if (workflow.Schedule.End != nil && now.After(*workflow.Schedule.End)) || (workflow.Schedule.End == nil && workflow.Schedule.Start != nil && now.After(*workflow.Schedule.Start)) { // if the start date is passed, then you can book
|
||||||
}
|
workflow.ScheduleActive = false
|
||||||
var access utils.Accessor
|
wfa.GenericRawUpdateOne(&workflow, id, wfa)
|
||||||
if t == tools.COMPUTE_RESOURCE {
|
|
||||||
access = resources.NewAccessor[*resources.ComputeResource](t, a.GetRequest(), func() utils.DBObject { return &resources.ComputeResource{} })
|
} // if the start date is passed, update the executions
|
||||||
} else if t == tools.PROCESSING_RESOURCE {
|
}
|
||||||
access = resources.NewAccessor[*resources.ProcessingResource](t, a.GetRequest(), func() utils.DBObject { return &resources.ProcessingResource{} })
|
wfa.execute(&workflow, false, true) // if no workspace is attached to the workflow, create it
|
||||||
} else if t == tools.STORAGE_RESOURCE {
|
return &workflow, 200, nil
|
||||||
access = resources.NewAccessor[*resources.StorageResource](t, a.GetRequest(), func() utils.DBObject { return &resources.StorageResource{} })
|
}
|
||||||
} else if t == tools.WORKFLOW_RESOURCE {
|
|
||||||
access = resources.NewAccessor[*resources.WorkflowResource](t, a.GetRequest(), func() utils.DBObject { return &resources.WorkflowResource{} })
|
// LoadAll loads all the workflows from the database
|
||||||
} else if t == tools.DATA_RESOURCE {
|
func (wfa workflowMongoAccessor) LoadAll() ([]utils.ShallowDBObject, int, error) {
|
||||||
access = resources.NewAccessor[*resources.DataResource](t, a.GetRequest(), func() utils.DBObject { return &resources.DataResource{} })
|
objs := []utils.ShallowDBObject{}
|
||||||
} else {
|
res_mongo, code, err := mongo.MONGOService.LoadAll(wfa.GetType())
|
||||||
wf.Graph.Clear(resource.GetID())
|
if err != nil {
|
||||||
}
|
wfa.Logger.Error().Msg("Could not retrieve any from db. Error: " + err.Error())
|
||||||
if error := utils.VerifyAccess(access, resource.GetID()); error != nil {
|
return nil, code, err
|
||||||
wf.Graph.Clear(resource.GetID())
|
}
|
||||||
|
var results []Workflow
|
||||||
|
if err = res_mongo.All(mongo.MngoCtx, &results); err != nil {
|
||||||
|
return nil, 404, err
|
||||||
|
}
|
||||||
|
for _, r := range results {
|
||||||
|
objs = append(objs, &r.AbstractObject) // only AbstractObject fields !
|
||||||
|
}
|
||||||
|
return objs, 200, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (wfa *workflowMongoAccessor) 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}},
|
||||||
|
},
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return wf
|
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 []Workflow
|
||||||
|
if err = res_mongo.All(mongo.MngoCtx, &results); err != nil {
|
||||||
|
return nil, 404, err
|
||||||
|
}
|
||||||
|
for _, r := range results {
|
||||||
|
objs = append(objs, &r)
|
||||||
|
}
|
||||||
|
return objs, 200, nil
|
||||||
}
|
}
|
||||||
|
23
models/workflow/workflow_schedule.go
Normal file
23
models/workflow/workflow_schedule.go
Normal file
@ -0,0 +1,23 @@
|
|||||||
|
package workflow
|
||||||
|
|
||||||
|
import "time"
|
||||||
|
|
||||||
|
// WorkflowSchedule is a struct that contains the scheduling information of a workflow
|
||||||
|
type ScheduleMode int
|
||||||
|
|
||||||
|
const (
|
||||||
|
TASK ScheduleMode = iota
|
||||||
|
SERVICE
|
||||||
|
)
|
||||||
|
|
||||||
|
/*
|
||||||
|
* WorkflowSchedule is a struct that contains the scheduling information of a workflow
|
||||||
|
* It contains the mode of the schedule (Task or Service), the name of the schedule, the start and end time of the schedule and the cron expression
|
||||||
|
*/
|
||||||
|
type WorkflowSchedule struct {
|
||||||
|
Mode int64 `json:"mode" bson:"mode" validate:"required"` // Mode is the mode of the schedule (Task or Service)
|
||||||
|
Name string `json:"name" bson:"name" validate:"required"` // Name is the name of the schedule
|
||||||
|
Start *time.Time `json:"start" bson:"start" validate:"required,ltfield=End"` // Start is the start time of the schedule, is required and must be less than the End time
|
||||||
|
End *time.Time `json:"end,omitempty" bson:"end,omitempty"` // End is the end time of the schedule
|
||||||
|
Cron string `json:"cron,omitempty" bson:"cron,omitempty"` // here the cron format : ss mm hh dd MM dw task
|
||||||
|
}
|
@ -12,7 +12,7 @@ func TestStoreOneWorkflow(t *testing.T) {
|
|||||||
AbstractObject: utils.AbstractObject{Name: "testWorkflow"},
|
AbstractObject: utils.AbstractObject{Name: "testWorkflow"},
|
||||||
}
|
}
|
||||||
|
|
||||||
wma := NewAccessor(nil)
|
wma := New()
|
||||||
id, _, _ := wma.StoreOne(&w)
|
id, _, _ := wma.StoreOne(&w)
|
||||||
|
|
||||||
assert.NotEmpty(t, id)
|
assert.NotEmpty(t, id)
|
||||||
@ -23,7 +23,7 @@ func TestLoadOneWorkflow(t *testing.T) {
|
|||||||
AbstractObject: utils.AbstractObject{Name: "testWorkflow"},
|
AbstractObject: utils.AbstractObject{Name: "testWorkflow"},
|
||||||
}
|
}
|
||||||
|
|
||||||
wma := NewAccessor(nil)
|
wma := New()
|
||||||
new_w, _, _ := wma.StoreOne(&w)
|
new_w, _, _ := wma.StoreOne(&w)
|
||||||
assert.Equal(t, w, new_w)
|
assert.Equal(t, w, new_w)
|
||||||
}
|
}
|
||||||
|
@ -1,149 +0,0 @@
|
|||||||
package workflow_execution_test
|
|
||||||
|
|
||||||
import (
|
|
||||||
"testing"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/enum"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/utils"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/workflow"
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/workflow_execution"
|
|
||||||
|
|
||||||
"github.com/google/uuid"
|
|
||||||
"github.com/stretchr/testify/assert"
|
|
||||||
"github.com/stretchr/testify/mock"
|
|
||||||
)
|
|
||||||
|
|
||||||
type MockAccessor struct {
|
|
||||||
mock.Mock
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *MockAccessor) LoadOne(id string) (interface{}, int, error) {
|
|
||||||
args := m.Called(id)
|
|
||||||
return args.Get(0), args.Int(1), args.Error(2)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestNewScheduler_ValidInput(t *testing.T) {
|
|
||||||
s := "2025-06-16T15:00:00"
|
|
||||||
e := "2025-06-16T17:00:00"
|
|
||||||
dur := 7200.0
|
|
||||||
cronStr := "0 0 * * * *"
|
|
||||||
|
|
||||||
sched := workflow_execution.NewScheduler(s, e, dur, cronStr)
|
|
||||||
|
|
||||||
assert.NotNil(t, sched)
|
|
||||||
assert.Equal(t, dur, sched.DurationS)
|
|
||||||
assert.Equal(t, cronStr, sched.Cron)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestNewScheduler_InvalidStart(t *testing.T) {
|
|
||||||
s := "invalid"
|
|
||||||
e := "2025-06-16T17:00:00"
|
|
||||||
dur := 7200.0
|
|
||||||
cronStr := "0 0 * * * *"
|
|
||||||
|
|
||||||
sched := workflow_execution.NewScheduler(s, e, dur, cronStr)
|
|
||||||
assert.Nil(t, sched)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestNewScheduler_InvalidEnd(t *testing.T) {
|
|
||||||
s := "2025-06-16T15:00:00"
|
|
||||||
e := "invalid"
|
|
||||||
dur := 7200.0
|
|
||||||
cronStr := "0 0 * * * *"
|
|
||||||
|
|
||||||
sched := workflow_execution.NewScheduler(s, e, dur, cronStr)
|
|
||||||
assert.NotNil(t, sched)
|
|
||||||
assert.Nil(t, sched.End)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestGetDates_NoCron(t *testing.T) {
|
|
||||||
start := time.Now()
|
|
||||||
end := start.Add(2 * time.Hour)
|
|
||||||
|
|
||||||
s := &workflow_execution.WorkflowSchedule{
|
|
||||||
Start: start,
|
|
||||||
End: &end,
|
|
||||||
}
|
|
||||||
|
|
||||||
schedule, err := s.GetDates()
|
|
||||||
|
|
||||||
assert.NoError(t, err)
|
|
||||||
assert.Len(t, schedule, 1)
|
|
||||||
assert.Equal(t, start, schedule[0].Start)
|
|
||||||
assert.Equal(t, end, *schedule[0].End)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestGetDates_InvalidCron(t *testing.T) {
|
|
||||||
start := time.Now()
|
|
||||||
end := start.Add(2 * time.Hour)
|
|
||||||
|
|
||||||
s := &workflow_execution.WorkflowSchedule{
|
|
||||||
Start: start,
|
|
||||||
End: &end,
|
|
||||||
Cron: "bad cron",
|
|
||||||
}
|
|
||||||
|
|
||||||
_, err := s.GetDates()
|
|
||||||
assert.Error(t, err)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestGetDates_ValidCron(t *testing.T) {
|
|
||||||
start := time.Now()
|
|
||||||
end := start.Add(10 * time.Minute)
|
|
||||||
|
|
||||||
s := &workflow_execution.WorkflowSchedule{
|
|
||||||
Start: start,
|
|
||||||
End: &end,
|
|
||||||
DurationS: 60,
|
|
||||||
Cron: "0 */2 * * * *",
|
|
||||||
}
|
|
||||||
|
|
||||||
dates, err := s.GetDates()
|
|
||||||
assert.NoError(t, err)
|
|
||||||
assert.Greater(t, len(dates), 0)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestGetExecutions_Success(t *testing.T) {
|
|
||||||
start := time.Now()
|
|
||||||
end := start.Add(1 * time.Hour)
|
|
||||||
ws := &workflow_execution.WorkflowSchedule{
|
|
||||||
UUID: uuid.New().String(),
|
|
||||||
Start: start,
|
|
||||||
End: &end,
|
|
||||||
}
|
|
||||||
|
|
||||||
wf := &workflow.Workflow{
|
|
||||||
AbstractObject: utils.AbstractObject{
|
|
||||||
UUID: uuid.New().String(),
|
|
||||||
Name: "TestWorkflow",
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
execs, err := ws.GetExecutions(wf)
|
|
||||||
assert.NoError(t, err)
|
|
||||||
assert.Greater(t, len(execs), 0)
|
|
||||||
assert.Equal(t, wf.UUID, execs[0].WorkflowID)
|
|
||||||
assert.Equal(t, ws.UUID, execs[0].ExecutionsID)
|
|
||||||
assert.Equal(t, enum.DRAFT, execs[0].State)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestSchedules_NoRequest(t *testing.T) {
|
|
||||||
ws := &workflow_execution.WorkflowSchedule{}
|
|
||||||
|
|
||||||
ws, wf, execs, err := ws.Schedules("someID", nil)
|
|
||||||
assert.Error(t, err)
|
|
||||||
assert.Nil(t, wf)
|
|
||||||
assert.Len(t, execs, 0)
|
|
||||||
assert.Equal(t, ws, ws)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Additional test stubs to be completed with gomock usage for:
|
|
||||||
// - CheckBooking
|
|
||||||
// - BookExecs
|
|
||||||
// - getBooking
|
|
||||||
// - Schedules (success path)
|
|
||||||
// - Planify mocking in CheckBooking
|
|
||||||
// - Peer interaction in BookExecs
|
|
||||||
// - Caller deep copy errors in getCallerCopy
|
|
||||||
// Will be continued...
|
|
@ -1,154 +0,0 @@
|
|||||||
package workflow_execution_test
|
|
||||||
|
|
||||||
import (
|
|
||||||
"testing"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"cloud.o-forge.io/core/oc-lib/models/common/enum"
|
|
||||||
"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"
|
|
||||||
"github.com/stretchr/testify/assert"
|
|
||||||
)
|
|
||||||
|
|
||||||
func (m *MockAccessor) DeleteOne(id string) (utils.DBObject, int, error) {
|
|
||||||
args := m.Called(id)
|
|
||||||
return nil, args.Int(1), args.Error(2)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *MockAccessor) Search(filters interface{}, search string, isDraft bool) ([]utils.ShallowDBObject, int, error) {
|
|
||||||
args := m.Called(filters, search, isDraft)
|
|
||||||
return args.Get(0).([]utils.ShallowDBObject), args.Int(1), args.Error(2)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestStoreDraftDefault(t *testing.T) {
|
|
||||||
exec := &workflow_execution.WorkflowExecution{}
|
|
||||||
exec.StoreDraftDefault()
|
|
||||||
assert.False(t, exec.IsDraft)
|
|
||||||
assert.Equal(t, enum.SCHEDULED, exec.State)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestCanUpdate_StateChange(t *testing.T) {
|
|
||||||
existing := &workflow_execution.WorkflowExecution{State: enum.DRAFT}
|
|
||||||
newExec := &workflow_execution.WorkflowExecution{State: enum.SCHEDULED}
|
|
||||||
canUpdate, updated := existing.CanUpdate(newExec)
|
|
||||||
assert.True(t, canUpdate)
|
|
||||||
assert.Equal(t, enum.SCHEDULED, updated.(*workflow_execution.WorkflowExecution).State)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestCanUpdate_SameState_Draft(t *testing.T) {
|
|
||||||
existing := &workflow_execution.WorkflowExecution{AbstractObject: utils.AbstractObject{IsDraft: true}, State: enum.DRAFT}
|
|
||||||
newExec := &workflow_execution.WorkflowExecution{AbstractObject: utils.AbstractObject{IsDraft: true}, State: enum.DRAFT}
|
|
||||||
canUpdate, _ := existing.CanUpdate(newExec)
|
|
||||||
assert.False(t, canUpdate)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestCanDelete_TrueIfDraft(t *testing.T) {
|
|
||||||
exec := &workflow_execution.WorkflowExecution{AbstractObject: utils.AbstractObject{IsDraft: true}}
|
|
||||||
assert.True(t, exec.CanDelete())
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestCanDelete_FalseIfNotDraft(t *testing.T) {
|
|
||||||
exec := &workflow_execution.WorkflowExecution{AbstractObject: utils.AbstractObject{IsDraft: false}}
|
|
||||||
assert.False(t, exec.CanDelete())
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestEquals_True(t *testing.T) {
|
|
||||||
d := time.Now()
|
|
||||||
exec1 := &workflow_execution.WorkflowExecution{ExecDate: d, WorkflowID: "123"}
|
|
||||||
exec2 := &workflow_execution.WorkflowExecution{ExecDate: d, WorkflowID: "123"}
|
|
||||||
assert.True(t, exec1.Equals(exec2))
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestEquals_False(t *testing.T) {
|
|
||||||
exec1 := &workflow_execution.WorkflowExecution{ExecDate: time.Now(), WorkflowID: "abc"}
|
|
||||||
exec2 := &workflow_execution.WorkflowExecution{ExecDate: time.Now().Add(time.Hour), WorkflowID: "def"}
|
|
||||||
assert.False(t, exec1.Equals(exec2))
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestArgoStatusToState_Success(t *testing.T) {
|
|
||||||
exec := &workflow_execution.WorkflowExecution{}
|
|
||||||
exec.ArgoStatusToState("succeeded")
|
|
||||||
assert.Equal(t, enum.SUCCESS, exec.State)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestArgoStatusToState_DefaultToFailure(t *testing.T) {
|
|
||||||
exec := &workflow_execution.WorkflowExecution{}
|
|
||||||
exec.ArgoStatusToState("unknown")
|
|
||||||
assert.Equal(t, enum.FAILURE, exec.State)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestGenerateID_AssignsUUID(t *testing.T) {
|
|
||||||
exec := &workflow_execution.WorkflowExecution{}
|
|
||||||
exec.GenerateID()
|
|
||||||
assert.NotEmpty(t, exec.UUID)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestGetName_ReturnsCorrectFormat(t *testing.T) {
|
|
||||||
time := time.Now()
|
|
||||||
exec := &workflow_execution.WorkflowExecution{AbstractObject: utils.AbstractObject{UUID: "abc"}, ExecDate: time}
|
|
||||||
assert.Contains(t, exec.GetName(), "abc")
|
|
||||||
assert.Contains(t, exec.GetName(), time.String())
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestVerifyAuth_AlwaysTrue(t *testing.T) {
|
|
||||||
exec := &workflow_execution.WorkflowExecution{}
|
|
||||||
assert.True(t, exec.VerifyAuth(nil))
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestUpdateOne_RejectsZeroState(t *testing.T) {
|
|
||||||
accessor := &workflow_execution.WorkflowExecutionMongoAccessor{}
|
|
||||||
set := &workflow_execution.WorkflowExecution{State: 0}
|
|
||||||
_, code, err := accessor.UpdateOne(set, "someID")
|
|
||||||
assert.Equal(t, 400, code)
|
|
||||||
assert.Error(t, err)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoadOne_DraftExpired_ShouldDelete(t *testing.T) {
|
|
||||||
// Normally would mock time.Now and delete call; for now we test structure
|
|
||||||
accessor := workflow_execution.NewAccessor(&tools.APIRequest{})
|
|
||||||
exec := &workflow_execution.WorkflowExecution{
|
|
||||||
ExecDate: time.Now().Add(-2 * time.Minute),
|
|
||||||
State: enum.DRAFT,
|
|
||||||
AbstractObject: utils.AbstractObject{UUID: "to-delete"},
|
|
||||||
}
|
|
||||||
_, _, _ = accessor.LoadOne(exec.GetID())
|
|
||||||
// No panic = good enough placeholder
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoadOne_ScheduledExpired_ShouldUpdateToForgotten(t *testing.T) {
|
|
||||||
accessor := workflow_execution.NewAccessor(&tools.APIRequest{})
|
|
||||||
exec := &workflow_execution.WorkflowExecution{
|
|
||||||
ExecDate: time.Now().Add(-2 * time.Minute),
|
|
||||||
State: enum.SCHEDULED,
|
|
||||||
AbstractObject: utils.AbstractObject{UUID: "to-forget"},
|
|
||||||
}
|
|
||||||
_, _, _ = accessor.LoadOne(exec.GetID())
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestDeleteOne_NotImplemented(t *testing.T) {
|
|
||||||
accessor := workflow_execution.NewAccessor(&tools.APIRequest{})
|
|
||||||
_, code, err := accessor.DeleteOne("someID")
|
|
||||||
assert.Equal(t, 404, code)
|
|
||||||
assert.Error(t, err)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestStoreOne_NotImplemented(t *testing.T) {
|
|
||||||
accessor := workflow_execution.NewAccessor(&tools.APIRequest{})
|
|
||||||
_, code, err := accessor.StoreOne(nil)
|
|
||||||
assert.Equal(t, 404, code)
|
|
||||||
assert.Error(t, err)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestCopyOne_NotImplemented(t *testing.T) {
|
|
||||||
accessor := workflow_execution.NewAccessor(&tools.APIRequest{})
|
|
||||||
_, code, err := accessor.CopyOne(nil)
|
|
||||||
assert.Equal(t, 404, code)
|
|
||||||
assert.Error(t, err)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestGetExecFilters_BasicPattern(t *testing.T) {
|
|
||||||
a := workflow_execution.NewAccessor(&tools.APIRequest{})
|
|
||||||
filters := a.GetExecFilters("foo")
|
|
||||||
assert.Contains(t, filters.Or["abstractobject.name"][0].Value, "foo")
|
|
||||||
}
|
|
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in New Issue
Block a user