Reduce the delay between two fetch feed timed task

This commit is contained in:
LittleSheep 2025-04-06 13:13:03 +08:00
parent b2f88c18c0
commit 1f27667b7e
3 changed files with 19 additions and 8 deletions

View File

@ -13,15 +13,14 @@ func adminTriggerScanTask(c *fiber.Ctx) error {
} }
var data struct { var data struct {
Eager bool `json:"eager"` Eager bool `json:"eager"`
Sources []string `json:"sources"`
} }
if err := exts.BindAndValidate(c, &data); err != nil { if err := exts.BindAndValidate(c, &data); err != nil {
return err return err
} }
go services.FetchFeedTimed() go services.FetchFeed(data.Eager)
return c.SendStatus(fiber.StatusOK) return c.SendStatus(fiber.StatusOK)
} }

View File

@ -22,9 +22,18 @@ func FetchFeedTimed() {
func FetchFeed(eager ...bool) { func FetchFeed(eager ...bool) {
var feeds []models.SubscriptionFeed var feeds []models.SubscriptionFeed
if err := database.C.Where("is_enabled = ?", true).Find(&feeds).Error; err != nil { if len(eager) > 0 && eager[0] {
log.Warn().Err(err).Msg("An error occurred when fetching feeds.") if err := database.C.Where("is_enabled = ?", true).Find(&feeds).Error; err != nil {
return log.Warn().Err(err).Msg("An error occurred when fetching feeds.")
return
}
} else {
if err := database.C.
Where("last_fetched_at IS NULL OR NOW() >= last_fetched_at + (pull_interval || ' hours')::interval").
Find(&feeds).Error; err != nil {
log.Warn().Err(err).Msg("An error occurred when fetching due feeds.")
return
}
} }
log.Info().Int("count", len(feeds)).Msg("Ready to fetch feeds...") log.Info().Int("count", len(feeds)).Msg("Ready to fetch feeds...")
@ -56,7 +65,10 @@ func FetchFeed(eager ...bool) {
count += len(result) count += len(result)
} }
database.C.Where("id IN ?", scannedFeed).Update("last_fetched_at", time.Now()) database.C.
Model(&models.SubscriptionFeed{}).
Where("id IN ?", scannedFeed).
Update("last_fetched_at", time.Now())
log.Info().Int("count", count).Msg("Scanned all feeds.") log.Info().Int("count", count).Msg("Scanned all feeds.")
} }

View File

@ -75,7 +75,7 @@ func main() {
// Configure timed tasks // Configure timed tasks
quartz := cron.New(cron.WithLogger(cron.VerbosePrintfLogger(&log.Logger))) quartz := cron.New(cron.WithLogger(cron.VerbosePrintfLogger(&log.Logger)))
quartz.AddFunc("@every 60m", services.DoAutoDatabaseCleanup) quartz.AddFunc("@every 60m", services.DoAutoDatabaseCleanup)
quartz.AddFunc("@midnight", services.FetchFeedTimed) quartz.AddFunc("@every 60m", services.FetchFeedTimed)
quartz.Start() quartz.Start()
// Server // Server