blockchain/internal/lib/websocket/connection.go

26 lines
391 B
Go

package websocket
import (
"github.com/lxzan/gws"
"net/url"
)
type Connection struct {
*gws.Conn
}
func NewWebSocketConnect(url url.URL, handler gws.Event) (*Connection, error) {
socket, _, err := gws.NewClient(handler, &gws.ClientOption{
Addr: url.String(),
})
if err != nil {
return nil, err
}
// 取得消息
go socket.ReadLoop()
return &Connection{
socket,
}, nil
}