58 lines
1.8 KiB
Go
58 lines
1.8 KiB
Go
package oclib
|
|
|
|
import (
|
|
"encoding/json"
|
|
|
|
"cloud.o-forge.io/core/oc-lib/models/resources"
|
|
"cloud.o-forge.io/core/oc-lib/models/resources/data"
|
|
"cloud.o-forge.io/core/oc-lib/models/resources/datacenter"
|
|
"cloud.o-forge.io/core/oc-lib/models/resources/processing"
|
|
"cloud.o-forge.io/core/oc-lib/models/resources/storage"
|
|
"cloud.o-forge.io/core/oc-lib/models/utils"
|
|
"cloud.o-forge.io/core/oc-lib/models/workflow/graph"
|
|
tool "cloud.o-forge.io/core/oc-lib/utils"
|
|
)
|
|
|
|
const WORKFLOW = "workflow"
|
|
|
|
type Workflow struct {
|
|
resources.AbstractResource
|
|
Graph graph.Graph `bson:"graph,omitempty" json:"graph,omitempty"`
|
|
Datas map[string]data.Data `bson:"datas,omitempty" json:"datas,omitempty"`
|
|
Storages map[string]storage.Storage `bson:"storages,omitempty" json:"storages,omitempty"`
|
|
Processing map[string]processing.Processing `bson:"processing,omitempty" json:"processing,omitempty"`
|
|
Datacenters map[string]datacenter.Datacenter `bson:"datacenters,omitempty" json:"datacenters,omitempty"`
|
|
Schedule WorkflowSchedule `bson:"schedule,omitempty" json:"schedule,omitempty"`
|
|
}
|
|
|
|
func (d *Workflow) GetAccessor(driver tool.Driver) utils.Accessor {
|
|
var data utils.Accessor
|
|
switch driver {
|
|
case tool.MONGO:
|
|
data = &WorkflowMongoAccessor{}
|
|
default:
|
|
data = &WorkflowMongoAccessor{}
|
|
}
|
|
data.SetLogger()
|
|
return data
|
|
}
|
|
|
|
func (dma *Workflow) Deserialize(j map[string]interface{}) utils.DBObject {
|
|
b, err := json.Marshal(j)
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
json.Unmarshal(b, dma)
|
|
return dma
|
|
}
|
|
|
|
func (w *Workflow) isDCLink(link graph.GraphLink) bool {
|
|
if _, exists := w.Datacenters[link.Destination.ID]; exists {
|
|
return true
|
|
} else if _, exists := w.Datacenters[link.Source.ID]; exists {
|
|
return true
|
|
}
|
|
|
|
return false
|
|
}
|