LumeWeb
/
libs5-go
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: implement GetIP

This commit is contained in:
Derrick Hammer 2024-01-15 11:15:11 -05:00
parent e7026459b4
commit fc10a265a7
Signed by: pcfreak30
GPG Key ID: C997C339BE476FF2
2 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
net/peer.go
View File

@ -81,7 +81,6 @@ func (b *BasePeer) End() error {
panic("must implement in child class") panic("must implement in child class")
} }
func (b *BasePeer) GetIP() string { func (b *BasePeer) GetIP() string {
//TODO implement me
panic("must implement in child class") panic("must implement in child class")
} }

11
net/ws.go
View File

@ -115,3 +115,14 @@ func (p *WebSocketPeer) SetChallenge(challenge []byte) {
func (p *WebSocketPeer) GetChallenge() []byte { func (p *WebSocketPeer) GetChallenge() []byte {
return p.challenge return p.challenge
} }
func (b *WebSocketPeer) GetIP() string {
ctx, cancel := context.WithCancel(context.Background())
netConn := websocket.NetConn(ctx, b.socket, websocket.MessageBinary)
ipAddr := netConn.RemoteAddr().String()
cancel()
return ipAddr
}