func TestHandleIncrMissBucket(t *testing.T) {
listener = stats.NewMemoryStatsListener()
listener.HandleEvent(events.NewBucketMissedEvent("test", "dyn", true))
listener.HandleEvent(events.NewBucketMissedEvent("test", "dyn", true))
listener.HandleEvent(events.NewBucketMissedEvent("test", "dyn", true))
scores := listener.Get("test", "dyn")
if scores.Hits != 0 || scores.Misses != 3 {
t.Fatalf("Bucket score was not accurate: %+v != [Hits=0, Misses=3]", scores)
}
}
func TestHandleIncrMissBucket(t *testing.T) {
namespace = randomNamespace()
listener.HandleEvent(events.NewBucketMissedEvent(namespace, "incr-miss", true))
listener.HandleEvent(events.NewBucketMissedEvent(namespace, "incr-miss", true))
listener.HandleEvent(events.NewBucketMissedEvent(namespace, "incr-miss", true))
scores := listener.Get(namespace, "incr-miss")
if scores.Hits != 0 || scores.Misses != 3 {
t.Fatalf("Bucket score was not accurate: %+v != [Hits=0, Misses=3]", scores)
}
}
func TestHandleNewMissBucket(t *testing.T) {
namespace = randomNamespace()
listener.HandleEvent(events.NewBucketMissedEvent(namespace, "new-miss-dyn", true))
scores := listener.Get(namespace, "new-miss-dyn")
if scores.Hits != 0 || scores.Misses != 1 {
t.Fatalf("Bucket score was not accurate: %+v != [Hits=0, Misses=1]", scores)
}
listener.HandleEvent(events.NewBucketMissedEvent(namespace, "nondyn", false))
scores = listener.Get(namespace, "nondyn")
if scores.Hits != 0 || scores.Misses != 0 {
t.Fatalf("Non-dynamic bucket score was not accurate: %+v != [Hits=0, Misses=0]", scores)
}
}
func TestHandleNewMissBucket(t *testing.T) {
listener = stats.NewMemoryStatsListener()
listener.HandleEvent(events.NewBucketMissedEvent("test", "dyn", true))
scores := listener.Get("test", "dyn")
if scores.Hits != 0 || scores.Misses != 1 {
t.Fatalf("Bucket score was not accurate: %+v != [Hits=0, Misses=1]", scores)
}
listener.HandleEvent(events.NewBucketMissedEvent("test", "nondyn", false))
scores = listener.Get("test", "nondyn")
if scores.Hits != 0 || scores.Misses != 0 {
t.Fatalf("Non-dynamic bucket score was not accurate: %+v != [Hits=0, Misses=0]", scores)
}
}
func (s *server) Allow(namespace, name string, tokensRequested int64, maxWaitMillisOverride int64, maxWaitTimeOverride bool) (time.Duration, error) {
s.RLock()
b, e := s.bucketContainer.FindBucket(namespace, name)
s.RUnlock()
if e != nil {
// Attempted to create a dynamic bucket and failed.
s.Emit(events.NewBucketMissedEvent(namespace, name, true))
return 0, newError("Cannot create dynamic bucket "+config.FullyQualifiedName(namespace, name), ER_TOO_MANY_BUCKETS)
}
if b == nil {
s.Emit(events.NewBucketMissedEvent(namespace, name, false))
return 0, newError("No such bucket "+config.FullyQualifiedName(namespace, name), ER_NO_BUCKET)
}
if b.Config().MaxTokensPerRequest < tokensRequested && b.Config().MaxTokensPerRequest > 0 {
s.Emit(events.NewTooManyTokensRequestedEvent(namespace, name, b.Dynamic(), tokensRequested))
return 0, newError(fmt.Sprintf("Too many tokens requested. Bucket %v:%v, tokensRequested=%v, maxTokensPerRequest=%v",
namespace, name, tokensRequested, b.Config().MaxTokensPerRequest),
ER_TOO_MANY_TOKENS_REQUESTED)
}
maxWaitTime := time.Millisecond
if maxWaitTimeOverride && maxWaitMillisOverride < b.Config().WaitTimeoutMillis {
// Use the max wait time override from the request.
maxWaitTime *= time.Duration(maxWaitMillisOverride)
} else {
// Fall back to the max wait time configured on the bucket.
maxWaitTime *= time.Duration(b.Config().WaitTimeoutMillis)
}
w, success := b.Take(tokensRequested, maxWaitTime)
if !success {
// Could not claim tokens within the given max wait time
s.Emit(events.NewTimedOutEvent(namespace, name, b.Dynamic(), tokensRequested))
return 0, newError(fmt.Sprintf("Timed out waiting on %v:%v", namespace, name), ER_TIMEOUT)
}
// The only positive result
s.Emit(events.NewTokensServedEvent(namespace, name, b.Dynamic(), tokensRequested, w))
return w, nil
}
func TestTop10Misses(t *testing.T) {
namespace = randomNamespace()
listener.HandleEvent(events.NewBucketMissedEvent(namespace, "misses-dyn-1", true))
listener.HandleEvent(events.NewBucketMissedEvent(namespace, "misses-dyn-2", true))
listener.HandleEvent(events.NewBucketMissedEvent(namespace, "misses-dyn-2", true))
listener.HandleEvent(events.NewBucketMissedEvent(namespace, "misses-dyn-2", true))
listener.HandleEvent(events.NewBucketMissedEvent(namespace, "misses-dyn-3", true))
listener.HandleEvent(events.NewBucketMissedEvent(namespace, "misses-dyn-3", true))
misses := listener.TopMisses(namespace)
correctMisses := []*stats.BucketScore{
&stats.BucketScore{"misses-dyn-2", 3},
&stats.BucketScore{"misses-dyn-3", 2},
&stats.BucketScore{"misses-dyn-1", 1}}
if !reflect.DeepEqual(misses, correctMisses) {
t.Fatalf("Misses top10 is not correct %+v", misses)
}
}
func TestTop10Misses(t *testing.T) {
listener = stats.NewMemoryStatsListener()
listener.HandleEvent(events.NewBucketMissedEvent("test", "dyn-1", true))
listener.HandleEvent(events.NewBucketMissedEvent("test", "dyn-2", true))
listener.HandleEvent(events.NewBucketMissedEvent("test", "dyn-2", true))
listener.HandleEvent(events.NewBucketMissedEvent("test", "dyn-2", true))
listener.HandleEvent(events.NewBucketMissedEvent("test", "dyn-3", true))
listener.HandleEvent(events.NewBucketMissedEvent("test", "dyn-3", true))
misses := listener.TopMisses("test")
correctMisses := []*stats.BucketScore{
&stats.BucketScore{"dyn-2", 3},
&stats.BucketScore{"dyn-3", 2},
&stats.BucketScore{"dyn-1", 1}}
if !reflect.DeepEqual(misses, correctMisses) {
t.Fatalf("Misses top10 is not correct %+v", misses)
}
}