fix scheduled vs enqueued metrics

lib/sidekiq/instrument/middleware/client.rb

@@ -12,12 +12,18 @@ def call(worker_class, job, queue, _redis_pool)
       klass = Object.const_get(worker_class.to_s)
       class_instance = klass.new
 
-      # This is needed because the ClientMiddleware is called twice for scheduled jobs
-      # - Once when it gets scheduled
-      # - Once when it gets dequeued for processing
-      # We only want to increment the enqueue metric when the job is scheduled and
-      # Sidekiq::Context.current[:class] is only ever set when the job is scheduled
-      if Sidekiq::Context.current[:class].present?
+      # Depending on the type of perform called, this method can be hit either
+      # once or twice for the same Job ID.
+      #
+      # perform_async:
+      #   - once when it is enqueued, with no job['at'] key
+      # perform_in:
+      #   - once when it is scheduled, with job['at'] key
+      #   - once when it is enqueued, without job['at'] key
+      if job['at'].present?
+        Statter.statsd.increment(metric_name(class_instance, 'schedule'))
+        Statter.dogstatsd&.increment('sidekiq.schedule', worker_dog_options(class_instance, job))
+      else
         WorkerMetrics.trace_workers_increment_counter(klass.name.underscore)
         Statter.statsd.increment(metric_name(class_instance, 'enqueue'))
         Statter.dogstatsd&.increment('sidekiq.enqueue', worker_dog_options(class_instance, job))

spec/sidekiq-instrument/client_middleware_spec.rb

@@ -28,11 +28,7 @@
       end
     end
 
-    context 'with Sidekiq::Context.current[:class] (job being enqueued)' do
-      before do
-        Sidekiq::Context.current[:class] = 'MyWorker'
-      end
-
+    context 'with nil job["at"] (perform_async)' do
       context 'without statsd_metric_name' do
         it 'increments the StatsD enqueue counter' do
           expect do
@@ -93,36 +89,53 @@
       end
     end
 
-    context 'without the Sidekiq::Context.current[:class] (job being dequeued)' do
-      before do
-        Sidekiq::Context.current[:class] = nil
-      end
+    context 'with job["at"] (perform_in)' do
+      context 'without statsd_metric_name' do
+        it 'increments the StatsD schedule counter' do
+          expect do
+            MyWorker.perform_in(1)
+          end.to trigger_statsd_increment('shared.sidekiq.default.MyWorker.schedule')
+        end
 
-      it 'does not increment the StatsD enqueue counter' do
-        expect do
-          MyWorker.perform_async
-        end.not_to trigger_statsd_increment('shared.sidekiq.default.MyWorker.enqueue')
+        it 'increments the DogStatsD schedule counter' do
+          expect(
+            Sidekiq::Instrument::Statter.dogstatsd
+          ).to receive(:increment).with('sidekiq.schedule', { tags: ['queue:default', 'worker:my_worker'] }).once
+          MyWorker.perform_in(1)
+        end
+
+        context 'with additional tag(s)' do
+          it 'increments DogStatsD schedule counter with additional tag(s)' do
+            tag = 'test_worker'
+
+            expect(
+              Sidekiq::Instrument::Statter.dogstatsd
+            ).to receive(:increment).with('sidekiq.schedule', { tags: ['queue:default', 'worker:my_worker', tag] }).once
+            MyWorker.set(tags: [tag]).perform_in(1)
+          end
+        end
       end
 
-      it 'does not increment the DogStatsD enqueue counter' do
-        expect(
-          Sidekiq::Instrument::Statter.dogstatsd
-        ).not_to receive(:increment).with('sidekiq.enqueue', { tags: ['queue:default', 'worker:my_worker'] })
-        MyWorker.perform_async
+      context 'with statsd_metric_name' do
+        it 'increments the schedule counter' do
+          expect do
+            MyOtherWorker.perform_in(1)
+          end.to trigger_statsd_increment('my_other_worker.schedule')
+        end
       end
 
-      context 'with WorkerMetrics.enabled true' do
+      context 'without optional DogStatsD client' do
         before do
-          Redis.new.flushall
-          Redis.new.hset(worker_metric_name, 'my_other_worker', 0)
+          @tmp = Sidekiq::Instrument::Statter.dogstatsd
+          Sidekiq::Instrument::Statter.dogstatsd = nil
         end
 
-        it 'does not increment the in_queue counter' do
-          Sidekiq::Instrument::WorkerMetrics.enabled = true
-          MyOtherWorker.perform_async
-          expect(Redis.new.hget(worker_metric_name, 'my_other_worker')).to eq('0')
-          MyOtherWorker.perform_async
-          expect(Redis.new.hget(worker_metric_name, 'my_other_worker')).to eq('0')
+        after do
+          Sidekiq::Instrument::Statter.dogstatsd = @tmp
+        end
+
+        it 'does not error' do
+          expect { MyWorker.perform_in(1) }.not_to raise_error
         end
       end
     end