From 40c3465ec513f99dc58514789b98e36f7581250e Mon Sep 17 00:00:00 2001 From: Adrian Brink Date: Mon, 26 Feb 2018 14:02:03 +0100 Subject: [PATCH] Add multiple routes simultaneously closes #329 --- baseapp/router.go | 6 ++++-- examples/basecoin/app/app.go | 5 +++-- 2 files changed, 7 insertions(+), 4 deletions(-) diff --git a/baseapp/router.go b/baseapp/router.go index b6b56792bd..a9d7547164 100644 --- a/baseapp/router.go +++ b/baseapp/router.go @@ -8,7 +8,7 @@ import ( // Router provides handlers for each transaction type. type Router interface { - AddRoute(r string, h sdk.Handler) + AddRoute(r string, h sdk.Handler) (rtr Router) Route(path string) (h sdk.Handler) } @@ -34,11 +34,13 @@ func NewRouter() *router { var isAlpha = regexp.MustCompile(`^[a-zA-Z]+$`).MatchString // AddRoute - TODO add description -func (rtr *router) AddRoute(r string, h sdk.Handler) { +func (rtr *router) AddRoute(r string, h sdk.Handler) Router { if !isAlpha(r) { panic("route expressions can only contain alphanumeric characters") } rtr.routes = append(rtr.routes, route{r, h}) + + return rtr } // Route - TODO add description diff --git a/examples/basecoin/app/app.go b/examples/basecoin/app/app.go index 58c25dda80..b26a9bdd45 100644 --- a/examples/basecoin/app/app.go +++ b/examples/basecoin/app/app.go @@ -53,8 +53,9 @@ func NewBasecoinApp(logger log.Logger, db dbm.DB) *BasecoinApp { // add handlers coinKeeper := bank.NewCoinKeeper(app.accountMapper) - app.Router().AddRoute("bank", bank.NewHandler(coinKeeper)) - app.Router().AddRoute("sketchy", sketchy.NewHandler()) + app.Router(). + AddRoute("bank", bank.NewHandler(coinKeeper)). + AddRoute("sketchy", sketchy.NewHandler()) // initialize BaseApp app.SetTxDecoder(app.txDecoder)