Upgrade context?

[vinhjaxt]

Hi,
First of all, thank you for great repo.
In my case, I want to pass some data to the ws handler like this:

var wsUpgrader = &fastws.Upgrader{
	Compress: true,
	Handler: func(c *fastws.Conn, yourCtx interface{}) {
		var (
			bf  []byte
			err error
		)
		myCtx := yourCtx.(*MyContext)

		c.ReadTimeout = 0
		c.WriteTimeout = 0

		for {
			_, bf, err = c.ReadMessage(bf[:0])
			if err != nil {
				/*
					if err != fastws.EOF {
						log.Println("error reading message:", err)
					}
					// */
				break
			}
			log.Println("UID send data:", myCtx.UID, string(bf))
		}
	},
	UpgradeHandler: func(ctx *fasthttp.RequestCtx, yourCtx interface{}) bool {
		myCtx := yourCtx.(*MyContext)
		myCtx.UID = "456"
		return true
	},
}

func requestHandler(ctx *fasthttp.RequestCtx) {
	myCtx := &MyContext{
		UID: "213",
	}
	wsUpgrader.Upgrade(ctx, myCtx)
}

[dgrr]

That's interesting. I can use something like fasthttp does with ctx.UserValue widely used by routers and other middlewares. Would you like me to add that?
I can't change the Handler method but I can add the UserValues to conn. Also, compression is not working yet, I will try to work on that this week.

[vinhjaxt]

Thank you for the quick response.
I think adding UserValues ​​to Conn will do the trick.

[dgrr]

Ok. I will add that for now. I will try to do other workaround later on

[dgrr]

You can do the following:

upgr := Upgrader{
	Origin: uri,
	Handler: func(conn *Conn) {
		v := conn.UserValue("custom")
		if v == nil {
			t.Fatal("custom is nil")
		}
		conn.WriteString(v.(string))
	},
}
s := fasthttp.Server{
	Handler: func(ctx *fasthttp.RequestCtx) {
		ctx.SetUserValue("custom", text)
		upgr.Upgrade(ctx)
	},
}

All the values will be copied from the fasthttp's RequestCtx

[dgrr]

https://github.com/dgrr/fastws/blob/master/examples/pass_value.go

[vinhjaxt]

Good work!!
Thank you!!