accessor

entrypoint.go

@@ -6,32 +6,38 @@ import (
 	"cloud.o-forge.io/core/oc-lib/models/resources"
 )
 
+type LibData int
+
 const (
-	INVALID    = resources.INVALID
-	DATA       = resources.DATA
-	PROCESSING = resources.PROCESSING
-	STORAGE    = resources.STORAGE
-	DATACENTER = resources.DATACENTER
-	WORKFLOW   = resources.WORKFLOW
+	INVALID    LibData = iota
+	DATA               = resources.DATA
+	PROCESSING         = resources.PROCESSING
+	STORAGE            = resources.STORAGE
+	DATACENTER         = resources.DATACENTER
+	WORKFLOW           = resources.WORKFLOW
 )
 
+func (d LibData) EnumIndex() int {
+	return int(d)
+}
+
 func Init() {
 	mongo.MONGOService.Init(models.GetModelsNames(), GetConfig())
 }
 
-func LoadOne(name string, id string) interface{} {
-	return models.Model(name).GetAccessor().LoadOne(id)
+func LoadOne(col LibData, id string) interface{} {
+	return models.Model(col.EnumIndex()).GetAccessor().LoadOne(id)
 }
 
-func UpdateOne(name string, set map[string]interface{}, id string) interface{} {
-	return models.Model(name).GetAccessor().UpdateOne(set, id)
+func UpdateOne(col LibData, set map[string]interface{}, id string) interface{} {
+	return models.Model(col.EnumIndex()).GetAccessor().UpdateOne(set, id)
 }
 
-func DeleteOne(name string, id string) interface{} {
-	return models.Model(name).GetAccessor().DeleteOne(id)
+func DeleteOne(col LibData, id string) interface{} {
+	return models.Model(col.EnumIndex()).GetAccessor().DeleteOne(id)
 }
 
-func StoreOne(name string, object map[string]interface{}) interface{} {
-	model := models.Model(name)
+func StoreOne(col LibData, object map[string]interface{}) interface{} {
+	model := models.Model(col.EnumIndex())
 	return model.GetAccessor().StoreOne(model.Deserialize(object))
 }