oc-lib/models/resources/workflow/workflow.go

59 lines
1.7 KiB
Go
Raw Normal View History

2024-07-18 11:51:12 +02:00
package oclib
import (
2024-07-18 14:39:54 +02:00
"encoding/json"
2024-07-23 08:59:39 +02:00
"slices"
2024-07-18 14:39:54 +02:00
2024-07-18 13:35:14 +02:00
"cloud.o-forge.io/core/oc-lib/models/resources"
2024-07-18 16:46:54 +02:00
"cloud.o-forge.io/core/oc-lib/models/resources/workflow/graph"
2024-07-18 13:35:14 +02:00
"cloud.o-forge.io/core/oc-lib/models/utils"
2024-07-18 11:51:12 +02:00
)
2024-07-19 10:54:58 +02:00
type AbstractWorkflow struct {
2024-07-23 08:59:39 +02:00
Graph *graph.Graph `bson:"graph,omitempty" json:"graph,omitempty"`
Datas []string `bson:"datas,omitempty" json:"datas,omitempty"`
Storages []string `bson:"storages,omitempty" json:"storages,omitempty"`
ProcessingResource []string `bson:"processing,omitempty" json:"processing,omitempty"`
Datacenters []string `bson:"datacenters,omitempty" json:"datacenters,omitempty"`
Workflows []string `bson:"workflows,omitempty" json:"workflows,omitempty"`
Schedule *WorkflowSchedule `bson:"schedule,omitempty" json:"schedule,omitempty"`
2024-07-19 10:54:58 +02:00
}
func (w *AbstractWorkflow) isDCLink(link graph.GraphLink) bool {
2024-07-23 08:59:39 +02:00
if slices.Contains(w.Datacenters, link.Destination.ID) || slices.Contains(w.Datacenters, link.Source.ID) {
2024-07-19 10:54:58 +02:00
return true
}
return false
}
type WorkflowResource struct {
2024-07-18 11:51:12 +02:00
resources.AbstractResource
2024-07-19 10:54:58 +02:00
AbstractWorkflow
WorkflowID string `bson:"workflow_id,omitempty" json:"workflow_id,omitempty"`
2024-07-18 11:51:12 +02:00
}
2024-07-19 10:54:58 +02:00
func (d *WorkflowResource) GetAccessor() utils.Accessor {
data := &WorkflowResourceMongoAccessor{}
data.SetLogger(utils.WORKFLOW_RESOURCE)
2024-07-18 11:51:12 +02:00
return data
}
2024-07-19 10:54:58 +02:00
func (dma *WorkflowResource) Deserialize(j map[string]interface{}) utils.DBObject {
2024-07-18 14:39:54 +02:00
b, err := json.Marshal(j)
if err != nil {
return nil
}
json.Unmarshal(b, dma)
return dma
}
2024-07-19 10:54:58 +02:00
func (dma *WorkflowResource) Serialize() map[string]interface{} {
2024-07-19 09:32:58 +02:00
var m map[string]interface{}
b, err := json.Marshal(dma)
if err != nil {
return nil
}
json.Unmarshal(b, dma)
return m
}