Move code to pkg

middlewares/ratelimiter/rate_limiter.go

@@ -1,54 +0,0 @@
-package ratelimiter
-
-import (
-	"context"
-	"net/http"
-	"time"
-
-	"github.com/containous/traefik/config"
-	"github.com/containous/traefik/middlewares"
-	"github.com/containous/traefik/tracing"
-	"github.com/opentracing/opentracing-go/ext"
-	"github.com/vulcand/oxy/ratelimit"
-	"github.com/vulcand/oxy/utils"
-)
-
-const (
-	typeName = "RateLimiterType"
-)
-
-type rateLimiter struct {
-	handler http.Handler
-	name    string
-}
-
-// New creates rate limiter middleware.
-func New(ctx context.Context, next http.Handler, config config.RateLimit, name string) (http.Handler, error) {
-	middlewares.GetLogger(ctx, name, typeName).Debug("Creating middleware")
-
-	extractFunc, err := utils.NewExtractor(config.ExtractorFunc)
-	if err != nil {
-		return nil, err
-	}
-
-	rateSet := ratelimit.NewRateSet()
-	for _, rate := range config.RateSet {
-		if err = rateSet.Add(time.Duration(rate.Period), rate.Average, rate.Burst); err != nil {
-			return nil, err
-		}
-	}
-
-	rl, err := ratelimit.New(next, extractFunc, rateSet)
-	if err != nil {
-		return nil, err
-	}
-	return &rateLimiter{handler: rl, name: name}, nil
-}
-
-func (r *rateLimiter) GetTracingInformation() (string, ext.SpanKindEnum) {
-	return r.name, tracing.SpanKindNoneEnum
-}
-
-func (r *rateLimiter) ServeHTTP(rw http.ResponseWriter, req *http.Request) {
-	r.handler.ServeHTTP(rw, req)
-}