diff --git a/models/client.go b/models/client.go index 23d9c77..7cacfb4 100644 --- a/models/client.go +++ b/models/client.go @@ -57,8 +57,11 @@ var upgrader = websocket.Upgrader{ EnableCompression: false, } -func NewClient(ip, id string, port uint, logger *logging.Logger) (*Client, error) { - u := url.URL{Scheme: "ws", Host: fmt.Sprintf("%s:%d", ip, port), Path: "status", RawQuery: "id=" + id} +func NewClient(id, ip, endpoint string, port uint, logger *logging.Logger) (*Client, error) { + if endpoint == "" { + endpoint = "status" + } + u := url.URL{Scheme: "ws", Host: fmt.Sprintf("%s:%d", ip, port), Path: endpoint, RawQuery: "id=" + id} c := &Client{ id: id, diff --git a/statusClient.go b/statusClient.go index 9053773..a862dd4 100644 --- a/statusClient.go +++ b/statusClient.go @@ -43,7 +43,7 @@ func NewStatusClient(serviceName, ip string, port uint, debug bool) (*StatusClie } // connect underlying websocket client - sc.client, err = models.NewClient(ip, serviceName, port, logger) + sc.client, err = models.NewClient(serviceName, ip, "", port, logger) if err != nil { return nil, err }