Gorilla-Websocket

[diiimo1973]

I would like to use the gorilla websocket.
I need the response writer. Where can I find the context?
Here an example of gin.

func WsConnectHandler(c *gin.Context) {
    if err := wsHandler(c.Writer, c.Request); err != nil {
        …..
    }
}

func wsHandler(w http.ResponseWriter, r *http.Request) error {
    ……
}

[vishr]

@diiimo2k11 I just tried example below, hope it helps.

package main

import (
    "log"

    "github.com/gorilla/websocket"
    "github.com/labstack/echo"
)

var (
    upgrader = websocket.Upgrader{}
)

func main() {
    e := echo.New()
    e.Get("/ws", func(c *echo.Context) error {
        conn, err := upgrader.Upgrade(c.Response(), c.Request(), nil)
        if err != nil {
            return err
        }
        defer conn.Close()
        for {
            mt, message, err := conn.ReadMessage()
            if err != nil {
                return err
            }
            log.Printf("recv: %s", message)
            err = conn.WriteMessage(mt, message)
            if err != nil {
                return err
            }
        }
    })
    e.Run(":1323")
}

[diiimo1973]

That works for me, thanks...
cors for websocket with CheckOrigin:

package main

import (
    "log"

    "github.com/gorilla/websocket"
    "github.com/labstack/echo"
)

var (
    // Websocket http upgrader
    upgrader = websocket.Upgrader{
        ReadBufferSize:  1024,
        WriteBufferSize: 1024,
        CheckOrigin: func(r *http.Request) bool {
            return true
        },
    }
)

func main() {
    e := echo.New()
    e.Get("/ws", func(c *echo.Context) error {
        conn, err := upgrader.Upgrade(c.Response(), c.Request(), nil)
        if err != nil {
            return err
        }
        defer conn.Close()
        for {
            mt, message, err := conn.ReadMessage()
            if err != nil {
                return err
            }
            log.Printf("recv: %s", message)
            err = conn.WriteMessage(mt, message)
            if err != nil {
                return err
            }
        }
    })
    e.Run(":1323")
}

[vishr]

Closing to mention in #269

[diiimo1973]

@vishr
Is still a bug with CheckOrigin.
When one submits a close event by the client, the following error:
2015/11/24 20:58:36 http: response.WriteHeader on hijacked connection
2015/11/24 20:58:36 http: response.Write on hijacked connection
ERROR|echo|websocket: close 1005
INFO|echo|::1 GET /ws 500 1.474592306s 0

[diiimo1973]

The problem is: return err
The error handler does not understand the error.
Better solution that works:

package main

import (
    "log"

    "net/http"
    "os"
    "strings"

    "github.com/gorilla/websocket"
    "github.com/labstack/echo"
)

var (
    // Websocket http upgrader
    upgrader = websocket.Upgrader{
        ReadBufferSize:  1024,
        WriteBufferSize: 1024,
        CheckOrigin: func(r *http.Request) bool {
            return true
        },
    }
)

func main() {

    log.SetOutput(os.Stdout)

    e := echo.New()
    e.Get("/ws", func(c *echo.Context) error {
        conn, err := upgrader.Upgrade(c.Response(), c.Request(), nil)
        if err != nil {
            log.Printf("error upgrade: %s\n", err.Error())
            return nil
        }

        log.Printf("websocket connect: %s\n", c.Request().RemoteAddr)

        defer func() {
            log.Printf("websocket close connection: %s\n", c.Request().RemoteAddr)
            conn.Close()
        }()
        for {
            mt, message, err := conn.ReadMessage()
            if err != nil {
                if strings.Contains(err.Error(), "close 1005") {
                    // Receive close event
                    log.Printf("receive close message from: %s\n", c.Request().RemoteAddr)
                } else {
                    log.Printf("error read: %s\n", err.Error())
                }

                break
            }
            log.Printf("recv: %s", message)
            err = conn.WriteMessage(mt, message)
            if err != nil {
                log.Printf("error write: %s\n", err.Error())
                break
            }
        }
        return nil
    })
    e.Run(":1323")
}

[vishr]

Cool, closing now.

[let4be]

How to use this with v2(beta)? I'm having a trouble calling upgrader.Upgrade method

[vishr]

@let4be I have updated WebSocket recipe https://echo.labstack.com/recipes/websocket, hope it helps.

[let4be]

@vishr thanks! Project is amazing