major change of websocket dbmHandler structure

dbm/webSocket.go

@@ -0,0 +1,91 @@
+package dbm
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/coder/websocket"
+	"github.com/coder/websocket/wsjson"
+	"github.com/gin-gonic/gin"
+	"github.com/zuadi/tecamino-dbm/models"
+)
+
+const (
+	OnCreate = "onCreate"
+	OnChange = "onChange"
+	OnDelete = "onDelete"
+)
+
+func (d *DBMHandler) WebSocket(c *gin.Context) {
+	// id, err := auth.GetIDFromAuth(c)
+	// if err != nil {
+	// 	d.Log.Error("dbmHandler.webSocket.Websocket", "error GetIDFromAuth: "+err.Error())
+	// 	return
+	// }
+	id := "test"
+
+	d.Log.Debug("dbmHandler.webSocket.Websocket", "authorization id token: "+id)
+
+	ctx, cancel := context.WithCancel(c.Request.Context())
+	defer cancel()
+	err := d.Clients.ConnectRecievingWsConnection(id, c)
+	defer d.Clients.RemoveClient(id)
+	if err != nil {
+		d.Log.Error("dbmHandler.webSocket.Websocket", "error connecting recieving websocket conection: "+err.Error())
+		return
+	}
+	defer d.Clients.DisconnectWsConnection(id, websocket.StatusInternalError, "Internal error")
+
+	//Read loop
+	for {
+		request, err := d.readJsonData(ctx, id)
+		if err != nil {
+			break
+		}
+
+		// Subscribe
+
+		if request.Subscribe != nil {
+			go func() {
+				for _, sub := range *request.Subscribe {
+					d.Subscribe(ctx, &sub, id)
+				}
+			}()
+		}
+
+		// Unsubscribe
+		if request.Unsubscribe != nil {
+			for _, unsub := range *request.Unsubscribe {
+				err = d.Unsubscribe(ctx, &unsub, id)
+				if err != nil {
+					d.Log.Error("dbmHandler.WebSocket unsubscribe", err)
+					break
+				}
+			}
+		}
+
+		if err != nil {
+			break
+		}
+	}
+}
+
+func (d *DBMHandler) readJsonData(ctx context.Context, id string) (*models.JsonData, error) {
+	var request models.JsonData
+	err := wsjson.Read(ctx, d.Clients[id].Conn, &request)
+	if err != nil {
+		code := websocket.CloseStatus(err)
+
+		switch code {
+		case websocket.StatusNormalClosure,
+			websocket.StatusGoingAway,
+			websocket.StatusNoStatusRcvd:
+			d.Log.Info("webSocket.readJsonData", fmt.Sprintf("WebSocket closed: %v (code: %v)\n", err, code))
+			return nil, err
+		default:
+			d.Log.Error("webSocket.readJsonData", fmt.Sprintf("WebSocket read error: %v (code: %v)\n", err, code))
+			return nil, err
+		}
+	}
+	return &request, nil
+}