lib/generators/gitlab/usage_metric_definition/redis_hll_generator.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26

# frozen_string_literal: true

require 'rails/generators'
require_relative '../usage_metric_definition_generator'

module Gitlab
  module UsageMetricDefinition
    class RedisHllGenerator < Rails::Generators::Base
      desc 'Generates a metric definition .yml file with defaults for Redis HLL.'

      argument :category, type: :string, desc: "Category name"
      argument :event, type: :string, desc: "Event name"

      def create_metrics
        Gitlab::UsageMetricDefinitionGenerator.start(["#{key_path}_weekly", '--dir', '7d'])
        Gitlab::UsageMetricDefinitionGenerator.start(["#{key_path}_monthly", '--dir', '28d'])
      end

      private

      def key_path
        "redis_hll_counters.#{category}.#{event}"
      end
    end
  end
end