Resource Model Apparel

models/resources/datacenter/datacenter_mongo_accessor.go

@@ -2,6 +2,7 @@ package datacenter
 
 import (
 	"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"
 )
 
@@ -37,7 +38,11 @@ func (dca *DatacenterMongoAccessor) LoadOne(id string) (utils.DBObject, int, err
 	}
 
 	res_mongo.Decode(&datacenter)
-
+	accessor := (&resource_model.ResourceModel{}).GetAccessor()
+	resources, _, err := accessor.Search(dca.GetType())
+	if err == nil && len(resources) > 0 {
+		datacenter.ResourceModel = resources[0].(*resource_model.ResourceModel)
+	}
 	return &datacenter, 200, nil
 }
 
@@ -52,7 +57,12 @@ func (wfa DatacenterMongoAccessor) LoadAll() ([]utils.ShallowDBObject, int, erro
 	if err = res_mongo.All(mongo.MngoCtx, &results); err != nil {
 		return nil, 404, err
 	}
+	accessor := (&resource_model.ResourceModel{}).GetAccessor()
+	resources, _, err := accessor.Search(wfa.GetType())
 	for _, r := range results {
+		if err == nil && len(resources) > 0 {
+			r.ResourceModel = resources[0].(*resource_model.ResourceModel)
+		}
 		objs = append(objs, &r.AbstractResource)
 	}
 	return objs, 200, nil
@@ -69,7 +79,12 @@ func (wfa *DatacenterMongoAccessor) Search(word string) ([]utils.ShallowDBObject
 	if err = res_mongo.All(mongo.MngoCtx, &results); err != nil {
 		return nil, 404, err
 	}
+	accessor := (&resource_model.ResourceModel{}).GetAccessor()
+	resources, _, err := accessor.Search(wfa.GetType())
 	for _, r := range results {
+		if err == nil && len(resources) > 0 {
+			r.ResourceModel = resources[0].(*resource_model.ResourceModel)
+		}
 		objs = append(objs, &r.AbstractResource)
 	}
 	return objs, 200, nil