You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
109 lines
2.7 KiB
109 lines
2.7 KiB
package pgc |
|
|
|
import ( |
|
"context" |
|
"math/rand" |
|
"sync" |
|
"time" |
|
|
|
"go-common/app/job/main/tv/conf" |
|
"go-common/app/job/main/tv/dao/app" |
|
"go-common/app/job/main/tv/dao/cms" |
|
"go-common/app/job/main/tv/dao/lic" |
|
playdao "go-common/app/job/main/tv/dao/playurl" |
|
model "go-common/app/job/main/tv/model/pgc" |
|
"go-common/library/log" |
|
"go-common/library/queue/databus" |
|
|
|
"go-common/app/job/main/tv/dao/ftp" |
|
|
|
"github.com/robfig/cron" |
|
) |
|
|
|
var ctx = context.Background() |
|
|
|
// Service struct of service. |
|
type Service struct { |
|
dao *app.Dao |
|
daoClosed bool // logic close the dao's DB |
|
playurlDao *playdao.Dao |
|
licDao *lic.Dao |
|
ftpDao *ftp.Dao |
|
cmsDao *cms.Dao |
|
c *conf.Config |
|
waiter *sync.WaitGroup // general waiter |
|
waiterConsumer *sync.WaitGroup |
|
contentSub *databus.Databus // consumer for state change |
|
cron *cron.Cron |
|
ResuEps []*model.Content |
|
ResuSns []*model.TVEpSeason |
|
resuRetry map[string]int |
|
} |
|
|
|
// New create service instance and return. |
|
func New(c *conf.Config) (s *Service) { |
|
s = &Service{ |
|
c: c, |
|
dao: app.New(c), |
|
playurlDao: playdao.New(c), |
|
licDao: lic.New(c), |
|
ftpDao: ftp.New(c), |
|
cmsDao: cms.New(c), |
|
daoClosed: false, |
|
waiter: new(sync.WaitGroup), |
|
waiterConsumer: new(sync.WaitGroup), |
|
contentSub: databus.New(c.ContentSub), |
|
cron: cron.New(), |
|
resuRetry: make(map[string]int), |
|
} |
|
rand.Seed(time.Now().UnixNano()) |
|
// flush Redis - zone list |
|
go s.ZoneIdx() |
|
if err := s.cron.AddFunc(s.c.Redis.CronPGC, s.ZoneIdx); err != nil { |
|
panic(err) |
|
} |
|
if err := s.cron.AddFunc(s.c.PlayControl.ProducerCron, s.refreshCache); err != nil { |
|
panic(err) |
|
} |
|
if err := s.cron.AddFunc(s.c.Cfg.Merak.Cron, s.cmsShelve); err != nil { |
|
panic(err) |
|
} |
|
s.cron.Start() |
|
go s.searchSugproc() // uploads the passed season's list to search sug's FTP |
|
go s.seaPgcContproc() // uploads pgc search content to sug's FTP |
|
s.waiter.Add(1) |
|
go s.syncEPs() |
|
s.waiter.Add(1) |
|
go s.resubEps() |
|
s.waiter.Add(1) |
|
go s.resubSns() |
|
s.waiter.Add(1) |
|
go s.syncSeason() |
|
s.waiter.Add(1) |
|
go s.delSeason() |
|
s.waiter.Add(1) |
|
go s.delCont() |
|
// Databus |
|
s.waiterConsumer.Add(1) |
|
go s.consumeContent() // consume Databus Message to update MC |
|
return |
|
} |
|
|
|
// Close dao. |
|
func (s *Service) Close() { |
|
if s.dao != nil { |
|
s.daoClosed = true |
|
log.Info("Dao Closed!") |
|
} |
|
log.Info("Crontab Closed!") |
|
s.cron.Stop() |
|
log.Info("Databus Closed!") |
|
s.contentSub.Close() |
|
log.Info("Wait Producer!") |
|
s.waiter.Wait() |
|
log.Info("Wait SyncMC Consumers") |
|
s.waiterConsumer.Wait() |
|
log.Info("Physical Dao Closed!") |
|
s.dao.Close() |
|
log.Info("tv-job has been closed.") |
|
}
|
|
|