60 lines
1.3 KiB
Go
60 lines
1.3 KiB
Go
package seo
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
"time"
|
|
"videoweb/app/skd/job/sitemap"
|
|
"videoweb/pub/common/log"
|
|
"videoweb/pub/seo/sitemapdirty"
|
|
|
|
"github.com/robfig/cron/v3"
|
|
)
|
|
|
|
var seoCron *cron.Cron
|
|
|
|
func init() {
|
|
seoCron = cron.New(cron.WithSeconds())
|
|
}
|
|
|
|
var (
|
|
dirtyMu sync.Mutex
|
|
// generatedDirtyAt 最近一次 sitemap 生成所对应的内容变更时间
|
|
generatedDirtyAt time.Time
|
|
)
|
|
|
|
// Start 启动 SEO 静态资源生成（sitemap）。
|
|
// 页面级搜索引擎推送由 web 进程的 seopush 任务负责，不在这里处理。
|
|
func Start() {
|
|
generatedDirtyAt = sitemapdirty.LastDirtyAt(context.Background())
|
|
sitemap.StartSitemapJob(6*time.Hour, sitemap.OutputDir)
|
|
|
|
// 短剧发布/更新/下架发生在 web 进程，这里检查它留下的变更标记，有变更才重新生成
|
|
_, _ = seoCron.AddFunc("0 */2 * * * ?", func() {
|
|
regenerateSitemapIfDirty()
|
|
})
|
|
seoCron.Start()
|
|
log.Info("seo skd started")
|
|
}
|
|
|
|
func regenerateSitemapIfDirty() {
|
|
if !dirtyMu.TryLock() {
|
|
return
|
|
}
|
|
defer dirtyMu.Unlock()
|
|
|
|
dirtyAt := sitemapdirty.LastDirtyAt(context.Background())
|
|
if dirtyAt.IsZero() || !dirtyAt.After(generatedDirtyAt) {
|
|
return
|
|
}
|
|
log.Info("seo: content changed, regenerate sitemap", log.Any("dirtyAt", dirtyAt))
|
|
sitemap.GenerateSitemap()
|
|
generatedDirtyAt = dirtyAt
|
|
}
|
|
|
|
func Stop() {
|
|
if seoCron != nil {
|
|
seoCron.Stop()
|
|
}
|
|
}
|