add one more command middleware handler pattern func(http.ResponseWriter, *http.Request, http.Handler)

Author: joeybloggs

context_test.go

@@ -271,6 +271,18 @@ func TestNativeHandlersAndParseForm(t *testing.T) {
 	code, body = request(GET, "/chain-handler", l6)
 	Equal(t, code, http.StatusOK)
 	Equal(t, body, "aok")
+
+	l7 := New()
+	l7.Get("/chain-handler", func(w http.ResponseWriter, r *http.Request, next http.Handler) {
+		w.Write([]byte("a"))
+		next.ServeHTTP(w, r)
+	}, func(c Context) {
+		c.Response().Write([]byte("ok"))
+	})
+
+	code, body = request(GET, "/chain-handler", l7)
+	Equal(t, code, http.StatusOK)
+	Equal(t, body, "aok")
 }
 
 func TestNativeHandlersAndParseMultiPartForm(t *testing.T) {

lars.go

@@ -194,7 +194,6 @@ func New() *LARS {
 	}
 
 	l.routeGroup.lars = l
-	// l.router = newRouter(l)
 	l.pool.New = func() interface{} {
 
 		c := l.contextFunc(l)

util.go

@@ -8,8 +8,8 @@ import (
 	"runtime"
 )
 
-// NativeChainHandler is used in native handler chains
-// example using nosurf crsf middleware nosurf.NewPure(lars.NativeChainHandlerFunc)
+// NativeChainHandler is used in native handler chain middleware
+// example using nosurf crsf middleware nosurf.NewPure(lars.NativeChainHandler)
 var NativeChainHandler = http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 
 	c := GetContext(w)
@@ -73,7 +73,15 @@ func (l *LARS) wrapHandler(h Handler) HandlerFunc {
 			}
 		}
 
-	case func(handler http.Handler) http.Handler:
+	case func(http.ResponseWriter, *http.Request, http.Handler):
+
+		return func(c Context) {
+			ctx := c.BaseContext()
+
+			h(ctx.response, ctx.request, NativeChainHandler)
+		}
+
+	case func(http.Handler) http.Handler:
 
 		hf := h(NativeChainHandler)