Update

LikDan · Nov 4, 2023 · dc47e36 · dc47e36
1 parent 46a50ef
commit dc47e36
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 4 deletions.
diff --git a/internal/schedule/controllers/controller.go b/internal/schedule/controllers/controller.go
@@ -506,8 +506,11 @@ func (s *controller) AddScheduleInfo(ctx context.Context, dto dto2.AddScheduleIn
 		StudyPlaceId: user.StudyPlaceInfo.ID,
 	}
 
-	err := s.repository.AddScheduleInfo(ctx, entry)
-	if err != nil {
+	if err := s.repository.RemoveScheduleInfo(ctx, user.StudyPlaceInfo.ID, dto.Date); err != nil {
+		return entities.ScheduleInfoEntry{}, err
+	}
+
+	if err := s.repository.AddScheduleInfo(ctx, entry); err != nil {
 		return entities.ScheduleInfoEntry{}, err
 	}
 

diff --git a/internal/schedule/repositories/repository.go b/internal/schedule/repositories/repository.go
@@ -44,6 +44,7 @@ type Repository interface {
 	FilterLessonMarks(ctx context.Context, lessonID primitive.ObjectID, marks []string) error
 
 	AddScheduleInfo(ctx context.Context, entry entities.ScheduleInfoEntry) error
+	RemoveScheduleInfo(ctx context.Context, studyPlaceID primitive.ObjectID, date time.Time) error
 }
 
 type repository struct {
@@ -175,8 +176,9 @@ return items;
 						"status":    "general",
 					},
 				},
-			}, "as": "generalLessons"},
-		},
+			},
+			"as": "generalLessons",
+		}},
 		bson.M{
 			"$project": bson.M{
 				"lessons": bson.M{"$cond": bson.M{
@@ -454,3 +456,8 @@ func (s *repository) AddScheduleInfo(ctx context.Context, entry entities.Schedul
 	_, err := s.schedule.InsertOne(ctx, entry)
 	return err
 }
+
+func (s *repository) RemoveScheduleInfo(ctx context.Context, studyPlaceID primitive.ObjectID, date time.Time) error {
+	_, err := s.schedule.DeleteMany(ctx, bson.M{"studyPlaceID": studyPlaceID, "date": date})
+	return err
+}