代码拉取完成,页面将自动刷新
package binance
import (
"fmt"
"time"
"github.com/gorilla/websocket"
)
// WsHandler handle raw websocket message
type WsHandler func(message []byte)
// ErrHandler handles errors
type ErrHandler func(err error)
// WsConfig webservice configuration
type WsConfig struct {
Endpoint string
}
func newWsConfig(endpoint string) *WsConfig {
return &WsConfig{
Endpoint: endpoint,
}
}
var wsServe = func(cfg *WsConfig, handler WsHandler, errHandler ErrHandler) (doneC, stopC chan struct{}, err error) {
//fmt.Println(cfg.Endpoint)
c, _, err := websocket.DefaultDialer.Dial(cfg.Endpoint, nil)
if err != nil {
return nil, nil, err
}
doneC = make(chan struct{})
stopC = make(chan struct{})
go func() {
// This function will exit either on error from
// websocket.Conn.ReadMessage or when the stopC channel is
// closed by the client.
defer close(doneC)
if WebsocketKeepalive {
keepAlive(c, WebsocketTimeout)
}
// Wait for the stopC channel to be closed. We do that in a
// separate goroutine because ReadMessage is a blocking
// operation.
silent := false
go func() {
select {
case <-stopC:
silent = true
case <-doneC:
}
c.Close()
}()
for {
code, message, err := c.ReadMessage()
if err != nil {
if !silent {
fmt.Println("error from 1236")
errHandler(err)
}
if websocket.IsCloseError(err, code) {
c, _, err = websocket.DefaultDialer.Dial(cfg.Endpoint, nil)
errHandler(err)
}
continue
}
handler(message)
}
}()
return
}
func keepAlive(c *websocket.Conn, timeout time.Duration) {
ticker := time.NewTicker(timeout)
lastResponse := time.Now()
c.SetPongHandler(func(msg string) error {
lastResponse = time.Now()
return nil
})
go func() {
defer ticker.Stop()
for {
deadline := time.Now().Add(10 * time.Second)
err := c.WriteControl(websocket.PingMessage, []byte{}, deadline)
if err != nil {
return
}
<-ticker.C
if time.Since(lastResponse) > timeout {
c.Close()
return
}
}
}()
}
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。