Move code to pkg

pkg/middlewares/recovery/recovery.go

@@ -0,0 +1,40 @@
+package recovery
+
+import (
+	"context"
+	"net/http"
+
+	"github.com/containous/traefik/pkg/middlewares"
+	"github.com/sirupsen/logrus"
+)
+
+const (
+	typeName = "Recovery"
+)
+
+type recovery struct {
+	next http.Handler
+	name string
+}
+
+// New creates recovery middleware.
+func New(ctx context.Context, next http.Handler, name string) (http.Handler, error) {
+	middlewares.GetLogger(ctx, name, typeName).Debug("Creating middleware")
+
+	return &recovery{
+		next: next,
+		name: name,
+	}, nil
+}
+
+func (re *recovery) ServeHTTP(rw http.ResponseWriter, req *http.Request) {
+	defer recoverFunc(middlewares.GetLogger(req.Context(), re.name, typeName), rw)
+	re.next.ServeHTTP(rw, req)
+}
+
+func recoverFunc(logger logrus.FieldLogger, rw http.ResponseWriter) {
+	if err := recover(); err != nil {
+		logger.Errorf("Recovered from panic in http handler: %+v", err)
+		http.Error(rw, http.StatusText(http.StatusInternalServerError), http.StatusInternalServerError)
+	}
+}