Update write websocket func

cmd/api/ws.go

@@ -64,22 +64,14 @@ func (ctx *Context) Close() {
 
 func (ctx *Context) Write(msg interface{}) {
 	ctx.mu.Lock()
-	defer ctx.mu.Unlock()
 
-	var err error
+	if data, ok := msg.([]byte); ok {
-
+		_ = ctx.Conn.WriteMessage(websocket.BinaryMessage, data)
-	switch msg := msg.(type) {
+	} else {
-	case *streamer.Message:
+		_ = ctx.Conn.WriteJSON(msg)
-		err = ctx.Conn.WriteJSON(msg)
-	case []byte:
-		err = ctx.Conn.WriteMessage(websocket.BinaryMessage, msg)
-	default:
-		return
 	}
 
-	if err != nil {
+	ctx.mu.Unlock()
-		//panic(err) // TODO: fix panic
-	}
 }
 
 func (ctx *Context) Error(err error) {