ntfy/server/topic.go

64 lines
1.4 KiB
Go
Raw Normal View History

2021-10-23 03:26:01 +02:00
package server
import (
"log"
"math/rand"
"sync"
)
2021-10-24 04:49:50 +02:00
// topic represents a channel to which subscribers can subscribe, and publishers
// can publish a message
2021-10-23 03:26:01 +02:00
type topic struct {
2021-12-09 04:57:31 +01:00
ID string
2021-11-03 02:09:49 +01:00
subscribers map[int]subscriber
2021-10-23 21:22:17 +02:00
mu sync.Mutex
2021-10-23 03:26:01 +02:00
}
2021-10-24 04:49:50 +02:00
// subscriber is a function that is called for every new message on a topic
2022-06-01 02:38:56 +02:00
type subscriber func(v *visitor, msg *message) error
2021-10-23 03:26:01 +02:00
// newTopic creates a new topic
2021-12-09 04:57:31 +01:00
func newTopic(id string) *topic {
2021-10-23 03:26:01 +02:00
return &topic{
2021-12-09 04:57:31 +01:00
ID: id,
2021-10-23 03:26:01 +02:00
subscribers: make(map[int]subscriber),
}
}
2021-11-09 20:48:25 +01:00
// Subscribe subscribes to this topic
2021-10-23 03:26:01 +02:00
func (t *topic) Subscribe(s subscriber) int {
t.mu.Lock()
defer t.mu.Unlock()
subscriberID := rand.Int()
t.subscribers[subscriberID] = s
return subscriberID
}
2021-11-09 20:48:25 +01:00
// Unsubscribe removes the subscription from the list of subscribers
2021-10-29 19:58:14 +02:00
func (t *topic) Unsubscribe(id int) {
2021-10-23 03:26:01 +02:00
t.mu.Lock()
defer t.mu.Unlock()
delete(t.subscribers, id)
}
2021-11-09 20:48:25 +01:00
// Publish asynchronously publishes to all subscribers
2022-06-01 02:38:56 +02:00
func (t *topic) Publish(v *visitor, m *message) error {
2021-11-09 20:48:25 +01:00
go func() {
t.mu.Lock()
defer t.mu.Unlock()
for _, s := range t.subscribers {
2022-06-01 02:38:56 +02:00
if err := s(v, m); err != nil {
2021-11-09 20:48:25 +01:00
log.Printf("error publishing message to subscriber")
}
2021-10-23 03:26:01 +02:00
}
2021-11-09 20:48:25 +01:00
}()
2021-10-23 03:26:01 +02:00
return nil
}
2021-11-09 20:48:25 +01:00
// Subscribers returns the number of subscribers to this topic
2021-11-03 02:09:49 +01:00
func (t *topic) Subscribers() int {
2021-10-29 19:58:14 +02:00
t.mu.Lock()
defer t.mu.Unlock()
2021-11-03 02:09:49 +01:00
return len(t.subscribers)
2021-10-23 03:26:01 +02:00
}