implementing download from id

schematic-service-go/api/schematicapi.go

@@ -0,0 +1,83 @@
+package api
+
+import (
+	"net/http"
+
+	"github.com/go-chi/chi"
+	"github.com/go-chi/render"
+	"github.com/willie68/schematic-service-go/dao"
+)
+
+//SchematicsRoutes getting all routes for the config endpoint
+func SchematicsRoutes() *chi.Mux {
+	router := chi.NewRouter()
+	router.Post("/", PostSchematicEndpoint)
+	router.Get("/", GetSchematicsEndpoint)
+	router.Delete("/{schematicId}", DeleteSchematicEndpoint)
+	router.Put("/{schematicId}", UpdateSchematicEndpoint)
+	router.Get("/{schematicId}", GetSchematicHandler)
+	router.Get("/{schematicId}/files/{fileId}", GetSchematicFileHandler)
+	return router
+}
+
+// GetTenantHandler gets a tenant
+func GetSchematicHandler(response http.ResponseWriter, req *http.Request) {
+	schematicID := chi.URLParam(req, "schematicId")
+	schematic, err := dao.GetSchematic(schematicID)
+	if err != nil {
+		Msg(response, http.StatusBadRequest, err.Error())
+		return
+	}
+	render.JSON(response, req, schematic)
+}
+
+// GetTenantHandler gets a tenant
+func GetSchematicFileHandler(response http.ResponseWriter, req *http.Request) {
+	//schematicID := chi.URLParam(req, "schematicId")
+	fileID := chi.URLParam(req, "fileId")
+	err := dao.GetFile(fileID, response)
+	if err != nil {
+		Msg(response, http.StatusBadRequest, err.Error())
+		return
+	}
+	//render.JSON(response, req, schematic)
+}
+
+// GetTenantsHandler gets all tenants
+func GetSchematicsEndpoint(response http.ResponseWriter, req *http.Request) {
+	/*
+		var tenants []dto.Tenant
+		err := dao.Get().All("tenants", &tenants)
+		if err != nil {
+			Msg(response, http.StatusBadRequest, err.Error())
+			return
+		}
+	*/
+	render.JSON(response, req, "tenants")
+}
+
+func PostSchematicEndpoint(response http.ResponseWriter, req *http.Request) {
+}
+
+func DeleteSchematicEndpoint(response http.ResponseWriter, req *http.Request) {
+	/*
+		schematicID := chi.URLParam(req, "schematicId")
+		if err != nil {
+			Msg(response, http.StatusBadRequest, err.Error())
+			return
+		}
+	*/
+	render.JSON(response, req, "tenant")
+}
+
+func UpdateSchematicEndpoint(response http.ResponseWriter, req *http.Request) {
+	/*
+		schematicID := chi.URLParam(req, "schematicId")
+		err := dao.Get().One("tenants", chi.URLParam(req, "schematicId"), &tenant)
+		if err != nil {
+			Msg(response, http.StatusBadRequest, err.Error())
+			return
+		}
+	*/
+	render.JSON(response, req, "tenant")
+}

schematic-service-go/cmd/service.go

@@ -81,6 +81,7 @@ func routes() *chi.Mux {
 		r.Mount(baseURL+"/config", api.ConfigRoutes())
 		r.Mount(baseURL+"/tags", api.TagsRoutes())
 		r.Mount(baseURL+"/manufacturers", api.ManufacturersRoutes())
+		r.Mount(baseURL+"/schematics", api.SchematicsRoutes())
 		r.Mount("/health", health.Routes())
 	})
 	return router

schematic-service-go/dao/mongodao.go

@@ -13,6 +13,7 @@ import (
 	"github.com/willie68/schematic-service-go/model"
 
 	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/bson/primitive"
 	"go.mongodb.org/mongo-driver/mongo"
 	"go.mongodb.org/mongo-driver/mongo/gridfs"
 	"go.mongodb.org/mongo-driver/mongo/options"
@@ -255,7 +256,7 @@ func AddFile(filename string, reader io.Reader) (string, error) {
 		return "", err
 	}
 	log.Printf("Write file to DB was successful. File id: %s \n", fileID)
-	id := fileID.String()
+	id := fileID.Hex()
 	return id, nil
 }
 
@@ -288,6 +289,35 @@ func CreateSchematic(schematic model.Schematic) (string, error) {
 	return result.InsertedID.(string), nil
 }
 
+// GetSchematic getting a sdingle schematic
+func GetSchematic(schematicID string) (model.Schematic, error) {
+	ctx, _ := context.WithTimeout(context.Background(), timeout)
+	schematicCollection := database.Collection(schematicsCollectionName)
+	result := schematicCollection.FindOne(ctx, bson.M{"_id": schematicID})
+	var schematic model.Schematic
+	if err := result.Decode(&schematic); err != nil {
+		log.Print(err)
+		return model.Schematic{}, err
+	} else {
+		return schematic, nil
+	}
+}
+
+func GetFile(fileid string, stream io.Writer) error {
+	objectID, err := primitive.ObjectIDFromHex(fileid)
+	if err != nil {
+		log.Print(err)
+		return err
+	}
+	dStream, err := bucket.DownloadToStream(objectID, stream)
+	if err != nil {
+		log.Print(err)
+		return err
+	}
+	fmt.Printf("File size to download: %v \n", dStream)
+	return nil
+}
+
 // CreateTag create a new tag in the storage
 func CreateTag(tag string) error {
 	tag = strings.ToLower(tag)