26 lines
391 B
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
|
||
|
|
}
|