maeqtt/session/expiry.go

package session

import (
	"time"

	"badat.dev/maeqtt/v2/subscription"
)

type Expiry struct {
	ExpiryInterval time.Duration 
	expireTimer    time.Timer    
	RemoveSessionChannel
}

func NewExpiry(channel RemoveSessionChannel) Expiry {
	expiry := Expiry {}

	expiry.RemoveSessionChannel = channel
	expiry.expireTimer = *time.NewTimer(time.Hour*9999)
	expiry.expireTimer.Stop()
	return expiry
}

// Channel for removing a session from the global state
type RemoveSessionChannel chan string

func (s *Session) expireSession() {
	subscription.Subscriptions.RemoveSubsForChannel(s.SubscriptionChannel)
	s.RemoveSessionChannel <- *s.ClientID
}

// newTime is nullable
func (s *Session) SetExpireTimer(newTime *uint32) {
	expiry := uint32(0)
	if newTime != nil {
		expiry = *newTime
	} else {
		expiry = uint32(0)
	}
	s.ExpiryInterval = time.Duration(expiry) * time.Second
}

func (s *Session) resetExpireTimer() {
	if s.ExpiryInterval == 0 {
		s.expireTimer.Stop()
	} else {
		s.expireTimer.Reset(s.ExpiryInterval)
	}
}