fix build

bytedance · Jul 24, 2024 · 9d83d3b · 9d83d3b
1 parent 85de8a8
commit 9d83d3b
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/lib/g3-io-ext/src/quic/limited_socket.rs b/lib/g3-io-ext/src/quic/limited_socket.rs
@@ -341,13 +341,13 @@ impl AsyncUdpSocket for LimitedUdpSocket {
         let l = unsafe { &mut *self.recv_state.get() };
         if l.limit.is_set() {
             let dur_millis = l.started.elapsed().as_millis() as u64;
-            let mut total_size_v = SmallVec::with_capacity(meta.len());
+            let mut total_size_v = SmallVec::<[usize; 16]>::with_capacity(meta.len());
             let mut total_size = 0;
             for b in bufs.iter() {
                 total_size += b.len();
                 total_size_v.push(total_size);
             }
-            match l.limit.check_packets(dur_millis, &total_size_v) {
+            match l.limit.check_packets(dur_millis, total_size_v.as_ref()) {
                 DatagramLimitAction::Advance(n) => {
                     match self.inner.poll_recv(cx, &mut bufs[0..n], &mut meta[0..n]) {
                         Poll::Ready(Ok(nr)) => {

diff --git a/lib/g3-io-ext/src/udp/recv.rs b/lib/g3-io-ext/src/udp/recv.rs
@@ -226,13 +226,13 @@ where
 
         if self.limit.is_set() {
             let dur_millis = self.started.elapsed().as_millis() as u64;
-            let mut total_size_v = SmallVec::with_capacity(hdr_v.len());
+            let mut total_size_v = SmallVec::<[usize; 16]>::with_capacity(hdr_v.len());
             let mut total_size = 0usize;
             for hdr in hdr_v.iter() {
                 total_size += hdr.iov.iter().map(|v| v.len()).sum::<usize>();
                 total_size_v.push(total_size);
             }
-            match self.limit.check_packets(dur_millis, &total_size_v) {
+            match self.limit.check_packets(dur_millis, total_size_v.as_ref()) {
                 DatagramLimitAction::Advance(n) => {
                     match self.inner.poll_batch_recvmsg(cx, &mut hdr_v[0..n]) {
                         Poll::Ready(Ok(count)) => {

diff --git a/lib/g3-io-ext/src/udp/send.rs b/lib/g3-io-ext/src/udp/send.rs
@@ -288,13 +288,13 @@ where
 
         if self.limit.is_set() {
             let dur_millis = self.started.elapsed().as_millis() as u64;
-            let mut total_size_v = SmallVec::with_capacity(msgs.len());
+            let mut total_size_v = SmallVec::<[usize; 16]>::with_capacity(msgs.len());
             let mut total_size = 0;
             for msg in msgs.iter() {
                 total_size += msg.iov.iter().map(|v| v.len()).sum::<usize>();
                 total_size_v.push(total_size);
             }
-            match self.limit.check_packets(dur_millis, &total_size_v) {
+            match self.limit.check_packets(dur_millis, total_size_v.as_ref()) {
                 DatagramLimitAction::Advance(n) => {
                     match self.inner.poll_batch_sendmsg(cx, &mut msgs[0..n]) {
                         Poll::Ready(Ok(count)) => {