diff --git a/cmd/main.go b/cmd/main.go index a886200..d2f2a4d 100644 --- a/cmd/main.go +++ b/cmd/main.go @@ -51,6 +51,7 @@ func initArticlesRoutes(r *gin.Engine) { articlesRouter.GET("/", internal.GetAllArticlesHandler) articlesRouter.GET("/:id", internal.GetArticleHandler) articlesRouter.POST("/:id", internal.AddArticleHandler) + articlesRouter.PATCH("/:id", internal.EditArticleHandler) articlesRouter.DELETE("/:id", internal.DeleteArticleHandler) } diff --git a/internal/articlesHandlers.go b/internal/articlesHandlers.go index 7e5da79..7d34805 100644 --- a/internal/articlesHandlers.go +++ b/internal/articlesHandlers.go @@ -70,3 +70,23 @@ func DeleteArticleHandler(c *gin.Context) { SendOk(c, fmt.Sprintf("%d articles were successfully deleted!", deleteCount)) } + +func EditArticleHandler(c *gin.Context) { + id := c.Params.ByName("id") + + var articleUpdate DocumentUpdate + articleUpdate.Filter = gin.H{"id_name": id} + c.BindJSON(&articleUpdate.Update) + + editCount, err := editDocument(articlesLocation, articleUpdate) + if err != nil { + SendBadRequest(c, err.Error()) + return + } + + if editCount != 0 { + SendOk(c, fmt.Sprintf("%d articles were successfully edited!", editCount)) + } else { + SendOk(c, "No articles were edited.") + } +} diff --git a/internal/database.go b/internal/database.go index da6e592..b190406 100644 --- a/internal/database.go +++ b/internal/database.go @@ -4,6 +4,7 @@ import ( "context" "errors" + "github.com/gin-gonic/gin" "go.mongodb.org/mongo-driver/mongo" "go.mongodb.org/mongo-driver/mongo/options" ) @@ -13,6 +14,11 @@ type Location struct { Collection string } +type DocumentUpdate struct { + Filter gin.H `json:"filter"` + Update gin.H `json:"update"` +} + func getNewClient() *mongo.Client { client, err := mongo.Connect( context.TODO(), @@ -44,7 +50,7 @@ func getDocuments(location Location, filter interface{}) ([][]byte, error) { cursor, err := collection.Find(context.TODO(), filter) if err != nil { - return results, errors.New("something is wrong with filter to find the document") + return results, errors.New("something is wrong with filter to find the document.") } for cursor.TryNext(context.TODO()) { results = append(results, cursor.Current) @@ -71,8 +77,24 @@ func deleteDocument(location Location, filter interface{}) (int64, error) { result, err := collection.DeleteOne(context.TODO(), filter) if err != nil { - return 0, errors.New("something is wrong with filter to delete the document") + return 0, errors.New("something is wrong with filter to delete the document.") } return result.DeletedCount, nil } + +func editDocument(location Location, jsons DocumentUpdate) (int64, error) { + client := getNewClient() + collection := client.Database(location.Database).Collection(location.Collection) + defer client.Disconnect(context.TODO()) + + result, err := collection.UpdateOne(context.TODO(), jsons.Filter, gin.H{"$set": jsons.Update}) + if err != nil { + return 0, err + } + if result.MatchedCount == 0 { + return 0, errors.New("cannot find document matching filter.") + } + + return result.ModifiedCount, nil +}