lingling/tiny-rdm
1
0
Fork 0
mirror of https://github.com/tiny-craft/tiny-rdm.git synced 2025-04-13 01:58:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

pref: redesign the throttle of pub/sub

Browse Source
This commit is contained in:
Lykin 2025-04-08 16:39:20 +08:00
parent 84d1141f8c
commit 16fea2b6cc
1 changed files with 14 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
backend/services/pubsub_service.go
View File

@ -120,8 +120,10 @@ func (p *pubsubService) StartSubscribe(server string) (resp types.JSResp) {
}
func (p *pubsubService) processSubscribe(mutex *sync.Mutex, ch <-chan *redis.Message, closeCh <-chan struct{}, eventName string) {
lastEmitTime := time.Now().Add(-1 * time.Minute)
cache := make([]subMessage, 0, 1000)
ticker := time.NewTicker(300 * time.Millisecond)
defer ticker.Stop()
for {
select {
case data := <-ch:
@ -134,10 +136,19 @@ func (p *pubsubService) processSubscribe(mutex *sync.Mutex, ch <-chan *redis.Mes
Channel: data.Channel,
Message: data.Payload,
})
if time.Now().Sub(lastEmitTime) > 300*time.Millisecond || len(cache) > 300 {
if len(cache) > 300 {
runtime.EventsEmit(p.ctx, eventName, cache)
cache = cache[:0:cap(cache)]
}
}()
case <-ticker.C:
func() {
mutex.Lock()
defer mutex.Unlock()
if len(cache) > 0 {
runtime.EventsEmit(p.ctx, eventName, cache)
cache = cache[:0:cap(cache)]
lastEmitTime = time.Now()
}
}()