34 lines
529 B
Go
34 lines
529 B
Go
|
package endpoint
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net"
|
||
|
|
|
||
|
|
"go.uber.org/multierr"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Uplink struct {
|
||
|
|
Proto NetProto
|
||
|
|
Listener net.Listener
|
||
|
|
PacketConn net.PacketConn
|
||
|
|
}
|
||
|
|
|
||
|
|
func (u Uplink) Addr() net.Addr {
|
||
|
|
if u.Listener != nil {
|
||
|
|
return u.Listener.Addr()
|
||
|
|
}
|
||
|
|
if u.PacketConn != nil {
|
||
|
|
return u.PacketConn.LocalAddr()
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (u Uplink) Close() (err error) {
|
||
|
|
if u.Listener != nil {
|
||
|
|
err = multierr.Append(err, u.Listener.Close())
|
||
|
|
}
|
||
|
|
if u.PacketConn != nil {
|
||
|
|
err = multierr.Append(err, u.PacketConn.Close())
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|