Merge pull request #749 from msaf1980/limits

limiter: preallocate timeout error
go-graphite · Dec 29, 2022 · 117040a · 117040a
2 parents d04db28 + 5f495d8
commit 117040a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/limiter/interface.go b/limiter/interface.go
@@ -2,8 +2,11 @@ package limiter
 
 import (
 	"context"
+	"errors"
 )
 
+var ErrTimeout = errors.New("timeout exceeded")
+
 type ServerLimiter interface {
 	Capacity() int
 	Enter(ctx context.Context, s string) error

diff --git a/limiter/limiter.go b/limiter/limiter.go
@@ -2,7 +2,6 @@ package limiter
 
 import (
 	"context"
-	"errors"
 )
 
 // ServerLimiter provides interface to limit amount of requests
@@ -44,7 +43,7 @@ func (sl RealLimiter) Enter(ctx context.Context, s string) error {
 	case sl.m[s] <- struct{}{}:
 		return nil
 	case <-ctx.Done():
-		return errors.New("timeout exceeded")
+		return ErrTimeout
 	}
 }