benjamyn/any-sync
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Change add to tryadd in dial

Browse Source
This commit is contained in:
mcrakhman 2023-04-12 19:16:48 +02:00
parent c77b9620c6
commit 77ad931cc3
No known key found for this signature in database
GPG Key ID: DED12CFEF5B8396B
2 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
net/streampool/sendpool.go
View File

@ -28,6 +28,10 @@ func (ss *execPool) Add(ctx context.Context, f ...func()) (err error) {
return ss.batch.Add(ctx, f...) return ss.batch.Add(ctx, f...)
} }
func (ss *execPool) TryAdd(f ...func()) (err error) {
return ss.batch.TryAdd(f...)
}
func (ss *execPool) sendLoop() { func (ss *execPool) sendLoop() {
for { for {
f, err := ss.batch.WaitOne(context.Background()) f, err := ss.batch.WaitOne(context.Background())

2
net/streampool/streampool.go
View File

@ -117,7 +117,7 @@ func (s *streamPool) Send(ctx context.Context, msg drpc.Message, peerGetter Peer
} }
} }
} }
return s.dial.Add(ctx, func() { return s.dial.TryAdd(func() {
peers, dialErr := peerGetter(ctx) peers, dialErr := peerGetter(ctx)
if dialErr != nil { if dialErr != nil {
log.InfoCtx(ctx, "can't get peers", zap.Error(dialErr)) log.InfoCtx(ctx, "can't get peers", zap.Error(dialErr))