yetangitu/Peertube
1
0
Fork 0
mirror of https://github.com/Chocobozzz/PeerTube.git synced 2025-10-03 09:49:20 +02:00
Code Issues Releases Wiki Activity

Optimize updating oauth tokens activity

Browse source
This commit is contained in:
Chocobozzz 2025-09-10 15:16:40 +02:00
parent efa32646ed
commit bbc1afada5
No known key found for this signature in database
GPG key ID: 583A612D890159BE
4 changed files with 10 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

17
server/core/lib/schedulers/update-token-session-scheduler.ts
View file

@ -3,7 +3,6 @@ import { SCHEDULER_INTERVALS_MS } from '../../initializers/constants.js'
import { AbstractScheduler } from './abstract-scheduler.js'
type UpdatePayload = {
id: number
lastActivityDate: Date
lastActivityIP: string
lastActivityDevice: string
@ -14,29 +13,27 @@ export class UpdateTokenSessionScheduler extends AbstractScheduler {
protected schedulerIntervalMs = SCHEDULER_INTERVALS_MS.UPDATE_TOKEN_SESSION
private readonly toUpdate = new Set<UpdatePayload>()
private toUpdate = new Map<number, UpdatePayload>()
private constructor () {
super()
}
addToUpdate (payload: UpdatePayload) {
this.toUpdate.add(payload)
addToUpdate (id: number, payload: UpdatePayload) {
this.toUpdate.set(id, payload)
}
protected async internalExecute () {
const toUpdate = Array.from(this.toUpdate)
this.toUpdate.clear()
const entriesToUpdate = this.toUpdate.entries()
this.toUpdate = new Map()
for (const payload of toUpdate) {
for (const [ id, payload ] of entriesToUpdate) {
await OAuthTokenModel.update({
lastActivityDate: payload.lastActivityDate,
lastActivityIP: payload.lastActivityIP,
lastActivityDevice: payload.lastActivityDevice
}, {
where: {
id: payload.id
},
where: { id },
// Prevent tokens cache invalidation, we don't update fields that are meaningful for this cache
hooks: false
})

3
server/core/middlewares/auth.ts
View file

@ -13,8 +13,7 @@ export function authenticate (req: express.Request, res: express.Response, next:
res.locals.oauth = { token }
res.locals.authenticated = true
UpdateTokenSessionScheduler.Instance.addToUpdate({
id: token.id,
UpdateTokenSessionScheduler.Instance.addToUpdate(token.id, {
lastActivityDate: new Date(),
lastActivityIP: req.ip,
lastActivityDevice: req.header('user-agent')