gitlab-exporter 11.0.1 → 11.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4a277e09b79fa17edf12f018a94a78c66cce4cc13cb4507ec6e8c82266ff4e0f
-  data.tar.gz: de009b471d90a44f3721e3acad8c6f25cfc3e82d3f0ad72a76e8e7b0bf9e6e45
+  metadata.gz: 1629fb16b7451c28e085202cc001961dc3f1ac66308146ca992a7eeaf179c6e2
+  data.tar.gz: a6158d5da5323e132a4a07e9b33238bb6b224658bc60a6885225c6d6a3729698
 SHA512:
-  metadata.gz: 88bfdd110c7360730d076f24b96681d5b705ab926fa3925b79850bb499cc3247e3566fc132cdb5ae6877d7e533569515dcd00bd7e7b5d7e8c98c0fc77de8104d
-  data.tar.gz: 2658a7a6f6f735782868b43a9c1b68d3f63fd1d5ace3f8d1553bc471c50bf340f03000daded30012fb86f901f1b94cb029788e45ccbaf22f0e831d1be91fb8de
+  metadata.gz: c1f03be5bb43004977d466b5dd4985a1626cc1cf430ee4d2137ad1c5983da527521ca3b3785e28e7daf20f18405b1bf43e9964106d60900b5c26e102e035d7c6
+  data.tar.gz: 5433100b784c35f2c66e676e2c361ed6945fbcd4cc29d77be831f4f0c7d2f2d4a9b6271aeb8492c7af13f266975e053322a84e288c12502ba65ff03996f13a25

data/.gitlab-ci.yml

@@ -6,12 +6,15 @@ include:
   - template: Security/SAST.gitlab-ci.yml  # https://gitlab.com/gitlab-org/gitlab-foss/blob/master/lib/gitlab/ci/templates/Security/SAST.gitlab-ci.yml
   - template: Security/Secret-Detection.gitlab-ci.yml # https://gitlab.com/gitlab-org/gitlab-foss/-/blob/master/lib/gitlab/ci/templates/Security/Secret-Detection.gitlab-ci.yml
 
+variables:
+  RUBY_VERSION: "2.7"
+
 stages:
   - test
   - dast
 
 default:
-  image: ruby:2.7
+  image: ruby:${RUBY_VERSION}
   cache:
     paths:
       - vendor
@@ -36,6 +39,10 @@ rspec:
   script:
     - bundle exec rspec spec -f d -c
   before_script: *before_scripts
+  parallel:
+    matrix:
+      - RUBY_VERSION: "2.7"
+      - RUBY_VERSION: "3.0"
 
 rubocop:
   script:

data/.ruby-version

@@ -1 +1 @@
-2.7.2
+2.7.4

data/Gemfile

@@ -2,6 +2,8 @@ source "https://rubygems.org"
 
 gemspec
 
+gem "webrick", "~> 1.7"
+
 group :test do
   gem "rspec", "~>3.5"
   gem "rubocop", "~>0.42"

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    gitlab-exporter (11.0.1)
+    gitlab-exporter (11.4.0)
       connection_pool (= 2.2.5)
       pg (= 1.2.3)
       puma (= 5.3.2)
@@ -9,7 +9,7 @@ PATH
       redis (= 4.1.4)
       redis-namespace (= 1.6.0)
       sidekiq (= 5.2.9)
-      sinatra (= 2.0.8.1)
+      sinatra (~> 2.1.0)
 
 GEM
   remote: https://rubygems.org/
@@ -19,7 +19,7 @@ GEM
     diff-lcs (1.3)
     mustermann (1.1.1)
       ruby2_keywords (~> 0.0.1)
-    nio4r (2.5.7)
+    nio4r (2.5.8)
     parallel (1.20.1)
     parser (3.0.0.0)
       ast (~> 2.4.1)
@@ -28,7 +28,7 @@ GEM
       nio4r (~> 2.0)
     quantile (0.2.1)
     rack (2.2.3)
-    rack-protection (2.0.8.1)
+    rack-protection (2.1.0)
       rack
     rainbow (3.0.0)
     redis (4.1.4)
