// NewChord creates Chord instance
func NewChord(group *Group, callback *signatures.TaskSignature) *Chord {
// Generate a UUID for the chord callback
callback.UUID = fmt.Sprintf("chord_%v", uuid.New())
// Add a chord callback to all tasks
for _, task := range group.Tasks {
task.ChordCallback = callback
}
return &Chord{Group: group, Callback: callback}
}
// Publish places a new message on the default queue
func (amqpBroker *AMQPBroker) Publish(signature *signatures.TaskSignature) error {
conn, channel, _, confirmsChan, err := amqpBroker.open()
if err != nil {
return err
}
defer amqpBroker.close(channel, conn)
message, err := json.Marshal(signature)
if err != nil {
return fmt.Errorf("JSON Encode Message: %v", err)
}
signature.AdjustRoutingKey(
amqpBroker.config.ExchangeType,
amqpBroker.config.BindingKey,
amqpBroker.config.DefaultQueue,
)
if err := channel.Publish(
amqpBroker.config.Exchange, // exchange
signature.RoutingKey, // routing key
false, // mandatory
false, // immediate
amqp.Publishing{
Headers: amqp.Table(signature.Headers),
ContentType: "application/json",
Body: message,
DeliveryMode: amqp.Persistent,
},
); err != nil {
return err
}
confirmed := <-confirmsChan
if confirmed.Ack {
return nil
}
return fmt.Errorf("Failed delivery of delivery tag: %v", confirmed.DeliveryTag)
}
// SendTask publishes a task to the default queue
func (server *Server) SendTask(signature *signatures.TaskSignature) (*backends.AsyncResult, error) {
// Make sure result backend is defined
if server.backend == nil {
return nil, errors.New("Result backend required")
}
// Auto generate a UUID if not set already
if signature.UUID == "" {
signature.UUID = fmt.Sprintf("task_%v", uuid.New())
}
// Set initial task state to PENDING
if err := server.backend.SetStatePending(signature); err != nil {
return nil, fmt.Errorf("Set State Pending: %v", err)
}
if err := server.broker.Publish(signature); err != nil {
return nil, fmt.Errorf("Publish Message: %v", err)
}
return backends.NewAsyncResult(signature, server.backend), nil
}
func TestAdjustRoutingKey(t *testing.T) {
var signature signatures.TaskSignature
signature = signatures.TaskSignature{
RoutingKey: "routing_key",
}
signature.AdjustRoutingKey("direct", "binding_key", "queue")
assert.Equal(t, "routing_key", signature.RoutingKey)
signature = signatures.TaskSignature{
RoutingKey: "",
}
signature.AdjustRoutingKey("direct", "binding_key", "queue")
assert.Equal(t, "binding_key", signature.RoutingKey)
signature = signatures.TaskSignature{
RoutingKey: "",
}
signature.AdjustRoutingKey("topic", "binding_key", "queue")
assert.Equal(t, "queue", signature.RoutingKey)
}