fix: Small perf improvement to generate_r

open-telemetry · Sep 13, 2023 · 9ea275b · 9ea275b
1 parent cf8a18b
commit 9ea275b
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 1 deletion.
diff --git a/api/benchmarks/id_generation_bench.rb b/api/benchmarks/id_generation_bench.rb
@@ -35,6 +35,16 @@ def generate_span_id_while
   id
 end
 
+def generate_r(trace_id)
+  x = trace_id[8, 8].unpack1('Q>') | 0x3
+  64 - x.bit_length
+end
+
+def generate_r_in_place(trace_id)
+  x = trace_id.unpack1('Q>', offset: 8) | 0x3
+  64 - x.bit_length
+end
+
 Benchmark.ipsa do |x|
   x.report('generate_trace_id') { generate_trace_id }
   x.report('generate_trace_id_while') { generate_trace_id_while }
@@ -46,3 +56,10 @@ def generate_span_id_while
   x.report('generate_span_id_while') { generate_span_id_while }
   x.compare!
 end
+
+Benchmark.ipsa do |x|
+  trace_id = generate_trace_id
+  x.report('generate_r') { generate_r(trace_id) }
+  x.report('generate_r_in_place') { generate_r_in_place(trace_id) }
+  x.compare!
+end
diff --git a/sdk_experimental/lib/opentelemetry/sdk/trace/samplers/consistent_probability_tracestate.rb b/sdk_experimental/lib/opentelemetry/sdk/trace/samplers/consistent_probability_tracestate.rb
@@ -106,7 +106,7 @@ def decimal(str)
           end
 
           def generate_r(trace_id)
-            x = trace_id[8, 8].unpack1('Q>') | 0x3
+            x = trace_id.unpack1('Q>', offset: 8) | 0x3
             64 - x.bit_length
           end
         end