@@ -61,19 +61,20 @@ GEM
     rubocop-ast (1.4.0)
       parser (>= 2.7.1.5)
     ruby-progressbar (1.11.0)
-    ruby2_keywords (0.0.4)
+    ruby2_keywords (0.0.5)
     sidekiq (5.2.9)
       connection_pool (~> 2.2, >= 2.2.2)
       rack (~> 2.0)
       rack-protection (>= 1.5.0)
       redis (>= 3.3.5, < 4.2)
-    sinatra (2.0.8.1)
+    sinatra (2.1.0)
       mustermann (~> 1.0)
-      rack (~> 2.0)
-      rack-protection (= 2.0.8.1)
+      rack (~> 2.2)
+      rack-protection (= 2.1.0)
       tilt (~> 2.0)
     tilt (2.0.10)
     unicode-display_width (1.7.0)
+    webrick (1.7.0)
 
 PLATFORMS
   ruby
@@ -83,6 +84,7 @@ DEPENDENCIES
   rspec (~> 3.5)
   rspec-expectations (~> 3.7.0)
   rubocop (~> 0.42)
+  webrick (~> 1.7)
 
 BUNDLED WITH
-   2.1.4
+   2.2.22

data/README.md

@@ -34,7 +34,7 @@ metrics.
       `git_pull_time_milliseconds`, `git_push_time_milliseconds`
     * git processes stats (see Process below)
 1. [Sidekiq](lib/gitlab_exporter/sidekiq.rb)
-	* Stats
+	* Stats (probe_stats)
 	  * `sidekiq_jobs_processed_total`
 	  * `sidekiq_jobs_failed_total`
 	  * `sidekiq_jobs_enqueued_size`
@@ -44,14 +44,21 @@ metrics.
 	  * `sidekiq_default_queue_latency_seconds`
 	  * `sidekiq_processes_size`
 	  * `sidekiq_workers_size`
-	* Queues
+	* Queues (probe_queues)
 	  * `sidekiq_queue_size`
 	  * `sidekiq_queue_paused`
 	  * `sidekiq_queue_latency_seconds`
-	* Jobs
+	* Jobs (probe_jobs_limit)
 	  * `sidekiq_enqueued_jobs`
+  * Workers (probe_workers)
 	  * `sidekiq_running_jobs`
+  * Retries (probe_retries)
 	  * `sidekiq_to_be_retried_jobs`
+  * Future Sets (probe_future_sets)
+    * `sidekiq_schedule_set_processing_delay_seconds`
+    * `sidekiq_schedule_set_backlog_count`
+    * `sidekiq_retry_set_processing_delay_seconds`
+    * `sidekiq_retry_set_backlog_count`
 
 ### Setup with GitLab Development Kit
 

data/gitlab-exporter.gemspec

@@ -27,7 +27,7 @@ Gem::Specification.new do |s|
   s.add_runtime_dependency "redis", "4.1.4"
   s.add_runtime_dependency "redis-namespace", "1.6.0"
   s.add_runtime_dependency "sidekiq", "5.2.9"
-  s.add_runtime_dependency "sinatra", "2.0.8.1"
+  s.add_runtime_dependency "sinatra", "~> 2.1.0"
 
   s.add_development_dependency "rspec", "~> 3.7.0"
   s.add_development_dependency "rspec-expectations", "~> 3.7.0"

data/lib/gitlab_exporter/database/base.rb

@@ -41,8 +41,8 @@ module GitLab
           conn.type_map_for_results = tm
         end
 
-        def initialize(args, logger: nil)
-          @connection_string = args[:connection_string]
+        def initialize(connection_string:, logger: nil, **opts) # rubocop:disable Lint/UnusedMethodArgument
+          @connection_string = connection_string
           @logger = logger
         end
 

data/lib/gitlab_exporter/database/bloat.rb

@@ -39,12 +39,11 @@ module GitLab
 
         attr_reader :metrics, :collector, :bloat_types
 
