traefik/pkg/middlewares/maxconnection/max_connection.go

package maxconnection

import (
	"context"
	"fmt"
	"net/http"

	"github.com/containous/traefik/v2/pkg/config/dynamic"
	"github.com/containous/traefik/v2/pkg/middlewares"
	"github.com/containous/traefik/v2/pkg/tracing"
	"github.com/opentracing/opentracing-go/ext"
	"github.com/vulcand/oxy/connlimit"
	"github.com/vulcand/oxy/utils"
)

const (
	typeName = "MaxConnection"
)

type maxConnection struct {
	handler http.Handler
	name    string
}

// New creates a max connection middleware.
func New(ctx context.Context, next http.Handler, maxConns dynamic.MaxConn, name string) (http.Handler, error) {
	middlewares.GetLogger(ctx, name, typeName).Debug("Creating middleware")

	extractFunc, err := utils.NewExtractor(maxConns.ExtractorFunc)
	if err != nil {
		return nil, fmt.Errorf("error creating connection limit: %v", err)
	}

	handler, err := connlimit.New(next, extractFunc, maxConns.Amount)
	if err != nil {
		return nil, fmt.Errorf("error creating connection limit: %v", err)
	}

	return &maxConnection{handler: handler, name: name}, nil
}

func (mc *maxConnection) GetTracingInformation() (string, ext.SpanKindEnum) {
	return mc.name, tracing.SpanKindNoneEnum
}

func (mc *maxConnection) ServeHTTP(rw http.ResponseWriter, req *http.Request) {
	mc.handler.ServeHTTP(rw, req)
}