add Exists schedule method

opsway-io · Feb 3, 2023 · 328b2d9 · 328b2d9
1 parent 85d75e4
commit 328b2d9
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/schedule.go b/schedule.go
@@ -25,6 +25,7 @@ var (
 type Schedule interface {
 	Add(ctx context.Context, task *Task, interval time.Duration, firstExecution time.Time) error
 	Remove(ctx context.Context, kind string, id string) error
+	Exists(ctx context.Context, kind string, id string) (bool, error)
 	RunNow(ctx context.Context, kind string, id string) error
 	On(ctx context.Context, kind string, handler func(ctx context.Context, task *Task)) error
 }
@@ -152,6 +153,22 @@ func (s *ScheduleImpl) Remove(ctx context.Context, kind string, id string) error
 	}
 }
 
+func (s *ScheduleImpl) Exists(ctx context.Context, kind string, id string) (bool, error) {
+	scheduleKey := s.taskScheduleKey(kind)
+
+	exists, err := s.redisClient.ZScore(ctx, scheduleKey, id).Result()
+
+	if err == redis.Nil {
+		return false, nil
+	}
+
+	if err != nil {
+		return false, err
+	}
+
+	return exists != 0, nil
+}
+
 func (s *ScheduleImpl) RunNow(ctx context.Context, kind string, id string) error {
 	script := redis.NewScript(`
 		local queueKey = KEYS[1]