func (me *backend) Done(meta dingo.Meta) (err error) {
var v int
err = func() (err error) {
var (
ok bool
ids map[string]int
)
me.ridsLock.Lock()
defer me.ridsLock.Unlock()
if ids, ok = me.rids[meta.Name()]; ok {
v, ok = ids[meta.ID()]
delete(ids, meta.ID())
}
if !ok {
err = errors.New("store id not found")
return
}
return
}()
if err == nil {
err = me.stores.Stop(v)
}
return
}
func (me *backend) Poll(meta dingo.Meta) (reports <-chan []byte, err error) {
quit, done, idx := me.stores.New(0)
me.ridsLock.Lock()
defer me.ridsLock.Unlock()
if v, ok := me.rids[meta.Name()]; ok {
v[meta.ID()] = idx
} else {
me.rids[meta.Name()] = map[string]int{meta.ID(): idx}
}
r := make(chan []byte, 10)
reports = r
go me._store_routine_(quit, done, me.stores.Events(), r, meta)
return
}
func getConsumerTag(meta dingo.Meta) string {
return fmt.Sprintf("dingo.consumer.%s.%s", meta.Name(), meta.ID())
}
func getRoutingKey(meta dingo.Meta) string {
return fmt.Sprintf("dingo.rkey.%s.%s", meta.Name(), meta.ID())
}
func getQueueName(meta dingo.Meta) string {
return fmt.Sprintf("dingo.q.%s.%s", meta.Name(), meta.ID())
}
func (me *backend) Poll(meta dingo.Meta) (reports <-chan []byte, err error) {
// bind to the queue for this task
tag, qName, rKey := getConsumerTag(meta), getQueueName(meta), getRoutingKey(meta)
quit, done, idx := me.stores.New(0)
me.ridsLock.Lock()
defer me.ridsLock.Unlock()
if v, ok := me.rids[meta.Name()]; ok {
v[meta.ID()] = idx
} else {
me.rids[meta.Name()] = map[string]int{meta.ID(): idx}
}
// acquire a free channel
ci, err := me.receiver.Channel()
if err != nil {
return
}
var (
dv <-chan amqp.Delivery
r chan []byte
)
defer func() {
if err != nil {
me.receiver.ReleaseChannel(ci)
} else {
go me._store_routine_(quit, done, me.stores.Events(), r, ci, dv, meta)
}
}()
// declare a queue for this task
_, err = ci.Channel.QueueDeclare(
qName, // name of queue
true, // durable
false, // auto-delete
false, // exclusive
false, // nowait
nil, // args
)
if err != nil {
return
}
// bind queue to result-exchange
err = ci.Channel.QueueBind(
qName, // name of queue
rKey, // routing key
"dingo.x.result", // name of exchange
false, // nowait
nil, // args
)
if err != nil {
return
}
dv, err = ci.Channel.Consume(
qName, // name of queue
tag, // consumer Tag
false, // autoAck
false, // exclusive
false, // noLocal
false, // noWait
nil, // args
)
if err != nil {
return
}
r = make(chan []byte, 10)
reports = r
return
}