coopgo-platform/gorush
7
0
Fork 0
Code Issues Pull Requests Projects Releases 1 Wiki Activity

chore(queue): add queue message interface (#602)

Browse Source 
Signed-off-by: Bo-Yi Wu <appleboy.tw@gmail.com>
This commit is contained in:
Bo-Yi Wu
2021-07-18 01:58:46 +08:00
committed by GitHub
parent d6b4a0ae39
commit 380162a38a
8 changed files with 31 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
queue/nsq/nsq.go
View File

@@ -102,7 +102,7 @@ func (s *Worker) Run(quit chan struct{}) error {
s.q.AddHandler(nsq.HandlerFunc(func(msg *nsq.Message) error {
wg.Add(1)
defer wg.Done()
var notification gorush.PushNotification
var notification *gorush.PushNotification
if err := json.Unmarshal(msg.Body, &notification); err != nil {
return err
}
@@ -136,8 +136,8 @@ func (s *Worker) Usage() int {
}
// Queue send notification to queue
func (s *Worker) Queue(job interface{}) error {
v, ok := job.(gorush.PushNotification)
func (s *Worker) Queue(job queue.QueuedMessage) error {
v, ok := job.(*gorush.PushNotification)
if !ok {
return errors.New("wrong type of job")
}