waveterm/pkg/eventbus/eventbus.go

89 lines
1.9 KiB
Go
Raw Normal View History

// Copyright 2024, Command Line Inc.
// SPDX-License-Identifier: Apache-2.0
package eventbus
import (
"sync"
"github.com/wavetermdev/thenextwave/pkg/waveobj"
)
2024-06-12 02:42:10 +02:00
type WSEventType struct {
EventType string `json:"eventtype"`
ORef string `json:"oref,omitempty"`
Data any `json:"data"`
}
2024-06-14 08:54:04 +02:00
const (
FileOp_Append = "append"
)
type WSFileEventData struct {
ZoneId string `json:"zoneid"`
FileName string `json:"filename"`
FileOp string `json:"fileop"`
Data64 string `json:"data64"`
}
type WindowWatchData struct {
2024-06-12 02:42:10 +02:00
WindowWSCh chan any
WaveWindowId string
WatchedORefs map[waveobj.ORef]bool
}
var globalLock = &sync.Mutex{}
2024-06-12 02:42:10 +02:00
var wsMap = make(map[string]*WindowWatchData) // websocketid => WindowWatchData
2024-06-12 02:42:10 +02:00
func RegisterWSChannel(connId string, windowId string, ch chan any) {
globalLock.Lock()
defer globalLock.Unlock()
2024-06-12 02:42:10 +02:00
wsMap[connId] = &WindowWatchData{
WindowWSCh: ch,
WaveWindowId: windowId,
WatchedORefs: make(map[waveobj.ORef]bool),
}
}
2024-06-12 02:42:10 +02:00
func UnregisterWSChannel(connId string) {
globalLock.Lock()
defer globalLock.Unlock()
2024-06-12 02:42:10 +02:00
delete(wsMap, connId)
}
2024-06-12 02:42:10 +02:00
func getWindowWatchesForWindowId(windowId string) []*WindowWatchData {
globalLock.Lock()
defer globalLock.Unlock()
2024-06-12 02:42:10 +02:00
var watches []*WindowWatchData
for _, wdata := range wsMap {
if wdata.WaveWindowId == windowId {
watches = append(watches, wdata)
}
}
2024-06-12 02:42:10 +02:00
return watches
}
2024-06-12 02:42:10 +02:00
func getAllWatches() []*WindowWatchData {
globalLock.Lock()
defer globalLock.Unlock()
watches := make([]*WindowWatchData, 0, len(wsMap))
for _, wdata := range wsMap {
watches = append(watches, wdata)
}
2024-06-12 02:42:10 +02:00
return watches
}
2024-06-12 02:42:10 +02:00
func SendEventToWindow(windowId string, event WSEventType) {
wwdArr := getWindowWatchesForWindowId(windowId)
for _, wdata := range wwdArr {
wdata.WindowWSCh <- event
}
}
2024-06-12 02:42:10 +02:00
func SendEvent(event WSEventType) {
wwdArr := getAllWatches()
for _, wdata := range wwdArr {
wdata.WindowWSCh <- event
}
}