-        def initialize(opts,
-                       metrics: PrometheusMetrics.new,
-                       collector: BloatCollector.new(connection_string: opts[:connection_string]),
-                       logger: nil)
+        def initialize(metrics: PrometheusMetrics.new,
+                       logger: nil,
+                       **opts)
           @metrics = metrics
-          @collector = collector
+          @collector = opts[:collector] || BloatCollector.new(**opts)
           @collector.logger = logger
           @bloat_types = opts[:bloat_types] || %i[btree table]
         end

data/lib/gitlab_exporter/database/ci_builds.rb

@@ -168,8 +168,8 @@ module GitLab
 
         DEFAULT_UNARCHIVED_TRACES_OFFSET_MINUTES = 1440
 
-        def initialize(opts, logger: nil)
-          super(opts, logger: logger)
+        def initialize(**opts)
+          super
 
           @created_builds_counting_disabled = opts[:created_builds_counting_disabled]
           @unarchived_traces_offset_minutes = opts[:unarchived_traces_offset_minutes]
@@ -294,13 +294,9 @@ module GitLab
 
       # The prober which is called when gathering metrics
       class CiBuildsProber
-        def initialize(opts, metrics: PrometheusMetrics.new, logger: nil)
+        def initialize(metrics: PrometheusMetrics.new, **opts)
           @metrics = metrics
-
-          collector_opts = { connection_string: opts[:connection_string],
-                             created_builds_counting_disabled: opts[:created_builds_counting_disabled],
-                             unarchived_traces_offset_minutes: opts[:unarchived_traces_offset_minutes] }
-          @collector = CiBuildsCollector.new(collector_opts, logger: logger)
+          @collector = CiBuildsCollector.new(**opts)
         end
 
         def probe_db
@@ -391,7 +387,7 @@ module GitLab
           labels[:namespace] = "" unless labels[:namespace]
 
           selected_labels = labels.select { |k, _| allowed_labels.include?(k) }.sort.to_h
-          @metrics.add(metric_name, value.to_f, selected_labels)
+          @metrics.add(metric_name, value.to_f, **selected_labels)
         end
 
         def unarchived_traces_metrics

data/lib/gitlab_exporter/database/remote_mirrors.rb

@@ -10,8 +10,8 @@ module GitLab
           FROM remote_mirrors WHERE project_id IN (%s) AND enabled = 't'
         SQL
 
-        def initialize(args)
-          super(args)
+        def initialize(**args)
+          super(**args)
 
           @project_ids = args[:project_ids]
         end
@@ -35,7 +35,7 @@ module GitLab
 
       # The prober which is called when gathering metrics
       class RemoteMirrorsProber
-        def initialize(opts, metrics: PrometheusMetrics.new, logger: nil) # rubocop:disable Lint/UnusedMethodArgument
+        def initialize(metrics: PrometheusMetrics.new, **opts)
           @metrics = metrics
           @collector = RemoteMirrorsCollector.new(
             connection_string: opts[:connection_string],

data/lib/gitlab_exporter/database/row_count.rb

@@ -131,13 +131,29 @@ module GitLab
               visibility_level: {},
               root: { definition: "(parent_id IS NULL)" }
             }
-          }
+          },
+          registry_gc_manifest_review_queue: { select: :gc_manifest_review_queue },
+          registry_gc_manifest_review_queue_overdue: {
+            select: :gc_manifest_review_queue,
+            where: "review_after < NOW()"
+          },
+          registry_gc_blob_review_queue: { select: :gc_blob_review_queue },
+          registry_gc_blob_review_queue_overdue: {
+            select: :gc_blob_review_queue,
+            where: "review_after < NOW()"
+          },
+          registry_top_level_namespaces: { select: :top_level_namespaces },
+          # Please note that the tables below are partitioned, so a SELECT COUNT(*)
+          # will scale poorly. Avoid using these in production:
+          registry_repositories: { select: :repositories },
+          registry_manifests: { select: :manifests },
+          registry_blobs: { select: :blobs }
         }.freeze
 
-        def initialize(args)
-          super(args)
+        def initialize(selected_queries: nil, **args)
+          super(**args)
 
