59 lines
2.1 KiB
Go
59 lines
2.1 KiB
Go
package processing
|
|
|
|
import (
|
|
"encoding/json"
|
|
|
|
resources "cloud.o-forge.io/core/oc-lib/models/resources"
|
|
"cloud.o-forge.io/core/oc-lib/models/utils"
|
|
tool "cloud.o-forge.io/core/oc-lib/utils"
|
|
)
|
|
|
|
type Processing struct {
|
|
resources.AbstractResource
|
|
Container string `bson:"container,omitempty" json:"container,omitempty"` // We could create a specific class for container, that could check if the name exists/is available
|
|
Repository string `bson:"repository,omitempty" json:"repository,omitempty"` // Indicate where to find the container image => Could add a struct handling authentication to the repo
|
|
Command string `bson:"command,omitempty" json:"command,omitempty"`
|
|
Arguments []string `bson:"arguments,omitempty" json:"arguments,omitempty"`
|
|
Environment []map[string]string `bson:"environment,omitempty" json:"environment,omitempty"` // a key/value struct is what ressembles the most a NAME=VALUE struct
|
|
|
|
ExecutionRequirements ExecutionRequirementsModel `json:"execution_requirements,omitempty"`
|
|
|
|
Price uint `bson:"price,omitempty" json:"price,omitempty"`
|
|
License string `bson:"license,omitempty" json:"license,omitempty"`
|
|
}
|
|
|
|
type ExecutionRequirementsModel struct {
|
|
CPUs uint `bson:"cp_us,omitempty" json:"cp_us,omitempty"`
|
|
GPUs uint `bson:"gp_us,omitempty" json:"gp_us,omitempty"`
|
|
RAM uint `bson:"ram,omitempty" json:"ram,omitempty"`
|
|
|
|
Parallel bool `bson:"parallel,omitempty" json:"parallel,omitempty"`
|
|
ScalingModel uint `bson:"scaling_model,omitempty" json:"scaling_model,omitempty"`
|
|
DiskIO string `bson:"disk_io,omitempty" json:"disk_io,omitempty"`
|
|
}
|
|
|
|
func (dma *Processing) Deserialize(j map[string]interface{}) utils.DBObject {
|
|
b, err := json.Marshal(j)
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
json.Unmarshal(b, dma)
|
|
return dma
|
|
}
|
|
|
|
func (p *Processing) GetType() resources.ResourceType {
|
|
return resources.PROCESSING
|
|
}
|
|
|
|
func (d *Processing) GetAccessor(driver tool.Driver) utils.Accessor {
|
|
var data utils.Accessor
|
|
switch driver {
|
|
case tool.MONGO:
|
|
data = &ProcessingMongoAccessor{}
|
|
default:
|
|
data = &ProcessingMongoAccessor{}
|
|
}
|
|
data.SetLogger()
|
|
return data
|
|
}
|