update config

Author: LyricTian

.travis.yml

@@ -4,6 +4,10 @@ go_import_path: github.com/go-session/gin-session
 go:
   - 1.9
 before_install:
-  - go get github.com/mattn/goveralls
+  - go get -t -v ./...
+
 script:
-  - $HOME/gopath/bin/goveralls -service=travis-ci
+  - go test -race -coverprofile=coverage.txt -covermode=atomic
+
+after_success:
+  - bash <(curl -s https://codecov.io/bash)

README.md

@@ -1,6 +1,6 @@
 # Session middleware for [Gin](https://github.com/gin-gonic/gin)
 
-[![Build][Build-Status-Image]][Build-Status-Url] [![Coverage][Coverage-Image]][Coverage-Url] [![ReportCard][reportcard-image]][reportcard-url] [![GoDoc][godoc-image]][godoc-url] [![License][license-image]][license-url]
+[![Build][Build-Status-Image]][Build-Status-Url] [![Codecov][codecov-image]][codecov-url] [![Coverage][Coverage-Image]][Coverage-Url] [![ReportCard][reportcard-image]][reportcard-url] [![GoDoc][godoc-image]][godoc-url] [![License][license-image]][license-url]
 
 ## Quick Start
 
@@ -77,8 +77,8 @@ $ ./server
 
 [Build-Status-Url]: https://travis-ci.org/go-session/gin-session
 [Build-Status-Image]: https://travis-ci.org/go-session/gin-session.svg?branch=master
-[Coverage-Url]: https://coveralls.io/github/go-session/gin-session?branch=master
-[Coverage-Image]: https://coveralls.io/repos/github/go-session/gin-session/badge.svg?branch=master
+[codecov-url]: https://codecov.io/gh/go-session/gin-session
+[codecov-image]: https://codecov.io/gh/go-session/gin-session/branch/master/graph/badge.svg
 [reportcard-url]: https://goreportcard.com/report/github.com/go-session/gin-session
 [reportcard-image]: https://goreportcard.com/badge/github.com/go-session/gin-session
 [godoc-url]: https://godoc.org/github.com/go-session/gin-session

session.go

@@ -8,27 +8,31 @@ import (
 	"gopkg.in/session.v2"
 )
 
+type (
+	// ErrorHandleFunc error handling function
+	ErrorHandleFunc func(*gin.Context, error)
+	// Config defines the config for Session middleware
+	Config struct {
+		// error handling when starting the session
+		ErrorHandleFunc ErrorHandleFunc
+		// keys stored in the context
+		StoreKey string
+	}
+)
+
 var (
-	// DefaultKey Keys stored in the context
-	DefaultKey      = "github.com/go-session/gin-session"
 	once            sync.Once
 	internalManager *session.Manager
-	internalError   ErrorHandleFunc
-)
+	storeKey        string
 
-func init() {
-	internalError = func(ctx *gin.Context, err error) {
-		ctx.AbortWithError(500, err)
+	// DefaultConfig is the default Recover middleware config.
+	DefaultConfig = Config{
+		ErrorHandleFunc: func(ctx *gin.Context, err error) {
+			ctx.AbortWithError(500, err)
+		},
+		StoreKey: "github.com/go-session/gin-session",
 	}
-}
-
-// ErrorHandleFunc error handling function
-type ErrorHandleFunc func(ctx *gin.Context, err error)
-
-// SetErrorHandler Set error handling
-func SetErrorHandler(handler ErrorHandleFunc) {
-	internalError = handler
-}
+)
 
 func manager(opt ...session.Option) *session.Manager {
 	once.Do(func() {
@@ -37,22 +41,36 @@ func manager(opt ...session.Option) *session.Manager {
 	return internalManager
 }
 
-// New Create a session middleware
+// New create a session middleware
 func New(opt ...session.Option) gin.HandlerFunc {
+	return NewWithConfig(DefaultConfig, opt...)
+}
+
+// NewWithConfig create a session middleware
+func NewWithConfig(config Config, opt ...session.Option) gin.HandlerFunc {
+	if config.ErrorHandleFunc == nil {
+		config.ErrorHandleFunc = DefaultConfig.ErrorHandleFunc
+	}
+
+	storeKey = config.StoreKey
+	if storeKey == "" {
+		storeKey = DefaultConfig.StoreKey
+	}
+
 	return func(ctx *gin.Context) {
 		store, err := manager(opt...).Start(context.Background(), ctx.Writer, ctx.Request)
 		if err != nil {
-			internalError(ctx, err)
+			config.ErrorHandleFunc(ctx, err)
 			return
 		}
-		ctx.Set(DefaultKey, store)
+		ctx.Set(storeKey, store)
 		ctx.Next()
 	}
 }
 
 // FromContext Get session storage from context
 func FromContext(ctx *gin.Context) session.Store {
-	return ctx.MustGet(DefaultKey).(session.Store)
+	return ctx.MustGet(storeKey).(session.Store)
 }
 
 // Destroy a session