mirror of
https://github.com/wyx2685/V2bX.git
synced 2026-02-04 04:30:08 +00:00
Merge pull request #120 from LinboLen/fix_limit_problem
fix: adjust rate limiting to use actual bytes transferred
This commit is contained in:
@@ -19,13 +19,19 @@ type Conn struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (c *Conn) Read(b []byte) (n int, err error) {
|
func (c *Conn) Read(b []byte) (n int, err error) {
|
||||||
c.limiter.Wait(int64(len(b)))
|
n, err = c.Conn.Read(b)
|
||||||
return c.Conn.Read(b)
|
if n > 0 {
|
||||||
|
c.limiter.Wait(int64(n))
|
||||||
|
}
|
||||||
|
return n, err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *Conn) Write(b []byte) (n int, err error) {
|
func (c *Conn) Write(b []byte) (n int, err error) {
|
||||||
c.limiter.Wait(int64(len(b)))
|
n, err = c.Conn.Write(b)
|
||||||
return c.Conn.Write(b)
|
if n > 0 {
|
||||||
|
c.limiter.Wait(int64(n))
|
||||||
|
}
|
||||||
|
return n, err
|
||||||
}
|
}
|
||||||
|
|
||||||
/*
|
/*
|
||||||
|
|||||||
Reference in New Issue
Block a user