Fixed djb2 arguments

Author: rads-1996

sdk/monitor/azure-monitor-opentelemetry-exporter/azure/monitor/opentelemetry/exporter/export/trace/_rate_limited_sampling.py

@@ -104,7 +104,7 @@ def should_sample(
                 return parent_result
 
         sampling_percentage = self._sampling_percentage_generator.get()
-        sampling_score = _get_DJB2_sample_score(format_trace_id(trace_id).lower())
+        sampling_score = _get_DJB2_sample_score(format_trace_id(trace_id).lower(), "rate_limited")
 
         if sampling_score < sampling_percentage:
             decision = Decision.RECORD_AND_SAMPLE

sdk/monitor/azure-monitor-opentelemetry-exporter/azure/monitor/opentelemetry/exporter/export/trace/_utils.py

@@ -340,7 +340,7 @@ def _get_url_for_http_request(attributes: Attributes) -> Optional[str]:
                 )
     return url
 
-def _get_DJB2_sample_score(trace_id_hex: str) -> float:
+def _get_DJB2_sample_score(trace_id_hex: str, sampler_type) -> float:
     # This algorithm uses 32bit integers
     hash_value = Int32(_SAMPLING_HASH)
     for char in trace_id_hex:
@@ -351,8 +351,12 @@ def _get_DJB2_sample_score(trace_id_hex: str) -> float:
     else:
         hash_value = abs(hash_value)
 
-    # divide by _INTEGER_MAX for value between 0 and 1, then multiply by 100 for percentage
-    return 100.0 * (float(hash_value) / _INTEGER_MAX)
+    if sampler_type == "rate_limited":
+        # divide by _INTEGER_MAX for value between 0 and 1, then multiply by 100 for percentage
+        return 100.0 * (float(hash_value) / _INTEGER_MAX)
+    else:
+        # divide by _INTEGER_MAX for value between 0 and 1 for sampling score
+        return float(hash_value) / _INTEGER_MAX 
 
 def _round_down_to_nearest(sampling_percentage: float) -> float:
     if sampling_percentage == 0:

sdk/monitor/azure-monitor-opentelemetry-exporter/tests/trace/test_rate_limited_sampling.py

@@ -469,7 +469,7 @@ def test_djb2_hash_consistency(self):
 
         trace_id = "test-trace-id-12345"
 
-        scores = [_get_DJB2_sample_score(trace_id) for _ in range(10)]
+        scores = [_get_DJB2_sample_score(trace_id, "rate_limited") for _ in range(10)]
 
         self.assertTrue(all(score == scores[0] for score in scores))
 
@@ -485,7 +485,7 @@ def test_djb2_hash_edge_cases(self):
 
         for trace_id in edge_cases:
             with self.subTest(trace_id=trace_id):
-                score = _get_DJB2_sample_score(trace_id)
+                score = _get_DJB2_sample_score(trace_id, "rate_limited")
                 self.assertIsInstance(score, float)
                 self.assertGreaterEqual(score, 0)
                 self.assertLess(score, 100)