dalu/cors
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix golint error.

Browse Source 
Signed-off-by: Bo-Yi Wu <appleboy.tw@gmail.com>
This commit is contained in:
Bo-Yi Wu 2016-10-20 09:06:20 +08:00
parent 14d8c87c42
commit 43adb82e7c
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
cors.go
View File

@ -8,6 +8,7 @@ import (
"github.com/gin-gonic/gin" "github.com/gin-gonic/gin"
) )
// Config represents all available options for the middleware.
type Config struct { type Config struct {
AllowAllOrigins bool AllowAllOrigins bool
@ -44,18 +45,22 @@ type Config struct {
MaxAge time.Duration MaxAge time.Duration
} }
// AddAllowMethods is allowed to add custom methods
func (c *Config) AddAllowMethods(methods ...string) { func (c *Config) AddAllowMethods(methods ...string) {
c.AllowMethods = append(c.AllowMethods, methods...) c.AllowMethods = append(c.AllowMethods, methods...)
} }
// AddAllowHeaders is allowed to add custom headers
func (c *Config) AddAllowHeaders(headers ...string) { func (c *Config) AddAllowHeaders(headers ...string) {
c.AllowHeaders = append(c.AllowHeaders, headers...) c.AllowHeaders = append(c.AllowHeaders, headers...)
} }
// AddExposeHeaders is allowed to add custom expose headers
func (c *Config) AddExposeHeaders(headers ...string) { func (c *Config) AddExposeHeaders(headers ...string) {
c.ExposeHeaders = append(c.ExposeHeaders, headers...) c.ExposeHeaders = append(c.ExposeHeaders, headers...)
} }
// Validate is check configuration of user defined.
func (c Config) Validate() error { func (c Config) Validate() error {
if c.AllowAllOrigins && (c.AllowOriginFunc != nil || len(c.AllowOrigins) > 0) { if c.AllowAllOrigins && (c.AllowOriginFunc != nil || len(c.AllowOrigins) > 0) {
return errors.New("conflict settings: all origins are allowed. AllowOriginFunc or AllowedOrigins is not needed") return errors.New("conflict settings: all origins are allowed. AllowOriginFunc or AllowedOrigins is not needed")
@ -71,6 +76,7 @@ func (c Config) Validate() error {
return nil return nil
} }
// DefaultConfig returns a generic default configuration mapped to localhost.
func DefaultConfig() Config { func DefaultConfig() Config {
return Config{ return Config{
AllowMethods: []string{"GET", "POST", "PUT", "HEAD"}, AllowMethods: []string{"GET", "POST", "PUT", "HEAD"},
@ -80,12 +86,14 @@ func DefaultConfig() Config {
} }
} }
// Default returns the location middleware with default configuration.
func Default() gin.HandlerFunc { func Default() gin.HandlerFunc {
config := DefaultConfig() config := DefaultConfig()
config.AllowAllOrigins = true config.AllowAllOrigins = true
return New(config) return New(config)
} }
// New returns the location middleware with user-defined custom configuration.
func New(config Config) gin.HandlerFunc { func New(config Config) gin.HandlerFunc {
cors := newCors(config) cors := newCors(config)
return func(c *gin.Context) { return func(c *gin.Context) {