Skip to content

Commit

Permalink
Modify API NewRedisCeleryBroker, add parameters queueName. Because i …
Browse files Browse the repository at this point in the history
…want to set queuename, call multiple tasks but they have same task-name
  • Loading branch information
YYYGH committed Aug 18, 2019
1 parent 17631e1 commit 6216127
Showing 1 changed file with 7 additions and 3 deletions.
10 changes: 7 additions & 3 deletions redis_broker.go
Original file line number Diff line number Diff line change
Expand Up @@ -38,10 +38,14 @@ func NewRedisPool(uri string) *redis.Pool {
}

// NewRedisCeleryBroker creates new RedisCeleryBroker based on given uri
func NewRedisCeleryBroker(uri string) *RedisCeleryBroker {
func NewRedisCeleryBroker(uri string, queuename string) *RedisCeleryBroker {
qName := "celery"
if queuename != "" {
qName = queuename
}
return &RedisCeleryBroker{
Pool: NewRedisPool(uri),
queueName: "celery",
queueName: qName,
}
}

Expand Down Expand Up @@ -72,7 +76,7 @@ func (cb *RedisCeleryBroker) GetCeleryMessage() (*CeleryMessage, error) {
return nil, fmt.Errorf("null message received from redis")
}
messageList := messageJSON.([]interface{})
if string(messageList[0].([]byte)) != "celery" {
if string(messageList[0].([]byte)) != cd.queueName {
return nil, fmt.Errorf("not a celery message: %v", messageList[0])
}
var message CeleryMessage
Expand Down

0 comments on commit 6216127

Please sign in to comment.