mirrors/dendrite
1
0
Fork 0
mirror of https://github.com/matrix-org/dendrite.git synced 2025-12-26 00:03:09 -06:00
Code Issues Packages Projects Releases Wiki Activity

Only update once per minute

Browse source
This commit is contained in:
Neil Alexander 2020-11-20 10:50:43 +00:00
parent 37dd2e5c5f
commit 4d2d55e1a4
No known key found for this signature in database
GPG key ID: A02A2019A2BB0944
1 changed files with 40 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

48
syncapi/sync/requestpool.go
View file

@ -20,6 +20,7 @@ import (
"context" "context"
"fmt" "fmt"
"net/http" "net/http"
"sync"
"time" "time"
"github.com/matrix-org/dendrite/clientapi/jsonerror" "github.com/matrix-org/dendrite/clientapi/jsonerror"
@ -41,6 +42,7 @@ type RequestPool struct {
notifier *Notifier notifier *Notifier
keyAPI keyapi.KeyInternalAPI keyAPI keyapi.KeyInternalAPI
rsAPI roomserverAPI.RoomserverInternalAPI rsAPI roomserverAPI.RoomserverInternalAPI
lastseen sync.Map
} }
// NewRequestPool makes a new RequestPool // NewRequestPool makes a new RequestPool
@ -48,7 +50,43 @@ func NewRequestPool(
db storage.Database, n *Notifier, userAPI userapi.UserInternalAPI, keyAPI keyapi.KeyInternalAPI, db storage.Database, n *Notifier, userAPI userapi.UserInternalAPI, keyAPI keyapi.KeyInternalAPI,
rsAPI roomserverAPI.RoomserverInternalAPI, rsAPI roomserverAPI.RoomserverInternalAPI,
) *RequestPool { ) *RequestPool {
return &RequestPool{db, userAPI, n, keyAPI, rsAPI} rp := &RequestPool{db, userAPI, n, keyAPI, rsAPI, sync.Map{}}
go rp.cleanLastSeen()
return rp
}
func (rp *RequestPool) cleanLastSeen() {
for {
rp.lastseen.Range(func(key interface{}, value interface{}) bool {
lastseen := value.(time.Time)
if time.Since(lastseen) > time.Minute*2 {
rp.lastseen.Delete(key)
fmt.Println("CLEANING", key)
}
return true
})
time.Sleep(time.Minute)
}
}
func (rp *RequestPool) updateLastSeen(req *http.Request, device *userapi.Device) {
value, _ := rp.lastseen.LoadOrStore(device.UserID+device.ID, time.Now().Add(-time.Minute))
lastseen := value.(time.Time)
if time.Since(lastseen) < time.Minute {
fmt.Println("TOO SOON")
return
}
fmt.Println("UPDATING")
lsreq := &userapi.PerformLastSeenUpdateRequest{
UserID: device.UserID,
DeviceID: device.ID,
RemoteAddr: req.RemoteAddr,
}
lsres := &userapi.PerformLastSeenUpdateResponse{}
go rp.userAPI.PerformLastSeenUpdate(req.Context(), lsreq, lsres) // nolint:errcheck
rp.lastseen.Store(device.UserID+device.ID, time.Now())
} }
// OnIncomingSyncRequest is called when a client makes a /sync request. This function MUST be // OnIncomingSyncRequest is called when a client makes a /sync request. This function MUST be
@ -74,13 +112,7 @@ func (rp *RequestPool) OnIncomingSyncRequest(req *http.Request, device *userapi.
"limit": syncReq.limit, "limit": syncReq.limit,
}) })
lsreq := &userapi.PerformLastSeenUpdateRequest{ rp.updateLastSeen(req, device)
UserID: device.UserID,
DeviceID: device.ID,
RemoteAddr: req.RemoteAddr,
}
lsres := &userapi.PerformLastSeenUpdateResponse{}
go rp.userAPI.PerformLastSeenUpdate(req.Context(), lsreq, lsres) // nolint:errcheck
currPos := rp.notifier.CurrentPosition() currPos := rp.notifier.CurrentPosition()