• Index
• Search by APIs
• Search by Words
• Project Search

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• discord-master
  • .github
    • ISSUE_TEMPLATE
      • feature_request.md
      • bug_report.md
    • FUNDING.yml
    • workflows
      • docker.yml
  • src
    • database
      • redis.ts
      • mongo-adapter.ts
      • README.md
      • db-stats.ts
      • database.ts
    • index.ts
    • freestuffbot.ts
    • lib
      • git-parser.ts
      • util.ts
      • parse-args.ts
      • permission-strings.ts
      • metrics.ts
      • README.md
      • guilddata-visualizer.ts
      • logger.ts
    • thirdparty
      • README.md
      • sentry
        • sentry.ts
    • types
      • controller.ts
      • plugins.ts
      • config.ts
      • context.ts
      • README.md
      • datastructs.ts
      • commands.ts
    • controller
      • experiments.ts
      • server.ts
      • manager.ts
      • README.md
      • remote-config.ts
    • bot
      • emojis.ts
      • states
        • help
          • main.ts
        • about
          • main.ts
        • admin
          • experiments.ts
          • print.ts
          • main.ts
        • settings
          • more.ts
          • language.ts
          • display.ts
          • filter.ts
          • channel.ts
          • main.ts
          • role.ts
      • announcement-manager.ts
      • message-distributor.ts
      • commands
        • free.ts
        • about.ts
        • vote.ts
        • help.ts
        • invite.ts
        • settings.ts
      • tracker.ts
      • components
        • admin
          • refetch.ts
          • main.ts
        • settings
          • guilddata
            • delete_confirm.ts
            • _.ts
            • delete.ts
            • delete_cancel.ts
          • role_change.ts
          • currency_change.ts
          • reaction_toggle.ts
          • language_change.ts
          • beta_toggle.ts
          • run_resend.ts
          • run_test.ts
          • price_change.ts
          • platforms_change.ts
          • channel_change.ts
          • trash_toggle.ts
          • theme_change.ts
      • database-manager.ts
      • README.md
  • gh.npmrc
  • config.docker.js
  • .npmrc
  • .editorconfig
  • LICENSE.md
  • README.md
  • package.json
  • CODE_OF_CONDUCT.md
  • scripts
    • register-commands.js
    • initdb.js
  • config.template.js
  • Dockerfile
  • .eslintrc.js
  • tsconfig.json
  • .gitignore
  • .dockerignore
  • .vscode
    • settings.json
  • docker-compose.yml

import { Counter, Gauge, Registry } from 'prom-client'
import { Request, Response } from 'express'
import { Core } from '..'


export default class Metrics {

  private static register = new Registry()

  public static counterGatewayEvents = new Counter({
    name: 'fsb_gateway_events',
    help: 'Keeps track of the total amount of incoming gateway events',
    labelNames: [ 'type' ]
  })

  public static counterInteractions = new Counter({
    name: 'fsb_interactions',
    help: 'Keeps track of the total amount of interactions',
    labelNames: [ 'type', 'name' ]
  })

  public static counterOutgoing = new Counter({
    name: 'fsb_outgoing',
    help: 'Keeps track of the total amount of announcement messages sent',
    labelNames: [ 'status' ]
  })

  public static counterApiResponses = new Counter({
    name: 'fsb_api_responses',
    help: 'Keep track of discord api responses',
    labelNames: [ 'status' ]
  })

  public static gaugeGatewayPing = new Gauge({
    name: 'fsb_gateway_ping',
    help: 'Shows the current gateway ping. Updated every 10 seconds.'
  })

  public static gaugeGuildCacheSize = new Gauge({
    name: 'fsb_caches_guilds_size',
    help: 'Amount of guilds currently cached on this worker. Updated every 10 seconds.'
  })

  public static gaugeUserCacheSize = new Gauge({
    name: 'fsb_caches_users_size',
    help: 'Amount of users currently cached on this worker. Updated every 10 seconds.'
  })

  public static gaugeChannelCacheSize = new Gauge({
    name: 'fsb_caches_channels_size',
    help: 'Amount of channels currently cached on this worker. Updated every 10 seconds.'
  })

  // public static gaugeCordoAIRC = new Gauge({
  //   name: 'fsb_cordo_airc',
  //   help: 'Amount of Active Interaction Reply Contexts (Cordo). Updated every 10 seconds.'
  // })

  //

  public static init() {
    Metrics.registerMetrics()
    Metrics.startCollectors()
  }

  private static registerMetrics() {
    // collectDefaultMetrics({ register: Metrics.register })

    Metrics.register.registerMetric(Metrics.counterGatewayEvents)
    Metrics.register.registerMetric(Metrics.counterInteractions)
    Metrics.register.registerMetric(Metrics.counterOutgoing)
    Metrics.register.registerMetric(Metrics.counterApiResponses)

    Metrics.register.registerMetric(Metrics.gaugeGatewayPing)
    Metrics.register.registerMetric(Metrics.gaugeGuildCacheSize)
    Metrics.register.registerMetric(Metrics.gaugeUserCacheSize)
    Metrics.register.registerMetric(Metrics.gaugeChannelCacheSize)
    // Metrics.register.registerMetric(Metrics.gaugeCordoAIRC)
  }

  private static startCollectors() {
    setInterval(() => {
      const ping = Core?.ws?.ping ?? -1
      Metrics.gaugeGatewayPing.set(ping)

      const guilds = Core?.guilds?.cache?.size ?? 0
      Metrics.gaugeGuildCacheSize.set(guilds)

      const users = Core?.users?.cache?.size ?? 0
      Metrics.gaugeUserCacheSize.set(users)

      const channels = Core?.channels?.cache?.size ?? 0
      Metrics.gaugeChannelCacheSize.set(channels)

      // const activeInteractionReplyContexts = CordoReplies.activeInteractionReplyContexts.length
      // Metrics.gaugeCordoAIRC.set(activeInteractionReplyContexts)
    }, 10e3)
  }

  //

  public static endpoint() {
    return async function (_req: Request, res: Response) {
      res
        .status(200)
        .header({ 'Content-Type': 'text/plain' })
        .send(await Metrics.register.metrics())
    }
  }

}