drlq92444
drlq92444
2018-10-24 08:30

websocket设置协议和来源

已采纳

i use this lib: https://github.com/gorilla/websocket how to set protocol and origin,

code :

func InitWSSClient(url1, protocol, origin string)  {
    u := url.URL{Scheme: "ws", Host: url1}

    fmt.Println(url1,u.String())
    d :=websocket.DefaultDialer


    //c, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
    c, _, err :=d.Dial(u.String(),nil)
    var limit int64 =1024 * 1024 *32

    if err != nil {
        log.Fatal("dial:", err)
    }
    c.SetReadLimit(limit)

    go readWSMessage(c)
    go sendWSmessage(c)

}

func readWSMessage(ws *websocket.Conn) error {
    for {
        _,message,err := ws.ReadMessage()
        if err != nil {
            log15.Error("receiveWebsocket ", "error", err)
            panic("error")

        }
        assignmentWebsocket(string(message))
    }
}

func sendWSmessage(ws *websocket.Conn) error {
    for {

        select {
        case params := <-WebsocketParams:
            fmt.Println("ws send messages", params)
            err :=ws.WriteMessage(websocket.TextMessage,[]byte(params))
            if err != nil {
                log15.Error("receiveWebsocket ", "error", err)
                panic("error")

            }
        }

    }

}

x.net.websocket set these element like this :

websocket.Dial(url, protocol, origin)

how to achieve this function in gorilla/websocket/, ths you help. i try many solutions is not working. if you know any solutions,

found the solution:

d := websocket.DefaultDialer
d.Subprotocols = []string{protocol}
header :=http.Header{}
header.Set("origin",origin)
  • 点赞
  • 写回答
  • 关注问题
  • 收藏
  • 复制链接分享
  • 邀请回答

1条回答

  • dtypj3308 dtypj3308 3年前

    i found the solution, you can set like this

    u := url.URL{Scheme: "ws", Host: url1}
    d := websocket.DefaultDialer
    d.Subprotocols = []string{protocol}
    header := http.Header{}
    header.Set("origin", origin)
    c, _, err := d.Dial(u.String(), header)
    
    点赞 评论 复制链接分享

相关推荐