summaryrefslogtreecommitdiff
path: root/internal/processing/streaming/streamnotification.go
diff options
context:
space:
mode:
authorLibravatar Tobi Smethurst <31960611+tsmethurst@users.noreply.github.com>2021-07-05 13:23:03 +0200
committerLibravatar GitHub <noreply@github.com>2021-07-05 13:23:03 +0200
commitd389e7b150df6ecd215c7b661b294ea153ad0103 (patch)
tree8739e3103cb5130875d903cc7fc72fd9db3b8434 /internal/processing/streaming/streamnotification.go
parentFix 404 contact (#74) (diff)
downloadgotosocial-d389e7b150df6ecd215c7b661b294ea153ad0103.tar.xz
Domain block (#76)
* start work on admin domain blocking * move stuff around + further work on domain blocks * move + restructure processor * prep work for deleting account * tidy * go fmt * formatting * domain blocking more work * check domain blocks way earlier on * progress on delete account * delete more stuff when an account is gone * and more... * domain blocky block block * get individual domain block, delete a block
Diffstat (limited to 'internal/processing/streaming/streamnotification.go')
-rw-r--r--internal/processing/streaming/streamnotification.go50
1 files changed, 50 insertions, 0 deletions
diff --git a/internal/processing/streaming/streamnotification.go b/internal/processing/streaming/streamnotification.go
new file mode 100644
index 000000000..24c8342ee
--- /dev/null
+++ b/internal/processing/streaming/streamnotification.go
@@ -0,0 +1,50 @@
+package streaming
+
+import (
+ "encoding/json"
+ "errors"
+ "fmt"
+
+ "github.com/sirupsen/logrus"
+ apimodel "github.com/superseriousbusiness/gotosocial/internal/api/model"
+ "github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
+)
+
+func (p *processor) StreamNotificationToAccount(n *apimodel.Notification, account *gtsmodel.Account) error {
+ l := p.log.WithFields(logrus.Fields{
+ "func": "StreamNotificationToAccount",
+ "account": account.ID,
+ })
+ v, ok := p.streamMap.Load(account.ID)
+ if !ok {
+ // no open connections so nothing to stream
+ return nil
+ }
+
+ streamsForAccount, ok := v.(*gtsmodel.StreamsForAccount)
+ if !ok {
+ return errors.New("stream map error")
+ }
+
+ notificationBytes, err := json.Marshal(n)
+ if err != nil {
+ return fmt.Errorf("error marshalling notification to json: %s", err)
+ }
+
+ streamsForAccount.Lock()
+ defer streamsForAccount.Unlock()
+ for _, stream := range streamsForAccount.Streams {
+ stream.Lock()
+ defer stream.Unlock()
+ if stream.Connected {
+ l.Debugf("streaming notification to stream id %s", stream.ID)
+ stream.Messages <- &gtsmodel.Message{
+ Stream: []string{stream.Type},
+ Event: "notification",
+ Payload: string(notificationBytes),
+ }
+ }
+ }
+
+ return nil
+}