From cb7298077d9abc6b2a48e7cc280f21ce968d0b83 Mon Sep 17 00:00:00 2001 From: Braeden Sowinski Date: Tue, 14 Jun 2022 14:10:01 -0700 Subject: [PATCH] limit priority --- controllers/updateController.go | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/controllers/updateController.go b/controllers/updateController.go index 9ac469d..b3e4d82 100644 --- a/controllers/updateController.go +++ b/controllers/updateController.go @@ -2054,10 +2054,20 @@ func UpdateContactPriority(c *fiber.Ctx) error { }) } + var priority int = data["priority"].(int) + + if priority > 10 || priority < 1 { + cancel() + return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{ + "success": false, + "message": "invalid priority", + }) + } + update_time, _ := time.Parse(time.RFC3339, time.Now().Format(time.RFC3339)) update := bson.M{ "$set": bson.M{ - "priority": data["priority"], + "priority": priority, "updated_at": update_time, }, }