implement new json_data model

client/client.go

@@ -0,0 +1,66 @@
+package client
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/coder/websocket"
+	"github.com/coder/websocket/wsjson"
+	json_data "github.com/tecamino/tecamino-json_data"
+	json_dataModels "github.com/tecamino/tecamino-json_data/models"
+	"github.com/tecamino/tecamino-logger/logging"
+)
+
+type Client struct {
+	conn   *websocket.Conn
+	ctx    context.Context
+	cancel context.CancelFunc
+	log    *logging.Logger
+}
+
+// Initialize new websocket server
+func NewClient(log *logging.Logger) *Client {
+	c := Client{
+		log: log,
+	}
+	c.ctx, c.cancel = context.WithCancel(context.Background())
+	return &c
+}
+
+func (c *Client) Connect(ip, id string, port uint) (err error) {
+	c.conn, _, err = websocket.Dial(c.ctx, fmt.Sprintf("ws://%s:%d/ws?id=%s", ip, port, id), nil)
+	return
+}
+
+func (c *Client) Disconnect() {
+	c.cancel()
+}
+func (c *Client) Subscribe(id string) error {
+
+	req := json_data.NewRequest()
+	req.AddDriverSubscription(".*", id, 0, true, false, false)
+	if err := wsjson.Write(c.ctx, c.conn, req); err != nil {
+		return err
+	}
+	return nil
+}
+
+func (c *Client) ReadJsonData() (response json_dataModels.Response, err error) {
+
+	err = wsjson.Read(c.ctx, c.conn, &response)
+	if err != nil {
+		code := websocket.CloseStatus(err)
+
+		switch code {
+		case websocket.StatusNormalClosure,
+			websocket.StatusGoingAway,
+			websocket.StatusNoStatusRcvd:
+			c.log.Info("webSocket.readJsonData", fmt.Sprintf("WebSocket closed: %v (code: %v)\n", err, code))
+			return
+		default:
+			c.log.Error("webSocket.readJsonData", fmt.Sprintf("WebSocket read error: %v (code: %v)\n", err, code))
+			return
+		}
+	}
+	return
+}