Include types from gonavitia/navitia

types/container_json.go

@@ -0,0 +1,57 @@
+package types
+
+import (
+	"encoding/json"
+	"sync"
+
+	"github.com/pkg/errors"
+)
+
+// UnmarshalJSON satisfies the json.Unmarshaller interface
+func (c *Container) UnmarshalJSON(b []byte) error {
+	// Set up a mutex
+	c.mu = &sync.RWMutex{}
+
+	// Unmarshal into a map
+	data := map[string]json.RawMessage{}
+	err := json.Unmarshal(b, &data)
+	if err != nil {
+		return errors.Wrap(err, "Couldn't unmarshal into a map")
+	}
+
+	// Create the error generator
+	gen := unmarshalErrorMaker{"Container", b}
+
+	// From a map, extract the ID, Name & EmbeddedType
+	if id, ok := data["id"]; ok {
+		err := json.Unmarshal(id, &c.ID)
+		if err != nil {
+			return gen.err(err, "ID", "id", id, "error while unmarshalling")
+		}
+	}
+	if name, ok := data["name"]; ok {
+		err := json.Unmarshal(name, &c.Name)
+		if err != nil {
+			return gen.err(err, "Name", "name", name, "error while unmarshalling")
+		}
+	}
+	if embeddedType, ok := data["embedded_type"]; ok {
+		err := json.Unmarshal(embeddedType, &c.EmbeddedType)
+		if err != nil {
+			return gen.err(err, "EmbeddedType", "embedded_type", embeddedType, "error while unmarshalling")
+		}
+	}
+	if quality, ok := data["quality"]; ok {
+		err := json.Unmarshal(quality, &c.Quality)
+		if err != nil {
+			return gen.err(err, "Quality", "quality", quality, "error while unmarshalling")
+		}
+	}
+
+	// Now, assign the embedded content to the Container
+	if embedded, ok := data[c.EmbeddedType]; ok {
+		c.embeddedJSON = embedded
+	}
+
+	return nil
+}