-          @selected_queries = Set.new(args[:selected_queries].map(&:to_sym)) unless args[:selected_queries].nil?
+          @selected_queries = Set.new(selected_queries.map(&:to_sym)) unless selected_queries.nil?
         end
 
         def run
@@ -200,12 +216,9 @@ module GitLab
 
       # The prober which is called when gathering metrics
       class RowCountProber
-        def initialize(opts, metrics: PrometheusMetrics.new, logger: nil) # rubocop:disable Lint/UnusedMethodArgument
+        def initialize(metrics: PrometheusMetrics.new, **opts)
           @metrics = metrics
-          @collector = RowCountCollector.new(
-            connection_string: opts[:connection_string],
-            selected_queries: opts[:selected_queries]
-          )
+          @collector = RowCountCollector.new(**opts)
         end
 
         def probe_db

data/lib/gitlab_exporter/database/tuple_stats.rb

@@ -25,9 +25,9 @@ module GitLab
 
       # Probes the DB specified by opts[:connection_string] for tuple stats, then converts them to metrics
       class TuplesProber
-        def initialize(opts, metrics: PrometheusMetrics.new, logger: nil)
+        def initialize(metrics: PrometheusMetrics.new, **opts)
           @metrics   = metrics
-          @collector = TupleStatsCollector.new(connection_string: opts[:connection_string], logger: logger)
+          @collector = TupleStatsCollector.new(**opts)
         end
 
         def probe_db

data/lib/gitlab_exporter/git.rb

@@ -64,10 +64,10 @@ module GitLab
     # Optionally takes a metrics object which by default is a PrometheusMetrics, useful to change the
     # metrics writer to something else.
     class GitProber
-      def initialize(opts, metrics: PrometheusMetrics.new, logger: nil) # rubocop:disable Lint/UnusedMethodArgument
+      def initialize(source:, metrics: PrometheusMetrics.new, labels: {}, **opts) # rubocop:disable Lint/UnusedMethodArgument
         @metrics = metrics
-        @labels = opts[:labels] || {}
-        @git = Git.new(opts[:source])
+        @labels = labels
+        @git = Git.new(source)
       end
 
       def probe_pull
@@ -87,9 +87,9 @@ module GitLab
 
     # A special prober for git processes
     class GitProcessProber
-      def initialize(opts, metrics: PrometheusMetrics.new, logger: nil) # rubocop:disable Lint/UnusedMethodArgument
-        @opts    = opts
+      def initialize(metrics: PrometheusMetrics.new, quantiles: nil, **opts) # rubocop:disable Lint/UnusedMethodArgument
         @metrics = metrics
+        @quantiles = quantiles
       end
 
       def probe_git # rubocop:disable Metrics/MethodLength
@@ -115,7 +115,7 @@ module GitLab
             {
               name: name,
               pid_or_pattern: pid,
-              quantiles: @opts[:quantiles]
+              quantiles: @quantiles
             },
             metrics: @metrics
           )

data/lib/gitlab_exporter/prober.rb

@@ -2,7 +2,7 @@ module GitLab
   module Exporter
     # A class to combine multiple probers into one
     class Prober
-      def initialize(prober_opts, metrics: PrometheusMetrics.new, logger: nil)
+      def initialize(metrics: PrometheusMetrics.new, logger: nil, **prober_opts)
         @prober_opts = prober_opts
         @metrics     = metrics
         @logger      = logger
@@ -13,7 +13,7 @@ module GitLab
       def probe_all
         @prober_opts.each do |_probe_name, params|
           Utils.wrap_in_array(params[:opts]).each do |opts|
-            prober = params[:class].new(opts, metrics: @metrics, logger: @logger)
+            prober = params[:class].new(metrics: @metrics, logger: @logger, **opts)
             params[:methods].each do |meth|
               prober.send(meth)
             end

data/lib/gitlab_exporter/process.rb

@@ -59,15 +59,15 @@ module GitLab
 
     # Probes a process for info then writes metrics to a target
     class ProcessProber
