go-packages/iris
1
0
Fork 0
mirror of https://github.com/kataras/iris.git synced 2025-01-23 10:41:03 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Websocket: added OnPong to Connection #986

Browse Source 
see issue #986


Former-commit-id: 964d393d40fc7e64a8bf02eae08d97fd5f847483
This commit is contained in:
Kirguir 2018-04-30 14:27:28 +03:00
parent 4eeffd07c7
commit 655a19ea74
1 changed files with 19 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
websocket/connection.go
View File

@ -133,6 +133,8 @@ type (
MessageFunc interface{} MessageFunc interface{}
// PingFunc is the callback which fires each ping // PingFunc is the callback which fires each ping
PingFunc func() PingFunc func()
// PongFunc is the callback which fires on pong message received
PongFunc func()
// Connection is the front-end API that you will use to communicate with the client side // Connection is the front-end API that you will use to communicate with the client side
Connection interface { Connection interface {
// Emitter implements EmitMessage & Emit // Emitter implements EmitMessage & Emit
@ -165,6 +167,8 @@ type (
OnError(ErrorFunc) OnError(ErrorFunc)
// OnPing registers a callback which fires on each ping // OnPing registers a callback which fires on each ping
OnPing(PingFunc) OnPing(PingFunc)
// OnPong registers a callback which fires on pong message received
OnPong(PongFunc)
// FireOnError can be used to send a custom error message to the connection // FireOnError can be used to send a custom error message to the connection
// //
// It does nothing more than firing the OnError listeners. It doesn't send anything to the client. // It does nothing more than firing the OnError listeners. It doesn't send anything to the client.
@ -221,6 +225,7 @@ type (
onRoomLeaveListeners []LeaveRoomFunc onRoomLeaveListeners []LeaveRoomFunc
onErrorListeners []ErrorFunc onErrorListeners []ErrorFunc
onPingListeners []PingFunc onPingListeners []PingFunc
onPongListeners []PongFunc
onNativeMessageListeners []NativeMessageFunc onNativeMessageListeners []NativeMessageFunc
onEventListeners map[string][]MessageFunc onEventListeners map[string][]MessageFunc
started bool started bool
@ -256,6 +261,7 @@ func newConnection(ctx context.Context, s *Server, underlineConn UnderlineConnec
onErrorListeners: make([]ErrorFunc, 0), onErrorListeners: make([]ErrorFunc, 0),
onNativeMessageListeners: make([]NativeMessageFunc, 0), onNativeMessageListeners: make([]NativeMessageFunc, 0),
onEventListeners: make(map[string][]MessageFunc, 0), onEventListeners: make(map[string][]MessageFunc, 0),
onPongListeners: make([]PongFunc, 0),
started: false, started: false,
ctx: ctx, ctx: ctx,
server: s, server: s,
@ -354,6 +360,13 @@ func (c *connection) fireOnPing() {
} }
} }
func (c *connection) fireOnPong() {
// fire the onPongListeners
for i := range c.onPongListeners {
c.onPongListeners[i]()
}
}
func (c *connection) startReader() { func (c *connection) startReader() {
conn := c.underline conn := c.underline
hasReadTimeout := c.server.config.ReadTimeout > 0 hasReadTimeout := c.server.config.ReadTimeout > 0
@ -363,6 +376,8 @@ func (c *connection) startReader() {
if hasReadTimeout { if hasReadTimeout {
conn.SetReadDeadline(time.Now().Add(c.server.config.ReadTimeout)) conn.SetReadDeadline(time.Now().Add(c.server.config.ReadTimeout))
} }
//fire all OnPong methods
go c.fireOnPong()
return nil return nil
}) })
@ -473,6 +488,10 @@ func (c *connection) OnPing(cb PingFunc) {
c.onPingListeners = append(c.onPingListeners, cb) c.onPingListeners = append(c.onPingListeners, cb)
} }
func (c *connection) OnPong(cb PongFunc) {
c.onPongListeners = append(c.onPongListeners, cb)
}
func (c *connection) FireOnError(errorMessage string) { func (c *connection) FireOnError(errorMessage string) {
for _, cb := range c.onErrorListeners { for _, cb := range c.onErrorListeners {
cb(errorMessage) cb(errorMessage)