lib/datadog/core/runtime/metrics.rb in datadog-2.1.0 vs lib/datadog/core/runtime/metrics.rb in datadog-2.2.0

@@ -138,10 +138,11 @@
 
         def gauge_if_not_nil(metric_name, metric_value)
           gauge(metric_name, metric_value) if metric_value
         end
 
+        # rubocop:disable Metrics/MethodLength
         def flush_yjit_stats
           # Only on Ruby >= 3.2
           try_flush do
             if Core::Environment::YJIT.available?
               gauge_if_not_nil(
@@ -174,12 +175,17 @@
               )
               gauge_if_not_nil(
                 Core::Runtime::Ext::Metrics::METRIC_YJIT_OUTLINED_CODE_SIZE,
                 Core::Environment::YJIT.outlined_code_size
               )
+              gauge_if_not_nil(
+                Core::Runtime::Ext::Metrics::METRIC_YJIT_YJIT_ALLOC_SIZE,
+                Core::Environment::YJIT.yjit_alloc_size
+              )
             end
           end
         end
+        # rubocop:enable Metrics/MethodLength
       end
     end
   end
 end