From 05dfbe6f8abb5a3b5b41a51594ed47c042318c82 Mon Sep 17 00:00:00 2001 From: "arraykeys@gmail.com" Date: Thu, 29 Nov 2018 11:23:24 +0800 Subject: [PATCH] a --- utils/iolimiter/iolimiter.go | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/utils/iolimiter/iolimiter.go b/utils/iolimiter/iolimiter.go index 8bc478f..7c1bdfa 100644 --- a/utils/iolimiter/iolimiter.go +++ b/utils/iolimiter/iolimiter.go @@ -102,6 +102,19 @@ func (s *conn) Write(p []byte) (int, error) { } return n, err } +func (s *conn) Close() error { + if s.Conn != nil { + e := s.Conn.Close() + s.Conn = nil + s.r = nil + s.w = nil + s.readLimiter = nil + s.writeLimiter = nil + s.ctx = nil + return e + } + return nil +} // NewReader returns a reader that implements io.Reader with rate limiting. func NewReader(r io.Reader) *Reader {