Function Handler

websocket/websocket.go:156–183

Handler returns an Iris handler to be served in a route of an Iris application. Accepts the neffos websocket server as its first input argument and optionally an Iris-specific `IDGenerator` as its second one. This SHOULD be the last handler in the route's chain as it hijacks the connection and the

(s *neffos.Server, idGenerator ...IDGenerator)

//
// This SHOULD be the last handler in the route's chain as it hijacks the connection and the context.
func Handler(s *neffos.Server, idGenerator ...IDGenerator) context.Handler {
	idGen := DefaultIDGenerator
	if len(idGenerator) > 0 {
		idGen = idGenerator[0]
	}

	if cb := s.OnDisconnect; cb != nil {
		s.OnDisconnect = func(c *neffos.Conn) {
			cb(c)
			manualReleaseWithoutResp(GetContext(c))
		}
	} else {
		s.OnDisconnect = func(c *neffos.Conn) {
			manualReleaseWithoutResp(GetContext(c))
		}
	}

	return func(ctx *context.Context) {
		if ctx.IsStopped() {
			// let the framework release it as always;
			// socket was not created so disconnect event will not called and the
			// DisablePoolRelease was not even called yet.
			return
		}

		Upgrade(ctx, idGen, s)
	}
}

// Upgrade upgrades the request and returns a new websocket Conn.
// Use `Handler` for higher-level implementation instead.