| 132 | } |
| 133 | |
| 134 | bool MmapLeakyBucket::HasToken() { |
| 135 | if (NULL == m_poLock || NULL == m_ptBucket) |
| 136 | return -1; |
| 137 | |
| 138 | m_poLock->WriteLockW(0); |
| 139 | |
| 140 | unsigned int tLatestRefillTime = (unsigned int) time(NULL); |
| 141 | tLatestRefillTime = tLatestRefillTime - tLatestRefillTime % (m_tConfig.tRefillPeriod); |
| 142 | |
| 143 | if (m_ptBucket->tLatestRefillTime != tLatestRefillTime) { |
| 144 | m_ptBucket->tLatestRefillTime = tLatestRefillTime; |
| 145 | m_ptBucket->iTokenCount = m_tConfig.iBucketSize; |
| 146 | } |
| 147 | |
| 148 | bool ret = m_ptBucket->iTokenCount > 0; |
| 149 | |
| 150 | m_poLock->Unlock(0); |
| 151 | |
| 152 | return ret; |
| 153 | } |
| 154 | |
| 155 | int MmapLeakyBucket::Apply(int iCount) { |
| 156 | if (NULL == m_poLock || NULL == m_ptBucket) |
no test coverage detected