62 lines
1.0 KiB
Go
62 lines
1.0 KiB
Go
package paychmgr
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/ipfs/go-cid"
|
|
)
|
|
|
|
type msgListener struct {
|
|
id string
|
|
cb func(c cid.Cid, err error)
|
|
}
|
|
|
|
type msgListeners struct {
|
|
lk sync.Mutex
|
|
listeners []*msgListener
|
|
}
|
|
|
|
func (ml *msgListeners) onMsg(mcid cid.Cid, cb func(error)) string {
|
|
ml.lk.Lock()
|
|
defer ml.lk.Unlock()
|
|
|
|
l := &msgListener{
|
|
id: uuid.New().String(),
|
|
cb: func(c cid.Cid, err error) {
|
|
if mcid.Equals(c) {
|
|
cb(err)
|
|
}
|
|
},
|
|
}
|
|
ml.listeners = append(ml.listeners, l)
|
|
return l.id
|
|
}
|
|
|
|
func (ml *msgListeners) fireMsgComplete(mcid cid.Cid, err error) {
|
|
ml.lk.Lock()
|
|
defer ml.lk.Unlock()
|
|
|
|
for _, l := range ml.listeners {
|
|
l.cb(mcid, err)
|
|
}
|
|
}
|
|
|
|
func (ml *msgListeners) unsubscribe(sub string) {
|
|
ml.lk.Lock()
|
|
defer ml.lk.Unlock()
|
|
|
|
for i, l := range ml.listeners {
|
|
if l.id == sub {
|
|
ml.removeListener(i)
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
func (ml *msgListeners) removeListener(i int) {
|
|
copy(ml.listeners[i:], ml.listeners[i+1:])
|
|
ml.listeners[len(ml.listeners)-1] = nil
|
|
ml.listeners = ml.listeners[:len(ml.listeners)-1]
|
|
}
|