mirror of
https://github.com/binwiederhier/ntfy.git
synced 2024-12-26 03:32:30 +01:00
Topic expiry attempt
This commit is contained in:
parent
57e1104afb
commit
8eae44ea61
2 changed files with 11 additions and 15 deletions
|
@ -40,7 +40,7 @@ func (s *Server) execManager() {
|
|||
if ev.IsTrace() {
|
||||
expiryMessage := ""
|
||||
if subs == 0 {
|
||||
expiryTime := time.Until(t.rateVisitorExpires)
|
||||
expiryTime := time.Until(t.expires)
|
||||
expiryMessage = ", expires in " + expiryTime.String()
|
||||
}
|
||||
ev.Trace("- topic %s: %d subscribers%s", t.ID, subs, expiryMessage)
|
||||
|
|
|
@ -8,17 +8,17 @@ import (
|
|||
)
|
||||
|
||||
const (
|
||||
rateVisitorExpiryDuration = 12 * time.Hour
|
||||
topicExpiryDuration = 6 * time.Hour
|
||||
)
|
||||
|
||||
// topic represents a channel to which subscribers can subscribe, and publishers
|
||||
// can publish a message
|
||||
type topic struct {
|
||||
ID string
|
||||
subscribers map[int]*topicSubscriber
|
||||
rateVisitor *visitor
|
||||
rateVisitorExpires time.Time
|
||||
mu sync.RWMutex
|
||||
ID string
|
||||
subscribers map[int]*topicSubscriber
|
||||
rateVisitor *visitor
|
||||
expires time.Time
|
||||
mu sync.RWMutex
|
||||
}
|
||||
|
||||
type topicSubscriber struct {
|
||||
|
@ -54,25 +54,18 @@ func (t *topic) Subscribe(s subscriber, visitor *visitor, cancel func()) int {
|
|||
func (t *topic) Stale() bool {
|
||||
t.mu.Lock()
|
||||
defer t.mu.Unlock()
|
||||
if t.rateVisitorExpires.Before(time.Now()) {
|
||||
t.rateVisitor = nil
|
||||
}
|
||||
return len(t.subscribers) == 0 && t.rateVisitor == nil
|
||||
return len(t.subscribers) == 0 && t.expires.Before(time.Now())
|
||||
}
|
||||
|
||||
func (t *topic) SetRateVisitor(v *visitor) {
|
||||
t.mu.Lock()
|
||||
defer t.mu.Unlock()
|
||||
t.rateVisitor = v
|
||||
t.rateVisitorExpires = time.Now().Add(rateVisitorExpiryDuration)
|
||||
}
|
||||
|
||||
func (t *topic) RateVisitor() *visitor {
|
||||
t.mu.Lock()
|
||||
defer t.mu.Unlock()
|
||||
if t.rateVisitorExpires.Before(time.Now()) {
|
||||
t.rateVisitor = nil
|
||||
}
|
||||
return t.rateVisitor
|
||||
}
|
||||
|
||||
|
@ -81,6 +74,9 @@ func (t *topic) Unsubscribe(id int) {
|
|||
t.mu.Lock()
|
||||
defer t.mu.Unlock()
|
||||
delete(t.subscribers, id)
|
||||
if len(t.subscribers) == 0 {
|
||||
t.expires = time.Now().Add(topicExpiryDuration)
|
||||
}
|
||||
}
|
||||
|
||||
// Publish asynchronously publishes to all subscribers
|
||||
|
|
Loading…
Reference in a new issue