From 11139e94eed1a6725a6219efa3c011881a706b57 Mon Sep 17 00:00:00 2001 From: Yingrong Zhao Date: Wed, 18 Mar 2020 13:51:39 -0400 Subject: [PATCH] more cleanup Change-Id: I256df930c969c4c81850d66d85a72488abd5529b --- drpcconn/conn.go | 3 --- drpcmanager/manager.go | 4 ---- drpcmetadata/doc.go | 6 ------ drpcmetadata/metadata.go | 6 +++--- drpcmetadata/proto/doc.go | 8 ++++++++ drpcmetadata/proto/metadata.pb.go | 31 ++++++++++++++--------------- drpcwire/packet.go | 15 +------------- go.mod | 1 - go.sum | 20 ------------------- internal/grpccompat/go.sum | 16 --------------- internal/integration/common_test.go | 6 +----- internal/integration/simple_test.go | 19 ++++++++---------- 12 files changed, 36 insertions(+), 99 deletions(-) create mode 100644 drpcmetadata/proto/doc.go diff --git a/drpcconn/conn.go b/drpcconn/conn.go index 339e714..157dd8d 100644 --- a/drpcconn/conn.go +++ b/drpcconn/conn.go @@ -16,9 +16,6 @@ import ( "storj.io/drpc/drpcwire" ) -// INVOKE_HEADER_VERSION_1 indicates version 1 of invoke header. -const INVOKE_HEADER_VERSION_1 = 1 - // Options controls configuration settings for a conn. type Options struct { // Manager controls the options we pass to the manager of this conn. diff --git a/drpcmanager/manager.go b/drpcmanager/manager.go index 14c99b8..53f100b 100644 --- a/drpcmanager/manager.go +++ b/drpcmanager/manager.go @@ -12,7 +12,6 @@ import ( "storj.io/drpc" "storj.io/drpc/drpcctx" "storj.io/drpc/drpcdebug" - "storj.io/drpc/drpcmetadata" "storj.io/drpc/drpcsignal" "storj.io/drpc/drpcstream" @@ -193,9 +192,6 @@ func (m *Manager) NewServerStream(ctx context.Context) (stream *drpcstream.Strea case drpcwire.KindInvoke: if metadata.ID.Stream == pkt.ID.Stream { - // we use the first two bytes as the version flag to indicate whether - // there's metadata stored in the invoke message. - // If so, we should store the metadata onto the stream context msg, err := drpcmetadata.Decode(metadata.Data) if err != nil { return nil, "", err diff --git a/drpcmetadata/doc.go b/drpcmetadata/doc.go index fa5360f..61f9a21 100644 --- a/drpcmetadata/doc.go +++ b/drpcmetadata/doc.go @@ -3,9 +3,3 @@ // Package drpcmetadata define the structure of the metadata supported by drpc library. package drpcmetadata - -import "github.com/spacemonkeygo/monkit/v3" - -var mon = monkit.Package() - -//go:generate bash -c "go install storj.io/drpc/cmd/protoc-gen-drpc && protoc --drpc_out=plugins=drpc:. ./proto/metadata.proto" diff --git a/drpcmetadata/metadata.go b/drpcmetadata/metadata.go index f18e313..94f6f6a 100644 --- a/drpcmetadata/metadata.go +++ b/drpcmetadata/metadata.go @@ -12,8 +12,8 @@ import ( ppb "storj.io/drpc/drpcmetadata/proto" ) -// INVOKE_METADATA_VERSION_1 indicates the version of InvokeMetadata message is 1. -const INVOKE_METADATA_VERSION_1 = 1 +// InvokMetadataVersion1 indicates the version of InvokeMetadata message is 1. +const InvokMetadataVersion1 = 1 // Metadata is a mapping from metadata key to value. type Metadata map[string]string @@ -40,7 +40,7 @@ func (md Metadata) AddPairs(ctx context.Context) context.Context { // Encode generates byte form of the metadata and appends it onto the passed in buffer. func (md Metadata) Encode(buffer []byte) ([]byte, error) { msg := ppb.InvokeMetadata{ - Version: INVOKE_METADATA_VERSION_1, + Version: InvokMetadataVersion1, Data: md, } diff --git a/drpcmetadata/proto/doc.go b/drpcmetadata/proto/doc.go new file mode 100644 index 0000000..6655be6 --- /dev/null +++ b/drpcmetadata/proto/doc.go @@ -0,0 +1,8 @@ +// Copyright (C) 2019 Storj Labs, Inc. +// See LICENSE for copying information. + +// Package proto defines the proto messages exposed by drpc for +// sending metadata across the wire. +package proto + +//go:generate bash -c "go install storj.io/drpc/cmd/protoc-gen-drpc && protoc --drpc_out=plugins=drpc:. metadata.proto" diff --git a/drpcmetadata/proto/metadata.pb.go b/drpcmetadata/proto/metadata.pb.go index c235df2..23e9a42 100644 --- a/drpcmetadata/proto/metadata.pb.go +++ b/drpcmetadata/proto/metadata.pb.go @@ -1,5 +1,5 @@ // Code generated by protoc-gen-gogo. DO NOT EDIT. -// source: proto/metadata.proto +// source: metadata.proto package proto @@ -32,7 +32,7 @@ func (m *InvokeMetadata) Reset() { *m = InvokeMetadata{} } func (m *InvokeMetadata) String() string { return proto.CompactTextString(m) } func (*InvokeMetadata) ProtoMessage() {} func (*InvokeMetadata) Descriptor() ([]byte, []int) { - return fileDescriptor_4891c0dff9aa45c8, []int{0} + return fileDescriptor_56d9f74966f40d04, []int{0} } func (m *InvokeMetadata) XXX_Unmarshal(b []byte) error { return xxx_messageInfo_InvokeMetadata.Unmarshal(m, b) @@ -71,19 +71,18 @@ func init() { proto.RegisterMapType((map[string]string)(nil), "metadata.InvokeMetadata.DataEntry") } -func init() { proto.RegisterFile("proto/metadata.proto", fileDescriptor_4891c0dff9aa45c8) } +func init() { proto.RegisterFile("metadata.proto", fileDescriptor_56d9f74966f40d04) } -var fileDescriptor_4891c0dff9aa45c8 = []byte{ - // 161 bytes of a gzipped FileDescriptorProto - 0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xe2, 0x12, 0x29, 0x28, 0xca, 0x2f, - 0xc9, 0xd7, 0xcf, 0x4d, 0x2d, 0x49, 0x4c, 0x49, 0x2c, 0x49, 0xd4, 0x03, 0x73, 0x85, 0x38, 0x60, - 0x7c, 0xa5, 0xd9, 0x8c, 0x5c, 0x7c, 0x9e, 0x79, 0x65, 0xf9, 0xd9, 0xa9, 0xbe, 0x50, 0x21, 0x21, - 0x09, 0x2e, 0xf6, 0xb2, 0xd4, 0xa2, 0xe2, 0xcc, 0xfc, 0x3c, 0x09, 0x46, 0x05, 0x46, 0x0d, 0xde, - 0x20, 0x18, 0x57, 0xc8, 0x8c, 0x8b, 0x05, 0xa4, 0x42, 0x82, 0x49, 0x81, 0x59, 0x83, 0xdb, 0x48, - 0x49, 0x0f, 0x6e, 0x2a, 0xaa, 0x09, 0x7a, 0x2e, 0x89, 0x25, 0x89, 0xae, 0x79, 0x25, 0x45, 0x95, - 0x41, 0x60, 0xf5, 0x52, 0xe6, 0x5c, 0x9c, 0x70, 0x21, 0x21, 0x01, 0x2e, 0xe6, 0xec, 0xd4, 0x4a, - 0xb0, 0xd1, 0x9c, 0x41, 0x20, 0xa6, 0x90, 0x08, 0x17, 0x6b, 0x59, 0x62, 0x4e, 0x69, 0xaa, 0x04, - 0x13, 0x58, 0x0c, 0xc2, 0xb1, 0x62, 0xb2, 0x60, 0x74, 0x62, 0x8f, 0x62, 0x05, 0x3b, 0x38, 0x89, - 0x0d, 0x4c, 0x19, 0x03, 0x02, 0x00, 0x00, 0xff, 0xff, 0x21, 0xb6, 0x4e, 0xe1, 0xcf, 0x00, 0x00, - 0x00, +var fileDescriptor_56d9f74966f40d04 = []byte{ + // 158 bytes of a gzipped FileDescriptorProto + 0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xe2, 0xe2, 0xcb, 0x4d, 0x2d, 0x49, + 0x4c, 0x49, 0x2c, 0x49, 0xd4, 0x2b, 0x28, 0xca, 0x2f, 0xc9, 0x17, 0xe2, 0x80, 0xf1, 0x95, 0x66, + 0x33, 0x72, 0xf1, 0x79, 0xe6, 0x95, 0xe5, 0x67, 0xa7, 0xfa, 0x42, 0x85, 0x84, 0x24, 0xb8, 0xd8, + 0xcb, 0x52, 0x8b, 0x8a, 0x33, 0xf3, 0xf3, 0x24, 0x18, 0x15, 0x18, 0x35, 0x78, 0x83, 0x60, 0x5c, + 0x21, 0x33, 0x2e, 0x16, 0x90, 0x0a, 0x09, 0x26, 0x05, 0x66, 0x0d, 0x6e, 0x23, 0x25, 0x3d, 0xb8, + 0xa9, 0xa8, 0x26, 0xe8, 0xb9, 0x24, 0x96, 0x24, 0xba, 0xe6, 0x95, 0x14, 0x55, 0x06, 0x81, 0xd5, + 0x4b, 0x99, 0x73, 0x71, 0xc2, 0x85, 0x84, 0x04, 0xb8, 0x98, 0xb3, 0x53, 0x2b, 0xc1, 0x46, 0x73, + 0x06, 0x81, 0x98, 0x42, 0x22, 0x5c, 0xac, 0x65, 0x89, 0x39, 0xa5, 0xa9, 0x12, 0x4c, 0x60, 0x31, + 0x08, 0xc7, 0x8a, 0xc9, 0x82, 0xd1, 0x89, 0x3d, 0x8a, 0x15, 0xec, 0xe0, 0x24, 0x36, 0x30, 0x65, + 0x0c, 0x08, 0x00, 0x00, 0xff, 0xff, 0x29, 0x18, 0xb7, 0xe6, 0xc9, 0x00, 0x00, 0x00, } diff --git a/drpcwire/packet.go b/drpcwire/packet.go index 79e30d5..8a4fdc5 100644 --- a/drpcwire/packet.go +++ b/drpcwire/packet.go @@ -3,10 +3,7 @@ package drpcwire -import ( - "bytes" - "fmt" -) +import "fmt" //go:generate stringer -type=Kind -trimprefix=Kind_ -output=packet_string.go @@ -161,13 +158,3 @@ func (p Packet) String() string { return fmt.Sprintf("", p.ID.Stream, p.ID.Message, p.Kind, len(p.Data)) } - -// IsVersioned checks wether a given packet contains versioning bytes. -// Currently, we use the first two bytes to indicate versioning. -func (p Packet) IsVersioned() bool { - if len(p.Data) < 3 { - return false - } - - return bytes.Equal(p.Data[:2], []byte{0, 0}) -} diff --git a/go.mod b/go.mod index cbe0451..4c560eb 100644 --- a/go.mod +++ b/go.mod @@ -7,5 +7,4 @@ require ( github.com/spacemonkeygo/monkit/v3 v3.0.4 github.com/zeebo/assert v1.1.0 github.com/zeebo/errs v1.2.2 - storj.io/monkit-jaeger v0.0.0-20200305222653-48145d285891 ) diff --git a/go.sum b/go.sum index 7c450e6..ac65c59 100644 --- a/go.sum +++ b/go.sum @@ -1,21 +1,11 @@ -github.com/apache/thrift v0.12.0 h1:pODnxUFNcjP9UTLZGTdeh+j16A8lJbRvD3rOtrk/7bs= -github.com/apache/thrift v0.12.0/go.mod h1:cp2SuWMxlEZw2r+iP2GNCdIi4C1qmUzdZFSVb+bacwQ= -github.com/codahale/hdrhistogram v0.0.0-20161010025455-3a0bb77429bd/go.mod h1:sE/e/2PUdi/liOCUjSTXgM1o87ZssimdTWN964YiIeI= -github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= github.com/gogo/protobuf v1.2.1 h1:/s5zKNz0uPFCZ5hddgPdo2TK2TVrUNMn0OOX8/aZMTE= github.com/gogo/protobuf v1.2.1/go.mod h1:hp+jE20tsWTFYpLwKvXlhS1hjn+gTNwPg2I6zVXpSg4= github.com/kisielk/errcheck v1.1.0/go.mod h1:EZBBE59ingxPouuu3KfxchcWSUPOHkagtvWXihfKN4Q= github.com/kisielk/gotool v1.0.0/go.mod h1:XhKaO+MFFWcvkIS/tQcRk01m1F5IRFswLeQ+oQHNcck= -github.com/opentracing/opentracing-go v1.1.0/go.mod h1:UkNAQd3GIcIGf0SeVgPpRdFStlNbqXla1AfSYxPUl2o= -github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0= -github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= -github.com/spacemonkeygo/errors v0.0.0-20171212215202-9064522e9fd1/go.mod h1:7NL9UAYQnRM5iKHUCld3tf02fKb5Dft+41+VckASUy0= github.com/spacemonkeygo/monkit/v3 v3.0.4 h1:Ay+PZirv+qfd4sqcT+X/U3BnC7AcIaqp/IXh0oV36k8= github.com/spacemonkeygo/monkit/v3 v3.0.4/go.mod h1:JcK1pCbReQsOsMKF/POFSZCq7drXFybgGmbc27tuwes= github.com/spacemonkeygo/monotime v0.0.0-20180824235756-e3f48a95f98a h1:8+cCjxhToanKmxLIbuyBNe2EnpgwhiivsIaRJstDRFA= github.com/spacemonkeygo/monotime v0.0.0-20180824235756-e3f48a95f98a/go.mod h1:ul4bvvnCOPZgq8w0nTkSmWVg/hauVpFS97Am1YM1XXo= -github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= -github.com/stretchr/testify v1.5.1/go.mod h1:5W2xD1RspED5o8YsWQXVCued0rvSQ+mT+I5cxcmMvtA= github.com/zeebo/assert v1.1.0 h1:hU1L1vLTHsnO8x8c9KAR5GmM5QscxHg5RNU5z5qbUWY= github.com/zeebo/assert v1.1.0/go.mod h1:Pq9JiuJQpG8JLJdtkwrJESF0Foym2/D9XMU5ciN/wJ0= github.com/zeebo/errs v1.2.2 h1:5NFypMTuSdoySVTqlNs1dEoU21QVamMQJxW/Fii5O7g= @@ -23,18 +13,8 @@ github.com/zeebo/errs v1.2.2/go.mod h1:sgbWHsvVuTPHcqJJGQ1WhI5KbWlHYz+2+2C/LSEtC golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= golang.org/x/net v0.0.0-20190923162816-aa69164e4478 h1:l5EDrHhldLYb3ZRHDUhXF7Om7MvYXnkV9/iQNo1lX6g= golang.org/x/net v0.0.0-20190923162816-aa69164e4478/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s= -golang.org/x/net v0.0.0-20200226121028-0de0cce0169b h1:0mm1VjtFUOIlE1SbDlwjYaDxZVDP2S5ou6y0gSgXHu8= -golang.org/x/net v0.0.0-20200226121028-0de0cce0169b/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s= golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a h1:1BGLXjeY4akVXGgbC9HugT3Jv3hCI0z56oJR5vAMgBU= golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= golang.org/x/text v0.3.0 h1:g61tztE5qeGQ89tm6NTjjM9VPIm088od1l6aSorWRWg= golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= golang.org/x/tools v0.0.0-20180221164845-07fd8470d635/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= -gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= -gopkg.in/spacemonkeygo/monkit-zipkin.v2 v2.0.0-20190419051645-ef9e56ba1633 h1:Z8UpEZC24VCgbDEi8Urgi0vgqWkstn6q8/h8GMJ8BgI= -gopkg.in/spacemonkeygo/monkit-zipkin.v2 v2.0.0-20190419051645-ef9e56ba1633/go.mod h1:pEbh8z/a1Jj9ZHhVsVhLfCZJKE/DVv3twibVY47FWPs= -gopkg.in/spacemonkeygo/monkit.v2 v2.0.0-20180827161543-6ebf5a752f9b h1:zkhw+LuPvld5/ZwIlpYzd4DX/r0KxTvhEpo+2lcrp9k= -gopkg.in/spacemonkeygo/monkit.v2 v2.0.0-20180827161543-6ebf5a752f9b/go.mod h1:6UQdi0rNB4YDNwVYP8fJNmjLNaU8MA84WCdXStx3Spo= -gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= -storj.io/monkit-jaeger v0.0.0-20200305222653-48145d285891 h1:8kEmGLyH6HtU+f/Kt9sDLZ9kY1UK1i29YhFI+qr8/NI= -storj.io/monkit-jaeger v0.0.0-20200305222653-48145d285891/go.mod h1:a018CXJ7hJPKYsMDyrdxwsYe1cq8w6691nSwNK5J5Ls= diff --git a/internal/grpccompat/go.sum b/internal/grpccompat/go.sum index 3242714..ce73e99 100644 --- a/internal/grpccompat/go.sum +++ b/internal/grpccompat/go.sum @@ -1,10 +1,7 @@ cloud.google.com/go v0.26.0/go.mod h1:aQUYkXzVsufM+DwF1aE+0xfcU+56JwCaLick0ClmMTw= github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU= -github.com/apache/thrift v0.12.0/go.mod h1:cp2SuWMxlEZw2r+iP2GNCdIi4C1qmUzdZFSVb+bacwQ= github.com/census-instrumentation/opencensus-proto v0.2.1/go.mod h1:f6KPmirojxKA12rnyqOA5BBL4O983OfeGPqjHWSTneU= github.com/client9/misspell v0.3.4/go.mod h1:qj6jICC3Q7zFZvVWo7KLAzC3yx5G7kyvSDkc90ppPyw= -github.com/codahale/hdrhistogram v0.0.0-20161010025455-3a0bb77429bd/go.mod h1:sE/e/2PUdi/liOCUjSTXgM1o87ZssimdTWN964YiIeI= -github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= github.com/envoyproxy/go-control-plane v0.9.0/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4= github.com/envoyproxy/protoc-gen-validate v0.1.0/go.mod h1:iSmxcyjqTsJpI2R4NaDN7+kN2VEUnK/pcBlmesArF7c= github.com/gogo/protobuf v1.2.1 h1:/s5zKNz0uPFCZ5hddgPdo2TK2TVrUNMn0OOX8/aZMTE= @@ -18,17 +15,11 @@ github.com/golang/protobuf v1.3.2/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5y github.com/google/go-cmp v0.2.0/go.mod h1:oXzfMopK8JAjlY9xF4vHSVASa0yLyX7SntLO5aqRK0M= github.com/kisielk/errcheck v1.1.0/go.mod h1:EZBBE59ingxPouuu3KfxchcWSUPOHkagtvWXihfKN4Q= github.com/kisielk/gotool v1.0.0/go.mod h1:XhKaO+MFFWcvkIS/tQcRk01m1F5IRFswLeQ+oQHNcck= -github.com/opentracing/opentracing-go v1.1.0/go.mod h1:UkNAQd3GIcIGf0SeVgPpRdFStlNbqXla1AfSYxPUl2o= -github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0= -github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= github.com/prometheus/client_model v0.0.0-20190812154241-14fe0d1b01d4/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA= -github.com/spacemonkeygo/errors v0.0.0-20171212215202-9064522e9fd1/go.mod h1:7NL9UAYQnRM5iKHUCld3tf02fKb5Dft+41+VckASUy0= github.com/spacemonkeygo/monkit/v3 v3.0.4 h1:Ay+PZirv+qfd4sqcT+X/U3BnC7AcIaqp/IXh0oV36k8= github.com/spacemonkeygo/monkit/v3 v3.0.4/go.mod h1:JcK1pCbReQsOsMKF/POFSZCq7drXFybgGmbc27tuwes= github.com/spacemonkeygo/monotime v0.0.0-20180824235756-e3f48a95f98a h1:8+cCjxhToanKmxLIbuyBNe2EnpgwhiivsIaRJstDRFA= github.com/spacemonkeygo/monotime v0.0.0-20180824235756-e3f48a95f98a/go.mod h1:ul4bvvnCOPZgq8w0nTkSmWVg/hauVpFS97Am1YM1XXo= -github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= -github.com/stretchr/testify v1.5.1/go.mod h1:5W2xD1RspED5o8YsWQXVCued0rvSQ+mT+I5cxcmMvtA= github.com/zeebo/assert v1.1.0 h1:hU1L1vLTHsnO8x8c9KAR5GmM5QscxHg5RNU5z5qbUWY= github.com/zeebo/assert v1.1.0/go.mod h1:Pq9JiuJQpG8JLJdtkwrJESF0Foym2/D9XMU5ciN/wJ0= github.com/zeebo/errs v1.2.2 h1:5NFypMTuSdoySVTqlNs1dEoU21QVamMQJxW/Fii5O7g= @@ -45,8 +36,6 @@ golang.org/x/net v0.0.0-20190311183353-d8887717615a h1:oWX7TPOiFAMXLq8o0ikBYfCJV golang.org/x/net v0.0.0-20190311183353-d8887717615a/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg= golang.org/x/net v0.0.0-20190923162816-aa69164e4478 h1:l5EDrHhldLYb3ZRHDUhXF7Om7MvYXnkV9/iQNo1lX6g= golang.org/x/net v0.0.0-20190923162816-aa69164e4478/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s= -golang.org/x/net v0.0.0-20200226121028-0de0cce0169b h1:0mm1VjtFUOIlE1SbDlwjYaDxZVDP2S5ou6y0gSgXHu8= -golang.org/x/net v0.0.0-20200226121028-0de0cce0169b/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s= golang.org/x/oauth2 v0.0.0-20180821212333-d2e6202438be/go.mod h1:N/0e6XlmueqKjAGxoOufVs8QHGRruUQn6yWY3a++T0U= golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= golang.org/x/sync v0.0.0-20181108010431-42b317875d0f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= @@ -71,10 +60,5 @@ google.golang.org/grpc v1.19.0/go.mod h1:mqu4LbDTu4XGKhr4mRzUsmM4RtVoemTSY81AxZi google.golang.org/grpc v1.23.0/go.mod h1:Y5yQAOtifL1yxbo5wqy6BxZv8vAUGQwXBOALyacEbxg= google.golang.org/grpc v1.25.1 h1:wdKvqQk7IttEw92GoRyKG2IDrUIpgpj6H6m81yfeMW0= google.golang.org/grpc v1.25.1/go.mod h1:c3i+UQWmh7LiEpx4sFZnkU36qjEYZ0imhYfXVyQciAY= -gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= -gopkg.in/spacemonkeygo/monkit-zipkin.v2 v2.0.0-20190419051645-ef9e56ba1633/go.mod h1:pEbh8z/a1Jj9ZHhVsVhLfCZJKE/DVv3twibVY47FWPs= -gopkg.in/spacemonkeygo/monkit.v2 v2.0.0-20180827161543-6ebf5a752f9b/go.mod h1:6UQdi0rNB4YDNwVYP8fJNmjLNaU8MA84WCdXStx3Spo= -gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= honnef.co/go/tools v0.0.0-20190102054323-c2f93a96b099/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4= honnef.co/go/tools v0.0.0-20190523083050-ea95bdfd59fc/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4= -storj.io/monkit-jaeger v0.0.0-20200305222653-48145d285891/go.mod h1:a018CXJ7hJPKYsMDyrdxwsYe1cq8w6691nSwNK5J5Ls= diff --git a/internal/integration/common_test.go b/internal/integration/common_test.go index aca9e70..12c9c8d 100644 --- a/internal/integration/common_test.go +++ b/internal/integration/common_test.go @@ -9,7 +9,6 @@ import ( "net" "github.com/zeebo/errs" - "storj.io/drpc/drpcconn" "storj.io/drpc/drpcctx" "storj.io/drpc/drpcerr" @@ -29,11 +28,8 @@ func createConnection(server DRPCServiceServer) (DRPCServiceClient, func()) { c1, c2 := net.Pipe() mux := drpcmux.New() - traceHandler := handler{ - mu: mux, - } _ = DRPCRegisterService(mux, server) - srv := drpcserver.New(&traceHandler) + srv := drpcserver.New(mux) ctx.Run(func(ctx context.Context) { _ = srv.ServeOne(ctx, c1) }) conn := drpcconn.New(c2) diff --git a/internal/integration/simple_test.go b/internal/integration/simple_test.go index 6b9f92b..dd8175e 100644 --- a/internal/integration/simple_test.go +++ b/internal/integration/simple_test.go @@ -8,31 +8,28 @@ import ( "io" "testing" - "github.com/spacemonkeygo/monkit/v3" "github.com/zeebo/assert" "storj.io/drpc/drpcctx" "storj.io/drpc/drpcerr" ) -var mon = monkit.Package() - func TestSimple(t *testing.T) { - tracker := drpcctx.NewTracker(context.Background()) - defer tracker.Wait() - defer tracker.Cancel() + ctx := drpcctx.NewTracker(context.Background()) + defer ctx.Wait() + defer ctx.Cancel() cli, close := createConnection(standardImpl) defer close() { - out, err := cli.Method1(tracker, &In{In: 1}) + out, err := cli.Method1(ctx, &In{In: 1}) assert.NoError(t, err) assert.DeepEqual(t, out, &Out{Out: 1}) } { - stream, err := cli.Method2(tracker) + stream, err := cli.Method2(ctx) assert.NoError(t, err) assert.NoError(t, stream.Send(&In{In: 2})) assert.NoError(t, stream.Send(&In{In: 2})) @@ -42,7 +39,7 @@ func TestSimple(t *testing.T) { } { - stream, err := cli.Method3(tracker, &In{In: 3}) + stream, err := cli.Method3(ctx, &In{In: 3}) assert.NoError(t, err) for { out, err := stream.Recv() @@ -55,7 +52,7 @@ func TestSimple(t *testing.T) { } { - stream, err := cli.Method4(tracker) + stream, err := cli.Method4(ctx) assert.NoError(t, err) assert.NoError(t, stream.Send(&In{In: 4})) assert.NoError(t, stream.Send(&In{In: 4})) @@ -73,7 +70,7 @@ func TestSimple(t *testing.T) { } { - _, err := cli.Method1(tracker, &In{In: 5}) + _, err := cli.Method1(ctx, &In{In: 5}) assert.Error(t, err) assert.Equal(t, drpcerr.Code(err), 5) }