Method get_buffer

src/protocol/buffer_pool.rs:158–191

Get a buffer from the pool or allocate a new one

(&self)

    
    /// Get a buffer from the pool or allocate a new one
    pub fn get_buffer(&self) -> PooledBytesMut {
        let mut inner = self.inner.lock();
        
        // Try to reuse a buffer from the pool
        if let Some(mut pooled_buffer) = inner.buffers.pop_front() {
            pooled_buffer.reset();
            
            inner.stats.buffers_reused += 1;
            inner.stats.total_bytes_reused += pooled_buffer.buffer.capacity() as u64;
            
            if self.config.enable_monitoring {
                debug!("Reused buffer from pool: capacity={}B, pool_size={}", 
                       pooled_buffer.buffer.capacity(), inner.buffers.len());
            }
            
            PooledBytesMut::new(pooled_buffer.buffer, Arc::clone(&self.inner), self.config.clone())
        } else {
            // Allocate a new buffer
            let buffer = BytesMut::with_capacity(self.config.initial_buffer_capacity);
            
            inner.stats.buffers_allocated += 1;
            inner.stats.total_bytes_allocated += buffer.capacity() as u64;
            
            // Record allocation in global memory monitor (skip in tests)
            #[cfg(not(test))]
            global_memory_monitor().record_buffer_allocation(buffer.capacity() as u64);
            
            if self.config.enable_monitoring {
                debug!("Allocated new buffer: capacity={}B", buffer.capacity());
            }
            
            PooledBytesMut::new(buffer, Arc::clone(&self.inner), self.config.clone())
        }
    }
    
    /// Return a buffer to the pool for reuse
    fn return_buffer(&self, buffer: BytesMut) {