Fix missed methods

tokio-rs · Oct 21, 2024 · b437e47 · b437e47
1 parent 9b25f46
commit b437e47
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/tokio/src/runtime/builder.rs b/tokio/src/runtime/builder.rs
@@ -1137,6 +1137,7 @@ impl Builder {
         ///
         /// [`enable_metrics_poll_time_histogram()`]: Builder::enable_metrics_poll_time_histogram
         #[deprecated(note = "`poll_count_histogram` related methods have been renamed `poll_time_histogram` to better reflect their functionality.")]
+        #[doc(hidden)]
         pub fn enable_metrics_poll_count_histogram(&mut self) -> &mut Self {
             self.enable_metrics_poll_time_histogram()
         }

diff --git a/tokio/src/runtime/metrics/runtime.rs b/tokio/src/runtime/metrics/runtime.rs
@@ -813,7 +813,6 @@ impl RuntimeMetrics {
         #[deprecated(note = "renamed to `poll_time_histogram_num_buckets`.")]
         pub fn poll_count_histogram_num_buckets(&self) -> usize {
             self.poll_time_histogram_num_buckets()
-
         }
 
         /// Returns the range of task poll times tracked by the given bucket.
@@ -933,8 +932,7 @@ impl RuntimeMetrics {
             ///
             /// [`poll_time_histogram_bucket_range()`]: crate::runtime::RuntimeMetrics::poll_time_histogram_bucket_range
             #[track_caller]
-            #[doc(hidden)]
-            pub fn poll_count_histogram_bucket_count(&self, worker: usize, bucket: usize) -> u64 {
+            pub fn poll_time_histogram_bucket_count(&self, worker: usize, bucket: usize) -> u64 {
                 self.handle
                     .inner
                     .worker_metrics(worker)
@@ -944,6 +942,12 @@ impl RuntimeMetrics {
                     .unwrap_or_default()
             }
 
+            #[doc(hidden)]
+            #[deprecated(note = "use `poll_time_histogram_bucket_count` instead")]
+            pub fn poll_count_histogram_bucket_count(&self, worker: usize, bucket: usize) -> u64 {
+                self.poll_time_histogram_bucket_count(worker, bucket)
+            }
+
             /// Returns the mean duration of task polls, in nanoseconds.
             ///
             /// This is an exponentially weighted moving average. Currently, this metric