gras: dont update item counts until after work

This lets API calls like get_consumed/produced
keep their values until after work is called.
The propagate tags overload need this to work.

1 files changed, 3 insertions, 2 deletions

diff --git a/lib/gras_impl/input_buffer_queues.hpp b/lib/gras_impl/input_buffer_queues.hpp
index 7d31b9a..b9bd1b7 100644
--- a/lib/gras_impl/input_buffer_queues.hpp
+++ b/lib/gras_impl/input_buffer_queues.hpp
@@ -77,7 +77,7 @@ struct InputBufferQueues
     }
 
     //! Call when input bytes consumed by work
-    void consume(const size_t i, const size_t bytes_consumed);
+    void consume(const size_t i, const size_t items_consumed);
 
     void resize(const size_t size);
 
@@ -330,8 +330,9 @@ GRAS_FORCE_INLINE void InputBufferQueues::push(const size_t i, const SBuffer &bu
 
 }
 
-GRAS_FORCE_INLINE void InputBufferQueues::consume(const size_t i, const size_t bytes_consumed)
+GRAS_FORCE_INLINE void InputBufferQueues::consume(const size_t i, const size_t items_consumed)
 {
+    const size_t bytes_consumed = items_consumed * _items_sizes[i];
     ASSERT(not _queues[i].empty());
     ASSERT((bytes_consumed % _items_sizes[i]) == 0);
     SBuffer &front = _queues[i].front();