60 lines
1.2 KiB
Go
60 lines
1.2 KiB
Go
|
package simple
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
|
||
|
|
"github.com/appleboy/gorush/config"
|
||
|
|
"github.com/appleboy/gorush/gorush"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Worker for simple queue using channel
|
||
|
|
type Worker struct {
|
||
|
|
cfg config.ConfYaml
|
||
|
|
queueNotification chan gorush.PushNotification
|
||
|
|
}
|
||
|
|
|
||
|
|
// Run start the worker
|
||
|
|
func (s *Worker) Run(_ chan struct{}) {
|
||
|
|
for notification := range s.queueNotification {
|
||
|
|
gorush.SendNotification(s.cfg, notification)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Stop worker
|
||
|
|
func (s *Worker) Stop() {
|
||
|
|
close(s.queueNotification)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Capacity for channel
|
||
|
|
func (s *Worker) Capacity() int {
|
||
|
|
return cap(s.queueNotification)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Usage for count of channel usage
|
||
|
|
func (s *Worker) Usage() int {
|
||
|
|
return len(s.queueNotification)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Enqueue send notification to queue
|
||
|
|
func (s *Worker) Enqueue(job interface{}) error {
|
||
|
|
select {
|
||
|
|
case s.queueNotification <- job.(gorush.PushNotification):
|
||
|
|
return nil
|
||
|
|
default:
|
||
|
|
return errors.New("max capacity reached")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Config update current config
|
||
|
|
func (s *Worker) Config(cfg config.ConfYaml) {
|
||
|
|
s.cfg = cfg
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewWorker for struct
|
||
|
|
func NewWorker(cfg config.ConfYaml) *Worker {
|
||
|
|
return &Worker{
|
||
|
|
cfg: cfg,
|
||
|
|
queueNotification: make(chan gorush.PushNotification, cfg.Core.QueueNum),
|
||
|
|
}
|
||
|
|
}
|