2021-06-15 17:13:40 +08:00
|
|
|
package net
|
2020-05-12 11:29:53 +08:00
|
|
|
|
|
|
|
import (
|
|
|
|
"bufio"
|
|
|
|
"net"
|
|
|
|
)
|
|
|
|
|
|
|
|
type BufferedConn struct {
|
|
|
|
r *bufio.Reader
|
|
|
|
net.Conn
|
|
|
|
}
|
|
|
|
|
|
|
|
func NewBufferedConn(c net.Conn) *BufferedConn {
|
2021-09-28 23:15:53 +08:00
|
|
|
if bc, ok := c.(*BufferedConn); ok {
|
|
|
|
return bc
|
|
|
|
}
|
2020-05-12 11:29:53 +08:00
|
|
|
return &BufferedConn{bufio.NewReader(c), c}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Reader returns the internal bufio.Reader.
|
|
|
|
func (c *BufferedConn) Reader() *bufio.Reader {
|
|
|
|
return c.r
|
|
|
|
}
|
|
|
|
|
|
|
|
// Peek returns the next n bytes without advancing the reader.
|
|
|
|
func (c *BufferedConn) Peek(n int) ([]byte, error) {
|
|
|
|
return c.r.Peek(n)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *BufferedConn) Read(p []byte) (int, error) {
|
|
|
|
return c.r.Read(p)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *BufferedConn) ReadByte() (byte, error) {
|
|
|
|
return c.r.ReadByte()
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *BufferedConn) UnreadByte() error {
|
|
|
|
return c.r.UnreadByte()
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *BufferedConn) Buffered() int {
|
|
|
|
return c.r.Buffered()
|
|
|
|
}
|