|
10 | 10 | //! }; |
11 | 11 | //! use http_body_util::Full; |
12 | 12 | //! use bytes::Bytes; |
13 | | -//! use std::sync::{Arc, atomic::{AtomicU64, Ordering}}; |
| 13 | +//! use std::sync::{Arc, atomic::{AtomicUsize, Ordering}}; |
14 | 14 | //! |
15 | 15 | //! # #[tokio::main] |
16 | 16 | //! # async fn main() -> Result<(), Box<dyn std::error::Error>> { |
|
21 | 21 | //! // A `MakeRequestId` that increments an atomic counter |
22 | 22 | //! #[derive(Clone, Default)] |
23 | 23 | //! struct MyMakeRequestId { |
24 | | -//! counter: Arc<AtomicU64>, |
| 24 | +//! counter: Arc<AtomicUsize>, |
25 | 25 | //! } |
26 | 26 | //! |
27 | 27 | //! impl MakeRequestId for MyMakeRequestId { |
|
68 | 68 | //! # }; |
69 | 69 | //! # use bytes::Bytes; |
70 | 70 | //! # use http_body_util::Full; |
71 | | -//! # use std::sync::{Arc, atomic::{AtomicU64, Ordering}}; |
| 71 | +//! # use std::sync::{Arc, atomic::{AtomicUsize, Ordering}}; |
72 | 72 | //! # #[tokio::main] |
73 | 73 | //! # async fn main() -> Result<(), Box<dyn std::error::Error>> { |
74 | 74 | //! # let handler = tower::service_fn(|request: Request<Full<Bytes>>| async move { |
75 | 75 | //! # Ok::<_, std::convert::Infallible>(Response::new(request.into_body())) |
76 | 76 | //! # }); |
77 | 77 | //! # #[derive(Clone, Default)] |
78 | 78 | //! # struct MyMakeRequestId { |
79 | | -//! # counter: Arc<AtomicU64>, |
| 79 | +//! # counter: Arc<AtomicUsize>, |
80 | 80 | //! # } |
81 | 81 | //! # impl MakeRequestId for MyMakeRequestId { |
82 | 82 | //! # fn make_request_id<B>(&mut self, request: &Request<B>) -> Option<RequestId> { |
|
122 | 122 | //! # }; |
123 | 123 | //! # use http_body_util::Full; |
124 | 124 | //! # use bytes::Bytes; |
125 | | -//! # use std::sync::{Arc, atomic::{AtomicU64, Ordering}}; |
| 125 | +//! # use std::sync::{Arc, atomic::{AtomicUsize, Ordering}}; |
126 | 126 | //! # #[tokio::main] |
127 | 127 | //! # async fn main() -> Result<(), Box<dyn std::error::Error>> { |
128 | 128 | //! # let handler = tower::service_fn(|request: Request<Full<Bytes>>| async move { |
129 | 129 | //! # Ok::<_, std::convert::Infallible>(Response::new(request.into_body())) |
130 | 130 | //! # }); |
131 | 131 | //! # #[derive(Clone, Default)] |
132 | 132 | //! # struct MyMakeRequestId { |
133 | | -//! # counter: Arc<AtomicU64>, |
| 133 | +//! # counter: Arc<AtomicUsize>, |
134 | 134 | //! # } |
135 | 135 | //! # impl MakeRequestId for MyMakeRequestId { |
136 | 136 | //! # fn make_request_id<B>(&mut self, request: &Request<B>) -> Option<RequestId> { |
@@ -483,12 +483,10 @@ mod tests { |
483 | 483 | use crate::test_helpers::Body; |
484 | 484 | use crate::ServiceBuilderExt as _; |
485 | 485 | use http::Response; |
| 486 | + use std::sync::atomic::AtomicUsize; |
486 | 487 | use std::{ |
487 | 488 | convert::Infallible, |
488 | | - sync::{ |
489 | | - atomic::{AtomicU64, Ordering}, |
490 | | - Arc, |
491 | | - }, |
| 489 | + sync::{atomic::Ordering, Arc}, |
492 | 490 | }; |
493 | 491 | use tower::{ServiceBuilder, ServiceExt}; |
494 | 492 |
|
@@ -574,7 +572,7 @@ mod tests { |
574 | 572 | } |
575 | 573 |
|
576 | 574 | #[derive(Clone, Default)] |
577 | | - struct Counter(Arc<AtomicU64>); |
| 575 | + struct Counter(Arc<AtomicUsize>); |
578 | 576 |
|
579 | 577 | impl MakeRequestId for Counter { |
580 | 578 | fn make_request_id<B>(&mut self, _request: &Request<B>) -> Option<RequestId> { |
|
0 commit comments