Beispiel #1
0
func TestEnforceQuota(t *testing.T) {
	defer metrics.Reset()
	client := int64(0x1234)
	quota := 1
	SetQuota(client, quota)

	for i := 0; i < 5; i++ {
		updateOps()
		for k := 0; k < quota*flushIntervalSecond; k++ {
			metrics.Counter(stats.ClientCounterName(client)).Add()
			if !HasQuota(client) {
				t.Errorf("#%d.%d: unexpectedly out of quota", i, k)
			}
		}

		metrics.Counter(stats.ClientCounterName(client)).Add()
		if HasQuota(client) {
			t.Errorf("#%d: unexpectedly have quota", i)
		}
	}
}
Beispiel #2
0
func HasQuota(clientID int64) bool {
	if _, ok := quotas[clientID]; !ok {
		return true
	}

	counters, _ := metrics.Snapshot()
	nops := counters[stats.ClientCounterName(clientID)]

	mu.Lock()
	defer mu.Unlock()
	return int(nops)-ops[clientID] <= quotas[clientID]*flushIntervalSecond
}