Extend routes

Author: roshangade

README.md

@@ -59,6 +59,27 @@ fmt.Println("Starting server.")
 http.ListenAndServe(":8080", api)
 ```
 
+
+## Extend routes
+```
+var user = rest.Extend("/user", api)
+
+user.Use(func(ctx *rest.Context) {
+    //User middleware will execute for /user/* routes
+})
+
+user.Get("/:uid/profile", func(ctx *rest.Context) {
+    ctx.JSON(`{"user": "profile"}`)
+})
+
+user.Get("/:uid", func(ctx *rest.Context) {
+    ctx.JSON(ctx.Params)
+})
+```
+
+###Pending 
+- Stop execution on timeout/abort
+
 ## Documentation
 https://godoc.org/github.com/go-rs/rest-api-framework
 

examples/server.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"fmt"
+	"log"
 	"net/http"
 	"strconv"
 	"time"
@@ -43,6 +44,7 @@ func main() {
 
 	api.Get("/foo", func(ctx *rest.Context) {
 		ctx.Throw("UNAUTHORIZED")
+		// can also use ctx.ThrowWithError("SERVER_ERROR", error)
 	})
 
 	// error handler
@@ -52,11 +54,15 @@ func main() {
 
 	fmt.Println("Starting server.")
 
+	tout := http.TimeoutHandler(api, 100*time.Millisecond, "timeout")
+
 	server := http.Server{
 		Addr:    ":8080",
-		Handler: api,
+		Handler: tout,
+	}
+
+	if err := server.ListenAndServe(); err != nil {
+		log.Fatalf("could not start server, %v", err)
 	}
 
-	err := server.ListenAndServe()
-	fmt.Println(err)
 }

examples/user/user.go

@@ -6,9 +6,7 @@ import (
 
 func Load(api *rest.API) {
 
-	var user rest.Namespace
-
-	user.Set("/user", api)
+	var user = rest.Extend("/user", api)
 
 	user.Use(func(ctx *rest.Context) {
 		println("User middleware > /user/*")

namespace.go

@@ -5,8 +5,9 @@
 package rest
 
 import (
-	"github.com/go-rs/rest-api-framework/utils"
 	"net/http"
+
+	"github.com/go-rs/rest-api-framework/utils"
 )
 
 // Namespace is used to extend routes with a specific prefix
@@ -16,10 +17,12 @@ type Namespace struct {
 }
 
 // Set method is used to set prefix with API
-func (n *Namespace) Set(prefix string, api *API) {
+func Extend(prefix string, api *API) *Namespace {
 	prefix = utils.TrimSuffix(prefix, "/")
-	n.prefix = prefix
-	n.api = api
+	return &Namespace{
+		prefix: prefix,
+		api:    api,
+	}
 }
 
 // Use method is used to set interceptors/middlewares for all requests,

namespace_test.go

@@ -9,11 +9,11 @@ import (
 )
 
 var api API
-var ns Namespace
+var ns *Namespace
 var handle Handler
 
-func TestNamespace_Set(t *testing.T) {
-	ns.Set("/test", &api)
+func TestExtend(t *testing.T) {
+	ns = Extend("/test", &api)
 
 	if ns.prefix != "/test" {
 		t.Error("Prefix is not set.")