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

TryAdd correctly

Browse Source
This commit is contained in:
mcrakhman 2023-06-19 11:10:56 +02:00
parent 16be33fc96
commit a9889a6245
No known key found for this signature in database
GPG Key ID: DED12CFEF5B8396B
2 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
net/streampool/stream.go
View File

@ -23,7 +23,7 @@ type stream struct {
} }
func (sr *stream) write(msg drpc.Message) (err error) { func (sr *stream) write(msg drpc.Message) (err error) {
return sr.queue.TryAdd(sr.stream.Context(), msg) return sr.queue.TryAdd(msg)
} }
func (sr *stream) readLoop() error { func (sr *stream) readLoop() error {

11
net/streampool/streampool_test.go
View File

@ -2,18 +2,19 @@ package streampool
import ( import (
"fmt" "fmt"
"sort"
"sync"
"sync/atomic"
"testing"
"time"
"github.com/anyproto/any-sync/net/peer" "github.com/anyproto/any-sync/net/peer"
"github.com/anyproto/any-sync/net/rpc/rpctest" "github.com/anyproto/any-sync/net/rpc/rpctest"
"github.com/anyproto/any-sync/net/streampool/testservice" "github.com/anyproto/any-sync/net/streampool/testservice"
"github.com/stretchr/testify/assert" "github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require" "github.com/stretchr/testify/require"
"golang.org/x/net/context" "golang.org/x/net/context"
"sort"
"storj.io/drpc" "storj.io/drpc"
"sync"
"sync/atomic"
"testing"
"time"
) )
var ctx = context.Background() var ctx = context.Background()