Line data    Source code

       1              : use utils::tracing_span_assert::check_fields_present;
       2              : 
       3              : mod extractors {
       4              :     use utils::tracing_span_assert::ConstExtractor;
       5              : 
       6              :     pub(super) const TENANT_ID: ConstExtractor = ConstExtractor::new("tenant_id");
       7              :     pub(super) const SHARD_ID: ConstExtractor = ConstExtractor::new("shard_id");
       8              :     pub(super) const TIMELINE_ID: ConstExtractor = ConstExtractor::new("timeline_id");
       9              : }
      10              : 
      11              : #[track_caller]
      12         4269 : pub(crate) fn debug_assert_current_span_has_tenant_id() {
      13         4269 :     if cfg!(debug_assertions) {
      14         4269 :         if let Err(missing) = check_fields_present!([&extractors::TENANT_ID, &extractors::SHARD_ID])
      15              :         {
      16            0 :             panic!("missing extractors: {missing:?}")
      17         4269 :         }
      18            0 :     }
      19         4269 : }
      20              : 
      21              : #[track_caller]
      22      5592178 : pub(crate) fn debug_assert_current_span_has_tenant_and_timeline_id() {
      23      5592178 :     if cfg!(debug_assertions) {
      24      5592178 :         if let Err(missing) = check_fields_present!([
      25      5592178 :             &extractors::TENANT_ID,
      26      5592178 :             &extractors::SHARD_ID,
      27      5592178 :             &extractors::TIMELINE_ID,
      28      5592178 :         ]) {
      29            0 :             panic!("missing extractors: {missing:?}")
      30      5592178 :         }
      31            0 :     }
      32      5592178 : }
      33              : 
      34              : #[track_caller]
      35      5563691 : pub(crate) fn debug_assert_current_span_has_tenant_and_timeline_id_no_shard_id() {
      36      5563691 :     if cfg!(debug_assertions) {
      37            0 :         if let Err(missing) =
      38      5563691 :             check_fields_present!([&extractors::TENANT_ID, &extractors::TIMELINE_ID,])
      39              :         {
      40            0 :             panic!("missing extractors: {missing:?}")
      41      5563691 :         }
      42            0 :     }
      43      5563691 : }