add new

idprm · Oct 19, 2024 · 814c6a8 · 814c6a8
1 parent 6a8a908
commit 814c6a8
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 5 deletions.
diff --git a/cmd/publisher.go b/cmd/publisher.go
@@ -389,11 +389,11 @@ var publisherScrapingPredictionCmd = &cobra.Command{
 			scheduleRepo := repository.NewScheduleRepository(db)
 			scheduleService := services.NewScheduleService(scheduleRepo)
 
-			if scheduleService.IsUnlocked(ACT_SCRAPING, timeNow) {
+			if scheduleService.IsUnlocked(ACT_PREDICTION, timeNow) {
 
 				scheduleService.UpdateLocked(
 					&entity.Schedule{
-						Name: ACT_SCRAPING,
+						Name: ACT_PREDICTION,
 					},
 				)
 
@@ -402,10 +402,10 @@ var publisherScrapingPredictionCmd = &cobra.Command{
 				}()
 			}
 
-			if scheduleService.IsUnlocked(ACT_SCRAPING, timeNow) {
+			if scheduleService.IsUnlocked(ACT_PREDICTION, timeNow) {
 				scheduleService.Update(
 					&entity.Schedule{
-						Name:       ACT_SCRAPING,
+						Name:       ACT_PREDICTION,
 						IsUnlocked: true,
 					},
 				)

diff --git a/cmd/root.go b/cmd/root.go
@@ -78,6 +78,7 @@ const (
 	ACT_SUB                  string = "SUB"
 	ACT_UNSUB                string = "UNSUB"
 	ACT_USER_LOSES           string = "USER_LOSES"
+	ACT_PREDICTION           string = "PREDICTION"
 	ACT_SCRAPING             string = "SCRAPING"
 )
 

diff --git a/internal/domain/repository/schedule_repository.go b/internal/domain/repository/schedule_repository.go
@@ -79,7 +79,7 @@ func (r *ScheduleRepository) Update(c *entity.Schedule) error {
 }
 
 func (r *ScheduleRepository) UpdateLocked(c *entity.Schedule) error {
-	err := r.db.Where("name = ?", c.Name).Update("is_unlocked", false).Error
+	err := r.db.Model(c).Where("name = ?", c.Name).Update("is_unlocked", false).Error
 	if err != nil {
 		return err
 	}