gaugendre/youtubebeat
1
0
Fork 0
Code Issues Pull requests Releases Activity

Stop the beat when no more videos to fetch

Browse source
This commit is contained in:
Gabriel Augendre 2018-11-18 18:46:54 +01:00
parent d0ded4fd02
commit 007dbdd3a9
1 changed files with 18 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
beater/youtubebeat.go
View file

@ -35,7 +35,7 @@ func New(b *beat.Beat, cfg *common.Config) (beat.Beater, error) {
return bt, nil return bt, nil
} }
func scrapeVideos(startId string, fieldsToSend chan common.MapStr) { func scrapeVideos(startId string, fieldsToSend chan common.MapStr, done chan bool) {
const BaseUrl = "https://www.youtube.com" const BaseUrl = "https://www.youtube.com"
const BaseSuffix = "/watch?v=" const BaseSuffix = "/watch?v="
@ -81,6 +81,7 @@ func scrapeVideos(startId string, fieldsToSend chan common.MapStr) {
videoCollector.Visit(BaseUrl + BaseSuffix + startId) videoCollector.Visit(BaseUrl + BaseSuffix + startId)
videoCollector.Wait() videoCollector.Wait()
done <- true
} }
// Run starts youtubebeat. // Run starts youtubebeat.
@ -94,7 +95,8 @@ func (bt *Youtubebeat) Run(b *beat.Beat) error {
} }
fieldsToSend := make(chan common.MapStr) fieldsToSend := make(chan common.MapStr)
go scrapeVideos(bt.config.StartId, fieldsToSend) done := make(chan bool)
go scrapeVideos(bt.config.StartId, fieldsToSend, done)
ticker := time.NewTicker(bt.config.Period) ticker := time.NewTicker(bt.config.Period)
for { for {
@ -103,7 +105,13 @@ func (bt *Youtubebeat) Run(b *beat.Beat) error {
return nil return nil
case <-ticker.C: case <-ticker.C:
} }
fields := <-fieldsToSend // Handle a SIGINT even when no more videos to fetch
select {
case <-done:
return nil
case <-bt.done:
return nil
case fields := <-fieldsToSend:
fields["type"] = b.Info.Name fields["type"] = b.Info.Name
event := beat.Event{ event := beat.Event{
Timestamp: time.Now(), Timestamp: time.Now(),
@ -113,6 +121,7 @@ func (bt *Youtubebeat) Run(b *beat.Beat) error {
logp.Info("Event sent") logp.Info("Event sent")
} }
} }
}
// Stop stops youtubebeat. // Stop stops youtubebeat.
func (bt *Youtubebeat) Stop() { func (bt *Youtubebeat) Stop() {