Added edition workflow

This commit is contained in:
Florian Sylvain
2022-12-27 18:45:21 +01:00
parent e6e15392d1
commit bb324d3df4
3 changed files with 45 additions and 2 deletions
+20
View File
@@ -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.")
}
}
+24 -2
View File
@@ -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
}