Chore: Listener should not expose original net.Listener

This commit is contained in:
Dreamacro
2021-06-13 23:05:22 +08:00
parent 6091fcdfec
commit f231a63e93
7 changed files with 33 additions and 33 deletions

View File

@ -17,10 +17,10 @@ import (
)
type Listener struct {
net.Listener
address string
closed bool
cache *cache.Cache
listener net.Listener
address string
closed bool
cache *cache.Cache
}
func New(addr string, in chan<- C.ConnContext) (*Listener, error) {
@ -31,7 +31,7 @@ func New(addr string, in chan<- C.ConnContext) (*Listener, error) {
hl := &Listener{l, addr, false, cache.New(30 * time.Second)}
go func() {
for {
c, err := hl.Accept()
c, err := hl.listener.Accept()
if err != nil {
if hl.closed {
break
@ -47,7 +47,7 @@ func New(addr string, in chan<- C.ConnContext) (*Listener, error) {
func (l *Listener) Close() {
l.closed = true
l.Listener.Close()
l.listener.Close()
}
func (l *Listener) Address() string {