Explorar o código

fix lint warnings

Darien Raymond %!s(int64=8) %!d(string=hai) anos
pai
achega
e8e7921613
Modificáronse 2 ficheiros con 6 adicións e 7 borrados
  1. 3 5
      transport/internet/kcp/connection.go
  2. 3 2
      transport/internet/kcp/segment.go

+ 3 - 5
transport/internet/kcp/connection.go

@@ -198,8 +198,6 @@ type Connection struct {
 
 	dataUpdater *Updater
 	pingUpdater *Updater
-
-	mergingWriter buf.Writer
 }
 
 // NewConnection create a new KCP connection between local and remote.
@@ -285,7 +283,7 @@ func (v *Connection) ReadMultiBuffer() (buf.MultiBuffer, error) {
 
 		duration := time.Minute
 		if !v.rd.IsZero() {
-			duration = v.rd.Sub(time.Now())
+			duration = time.Until(v.rd)
 			if duration < 0 {
 				return nil, ErrIOTimeout
 			}
@@ -322,7 +320,7 @@ func (v *Connection) Read(b []byte) (int, error) {
 
 		duration := time.Minute
 		if !v.rd.IsZero() {
-			duration = v.rd.Sub(time.Now())
+			duration = time.Until(v.rd)
 			if duration < 0 {
 				return 0, ErrIOTimeout
 			}
@@ -358,7 +356,7 @@ func (v *Connection) Write(b []byte) (int, error) {
 
 		duration := time.Minute
 		if !v.wd.IsZero() {
-			duration = v.wd.Sub(time.Now())
+			duration = time.Until(v.wd)
 			if duration < 0 {
 				return totalWritten, ErrIOTimeout
 			}

+ 3 - 2
transport/internet/kcp/segment.go

@@ -1,6 +1,7 @@
 package kcp
 
 import (
+	"v2ray.com/core/common"
 	"v2ray.com/core/common/buf"
 	"v2ray.com/core/common/serial"
 )
@@ -65,9 +66,9 @@ func (v *DataSegment) SetData(data []byte) {
 	if v.Data == nil {
 		v.Data = buf.New()
 	}
-	v.Data.Reset(func(b []byte) (int, error) {
+	common.Must(v.Data.Reset(func(b []byte) (int, error) {
 		return copy(b, data), nil
-	})
+	}))
 }
 
 func (v *DataSegment) Bytes() buf.Supplier {