[core] CUnitQueue: get size and capacity

maxsharabayko · Jun 9, 2020 · 78b5fcf · 78b5fcf
1 parent 49f5ec6
commit 78b5fcf
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/srtcore/queue.cpp b/srtcore/queue.cpp
@@ -164,7 +164,7 @@ int CUnitQueue::increase()
     char *   tempb = NULL;
 
     // all queues have the same size
-    int size = m_pQEntry->m_iSize;
+    const int size = m_pQEntry->m_iSize;
 
     try
     {

diff --git a/srtcore/queue.h b/srtcore/queue.h
@@ -101,6 +101,10 @@ class CUnitQueue
 
    int shrink();
 
+public:
+   int size() const     { return m_iSize - m_iCount; }
+   int capacity() const { return m_iSize; }
+
 public:     // Operations on units
 
       /// find an available unit for incoming packet.
@@ -131,9 +135,9 @@ class CUnitQueue
    CUnit* m_pAvailUnit;         // recent available unit
 
    int m_iSize;			// total size of the unit queue, in number of packets
-   int m_iCount;		// total number of valid packets in the queue
+   int m_iCount;		   // total number of valid (occupied) packets in the queue
 
-   int m_iMSS;			// unit buffer size
+   int m_iMSS;			   // unit buffer size
    int m_iIPversion;		// IP version
 
 private: