Skip to content

Commit

Permalink
upgrade deps + uvarint delimited writer/reader.
Browse files Browse the repository at this point in the history
  • Loading branch information
raulk committed Jul 30, 2020
1 parent 81b0285 commit 1bfba46
Show file tree
Hide file tree
Showing 4 changed files with 364 additions and 35 deletions.
7 changes: 3 additions & 4 deletions dht_net.go
Original file line number Diff line number Diff line change
Expand Up @@ -11,12 +11,11 @@ import (
"github.com/libp2p/go-libp2p-core/helpers"
"github.com/libp2p/go-libp2p-core/network"
"github.com/libp2p/go-libp2p-core/peer"
"github.com/libp2p/go-msgio/protoio"

"github.com/libp2p/go-libp2p-kad-dht/metrics"
pb "github.com/libp2p/go-libp2p-kad-dht/pb"

ggio "github.com/gogo/protobuf/io"

"github.com/libp2p/go-msgio"
"go.opencensus.io/stats"
"go.opencensus.io/tag"
Expand All @@ -34,15 +33,15 @@ var ErrReadTimeout = fmt.Errorf("timed out reading response")
// packet for every single write.
type bufferedDelimitedWriter struct {
*bufio.Writer
ggio.WriteCloser
protoio.WriteCloser
}

var writerPool = sync.Pool{
New: func() interface{} {
w := bufio.NewWriter(nil)
return &bufferedDelimitedWriter{
Writer: w,
WriteCloser: ggio.NewDelimitedWriter(w),
WriteCloser: protoio.NewDelimitedWriter(w),
}
},
}
Expand Down
25 changes: 13 additions & 12 deletions ext_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -12,13 +12,14 @@ import (
"github.com/libp2p/go-libp2p-core/protocol"
"github.com/libp2p/go-libp2p-core/routing"

pb "github.com/libp2p/go-libp2p-kad-dht/pb"
record "github.com/libp2p/go-libp2p-record"
swarmt "github.com/libp2p/go-libp2p-swarm/testing"
bhost "github.com/libp2p/go-libp2p/p2p/host/basic"
mocknet "github.com/libp2p/go-libp2p/p2p/net/mock"
"github.com/libp2p/go-msgio/protoio"

pb "github.com/libp2p/go-libp2p-kad-dht/pb"

ggio "github.com/gogo/protobuf/io"
u "github.com/ipfs/go-ipfs-util"
)

Expand Down Expand Up @@ -144,8 +145,8 @@ func TestGetFailures(t *testing.T) {
host2.SetStreamHandler(proto, func(s network.Stream) {
defer s.Close()

pbr := ggio.NewDelimitedReader(s, network.MessageSizeMax)
pbw := ggio.NewDelimitedWriter(s)
pbr := protoio.NewDelimitedReader(s, network.MessageSizeMax)
pbw := protoio.NewDelimitedWriter(s)

pmes := new(pb.Message)
if err := pbr.ReadMsg(pmes); err != nil {
Expand Down Expand Up @@ -199,8 +200,8 @@ func TestGetFailures(t *testing.T) {
}
defer s.Close()

pbr := ggio.NewDelimitedReader(s, network.MessageSizeMax)
pbw := ggio.NewDelimitedWriter(s)
pbr := protoio.NewDelimitedReader(s, network.MessageSizeMax)
pbw := protoio.NewDelimitedWriter(s)

if err := pbw.WriteMsg(&req); err != nil {
t.Fatal(err)
Expand Down Expand Up @@ -250,8 +251,8 @@ func TestNotFound(t *testing.T) {
host.SetStreamHandler(proto, func(s network.Stream) {
defer s.Close()

pbr := ggio.NewDelimitedReader(s, network.MessageSizeMax)
pbw := ggio.NewDelimitedWriter(s)
pbr := protoio.NewDelimitedReader(s, network.MessageSizeMax)
pbw := protoio.NewDelimitedWriter(s)

pmes := new(pb.Message)
if err := pbr.ReadMsg(pmes); err != nil {
Expand Down Expand Up @@ -347,8 +348,8 @@ func TestLessThanKResponses(t *testing.T) {
host.SetStreamHandler(proto, func(s network.Stream) {
defer s.Close()

pbr := ggio.NewDelimitedReader(s, network.MessageSizeMax)
pbw := ggio.NewDelimitedWriter(s)
pbr := protoio.NewDelimitedReader(s, network.MessageSizeMax)
pbw := protoio.NewDelimitedWriter(s)

pmes := new(pb.Message)
if err := pbr.ReadMsg(pmes); err != nil {
Expand Down Expand Up @@ -416,8 +417,8 @@ func TestMultipleQueries(t *testing.T) {
hosts[1].SetStreamHandler(proto, func(s network.Stream) {
defer s.Close()

pbr := ggio.NewDelimitedReader(s, network.MessageSizeMax)
pbw := ggio.NewDelimitedWriter(s)
pbr := protoio.NewDelimitedReader(s, network.MessageSizeMax)
pbw := protoio.NewDelimitedWriter(s)

pmes := new(pb.Message)
if err := pbr.ReadMsg(pmes); err != nil {
Expand Down
54 changes: 36 additions & 18 deletions go.mod
Original file line number Diff line number Diff line change
Expand Up @@ -3,36 +3,54 @@ module github.com/libp2p/go-libp2p-kad-dht
go 1.14

require (
github.com/davidlazar/go-crypto v0.0.0-20200604182044-b73af7476f6c // indirect
github.com/gogo/protobuf v1.3.1
github.com/google/gopacket v1.1.17
github.com/google/gopacket v1.1.18
github.com/google/uuid v1.1.1
github.com/hashicorp/go-multierror v1.1.0
github.com/hashicorp/golang-lru v0.5.4
github.com/ipfs/go-cid v0.0.5
github.com/ipfs/go-cid v0.0.7
github.com/ipfs/go-datastore v0.4.4
github.com/ipfs/go-detect-race v0.0.1
github.com/ipfs/go-ipfs-util v0.0.1
github.com/ipfs/go-ipfs-util v0.0.2
github.com/ipfs/go-ipns v0.0.2
github.com/ipfs/go-log v1.0.4
github.com/ipfs/go-log/v2 v2.1.1 // indirect
github.com/jbenet/goprocess v0.1.4
github.com/libp2p/go-eventbus v0.1.0
github.com/libp2p/go-libp2p v0.8.2
github.com/libp2p/go-libp2p-core v0.5.4
github.com/libp2p/go-eventbus v0.2.1
github.com/libp2p/go-libp2p v0.10.0
github.com/libp2p/go-libp2p-autonat v0.3.0 // indirect
github.com/libp2p/go-libp2p-circuit v0.3.0 // indirect
github.com/libp2p/go-libp2p-core v0.6.1
github.com/libp2p/go-libp2p-discovery v0.5.0 // indirect
github.com/libp2p/go-libp2p-kbucket v0.4.2
github.com/libp2p/go-libp2p-peerstore v0.2.4
github.com/libp2p/go-libp2p-record v0.1.2
github.com/libp2p/go-libp2p-peerstore v0.2.6
github.com/libp2p/go-libp2p-record v0.1.3
github.com/libp2p/go-libp2p-routing-helpers v0.2.3
github.com/libp2p/go-libp2p-swarm v0.2.3
github.com/libp2p/go-libp2p-testing v0.1.1
github.com/libp2p/go-msgio v0.0.4
github.com/libp2p/go-netroute v0.1.2
github.com/libp2p/go-libp2p-swarm v0.2.8
github.com/libp2p/go-libp2p-testing v0.1.2-0.20200422005655-8775583591d8
github.com/libp2p/go-maddr-filter v0.1.0 // indirect
github.com/libp2p/go-mplex v0.1.3 // indirect
github.com/libp2p/go-msgio v0.0.6
github.com/libp2p/go-netroute v0.1.3
github.com/libp2p/go-reuseport-transport v0.0.4 // indirect
github.com/libp2p/go-sockaddr v0.1.0 // indirect
github.com/libp2p/go-yamux v1.3.8 // indirect
github.com/multiformats/go-base32 v0.0.3
github.com/multiformats/go-multiaddr v0.2.1
github.com/multiformats/go-multiaddr v0.2.2
github.com/multiformats/go-multiaddr-net v0.1.5
github.com/multiformats/go-multihash v0.0.13
github.com/multiformats/go-multistream v0.1.1
github.com/stretchr/testify v1.5.1
github.com/multiformats/go-multihash v0.0.14
github.com/multiformats/go-multistream v0.1.2
github.com/opentracing/opentracing-go v1.2.0 // indirect
github.com/stretchr/testify v1.6.1
github.com/wangjia184/sortedset v0.0.0-20160527075905-f5d03557ba30 // indirect
github.com/whyrusleeping/go-keyspace v0.0.0-20160322163242-5b898ac5add1
go.opencensus.io v0.22.3
go.uber.org/zap v1.14.1
go.opencensus.io v0.22.4
go.uber.org/zap v1.15.0
golang.org/x/crypto v0.0.0-20200728195943-123391ffb6de // indirect
golang.org/x/net v0.0.0-20200707034311-ab3426394381 // indirect
golang.org/x/sys v0.0.0-20200728102440-3e129f6d46b1 // indirect
golang.org/x/text v0.3.3 // indirect
google.golang.org/grpc v1.30.0 // indirect
gopkg.in/yaml.v3 v3.0.0-20200615113413-eeeca48fe776 // indirect
)
Loading

0 comments on commit 1bfba46

Please sign in to comment.