organize + graph

models/resources/data/data.go

@@ -0,0 +1,29 @@
+package data
+
+import (
+	resources "oc-lib/models/resources"
+	"oc-lib/models/utils"
+)
+
+type Data struct {
+	resources.AbstractResource
+	Protocols []string `json:"protocol,omitempty" bson:"protocol,omitempty"` //TODO Enum type
+	DataType  string   `json:"datatype" required:"true" bson:"datatype"`
+	Example   string   `json:"example" bson:"example" required:"true" validate:"required" description:"base64 encoded data"`
+}
+
+func (d *Data) GetType() resources.ResourceType {
+	return resources.DATA
+}
+
+func (d *Data) GetAccessor(driver utils.Driver) utils.Accessor {
+	var data utils.Accessor
+	switch driver {
+	case utils.MONGO:
+		data = &DataMongoAccessor{}
+	default:
+		data = &DataMongoAccessor{}
+	}
+	data.SetLogger()
+	return data
+}

models/resources/data/data_mongo_accessor.go

@@ -0,0 +1,39 @@
+package data
+
+import (
+	mongo "oc-lib/dbs/mongo"
+	"oc-lib/models/utils"
+)
+
+type DataMongoAccessor struct {
+	utils.AbstractAccessor
+}
+
+func (dma *DataMongoAccessor) DeleteOne(id string) utils.DBObject {
+	return dma.GenericDeleteOne(id, dma)
+}
+
+func (dma *DataMongoAccessor) UpdateOne(set map[string]interface{}, id string) utils.DBObject {
+	return dma.GenericUpdateOne(set, id, dma)
+}
+
+func (dma *DataMongoAccessor) StoreOne(data utils.DBObject) utils.DBObject {
+	id, err := mongo.StoreOne(data.(*Data), "data")
+	if err != nil {
+		dma.Logger.Error().Msg("Could not store " + data.GetName() + " to db. Error: " + err.Error())
+		return &Data{}
+	}
+	return dma.LoadOne(id)
+}
+
+func (dma *DataMongoAccessor) LoadOne(id string) utils.DBObject {
+	var data Data
+	res_mongo, err := mongo.LoadOne(id, "data")
+	if err != nil {
+		dma.Logger.Error().Msg("Could not retrieve " + id + " from db. Error: " + err.Error())
+		return &Data{}
+	}
+	res_mongo.Decode(&data)
+
+	return &data
+}

models/resources/data/data_test.go

@@ -0,0 +1,45 @@
+package data
+
+import (
+	resources "oc-lib/models/resources"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestStoreOneData(t *testing.T) {
+	d := Data{DataType: "jpeg", Example: "123456",
+		AbstractResource: resources.AbstractResource{
+			Uuid:        "123",
+			Name:        "testData",
+			Description: "Lorem Ipsum",
+			Logo:        "azerty.com",
+			Owner:       "toto",
+			OwnerLogo:   "totoLogo",
+			SourceUrl:   "azerty.fr",
+		},
+	}
+
+	dma := DataMongoAccessor{}
+	id := dma.StoreOne(&d)
+
+	assert.NotEmpty(t, id)
+}
+
+func TestLoadOneDate(t *testing.T) {
+	d := Data{DataType: "jpeg", Example: "123456",
+		AbstractResource: resources.AbstractResource{
+			Uuid:        "123",
+			Name:        "testData",
+			Description: "Lorem Ipsum",
+			Logo:        "azerty.com",
+			Owner:       "toto",
+			OwnerLogo:   "totoLogo",
+			SourceUrl:   "azerty.fr",
+		},
+	}
+
+	dma := DataMongoAccessor{}
+	new_d := dma.StoreOne(&d)
+	assert.Equal(t, d, new_d)
+}