Merge pull request #149 from zhu327/feat_cors

feat: application add UseGlobal method

Author: john-deng

pkg/app/application.go

@@ -32,6 +32,7 @@ import (
 	"github.com/hidevopsio/hiboot/pkg/system/scheduler"
 	"github.com/hidevopsio/hiboot/pkg/utils/cmap"
 	"github.com/hidevopsio/hiboot/pkg/utils/io"
+	"github.com/kataras/iris/core/router"
 )
 
 const (
@@ -60,6 +61,7 @@ type ApplicationContext interface {
 	Use(handlers ...context.Handler)
 	GetProperty(name string) (value interface{}, ok bool)
 	GetInstance(params ...interface{}) (instance interface{})
+	WrapRouter(handler router.WrapperFunc)
 }
 
 // BaseApplication is the base application
@@ -83,7 +85,7 @@ var (
 	configContainer    []*factory.MetaData
 	componentContainer []*factory.MetaData
 	// Profiles include profiles initially
-	Profiles           []string
+	Profiles []string
 
 	// ErrInvalidObjectType indicates that configuration type is invalid
 	ErrInvalidObjectType = errors.New("[app] invalid Configuration type, one of app.Configuration need to be embedded")

pkg/app/fake/application.go

@@ -15,7 +15,10 @@
 // Package fake provides fake.ApplicationContext for unit testing
 package fake
 
-import "github.com/hidevopsio/hiboot/pkg/app/web/context"
+import (
+	"github.com/hidevopsio/hiboot/pkg/app/web/context"
+	"github.com/kataras/iris/core/router"
+)
 
 // ApplicationContext application context
 type ApplicationContext struct {
@@ -40,3 +43,6 @@ func (a *ApplicationContext) GetProperty(name string) (value interface{}, ok boo
 func (a *ApplicationContext) GetInstance(params ...interface{}) (instance interface{}) {
 	return
 }
+
+func (a *ApplicationContext) WrapRouter(handler router.WrapperFunc) {
+}

pkg/app/web/application.go

@@ -22,12 +22,13 @@ import (
 	"regexp"
 	"time"
 
-	"github.com/kataras/iris"
 	"github.com/hidevopsio/hiboot/pkg/app"
 	"github.com/hidevopsio/hiboot/pkg/app/web/context"
 	"github.com/hidevopsio/hiboot/pkg/at"
 	"github.com/hidevopsio/hiboot/pkg/log"
 	"github.com/hidevopsio/hiboot/pkg/utils/io"
+	"github.com/kataras/iris"
+	"github.com/kataras/iris/core/router"
 )
 
 const (
@@ -190,6 +191,10 @@ func (a *application) Use(handlers ...context.Handler) {
 	}
 }
 
+func (a *application) WrapRouter(handler router.WrapperFunc) {
+	a.webApp.WrapRouter(handler)
+}
+
 func (a *application) initialize(controllers ...interface{}) (err error) {
 	io.EnsureWorkDir(3, "config/application.yml")