Function Upgrade

websocket/websocket.go:187–204

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

(ctx *context.Context, idGen IDGenerator, s *neffos.Server)

// Upgrade upgrades the request and returns a new websocket Conn.
// Use `Handler` for higher-level implementation instead.
func Upgrade(ctx *context.Context, idGen IDGenerator, s *neffos.Server) *neffos.Conn {
	/* Do NOT return the error as it is captured on the OnUpgradeError listener,
	the end-developer should not be able to write to this http client directly. */
	ctx.DisablePoolRelease()

	conn, upgradeErr := s.Upgrade(ctx.ResponseWriter(), ctx.Request(), func(socket neffos.Socket) neffos.Socket {
		return &socketWrapper{
			Socket: socket,
			ctx:    ctx,
		}
	}, wrapIDGenerator(idGen)(ctx))

	if upgradeErr != nil {
		manualReleaseWithoutResp(ctx)
	}

	return conn
}

func manualReleaseWithoutResp(ctx *context.Context) {
	ctx.ResponseWriter().EndResponse()                   // relases the response writer (common, recorder & compress).