45 lines
684 B
Go
45 lines
684 B
Go
|
package publisher
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"log"
|
||
|
|
|
||
|
|
"lmika.dev/lmika/weiro/models"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Queue struct {
|
||
|
|
publisher *Publisher
|
||
|
|
pending chan models.Site
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewQueue(publisher *Publisher) *Queue {
|
||
|
|
return &Queue{
|
||
|
|
publisher: publisher,
|
||
|
|
pending: make(chan models.Site, 1),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (q *Queue) Queue(site models.Site) bool {
|
||
|
|
select {
|
||
|
|
case q.pending <- site:
|
||
|
|
return true
|
||
|
|
default:
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (q *Queue) Start(ctx context.Context) {
|
||
|
|
go func() {
|
||
|
|
for {
|
||
|
|
select {
|
||
|
|
case site := <-q.pending:
|
||
|
|
if err := q.publisher.Publish(ctx, site); err != nil {
|
||
|
|
log.Printf("error publishing site: %v", err)
|
||
|
|
}
|
||
|
|
case <-ctx.Done():
|
||
|
|
return
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
}
|