diff --git a/address.go b/address.go index ad78f02..228d0fd 100644 --- a/address.go +++ b/address.go @@ -14,6 +14,7 @@ type Address interface { SetPort(port uint16) String() string + GetPort() uint16 } type enetAddress struct { @@ -49,6 +50,10 @@ func (addr *enetAddress) String() string { return ret } +func (addr *enetAddress) GetPort() uint16 { + return uint16(addr.cAddr.port) +} + // NewAddress creates a new address func NewAddress(ip string, port uint16) Address { ret := enetAddress{}