getUniqueDocument implementation

internal/articlesHandlers.go

@@ -19,18 +19,14 @@ func GetAllArticlesHandler(c *gin.Context) {
 
 func GetArticleHandler(c *gin.Context) {
 	articleID := c.Params.ByName("id")
-	articles, err := getDocuments(articlesLocation,
+	article, err := getUniqueDocument(articlesLocation,
 		bson.D{{Key: "id_name", Value: articleID}})
 	if err != nil {
-		SendBadRequest(c, err.Error())
-		return
-	}
-	if len(articles) == 0 {
 		SendBadRequest(c, "The ID provided doesn't match any article.")
 		return
 	}
 	var parsedArticle Article
-	bson.Unmarshal(articles[0], &parsedArticle)
+	bson.Unmarshal(article, &parsedArticle)
 	c.JSON(http.StatusOK, parsedArticle)
 }
 

internal/database.go

@@ -52,6 +52,18 @@ func getDocuments(location Location, filter interface{}) ([][]byte, error) {
 	return results, nil
 }
 
+func getUniqueDocument(location Location, filter interface{}) ([]byte, error) {
+	client := getNewClient()
+	collection := client.Database(location.Database).Collection(location.Collection)
+	defer client.Disconnect(context.TODO())
+
+	singleResult := collection.FindOne(context.TODO(), filter)
+	if singleResult.Err() != nil {
+		return nil, errors.New("no document was found, check the filter or the location.")
+	}
+	return singleResult.DecodeBytes()
+}
+
 func deleteDocument(location Location, filter interface{}) (int64, error) {
 	client := getNewClient()
 	collection := client.Database(location.Database).Collection(location.Collection)