-      def initialize(options, metrics: PrometheusMetrics.new, logger: nil) # rubocop:disable Lint/UnusedMethodArgument
+      def initialize(name:, metrics: PrometheusMetrics.new, quantiles: false, **options)
         @metrics = metrics
-        @name    = options[:name]
+        @name    = name
         @pids    = if options[:pid_or_pattern] =~ /^\d+$/
                      [options[:pid_or_pattern]]
                    else
                      Utils.pgrep(options[:pid_or_pattern])
                    end
-        @use_quantiles = options.fetch(:quantiles, false)
+        @use_quantiles = quantiles
       end
 
       def probe_stat

data/lib/gitlab_exporter/ruby.rb

@@ -4,9 +4,9 @@ module GitLab
   module Exporter
     # Probes a current process GC for info then writes metrics to a target
     class RubyProber
-      def initialize(options, metrics: PrometheusMetrics.new, logger: nil) # rubocop:disable Lint/UnusedMethodArgument
+      def initialize(metrics: PrometheusMetrics.new, quantiles: false, **opts) # rubocop:disable Lint/UnusedMethodArgument
         @metrics = metrics
-        @use_quantiles = options.fetch(:quantiles, false)
+        @use_quantiles = quantiles
       end
 
       def probe_gc

data/lib/gitlab_exporter/sidekiq.rb

@@ -1,4 +1,5 @@
 require "sidekiq/api"
+require "sidekiq/scheduled"
 require "digest"
 
 module GitLab
@@ -32,7 +33,7 @@ module GitLab
         end
       end
 
-      def initialize(opts, metrics: PrometheusMetrics.new, logger: nil)
+      def initialize(metrics: PrometheusMetrics.new, logger: nil, **opts)
         @opts    = opts
         @metrics = metrics
         @logger  = logger
@@ -76,6 +77,27 @@ module GitLab
         self
       end
 
+      def probe_future_sets
+        now = Time.now.to_f
+        with_sidekiq do
+          Sidekiq.redis do |conn|
+            Sidekiq::Scheduled::SETS.each do |set|
+              # Default to 0; if all jobs are due in the future, there is no "negative" delay.
+              delay = 0
+
+              _job, timestamp = conn.zrangebyscore(set, "-inf", now.to_s, limit: [0, 1], withscores: true).first
+              delay = now - timestamp if timestamp
+
+              @metrics.add("sidekiq_#{set}_set_processing_delay_seconds", delay)
+
+              # zcount is O(log(N)) (prob. binary search), so is still quick even with large sets
+              @metrics.add("sidekiq_#{set}_set_backlog_count",
+                           conn.zcount(set, "-inf", now.to_s))
+            end
+          end
+        end
+      end
+
       # Count worker classes present in Sidekiq queues. This only looks at the
       # first PROBE_JOBS_LIMIT jobs in each queue. This means that we run a
       # single LRANGE command for each queue, which does not block other

data/lib/gitlab_exporter/version.rb

@@ -1,5 +1,5 @@
 module GitLab
   module Exporter
-    VERSION = "11.0.1".freeze
+    VERSION = "11.4.0".freeze
   end
 end

data/lib/gitlab_exporter/web_exporter.rb

@@ -89,7 +89,7 @@ module GitLab
 
             get "/#{probe_name}" do
               content_type "text/plain; version=0.0.4"
-              prober = Prober.new(opts, metrics: PrometheusMetrics.new(include_timestamp: false), logger: logger)
+              prober = Prober.new(metrics: PrometheusMetrics.new(include_timestamp: false), logger: logger, **opts)
 
               prober.probe_all
               prober.write_to(response)

data/spec/database/bloat_spec.rb

@@ -26,7 +26,7 @@ describe GitLab::Exporter::Database::BloatCollector do
 end
 
 describe GitLab::Exporter::Database::BloatProber do
-  let(:opts) { { bloat_types: %i[btree table] } }
+  let(:opts) { { bloat_types: %i[btree table], connection_string: "" } }
   let(:metrics) { double("PrometheusMetrics", add: nil) }
   let(:collector) { double("BloatCollector", run: data) }
 
