From c60e7f52b77bb99eca545a11312cfacead1745a0 Mon Sep 17 00:00:00 2001
From: Nikolay Kim <fafhrd91@gmail.com>
Date: Mon, 8 Apr 2024 12:28:42 +0500
Subject: [PATCH] Move more code under mpool feature (#342)

---
 ntex-bytes/CHANGELOG.md |  4 ++++
 ntex-bytes/Cargo.toml   |  2 +-
 ntex-bytes/src/bytes.rs |  4 ++++
 ntex-bytes/src/pool.rs  | 42 ++++++++++++++++++++++++++++-------------
 4 files changed, 38 insertions(+), 14 deletions(-)

diff --git a/ntex-bytes/CHANGELOG.md b/ntex-bytes/CHANGELOG.md
index 4f8ebe89..50da5647 100644
--- a/ntex-bytes/CHANGELOG.md
+++ b/ntex-bytes/CHANGELOG.md
@@ -1,5 +1,9 @@
 # Changes
 
+## [0.1.27] (2024-04-08)
+
+* Move more code under mpool feature
+
 ## [0.1.26] (2024-04-04)
 
 * Make memory pools optional
diff --git a/ntex-bytes/Cargo.toml b/ntex-bytes/Cargo.toml
index 9d88d9ff..8d3eeb3a 100644
--- a/ntex-bytes/Cargo.toml
+++ b/ntex-bytes/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "ntex-bytes"
-version = "0.1.26"
+version = "0.1.27"
 authors = ["Nikolay Kim <fafhrd91@gmail.com>", "Carl Lerche <me@carllerche.com>"]
 description = "Types and traits for working with bytes (bytes crate fork)"
 documentation = "https://docs.rs/ntex-bytes"
diff --git a/ntex-bytes/src/bytes.rs b/ntex-bytes/src/bytes.rs
index d62e6e4c..77267307 100644
--- a/ntex-bytes/src/bytes.rs
+++ b/ntex-bytes/src/bytes.rs
@@ -1592,6 +1592,7 @@ impl BytesMut {
         self.chunk().iter()
     }
 
+    #[cfg(feature = "mpool")]
     pub(crate) fn move_to_pool(&mut self, pool: PoolRef) {
         self.inner.move_to_pool(pool);
     }
@@ -2417,6 +2418,7 @@ impl BytesVec {
         self.chunk().iter()
     }
 
+    #[cfg(feature = "mpool")]
     pub(crate) fn move_to_pool(&mut self, pool: PoolRef) {
         self.inner.move_to_pool(pool);
     }
@@ -2686,6 +2688,7 @@ impl InnerVec {
         }
     }
 
+    #[cfg(feature = "mpool")]
     #[inline]
     fn move_to_pool(&mut self, pool: PoolRef) {
         unsafe {
@@ -3102,6 +3105,7 @@ impl Inner {
         }
     }
 
+    #[cfg(feature = "mpool")]
     #[inline]
     fn move_to_pool(&mut self, pool: PoolRef) {
         let kind = self.kind();
diff --git a/ntex-bytes/src/pool.rs b/ntex-bytes/src/pool.rs
index dc07147d..33e7b86f 100644
--- a/ntex-bytes/src/pool.rs
+++ b/ntex-bytes/src/pool.rs
@@ -1,8 +1,11 @@
 #![allow(clippy::type_complexity)]
-use std::sync::atomic::{AtomicBool, AtomicUsize, Ordering::Relaxed};
+#[cfg(feature = "mpool")]
+use std::sync::atomic::AtomicBool;
+use std::sync::atomic::{AtomicUsize, Ordering::Relaxed};
 use std::task::{Context, Poll};
 use std::{cell::Cell, cell::RefCell, fmt, future::Future, pin::Pin, ptr, rc::Rc};
 
+#[cfg(feature = "mpool")]
 use futures_core::task::__internal::AtomicWaker;
 
 use crate::{BufMut, BytesMut, BytesVec};
@@ -35,7 +38,9 @@ bitflags::bitflags! {
 
 struct MemoryPool {
     id: PoolId,
+    #[cfg(feature = "mpool")]
     waker: AtomicWaker,
+    #[cfg(feature = "mpool")]
     waker_alive: AtomicBool,
     #[cfg(feature = "mpool")]
     waiters: RefCell<mpool::Waiters>,
@@ -188,13 +193,15 @@ impl PoolRef {
     }
 
     #[inline]
-    pub fn move_in(self, buf: &mut BytesMut) {
-        buf.move_to_pool(self);
+    pub fn move_in(self, _buf: &mut BytesMut) {
+        #[cfg(feature = "mpool")]
+        _buf.move_to_pool(self);
     }
 
     #[inline]
-    pub fn move_vec_in(self, buf: &mut BytesVec) {
-        buf.move_to_pool(self);
+    pub fn move_vec_in(self, _buf: &mut BytesVec) {
+        #[cfg(feature = "mpool")]
+        _buf.move_to_pool(self);
     }
 
     #[inline]
@@ -361,21 +368,28 @@ impl PoolRef {
     }
 
     #[inline]
-    pub(crate) fn acquire(self, size: usize) {
-        let prev = self.0.size.fetch_add(size, Relaxed);
-        if self.0.waker_alive.load(Relaxed) {
-            self.wake_driver(prev + size)
+    pub(crate) fn acquire(self, _size: usize) {
+        #[cfg(feature = "mpool")]
+        {
+            let prev = self.0.size.fetch_add(_size, Relaxed);
+            if self.0.waker_alive.load(Relaxed) {
+                self.wake_driver(prev + _size)
+            }
         }
     }
 
     #[inline]
-    pub(crate) fn release(self, size: usize) {
-        let prev = self.0.size.fetch_sub(size, Relaxed);
-        if self.0.waker_alive.load(Relaxed) {
-            self.wake_driver(prev - size)
+    pub(crate) fn release(self, _size: usize) {
+        #[cfg(feature = "mpool")]
+        {
+            let prev = self.0.size.fetch_sub(_size, Relaxed);
+            if self.0.waker_alive.load(Relaxed) {
+                self.wake_driver(prev - _size)
+            }
         }
     }
 
+    #[cfg(feature = "mpool")]
     fn wake_driver(self, allocated: usize) {
         let l = self.0.window_l.get();
         let h = self.0.window_h.get();
@@ -428,7 +442,9 @@ impl MemoryPool {
     fn create(id: PoolId) -> &'static MemoryPool {
         Box::leak(Box::new(MemoryPool {
             id,
+            #[cfg(feature = "mpool")]
             waker: AtomicWaker::new(),
+            #[cfg(feature = "mpool")]
             waker_alive: AtomicBool::new(false),
             #[cfg(feature = "mpool")]
             waiters: RefCell::new(mpool::Waiters::new()),