Move code to pkg

pkg/middlewares/buffering/buffering.go

@@ -0,0 +1,54 @@
+package buffering
+
+import (
+	"context"
+	"net/http"
+
+	"github.com/containous/traefik/pkg/config"
+	"github.com/containous/traefik/pkg/middlewares"
+	"github.com/containous/traefik/pkg/tracing"
+	"github.com/opentracing/opentracing-go/ext"
+	oxybuffer "github.com/vulcand/oxy/buffer"
+)
+
+const (
+	typeName = "Buffer"
+)
+
+type buffer struct {
+	name   string
+	buffer *oxybuffer.Buffer
+}
+
+// New creates a buffering middleware.
+func New(ctx context.Context, next http.Handler, config config.Buffering, name string) (http.Handler, error) {
+	logger := middlewares.GetLogger(ctx, name, typeName)
+	logger.Debug("Creating middleware")
+	logger.Debug("Setting up buffering: request limits: %d (mem), %d (max), response limits: %d (mem), %d (max) with retry: '%s'",
+		config.MemRequestBodyBytes, config.MaxRequestBodyBytes, config.MemResponseBodyBytes, config.MaxResponseBodyBytes, config.RetryExpression)
+
+	oxyBuffer, err := oxybuffer.New(
+		next,
+		oxybuffer.MemRequestBodyBytes(config.MemRequestBodyBytes),
+		oxybuffer.MaxRequestBodyBytes(config.MaxRequestBodyBytes),
+		oxybuffer.MemResponseBodyBytes(config.MemResponseBodyBytes),
+		oxybuffer.MaxResponseBodyBytes(config.MaxResponseBodyBytes),
+		oxybuffer.CondSetter(len(config.RetryExpression) > 0, oxybuffer.Retry(config.RetryExpression)),
+	)
+	if err != nil {
+		return nil, err
+	}
+
+	return &buffer{
+		name:   name,
+		buffer: oxyBuffer,
+	}, nil
+}
+
+func (b *buffer) GetTracingInformation() (string, ext.SpanKindEnum) {
+	return b.name, tracing.SpanKindNoneEnum
+}
+
+func (b *buffer) ServeHTTP(rw http.ResponseWriter, req *http.Request) {
+	b.buffer.ServeHTTP(rw, req)
+}