Merge pull request #7325 from kobergj/SEEImprovements

Improve SSE Notifications
This commit is contained in:
kobergj
2023-09-22 12:45:11 +02:00
committed by GitHub
4 changed files with 39 additions and 19 deletions
+6
View File
@@ -0,0 +1,6 @@
package service
// FileReadyEvent is emitted when the postprocessing of a file is finished
type FileReadyEvent struct {
ItemID string `json:"itemid"`
}
+11 -14
View File
@@ -17,12 +17,6 @@ import (
"go.opentelemetry.io/otel/trace"
)
// ClientNotification is the event the clientlog service is sending to the client
type ClientNotification struct {
Type string
ItemID string
}
// ClientlogService is the service responsible for user activities
type ClientlogService struct {
log log.Logger
@@ -93,8 +87,9 @@ func (cl *ClientlogService) processEvent(event events.Event) {
}
var (
users []string
noti ClientNotification
users []string
evType string
data interface{}
)
switch e := event.Event.(type) {
default:
@@ -106,8 +101,10 @@ func (cl *ClientlogService) processEvent(event events.Event) {
return
}
noti.Type = "postprocessing-finished"
noti.ItemID = storagespace.FormatResourceID(*info.GetId())
evType = "postprocessing-finished"
data = FileReadyEvent{
ItemID: storagespace.FormatResourceID(*info.GetId()),
}
users, err = utils.GetSpaceMembers(ctx, info.GetSpace().GetId().GetOpaqueId(), gwc, utils.ViewerRole)
}
@@ -119,22 +116,22 @@ func (cl *ClientlogService) processEvent(event events.Event) {
// II) instruct sse service to send the information
for _, id := range users {
if err := cl.sendSSE(id, noti); err != nil {
if err := cl.sendSSE(id, evType, data); err != nil {
cl.log.Error().Err(err).Str("userID", id).Str("eventid", event.ID).Msg("failed to store event for user")
return
}
}
}
func (cl *ClientlogService) sendSSE(userid string, noti ClientNotification) error {
b, err := json.Marshal(noti)
func (cl *ClientlogService) sendSSE(userid string, evType string, data interface{}) error {
b, err := json.Marshal(data)
if err != nil {
return err
}
return events.Publish(context.Background(), cl.publisher, events.SendSSE{
UserID: userid,
Type: "clientlog-notification",
Type: evType,
Message: b,
})
}