@@ -43,7 +43,7 @@ describe GitLab::Exporter::Database::BloatProber do
   end
 
   describe "#probe_db" do
-    subject { described_class.new(opts, metrics: metrics, collector: collector, logger: STDOUT).probe_db }
+    subject { described_class.new(metrics: metrics, collector: collector, logger: STDOUT, **opts).probe_db }
 
     before do
       expect(collector).to receive(:logger=).with(STDOUT)
@@ -92,7 +92,7 @@ describe GitLab::Exporter::Database::BloatProber do
   describe "#write_to" do
     let(:target) { double }
     let(:metrics) { double("PrometheusMetrics", to_s: double) }
-    subject { described_class.new(opts, metrics: metrics).write_to(target) }
+    subject { described_class.new(metrics: metrics, **opts).write_to(target) }
 
     it "writes to given target" do
       expect(target).to receive(:write).with(metrics.to_s)

data/spec/database/ci_builds_spec.rb

@@ -216,11 +216,12 @@ describe GitLab::Exporter::Database do
   describe GitLab::Exporter::Database::CiBuildsProber do
     let(:writer) { StringIO.new }
     let(:prober) do
-      opts = { connection_string: "host=localhost",
-               created_builds_counting_disabled: created_builds_counting_disabled,
-               unarchived_traces_offset_minutes: unarchived_traces_offset_minutes }
-      described_class.new(opts,
-                          metrics: GitLab::Exporter::PrometheusMetrics.new(include_timestamp: false))
+      described_class.new(
+        connection_string: "host=localhost",
+        created_builds_counting_disabled: created_builds_counting_disabled,
+        unarchived_traces_offset_minutes: unarchived_traces_offset_minutes,
+        metrics: GitLab::Exporter::PrometheusMetrics.new(include_timestamp: false)
+      )
     end
 
     before do

data/spec/git_spec.rb

@@ -34,17 +34,17 @@ context "With valid pair of repositories" do
   end
 
   describe GitLab::Exporter::GitProber do
-    let(:options) { GitProberOptions.new(repos.cloned_repo, {}) }
+    let(:options) { { source: repos.cloned_repo, labels: {} } }
     let(:output) { StringIO.new }
 
     it "probes and monitors a pull" do
-      prober = GitLab::Exporter::GitProber.new(options)
+      prober = GitLab::Exporter::GitProber.new(**options)
       prober.probe_pull.write_to(output)
       expect(output.string).to match(/git_pull_time_milliseconds \d+ \d+/)
     end
 
     it "probes and monitors a push" do
-      prober = GitLab::Exporter::GitProber.new(options)
+      prober = GitLab::Exporter::GitProber.new(**options)
       prober.probe_push.write_to(output)
       expect(output.string).to match(/git_push_time_milliseconds \d+ \d+/)
     end

data/spec/ruby_spec.rb

@@ -3,7 +3,7 @@ require "gitlab_exporter/ruby"
 require "gitlab_exporter/prober"
 
 describe GitLab::Exporter::RubyProber do
-  let(:prober) { GitLab::Exporter::Prober.new(options) }
+  let(:prober) { GitLab::Exporter::Prober.new(**options) }
 
   let(:options) do
     {

data/spec/spec_helper.rb

@@ -40,8 +40,6 @@ class GitRepoBuilder
   end
 end
 
-GitProberOptions = Struct.new(:source, :labels)
-
 class CLIArgs
   def initialize(args)
     @arguments = args

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: gitlab-exporter
 version: !ruby/object:Gem::Version
-  version: 11.0.1
+  version: 11.4.0
 platform: ruby
 authors:
 - Pablo Carranza
@@ -112,16 +112,16 @@ dependencies:
   name: sinatra
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - '='
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: 2.0.8.1
+        version: 2.1.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - '='
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: 2.0.8.1
+        version: 2.1.0
 - !ruby/object:Gem::Dependency
   name: rspec
   requirement: !ruby/object:Gem::Requirement