heat estimation change request supports multiple events

controller/heat_controller.go

@@ -27,7 +27,7 @@ func heatController() {
 
 	router.POST("/heat", addHeat)
 	router.POST("/heat/import", importHeat)
-	router.POST("/heat/meet/:meet_id/event/:event_id/start_estimation_date", updateHeatsStartEstimationDate)
+	router.POST("/heat/meet/:meet_id/events/start_estimation_date", updateHeatsStartEstimationDate)
 	router.POST("/heat/:id/time", updateHeatTime)
 
 	router.PUT("/heat", updateHeat)
@@ -315,19 +315,13 @@ func updateHeatsStartEstimationDate(c *gin.Context) {
 		return
 	}
 
-	event, err1 := strconv.Atoi(c.Param("event_id"))
-	if err1 != nil {
-		c.IndentedJSON(http.StatusBadRequest, gin.H{"message": "given event_id is not of type number"})
-		return
-	}
-
 	var request dto.HeatEstimationDateRequest
 	if err := c.BindJSON(&request); err != nil {
 		c.IndentedJSON(http.StatusInternalServerError, gin.H{"message": err.Error()})
 		return
 	}
 
-	info, err := service.UpdateHeatsEstimationDateByMeetingAndEvent(meeting, event, request.Time, request.UpdateTimeZone)
+	info, err := service.UpdateHeatsEstimationDateByMeetingAndEvent(meeting, request.Events, request.Time, request.UpdateTimeZone)
 	if err != nil {
 		c.IndentedJSON(http.StatusNotFound, gin.H{"message": err.Error()})
 		return

dto/heat_estimation_date_request.go

@@ -4,5 +4,6 @@ import "time"
 
 type HeatEstimationDateRequest struct {
 	Time           time.Time `json:"time"`
+	Events         []int     `json:"events"`
 	UpdateTimeZone bool      `json:"update_time_zone"`
 }

service/heat_service.go

@@ -152,6 +152,16 @@ func GetHeatsByMeetingAndEvent(id string, event int) ([]model.Heat, error) {
 	return getHeatsByBsonDocument(bson.D{{"meeting", id}, {"event", event}})
 }
 
+func GetHeatsByMeetingAndEvents(id string, events []int) ([]model.Heat, error) {
+	return getHeatsByBsonDocument(
+		bson.M{
+			"$and": []interface{}{
+				bson.M{"meeting": id},
+				bson.M{"event": bson.M{"$in": events}},
+			},
+		})
+}
+
 func GetHeatById(id primitive.ObjectID) (model.Heat, error) {
 	return getHeatByBsonDocument(bson.D{{"_id", id}})
 }
@@ -450,8 +460,18 @@ func UpdateHeatTimes(id primitive.ObjectID, time time.Time, timeType string) (mo
 	return UpdateHeat(heat)
 }
 
-func UpdateHeatsEstimationDateByMeetingAndEvent(meeting string, event int, t time.Time, updateTimeZone bool) ([]model.Heat, error) {
-	heats, err := GetHeatsByMeetingAndEvent(meeting, event)
+func UpdateHeatsEstimationDateByMeetingAndEvent(meeting string, events []int, t time.Time, updateTimeZone bool) ([]model.Heat, error) {
+	var heats []model.Heat
+	var err error
+	if len(events) <= 0 {
+		heats, err = GetHeatsByMeeting(meeting)
+		println("changing heat date and timezone for ALL events")
+	} else {
+		heats, err = GetHeatsByMeetingAndEvents(meeting, events)
+		println("changing heat date and timezone for events:")
+		println(events)
+	}
+
 	if err != nil {
 		return []model.Heat{}, err
 	}