38 lines
624 B
Go
38 lines
624 B
Go
|
package websocket
|
||
|
|
|
||
|
|
import (
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"github.com/gorilla/websocket"
|
||
|
|
)
|
||
|
|
|
||
|
|
type WebSocket struct {
|
||
|
|
id string
|
||
|
|
connection *websocket.Conn
|
||
|
|
}
|
||
|
|
|
||
|
|
func (socket *WebSocket) Address() *string {
|
||
|
|
remote := socket.connection.RemoteAddr()
|
||
|
|
address := strings.SplitN(remote.String(), ":", -1)
|
||
|
|
if len(address[0]) < 1 {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
return &address[0]
|
||
|
|
}
|
||
|
|
|
||
|
|
func (socket *WebSocket) Send(v interface{}) error {
|
||
|
|
if socket.connection == nil {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
return socket.connection.WriteJSON(v)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (socket *WebSocket) Destroy() error {
|
||
|
|
if socket.connection == nil {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
return socket.connection.Close()
|
||
|
|
}
|