import { CONFIG } from '@server/initializers/config' import { UserModel } from '@server/models/account/user' import { ActorFollowModel } from '@server/models/activitypub/actor-follow' import { VideoRedundancyModel } from '@server/models/redundancy/video-redundancy' import { VideoModel } from '@server/models/video/video' import { VideoChannelModel } from '@server/models/video/video-channel' import { VideoCommentModel } from '@server/models/video/video-comment' import { VideoFileModel } from '@server/models/video/video-file' import { VideoPlaylistModel } from '@server/models/video/video-playlist' import { ActivityType, ServerStats, VideoRedundancyStrategyWithManual } from '@shared/models' class StatsManager { private static instance: StatsManager private readonly instanceStartDate = new Date() private inboxMessages = { processed: 0, errors: 0, successes: 0, waiting: 0, errorsPerType: this.buildAPPerType(), successesPerType: this.buildAPPerType() } private constructor () {} updateInboxWaiting (inboxMessagesWaiting: number) { this.inboxMessages.waiting = inboxMessagesWaiting } addInboxProcessedSuccess (type: ActivityType) { this.inboxMessages.processed++ this.inboxMessages.successes++ this.inboxMessages.successesPerType[type]++ } addInboxProcessedError (type: ActivityType) { this.inboxMessages.processed++ this.inboxMessages.errors++ this.inboxMessages.errorsPerType[type]++ } async getStats () { const { totalLocalVideos, totalLocalVideoViews, totalVideos } = await VideoModel.getStats() const { totalLocalVideoComments, totalVideoComments } = await VideoCommentModel.getStats() const { totalUsers, totalDailyActiveUsers, totalWeeklyActiveUsers, totalMonthlyActiveUsers } = await UserModel.getStats() const { totalInstanceFollowers, totalInstanceFollowing } = await ActorFollowModel.getStats() const { totalLocalVideoFilesSize } = await VideoFileModel.getStats() const { totalLocalVideoChannels, totalLocalDailyActiveVideoChannels, totalLocalWeeklyActiveVideoChannels, totalLocalMonthlyActiveVideoChannels } = await VideoChannelModel.getStats() const { totalLocalPlaylists } = await VideoPlaylistModel.getStats() const videosRedundancyStats = await this.buildRedundancyStats() const data: ServerStats = { totalUsers, totalDailyActiveUsers, totalWeeklyActiveUsers, totalMonthlyActiveUsers, totalLocalVideos, totalLocalVideoViews, totalLocalVideoComments, totalLocalVideoFilesSize, totalVideos, totalVideoComments, totalLocalVideoChannels, totalLocalDailyActiveVideoChannels, totalLocalWeeklyActiveVideoChannels, totalLocalMonthlyActiveVideoChannels, totalLocalPlaylists, totalInstanceFollowers, totalInstanceFollowing, videosRedundancy: videosRedundancyStats, ...this.buildAPStats() } return data } private buildActivityPubMessagesProcessedPerSecond () { const now = new Date() const startedSeconds = (now.getTime() - this.instanceStartDate.getTime()) / 1000 return this.inboxMessages.processed / startedSeconds } private buildRedundancyStats () { const strategies = CONFIG.REDUNDANCY.VIDEOS.STRATEGIES .map(r => ({ strategy: r.strategy as VideoRedundancyStrategyWithManual, size: r.size })) strategies.push({ strategy: 'manual', size: null }) return Promise.all( strategies.map(r => { return VideoRedundancyModel.getStats(r.strategy) .then(stats => Object.assign(stats, { strategy: r.strategy, totalSize: r.size })) }) ) } private buildAPPerType () { return { Create: 0, Update: 0, Delete: 0, Follow: 0, Accept: 0, Reject: 0, Announce: 0, Undo: 0, Like: 0, Dislike: 0, Flag: 0, View: 0 } } private buildAPStats () { return { totalActivityPubMessagesProcessed: this.inboxMessages.processed, totalActivityPubMessagesSuccesses: this.inboxMessages.successes, // Dirty, but simpler and with type checking totalActivityPubCreateMessagesSuccesses: this.inboxMessages.successesPerType.Create, totalActivityPubUpdateMessagesSuccesses: this.inboxMessages.successesPerType.Update, totalActivityPubDeleteMessagesSuccesses: this.inboxMessages.successesPerType.Delete, totalActivityPubFollowMessagesSuccesses: this.inboxMessages.successesPerType.Follow, totalActivityPubAcceptMessagesSuccesses: this.inboxMessages.successesPerType.Accept, totalActivityPubRejectMessagesSuccesses: this.inboxMessages.successesPerType.Reject, totalActivityPubAnnounceMessagesSuccesses: this.inboxMessages.successesPerType.Announce, totalActivityPubUndoMessagesSuccesses: this.inboxMessages.successesPerType.Undo, totalActivityPubLikeMessagesSuccesses: this.inboxMessages.successesPerType.Like, totalActivityPubDislikeMessagesSuccesses: this.inboxMessages.successesPerType.Dislike, totalActivityPubFlagMessagesSuccesses: this.inboxMessages.successesPerType.Flag, totalActivityPubViewMessagesSuccesses: this.inboxMessages.successesPerType.View, totalActivityPubCreateMessagesErrors: this.inboxMessages.errorsPerType.Create, totalActivityPubUpdateMessagesErrors: this.inboxMessages.errorsPerType.Update, totalActivityPubDeleteMessagesErrors: this.inboxMessages.errorsPerType.Delete, totalActivityPubFollowMessagesErrors: this.inboxMessages.errorsPerType.Follow, totalActivityPubAcceptMessagesErrors: this.inboxMessages.errorsPerType.Accept, totalActivityPubRejectMessagesErrors: this.inboxMessages.errorsPerType.Reject, totalActivityPubAnnounceMessagesErrors: this.inboxMessages.errorsPerType.Announce, totalActivityPubUndoMessagesErrors: this.inboxMessages.errorsPerType.Undo, totalActivityPubLikeMessagesErrors: this.inboxMessages.errorsPerType.Like, totalActivityPubDislikeMessagesErrors: this.inboxMessages.errorsPerType.Dislike, totalActivityPubFlagMessagesErrors: this.inboxMessages.errorsPerType.Flag, totalActivityPubViewMessagesErrors: this.inboxMessages.errorsPerType.View, totalActivityPubMessagesErrors: this.inboxMessages.errors, activityPubMessagesProcessedPerSecond: this.buildActivityPubMessagesProcessedPerSecond(), totalActivityPubMessagesWaiting: this.inboxMessages.waiting } } static get Instance () { return this.instance || (this.instance = new this()) } } // --------------------------------------------------------------------------- export { StatsManager }