From 4f738345f9a19d7adac70aaf77ba2c166afa6eb1 Mon Sep 17 00:00:00 2001
From: Alexander Pardes <alexpardes@gmail.com>
Date: Sat, 12 Dec 2020 14:37:48 -0800
Subject: [PATCH] Add GetPort to Address

---
 address.go | 5 +++++
 1 file changed, 5 insertions(+)

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{}