organize + graph

models/resources/datacenter/datacenter.go

@@ -0,0 +1,52 @@
+package datacenter
+
+import (
+	resources "oc-lib/models/resources"
+	"oc-lib/models/utils"
+)
+
+type Datacenter struct {
+	resources.AbstractResource
+	Owner        string `bson:"owner" json:"owner" required:"true"`
+	BookingPrice int    `bson:"booking_price" json:"booking_price" required:"true"`
+
+	CPU DatacenterCpuModel    `bson:"cpu,omitempty" json:"cpu,omitempty"`
+	RAM DatacenterMemoryModel `bson:"ram,omitempty" json:"ram,omitempty"`
+	GPU []DatacenterGpuModel  `bson:"gpu,omitempty" json:"gpu,omitempty"`
+}
+
+type DatacenterCpuModel struct {
+	Cores         uint   `bson:"cores,omitempty" json:"cores,omitempty"`               //TODO: validate
+	Architecture  string `bson:"architecture,omitempty" json:"architecture,omitempty"` //TOOD: enum
+	Shared        bool   `bson:"shared,omitempty" json:"shared,omitempty"`
+	MinimumMemory uint   `bson:"minimum_memory,omitempty" json:"minimum_memory,omitempty"`
+	Platform      string `bson:"platform,omitempty" json:"platform,omitempty"`
+}
+
+type DatacenterMemoryModel struct {
+	Size uint `bson:"size,omitempty" json:"size,omitempty" description:"Units in MB"`
+	Ecc  bool `bson:"ecc,omitempty" json:"ecc,omitempty"`
+}
+
+type DatacenterGpuModel struct {
+	CudaCores   uint   `bson:"cuda_cores,omitempty" json:"cuda_cores,omitempty"`
+	Model       string `bson:"model,omitempty" json:"model,omitempty"`
+	Memory      uint   `bson:"memory,omitempty" json:"memory,omitempty" description:"Units in MB"`
+	TensorCores uint   `bson:"tensor_cores,omitempty" json:"tensor_cores,omitempty"`
+}
+
+func (d *Datacenter) GetType() resources.ResourceType {
+	return resources.DATACENTER
+}
+
+func (d *Datacenter) GetAccessor(driver utils.Driver) utils.Accessor {
+	var data utils.Accessor
+	switch driver {
+	case utils.MONGO:
+		data = &DatacenterMongoAccessor{}
+	default:
+		data = &DatacenterMongoAccessor{}
+	}
+	data.SetLogger()
+	return data
+}

models/resources/datacenter/datacenter_mongo_accessor.go

@@ -0,0 +1,43 @@
+package datacenter
+
+import (
+	"oc-lib/dbs/mongo"
+	logs "oc-lib/logs"
+	"oc-lib/models/utils"
+)
+
+type DatacenterMongoAccessor struct {
+	utils.AbstractAccessor
+}
+
+func (dma *DatacenterMongoAccessor) DeleteOne(id string) utils.DBObject {
+	return dma.GenericDeleteOne(id, dma)
+}
+
+func (dma *DatacenterMongoAccessor) UpdateOne(set map[string]interface{}, id string) utils.DBObject {
+	return dma.GenericUpdateOne(set, id, dma)
+}
+
+func (dma *DatacenterMongoAccessor) StoreOne(data utils.DBObject) utils.DBObject {
+	id, err := mongo.StoreOne(data.(*Datacenter), "data")
+	if err != nil {
+		dma.Logger.Error().Msg("Could not store " + data.GetName() + " to db. Error: " + err.Error())
+		return &Datacenter{}
+	}
+	return dma.LoadOne(id)
+}
+
+func (dca *DatacenterMongoAccessor) LoadOne(id string) utils.DBObject {
+	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
+}

models/resources/datacenter/datacenter_test.go

@@ -0,0 +1,46 @@
+package datacenter
+
+import (
+	resources "oc-lib/models/resources"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestStoreOneDatacenter(t *testing.T) {
+	dc := Datacenter{Owner: "toto", BookingPrice: 123,
+		AbstractResource: resources.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: resources.AbstractResource{
+			Uuid:        "123",
+			Name:        "testDatacenter",
+			Description: "Lorem Ipsum",
+			Logo:        "azerty.com",
+			Owner:       "toto",
+			OwnerLogo:   "totoLogo",
+			SourceUrl:   "azerty.fr",
+		},
+	}
+
+	dcma := DatacenterMongoAccessor{}
+	new_dc := dcma.StoreOne(&dc)
+
+	assert.Equal(t, dc, new_dc)
+}