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
}