27 lines
388 B
Go
27 lines
388 B
Go
|
package floodsub
|
||
|
|
|
||
|
|
type Subscription struct {
|
||
|
|
topic string
|
||
|
|
ch chan *Message
|
||
|
|
cancelCh chan<- *Subscription
|
||
|
|
err error
|
||
|
|
}
|
||
|
|
|
||
|
|
func (sub *Subscription) Topic() string {
|
||
|
|
return sub.topic
|
||
|
|
}
|
||
|
|
|
||
|
|
func (sub *Subscription) Next() (*Message, error) {
|
||
|
|
msg, ok := <-sub.ch
|
||
|
|
|
||
|
|
if !ok {
|
||
|
|
return msg, sub.err
|
||
|
|
}
|
||
|
|
|
||
|
|
return msg, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (sub *Subscription) Cancel() {
|
||
|
|
sub.cancelCh <- sub
|
||
|
|
}
|