vbatts/ntfy
1
0
Fork 0
Code Pull requests Activity

Add firebase subscriber to topics from cache upon initialization; stricter rate limits

Browse source
This commit is contained in:
Philipp Heckel 2021-11-03 21:16:07 -04:00
parent eef85c0955
commit 5948f39a53
2 changed files with 8 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
config/config.go
View file

@ -15,12 +15,12 @@ const (
)
// Defines all the limits
// - request limit: max number of PUT/GET/.. requests (here: 50 requests bucket, replenished at a rate of 1 per second)
// - request limit: max number of PUT/GET/.. requests (here: 50 requests bucket, replenished at a rate of one per 10 seconds)
// - global topic limit: max number of topics overall
// - subscription limit: max number of subscriptions (active HTTP connections) per per-visitor/IP
var (
defaultGlobalTopicLimit = 5000
defaultVisitorRequestLimit = rate.Every(time.Second)
defaultVisitorRequestLimit = rate.Every(10 * time.Second)
defaultVisitorRequestLimitBurst = 50
defaultVisitorSubscriptionLimit = 30
)

7
server/server.go
View file

@ -47,7 +47,7 @@ func (e errHTTP) Error() string {
}
const (
messageLimit = 1024
messageLimit = 512
)
var (
@ -86,6 +86,11 @@ func New(conf *config.Config) (*Server, error) {
if err != nil {
return nil, err
}
for _, t := range topics {
if firebaseSubscriber != nil {
t.Subscribe(firebaseSubscriber)
}
}
return &Server{
config: conf,
cache: cache,