luzifer/twitch-manager
1
0
Fork 0
mirror of https://github.com/Luzifer/twitch-manager.git synced 2024-11-12 19:12:41 +00:00
Code Issues Projects Releases Wiki Activity

Clarify comunication interface

Browse source
This commit is contained in:
Knut Ahlers 2020-11-23 09:54:02 +01:00
parent c902fbe230
commit 0c41f447f7
Signed by: luzifer
GPG key ID: 0066F03ED215AD7D
1 changed files with 16 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
api.go
View file

@ -16,20 +16,24 @@ import (
"github.com/gorilla/websocket"
)
const (
msgTypeStore string = "store"
)
const msgTypeStore string = "store"
var subscriptions = newSubscriptionStore()
type socketMessage struct {
Payload interface{} `json:"payload"`
Type string `json:"type"`
Version string `json:"version"`
}
type subcriptionStore struct {
socketSubscriptions map[string]func(interface{}) error
socketSubscriptions map[string]func(socketMessage) error
socketSubscriptionsLock *sync.RWMutex
}
func newSubscriptionStore() *subcriptionStore {
return &subcriptionStore{
socketSubscriptions: map[string]func(msg interface{}) error{},
socketSubscriptions: map[string]func(socketMessage) error{},
socketSubscriptionsLock: new(sync.RWMutex),
}
}
@ -47,7 +51,7 @@ func (s subcriptionStore) SendAllSockets(msgType string, msg interface{}) error
return nil
}
func (s *subcriptionStore) SubscribeSocket(id string, hdl func(interface{}) error) {
func (s *subcriptionStore) SubscribeSocket(id string, hdl func(socketMessage) error) {
s.socketSubscriptionsLock.Lock()
defer s.socketSubscriptionsLock.Unlock()
@ -61,7 +65,7 @@ func (s *subcriptionStore) UnsubscribeSocket(id string) {
delete(s.socketSubscriptions, id)
}
func compileSocketMessage(msgType string, msg interface{}) map[string]interface{} {
func compileSocketMessage(msgType string, msg interface{}) socketMessage {
assetVersionsLock.RLock()
defer assetVersionsLock.RUnlock()
@ -75,10 +79,10 @@ func compileSocketMessage(msgType string, msg interface{}) map[string]interface{
ver := fmt.Sprintf("%x", hash.Sum(nil))
return map[string]interface{}{
"payload": msg,
"type": msgType,
"version": ver,
return socketMessage{
Payload: msg,
Type: msgType,
Version: ver,
}
}
@ -106,7 +110,7 @@ func handleUpdateSocket(w http.ResponseWriter, r *http.Request) {
connLock = new(sync.Mutex)
id = uuid.Must(uuid.NewV4()).String()
)
subscriptions.SubscribeSocket(id, func(msg interface{}) error {
subscriptions.SubscribeSocket(id, func(msg socketMessage) error {
connLock.Lock()
defer connLock.Unlock()