accessor

models/resources/datacenter/datacenter.go

@@ -1,9 +1,10 @@
 package datacenter
 
 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 Datacenter struct {
@@ -36,18 +37,21 @@ type DatacenterGpuModel struct {
 	TensorCores uint   `bson:"tensor_cores,omitempty" json:"tensor_cores,omitempty"`
 }
 
+func (dma *Datacenter) Deserialize(j map[string]interface{}) utils.DBObject {
+	b, err := json.Marshal(j)
+	if err != nil {
+		return nil
+	}
+	json.Unmarshal(b, dma)
+	return dma
+}
+
 func (d *Datacenter) GetType() resources.ResourceType {
 	return resources.DATACENTER
 }
 
-func (d *Datacenter) GetAccessor(driver tool.Driver) utils.Accessor {
-	var data utils.Accessor
-	switch driver {
-	case tool.MONGO:
-		data = &DatacenterMongoAccessor{}
-	default:
-		data = &DatacenterMongoAccessor{}
-	}
+func (d *Datacenter) GetAccessor() utils.Accessor {
+	data := &DatacenterMongoAccessor{}
 	data.SetLogger()
 	return data
 }