From aaf226f3d19880a61e35a802ad3ba1b9360e3641 Mon Sep 17 00:00:00 2001 From: parkerzhu Date: Tue, 16 Jul 2019 01:50:40 +0800 Subject: [PATCH] add local addr and remote addr function to session --- session.go | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/session.go b/session.go index cf1a8c4..13b0884 100644 --- a/session.go +++ b/session.go @@ -1,6 +1,7 @@ package melody import ( + "net" "net/http" "sync" "time" @@ -220,11 +221,11 @@ func (s *Session) IsClosed() bool { } // LocalAddr returns the local addr of the connection. -func (s *session) LocalAddr() net.Addr { - return conn.LocalAddr() +func (s *Session) LocalAddr() net.Addr { + return s.conn.LocalAddr() } // RemoteAddr returns the remote addr of the connection. -func (s *session) RemoteAddr() net.Addr { - return conn.RemoteAddr() +func (s *Session) RemoteAddr() net.Addr { + return s.conn.RemoteAddr() }