packaged every model

datacenter/datacenter.go

@@ -0,0 +1,39 @@
+package datacenter
+
+import oclib "oc-lib"
+
+type Datacenter struct {
+	oclib.AbstractResource `json:"resource" required:"true"`
+
+	Owner        string `json:"owner" required:"true"`
+	BookingPrice int    `json:"booking_price" required:"true"`
+
+	CPU DatacenterCpuModel    `json:"cpu,omitempty"`
+	RAM DatacenterMemoryModel `json:"ram,omitempty"`
+	GPU []DatacenterGpuModel  `json:"gpu,omitempty"`
+}
+
+type DatacenterCpuModel struct {
+	Cores         uint   `json:"cores,omitempty"`        //TODO: validate
+	Architecture  string `json:"architecture,omitempty"` //TOOD: enum
+	Shared        bool   `json:"shared,omitempty"`
+	MinimumMemory uint   `json:"minimum_memory,omitempty"`
+	Platform      string `json:"platform,omitempty"`
+}
+
+type DatacenterMemoryModel struct {
+	Size uint `json:"size,omitempty" description:"Units in MB"`
+	Ecc  bool `json:"ecc,omitempty"`
+}
+
+type DatacenterGpuModel struct {
+	CudaCores   uint   `json:"cuda_cores,omitempty"`
+	Model       string `json:"model,omitempty"`
+	Memory      uint   `json:"memory,omitempty" description:"Units in MB"`
+	TensorCores uint   `json:"tensor_cores,omitempty"`
+}
+
+
+func (d *Datacenter) GetType() oclib.ResourceType{
+	return oclib.DATACENTER
+}

datacenter/datacenter_mongo_accessor.go

@@ -0,0 +1,33 @@
+package datacenter
+
+import (
+	logs "oc-lib/logs"
+	"oc-lib/mongo"
+)
+
+type DatacenterMongoAccessor struct{}
+
+func (dca *DatacenterMongoAccessor) StoreOne(datacenter Datacenter) string {
+    id, err := mongo.StoreOne(datacenter, "datacenter")
+    if err != nil {
+        l := logs.CreateLogger("oclib", "")
+        l.Error().Msg("Could not store " + datacenter.Name + " to db. Error: " + err.Error())
+        return ""
+    }
+    return id
+}
+
+func (dca *DatacenterMongoAccessor) LoadOne(id string) Datacenter {
+    var datacenter Datacenter
+
+    res_mongo, err := mongo.LoadOne(id, "datacenter")
+    if err != nil {
+        l := logs.CreateLogger("oclib", "")
+        l.Error().Msg("Could not retrieve " + id + " from db. Error: " + err.Error())
+        return Datacenter{}
+    }
+
+    res_mongo.Decode(&datacenter)
+
+    return datacenter
+}

datacenter/datacenter_test.go

@@ -0,0 +1,47 @@
+package datacenter
+
+import (
+	oclib "oc-lib"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestStoreOneDatacenter(t *testing.T){
+	dc := Datacenter{Owner: "toto", BookingPrice: 123, 
+				AbstractResource: oclib.AbstractResource{
+					Uuid: "123",
+					Name: "testDatacenter",
+					Description: "Lorem Ipsum",
+					Logo : "azerty.com",
+					Owner: "toto",
+					OwnerLogo: "totoLogo",
+					SourceUrl: "azerty.fr",
+				},
+			}
+
+	dcma := DatacenterMongoAccessor{}
+	id := dcma.StoreOne(dc)
+
+	assert.NotEmpty(t, id)
+}
+
+func TestLoadOneDatacenter(t *testing.T){
+	dc := Datacenter{Owner: "toto", BookingPrice: 123, 
+				AbstractResource: oclib.AbstractResource{
+					Uuid: "123",
+					Name: "testDatacenter",
+					Description: "Lorem Ipsum",
+					Logo : "azerty.com",
+					Owner: "toto",
+					OwnerLogo: "totoLogo",
+					SourceUrl: "azerty.fr",
+				},
+			}
+	
+	dcma := DatacenterMongoAccessor{}
+	id := dcma.StoreOne(dc)
+	new_dc := dcma.LoadOne(id)
+
+	assert.Equal(t,dc, new_dc)
+}