From 3f9814e6497e908df6b890d964fc3b1ebc434fe8 Mon Sep 17 00:00:00 2001
From: mr <morgane.roques@irt-saintexupery.com>
Date: Fri, 19 Jul 2024 13:27:34 +0200
Subject: [PATCH] test *

---
 entrypoint.go                             | 4 ++++
 models/resources/data/data.go             | 2 ++
 models/resources/datacenter/datacenter.go | 2 ++
 models/resources/processing/processing.go | 2 ++
 models/resources/storage/storage.go       | 2 ++
 models/resources/workflow/workflow.go     | 2 ++
 models/utils/abstracts.go                 | 4 ++++
 models/utils/interfaces.go                | 2 ++
 models/workflow/workflow.go               | 2 ++
 9 files changed, 22 insertions(+)

diff --git a/entrypoint.go b/entrypoint.go
index 94e9de0..dfc3c08 100644
--- a/entrypoint.go
+++ b/entrypoint.go
@@ -30,6 +30,10 @@ type LibData struct {
 	Err          string         `bson:"error" json:"error"`
 }
 
+func (lb LibData) BindID(id string) {
+	lb.DataResource.BindID(id)
+}
+
 func Init(appName string) {
 	logs.SetAppName(appName)
 	logs.SetLogger(logs.CreateLogger("main", ""))
diff --git a/models/resources/data/data.go b/models/resources/data/data.go
index 9ff6dc4..e941560 100644
--- a/models/resources/data/data.go
+++ b/models/resources/data/data.go
@@ -14,6 +14,8 @@ type DataResource struct {
 	Example   string   `json:"example" bson:"example" required:"true" validate:"required" description:"base64 encoded data"`
 }
 
+func (d *DataResource) BindID(id string) {}
+
 func (dma *DataResource) Deserialize(j map[string]interface{}) utils.DBObject {
 	b, err := json.Marshal(j)
 	if err != nil {
diff --git a/models/resources/datacenter/datacenter.go b/models/resources/datacenter/datacenter.go
index 3c0ccbb..243260f 100644
--- a/models/resources/datacenter/datacenter.go
+++ b/models/resources/datacenter/datacenter.go
@@ -37,6 +37,8 @@ type DatacenterGpuModel struct {
 	TensorCores uint   `bson:"tensor_cores,omitempty" json:"tensor_cores,omitempty"`
 }
 
+func (d *DatacenterResource) BindID(id string) {}
+
 func (dma *DatacenterResource) Deserialize(j map[string]interface{}) utils.DBObject {
 	b, err := json.Marshal(j)
 	if err != nil {
diff --git a/models/resources/processing/processing.go b/models/resources/processing/processing.go
index e08aaf5..8032fa3 100644
--- a/models/resources/processing/processing.go
+++ b/models/resources/processing/processing.go
@@ -31,6 +31,8 @@ type ExecutionRequirementsModel struct {
 	DiskIO       string `bson:"disk_io,omitempty" json:"disk_io,omitempty"`
 }
 
+func (d *ProcessingResource) BindID(id string) {}
+
 func (dma *ProcessingResource) Deserialize(j map[string]interface{}) utils.DBObject {
 	b, err := json.Marshal(j)
 	if err != nil {
diff --git a/models/resources/storage/storage.go b/models/resources/storage/storage.go
index f6e7b90..db9d3a7 100644
--- a/models/resources/storage/storage.go
+++ b/models/resources/storage/storage.go
@@ -24,6 +24,8 @@ type StorageResource struct {
 	BookingPrice uint   `bson:"booking_price,omitempty" json:"booking_price,omitempty"`
 }
 
+func (d *StorageResource) BindID(id string) {}
+
 func (dma *StorageResource) Deserialize(j map[string]interface{}) utils.DBObject {
 	b, err := json.Marshal(j)
 	if err != nil {
diff --git a/models/resources/workflow/workflow.go b/models/resources/workflow/workflow.go
index d1fe583..6eaa3a9 100644
--- a/models/resources/workflow/workflow.go
+++ b/models/resources/workflow/workflow.go
@@ -57,6 +57,8 @@ type WorkflowResource struct {
 	WorkflowID string `bson:"workflow_id,omitempty" json:"workflow_id,omitempty"`
 }
 
+func (d *WorkflowResource) BindID(id string) { d.WorkflowID = id }
+
 func (d *WorkflowResource) GetAccessor() utils.Accessor {
 	data := &WorkflowResourceMongoAccessor{}
 	data.SetLogger(utils.WORKFLOW_RESOURCE)
diff --git a/models/utils/abstracts.go b/models/utils/abstracts.go
index f6b2f63..ecdf2f7 100644
--- a/models/utils/abstracts.go
+++ b/models/utils/abstracts.go
@@ -17,6 +17,10 @@ type AbstractObject struct {
 	Name string `json:"name" required:"true" bson:"name" validate:"required"`
 }
 
+func (ao *AbstractObject) GetID() string {
+	return ao.UUID
+}
+
 func (dma *AbstractObject) ObjDeserialize(j map[string]interface{}) *AbstractObject {
 	b, err := json.Marshal(j)
 	if err != nil {
diff --git a/models/utils/interfaces.go b/models/utils/interfaces.go
index f26a73d..74bcc2c 100644
--- a/models/utils/interfaces.go
+++ b/models/utils/interfaces.go
@@ -2,6 +2,8 @@ package utils
 
 type DBObject interface {
 	GenerateID()
+	GetID() string
+	BindID(id string)
 	GetName() string
 	Deserialize(j map[string]interface{}) DBObject
 	Serialize() map[string]interface{}
diff --git a/models/workflow/workflow.go b/models/workflow/workflow.go
index a3da99f..9d85cb8 100644
--- a/models/workflow/workflow.go
+++ b/models/workflow/workflow.go
@@ -13,6 +13,8 @@ type Workflow struct {
 	w.AbstractWorkflow
 }
 
+func (d *Workflow) BindID(id string) {}
+
 func (d *Workflow) GetName() string {
 	return d.Name
 }