websocket.go

package main

import (
	"encoding/json"
	"time"

	ws "github.com/dictor/wswrapper"
	"github.com/sirupsen/logrus"
)

type (
	generalMessage struct {
		Type string      `json:"type"`
		Data interface{} `json:"data"`
	}
)

func wsEvent(evt *ws.WebsocketEvent) {
	switch evt.Kind {
	case ws.EVENT_RECIEVE:
		//when some message received.
	case ws.EVENT_REGISTER:
		globalLogger.WithFields(logrus.Fields{
			"id":             evt.Client.Id(),
			"remote_address": evt.Client.Connection().RemoteAddr().String(),
			"local_address":  evt.Client.Connection().LocalAddr().String(),
		}).Info("websocker client registered")
	case ws.EVENT_UNREGISTER:
		globalLogger.WithFields(logrus.Fields{
			"id":             evt.Client.Id(),
			"remote_address": evt.Client.Connection().RemoteAddr().String(),
			"local_address":  evt.Client.Connection().LocalAddr().String(),
		}).Info("websocker client unregistered")
	case ws.EVENT_ERROR:
		globalLogger.WithField("error", evt.Err).Errorln("websocket error")
	}
}

func broadcastData(h *ws.WebsocketHub, sendQueue chan []byte, interval int) {
	defer ValueMutex.Unlock()
	for {
		ValueMutex.Lock()
		data, err := json.Marshal(generalMessage{
			Type: "value",
			Data: Value,
		})
		ValueMutex.Unlock()
		if err != nil {
			globalLogger.WithField("error", err).Errorln("error caused while value broadcast")
		}
		h.SendAll(data)

	emptyLoop:
		for {
			select {
			case q := <-sendQueue:
				h.SendAll(q)
			default:
				break emptyLoop
			}
		}

		time.Sleep(time.Millisecond * time.Duration(interval))
	}
}