fix: support cancel cronjob

Author: MuZhou233

internal/biz/bizchesed/chesed.go

@@ -73,7 +73,7 @@ func NewChesed(
 		modelbinah.DownloadEmpty,
 		nil,
 	)
-	err := cron.BySeconds("ChesedScanImage", 60, c.ScanImage, context.Background()) //nolint:mnd //TODO
+	_, err := cron.NewJobBySeconds("ChesedScanImage", 60, c.ScanImage, context.Background()) //nolint:mnd //TODO
 	if err != nil {
 		return nil, err
 	}

internal/biz/bizyesod/yesod.go

@@ -62,7 +62,12 @@ func NewYesod(
 		feedOwner:          feedOwner,
 		builtinFeedActions: builtinFeedActions,
 	}
-	err = cron.BySeconds("YesodPullFeeds", 60, y.PullFeeds, context.Background()) //nolint:mnd // hard code min interval
+	_, err = cron.NewJobBySeconds(
+		"YesodPullFeeds",
+		60, //nolint:mnd // hard code min interval
+		y.PullFeeds,
+		context.Background(),
+	)
 	if err != nil {
 		return nil, err
 	}

internal/lib/libcron/cron.go

@@ -38,15 +38,32 @@ func (c *Cron) Stop(ctx context.Context) error {
 	return c.scheduler.StopJobs()
 }
 
-func (c *Cron) BySeconds(name string, seconds int, jobFunc interface{}, params ...interface{}) error {
-	return c.Duration(name, time.Duration(seconds)*time.Second, jobFunc, params...)
+func (c *Cron) NewJobBySeconds(name string, seconds int, jobFunc interface{}, params ...interface{}) (*Job, error) {
+	return c.NewJobByDuration(name, time.Duration(seconds)*time.Second, jobFunc, params...)
 }
 
-func (c *Cron) Duration(name string, duration time.Duration, jobFunc interface{}, params ...interface{}) error {
-	_, err := c.scheduler.NewJob(
+func (c *Cron) NewJobByDuration(
+	name string,
+	duration time.Duration,
+	jobFunc interface{},
+	params ...interface{},
+) (*Job, error) {
+	job, err := c.scheduler.NewJob(
 		gocron.DurationJob(duration),
 		gocron.NewTask(jobFunc, params...),
 		gocron.WithName(name),
 	)
-	return err
+	return &Job{
+		cron: c,
+		job:  job,
+	}, err
+}
+
+type Job struct {
+	cron *Cron
+	job  gocron.Job
+}
+
+func (c *Job) Cancel() error {
+	return c.cron.scheduler.RemoveJob(c.job.ID())
 }

internal/service/supervisor/supervisor.go

@@ -37,6 +37,7 @@ var ProviderSet = wire.NewSet(NewSupervisorService)
 type SupervisorService struct {
 	s            *bizsupervisor.Supervisor
 	cron         *libcron.Cron
+	job          *libcron.Job
 	porter       *client.Porter
 	t            *data.TipherethRepo
 	auth         *libauth.Auth
@@ -65,6 +66,7 @@ func NewSupervisorService(
 	res := SupervisorService{
 		s:            s,
 		cron:         cron,
+		job:          nil,
 		porter:       porter,
 		t:            t,
 		auth:         auth,
@@ -80,7 +82,7 @@ func NewSupervisorService(
 }
 
 func (s *SupervisorService) Start(ctx context.Context) error {
-	err := s.cron.Duration(
+	job, err := s.cron.NewJobByDuration(
 		"SupervisorService Heartbeat",
 		s.getHeartbeatInterval(),
 		func() {
@@ -93,10 +95,17 @@ func (s *SupervisorService) Start(ctx context.Context) error {
 	if err != nil {
 		return fmt.Errorf("failed to register cron: %w", err)
 	}
+	s.job = job
 	return nil
 }
 
 func (s *SupervisorService) Stop(ctx context.Context) error {
+	if s.job != nil {
+		if err := s.job.Cancel(); err != nil {
+			return fmt.Errorf("failed to stop cron job: %w", err)
+		}
+		s.job = nil
+	}
 	return nil
 }