Implements #init_label_set method for all metric classes

Author: Joao Amaro

lib/prometheus/client/histogram.rb

@@ -94,6 +94,23 @@ def values
         end
       end
 
+      def init_label_set(labels)
+        base_label_set = label_set_for(labels)
+
+        @store.synchronize do
+          [*@buckets, "+Inf"].each do |bucket|
+            # This is basically faster than doing `.merge`
+            bucket_label_set = base_label_set.dup
+            bucket_label_set[:le] = bucket.to_s
+            sum_label_set = base_label_set.dup
+            sum_label_set[:le] = "sum"
+
+            @store.set(labels: bucket_label_set, val: 0)
+            @store.set(labels: sum_label_set, val: 0)
+          end
+        end
+      end
+
       private
 
       # Modifies the passed in parameter

lib/prometheus/client/metric.rb

@@ -55,6 +55,10 @@ def with_labels(labels)
                        store_settings: @store_settings)
       end
 
+      def init_label_set(labels)
+        @store.set(labels: label_set_for(labels), val: 0)
+      end
+
       # Returns all label sets with their values
       def values
         @store.all_values

lib/prometheus/client/summary.rb

@@ -45,6 +45,15 @@ def values
         end
       end
 
+      def init_label_set(labels)
+        base_label_set = label_set_for(labels)
+
+        @store.synchronize do
+          @store.set(labels: base_label_set.merge(quantile: "count"), val: 0)
+          @store.set(labels: base_label_set.merge(quantile: "sum"), val: 0)
+        end
+      end
+
       private
 
       def reserved_labels

spec/prometheus/client/counter_spec.rb

@@ -79,4 +79,16 @@
       end.to change { counter.get }.by(100.0)
     end
   end
+
+  describe '#init_label_set' do
+    let(:expected_labels) { [:test] }
+
+    it 'initializes the metric for a given label set' do
+      expect(counter.values).to eql({})
+
+      counter.init_label_set(test: 'value')
+
+      expect(counter.values).to eql({test: 'value'} => 0.0)
+    end
+  end
 end

spec/prometheus/client/gauge_spec.rb

@@ -161,4 +161,16 @@
       end.to change { gauge.get }.by(-100.0)
     end
   end
+
+  describe '#init_label_set' do
+    let(:expected_labels) { [:test] }
+
+    it 'initializes the metric for a given label set' do
+      expect(gauge.values).to eql({})
+
+      gauge.init_label_set(test: 'value')
+
+      expect(gauge.values).to eql({test: 'value'} => 0.0)
+    end
+  end
 end

spec/prometheus/client/histogram_spec.rb

@@ -118,4 +118,20 @@
       )
     end
   end
+
+  describe '#init_label_set' do
+    let(:expected_labels) { [:status] }
+
+    it 'initializes the metric for a given label set' do
+      expect(histogram.values).to eql({})
+
+      histogram.init_label_set(status: 'bar')
+      histogram.init_label_set(status: 'foo')
+
+      expect(histogram.values).to eql(
+        { status: 'bar' } => { "2.5" => 0.0, "5" => 0.0, "10" => 0.0, "+Inf" => 0.0, "sum" => 0.0 },
+        { status: 'foo' } => { "2.5" => 0.0, "5" => 0.0, "10" => 0.0, "+Inf" => 0.0, "sum" => 0.0 },
+      )
+    end
+  end
 end

spec/prometheus/client/summary_spec.rb

@@ -99,4 +99,20 @@
       )
     end
   end
+
+  describe '#init_label_set' do
+    let(:expected_labels) { [:status] }
+
+    it 'initializes the metric for a given label set' do
+      expect(summary.values).to eql({})
+
+      summary.init_label_set(status: 'bar')
+      summary.init_label_set(status: 'foo')
+
+      expect(summary.values).to eql(
+        { status: 'bar' } => { "count" => 0.0, "sum" => 0.0 },
+        { status: 'foo' } => { "count" => 0.0, "sum" => 0.0 },
+      )
+    end
+  end
 end