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

@ -12,9 +12,9 @@ import (
)
type Listener struct {
net.Listener
address string
closed bool
listener net.Listener
address string
closed bool
}
func New(addr string, in chan<- C.ConnContext) (*Listener, error) {
@ -42,7 +42,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 {

View File

@ -12,9 +12,9 @@ import (
)
type UDPListener struct {
net.PacketConn
address string
closed bool
packetConn net.PacketConn
address string
closed bool
}
func NewUDP(addr string, in chan<- *inbound.PacketAdapter) (*UDPListener, error) {
@ -48,7 +48,7 @@ func NewUDP(addr string, in chan<- *inbound.PacketAdapter) (*UDPListener, error)
func (l *UDPListener) Close() error {
l.closed = true
return l.PacketConn.Close()
return l.packetConn.Close()
}
func (l *UDPListener) Address() string {