Gin compatabilty

[nmec]

I'm trying to integrate this into my existing app which is using the Gin framework, but I'm unable to establish a connection from the browser. Any ideas how I can get this working?

Server

package main

import (
	"context"
	"log"
	"net/http"
	"time"

	"github.com/gin-gonic/gin"
	"nhooyr.io/websocket"
	"nhooyr.io/websocket/wsjson"
)

func main() {
	router := gin.New()

	router.GET("/ws", handler)

	router.Run(":7000")
}

func handler(c *gin.Context) {
	log.Println("upgrading")
	conn, wsErr := websocket.Accept(c.Writer, c.Request, &websocket.AcceptOptions{
		InsecureSkipVerify: true,
	})

	if wsErr != nil {
		c.AbortWithError(http.StatusInternalServerError, wsErr)
		return
	}

	defer conn.Close(websocket.StatusInternalError, "Closed unexepetedly")

	log.Println("ready")

	ctx, cancel := context.WithTimeout(c.Request.Context(), time.Minute)
	defer cancel()
	ctx = conn.CloseRead(ctx)

	tick := time.NewTicker(time.Second * 5)
	defer tick.Stop()

	for {
		select {
		case <-ctx.Done():
			log.Println("done")
			conn.Close(websocket.StatusNormalClosure, "")
			return
		case <-tick.C:
			writeErr := wsjson.Write(ctx, conn, map[string]interface{}{
				"msg": "hello",
			})
			if writeErr != nil {
				log.Println(writeErr)
				return
			}
		}
	}
}

Client

const ws = new WebSocket('ws://localhost:7000/ws');
ws.onopen = console.log;
ws.onmessage = console.log;

JS Error
WebSocket connection to 'ws://localhost:7000/ws' failed: Error during WebSocket handshake: net::ERR_INVALID_HTTP_RESPONSE

Server logs

[GIN-debug] [WARNING] Running in "debug" mode. Switch to "release" mode in production.
 - using env:   export GIN_MODE=release
 - using code:  gin.SetMode(gin.ReleaseMode)

[GIN-debug] GET    /ws                       --> main.handler (1 handlers)
[GIN-debug] Listening and serving HTTP on :7000
2019/10/25 15:17:49 upgrading
2019/10/25 15:17:49 ready
2019/10/25 15:17:54 done

[nhooyr]

What does the Go program log?

[nhooyr]

Also, print wsErr.

[nmec]

Thanks for the speedy reply - I've updated the description with the server logs.

[nhooyr]

Mhmm. I believe the issue is that gin doesn't allow writing a response before hijacking the connection the way net/http does. Gorilla works because it hijacks first, then writes the handshake response directly to the tcp connection. We could do this as well but it'd be janky.

What do you get in the network tab of the browser? What does the HTTP response look like?

[nmec]

HTTP Request:

HTTP Response:

[nhooyr]

Got it, what does the headers tab look like for the response?

[nmec]

No response headers are shown, only the request.

[nhooyr]

Cool, lines up with my reasoning.

Looks like gin deviates from net/http for some reason. Have to call an additional WriteHeaderNow method to write the header.

https://github.com/gin-gonic/gin/blob/094f9a9105f8e7b971a01a64677eef5a1f7bcd9b/response_writer.go#L69

Opened gin-gonic/gin#2108

Will fix tonight.

[nmec]

Amazing, thanks so much for the quick responses.

[nmec]

Not sure if this is of use to you to fix the issue, but I just tried gobwas/ws for the heck of it and it works perfectly:

package main

import (
	"encoding/json"
	"log"
	"time"

	"github.com/gin-gonic/gin"
	"github.com/gobwas/ws"
	"github.com/gobwas/ws/wsutil"
)

func main() {
	router := gin.Default()

	router.GET("/ws", handler)

	router.Run(":7000")
}

func handler(c *gin.Context) {
	log.Println("upgrading")
	conn, _, _, err := ws.UpgradeHTTP(c.Request, c.Writer)
	if err != nil {
		log.Println(err)
	}

	log.Println("ready")

	go func() {
		tick := time.NewTicker(time.Second * 5)
		defer tick.Stop()
		defer conn.Close()

		for {
			select {
			case <-tick.C:
				log.Println("tick")
				data, _ := json.Marshal(map[string]interface{}{
					"msg": "hello",
				})

				err = wsutil.WriteServerMessage(conn, ws.OpText, data)
				if err != nil {
					log.Println(err)
				}

			}
		}
	}()
}

[nhooyr]

If you add w.(interface{ WriteHeaderNow() }.WriteHeaderNow()) on this line https://github.com/nhooyr/websocket/blob/8bf88772e1209d96a00bc38276cda458661ebe2b/handshake.go#L143 things should work. Could you test it?

[nhooyr]

gohack may be helpful. https://github.com/rogpeppe/gohack