mirror of https://github.com/Chocobozzz/PeerTube
parent
0f21769205
commit
40a6dcb632
|
@ -382,7 +382,7 @@ class Redis {
|
||||||
return 'verify-email-registration-' + registrationId
|
return 'verify-email-registration-' + registrationId
|
||||||
}
|
}
|
||||||
|
|
||||||
private generateIPViewKey (ip: string, videoUUID: string) {
|
generateIPViewKey (ip: string, videoUUID: string) {
|
||||||
return `views-${videoUUID}-${ip}`
|
return `views-${videoUUID}-${ip}`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -5,11 +5,18 @@ import { getServerActor } from '@server/models/application/application.js'
|
||||||
import { MVideo, MVideoImmutable } from '@server/types/models/index.js'
|
import { MVideo, MVideoImmutable } from '@server/types/models/index.js'
|
||||||
import { buildUUID } from '@peertube/peertube-node-utils'
|
import { buildUUID } from '@peertube/peertube-node-utils'
|
||||||
import { Redis } from '../../redis.js'
|
import { Redis } from '../../redis.js'
|
||||||
|
import { LRUCache } from 'lru-cache'
|
||||||
|
import { VIEW_LIFETIME } from '@server/initializers/constants.js'
|
||||||
|
|
||||||
const lTags = loggerTagsFactory('views')
|
const lTags = loggerTagsFactory('views')
|
||||||
|
|
||||||
export class VideoViews {
|
export class VideoViews {
|
||||||
|
|
||||||
|
private readonly viewsCache = new LRUCache<string, boolean>({
|
||||||
|
max: 10_000,
|
||||||
|
ttl: VIEW_LIFETIME.VIEW
|
||||||
|
})
|
||||||
|
|
||||||
async addLocalView (options: {
|
async addLocalView (options: {
|
||||||
video: MVideoImmutable
|
video: MVideoImmutable
|
||||||
ip: string
|
ip: string
|
||||||
|
@ -21,10 +28,10 @@ export class VideoViews {
|
||||||
|
|
||||||
if (!await this.hasEnoughWatchTime(video, watchTime)) return false
|
if (!await this.hasEnoughWatchTime(video, watchTime)) return false
|
||||||
|
|
||||||
const viewExists = await Redis.Instance.doesVideoIPViewExist(ip, video.uuid)
|
const viewExists = await this.doesVideoIPViewExist(ip, video.uuid)
|
||||||
if (viewExists) return false
|
if (viewExists) return false
|
||||||
|
|
||||||
await Redis.Instance.setIPVideoView(ip, video.uuid)
|
await this.setIPVideoView(ip, video.uuid)
|
||||||
|
|
||||||
await this.addView(video)
|
await this.addView(video)
|
||||||
|
|
||||||
|
@ -67,4 +74,19 @@ export class VideoViews {
|
||||||
// Check more than 50% of the video is watched
|
// Check more than 50% of the video is watched
|
||||||
return duration / watchTime < 2
|
return duration / watchTime < 2
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private doesVideoIPViewExist (ip: string, videoUUID: string) {
|
||||||
|
const key = Redis.Instance.generateIPViewKey(ip, videoUUID)
|
||||||
|
const value = this.viewsCache.has(key)
|
||||||
|
if (value === true) return Promise.resolve(true)
|
||||||
|
|
||||||
|
return Redis.Instance.doesVideoIPViewExist(ip, videoUUID)
|
||||||
|
}
|
||||||
|
|
||||||
|
private setIPVideoView (ip: string, videoUUID: string) {
|
||||||
|
const key = Redis.Instance.generateIPViewKey(ip, videoUUID)
|
||||||
|
this.viewsCache.set(key, true)
|
||||||
|
|
||||||
|
return Redis.Instance.setIPVideoView(ip, videoUUID)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue