logos-messaging/go-libp2p-pubsub
1
0
Fork 0
mirror of https://github.com/logos-messaging/go-libp2p-pubsub.git synced 2026-01-04 05:43:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make close concurrent safe

Browse Source
This commit is contained in:
gammazero 2021-06-14 14:03:52 -07:00 committed by vyzo
parent e25f98c402
commit 55d412efa7
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
subscription.go
View File

@ -2,6 +2,7 @@ package pubsub
import ( import (
"context" "context"
"sync"
) )
// Subscription handles the details of a particular Topic subscription. // Subscription handles the details of a particular Topic subscription.
@ -10,9 +11,9 @@ type Subscription struct {
topic string topic string
ch chan *Message ch chan *Message
cancelCh chan<- *Subscription cancelCh chan<- *Subscription
closed bool
ctx context.Context ctx context.Context
err error err error
once sync.Once
} }
// Topic returns the topic string associated with the Subscription // Topic returns the topic string associated with the Subscription
@ -44,8 +45,7 @@ func (sub *Subscription) Cancel() {
} }
func (sub *Subscription) close() { func (sub *Subscription) close() {
if !sub.closed { sub.once.Do(func() {
close(sub.ch) close(sub.ch)
sub.closed = true })
}
} }