Skip to content

Commit

Permalink
Fix word and minor changes
Browse files Browse the repository at this point in the history
Signed-off-by: dmaiocchi <[email protected]>
  • Loading branch information
MalloZup committed Jan 29, 2021
1 parent 2e2db6f commit 733f36f
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 8 deletions.
12 changes: 5 additions & 7 deletions providers/tokenbucket/examples_test.go
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
package tokenbucket

import (
middleware "github.com/grpc-ecosystem/go-grpc-middleware/v2"
grpc_ratelimit "github.com/grpc-ecosystem/go-grpc-middleware/v2/interceptors/ratelimit"
"github.com/juju/ratelimit"
"google.golang.org/grpc"
Expand All @@ -17,16 +16,15 @@ const (

// Simple example of server initialization code.
func Example() {

limiter := TockenBucketInterceptor{}
limiter := TokenBucketInterceptor{}
limiter.tokenBucket = ratelimit.NewBucket(rate, int64(tokenCapacity))

_ = grpc.NewServer(
middleware.WithUnaryServerChain(
grpc_ratelimit.UnaryServerInterceptor(limiter),
grpc.ChainUnaryInterceptor(
grpc_ratelimit.UnaryServerInterceptor(&limiter),
),
middleware.WithStreamServerChain(
grpc_ratelimit.UnaryServerInterceptor(limiter),
grpc.ChainStreamInterceptor(
grpc_ratelimit.StreamServerInterceptor(&limiter),
),
)
}
2 changes: 1 addition & 1 deletion providers/tokenbucket/limiter.go
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@ type TokenBucketInterceptor struct {
tokenBucket *ratelimit.Bucket
}

// Limit Implement Limiter interface
// Limit implements Limiter interface.
func (r *TokenBucketInterceptor) Limit(_ context.Context) error {
// Take one token per request. This call doesn't block.
tokenRes := r.tokenBucket.TakeAvailable(1)
Expand Down

0 comments on commit 733f36f

Please sign in to comment.