65 lines
1.5 KiB
Go
65 lines
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"log"
|
|
"net/http"
|
|
|
|
"github.com/r3labs/sse"
|
|
)
|
|
|
|
type RealtimeListener struct {
|
|
Url string
|
|
Collections []string
|
|
Chan chan string
|
|
client *sse.Client
|
|
}
|
|
|
|
type Subscription struct {
|
|
ClientId string `json:"clientId"`
|
|
Subscriptions []string `json:"subscriptions"`
|
|
}
|
|
|
|
func (listener RealtimeListener) handlePbConnect(msg *sse.Event) {
|
|
var connectEvent ConnectEvent
|
|
err := json.Unmarshal(msg.Data, &connectEvent)
|
|
if err != nil {
|
|
log.Printf("Error unmarshalling PB_CONNECT event: %v\n", err)
|
|
return
|
|
}
|
|
log.Printf("Client ID: %v\n", connectEvent.ClientId)
|
|
|
|
subscription := Subscription{
|
|
ClientId: connectEvent.ClientId,
|
|
Subscriptions: []string{"workshop_link"},
|
|
}
|
|
|
|
body, err := json.Marshal(subscription)
|
|
if err != nil {
|
|
log.Printf("Error marshalling subscription body: %v\n", err)
|
|
return
|
|
}
|
|
|
|
resp, err := http.Post(POCKETBASE_REALTIME, "application/json", bytes.NewBuffer(body))
|
|
if err != nil {
|
|
log.Printf("Error posting subscription: %v\n", err)
|
|
return
|
|
}
|
|
defer resp.Body.Close()
|
|
}
|
|
|
|
func (listener RealtimeListener) handlePbEvent(msg *sse.Event) {
|
|
log.Println(msg)
|
|
// listener.Chan <- msg.Data
|
|
}
|
|
|
|
func (listener RealtimeListener) initialize() {
|
|
listener.client = sse.NewClient(listener.Url)
|
|
listener.client.Subscribe("PB_CONNECT", listener.handlePbConnect)
|
|
|
|
for _, collection := range listener.Collections {
|
|
listener.client.Subscribe(collection, listener.handlePbEvent)
|
|
}
|
|
}
|