gin/mode.go

74 lines
1.4 KiB
Go
Raw Normal View History

2014-08-29 17:49:50 +00:00
// Copyright 2014 Manu Martinez-Almeida. All rights reserved.
// Use of this source code is governed by a MIT style
// license that can be found in the LICENSE file.
package gin
import (
2016-04-15 08:12:07 +00:00
"io"
"os"
2015-04-07 16:37:17 +00:00
"github.com/gin-gonic/gin/binding"
)
2015-05-22 14:55:16 +00:00
const ENV_GIN_MODE = "GIN_MODE"
const (
DebugMode string = "debug"
ReleaseMode string = "release"
2014-08-20 23:01:05 +00:00
TestMode string = "test"
)
const (
2017-03-11 13:35:29 +00:00
debugCode = iota
releaseCode
testCode
)
// DefaultWriter is the default io.Writer used the Gin for debug output and
// middleware output like Logger() or Recovery().
// Note that both Logger and Recovery provides custom ways to configure their
// output io.Writer.
// To support coloring in Windows use:
2016-01-26 21:40:29 +00:00
// import "github.com/mattn/go-colorable"
// gin.DefaultWriter = colorable.NewColorableStdout()
2016-04-15 08:12:07 +00:00
var DefaultWriter io.Writer = os.Stdout
var DefaultErrorWriter io.Writer = os.Stderr
2016-04-14 23:16:46 +00:00
var ginMode = debugCode
var modeName = DebugMode
2014-10-08 23:40:42 +00:00
func init() {
2015-05-22 14:55:16 +00:00
mode := os.Getenv(ENV_GIN_MODE)
if len(mode) == 0 {
2014-10-08 23:40:42 +00:00
SetMode(DebugMode)
} else {
2015-05-22 14:55:16 +00:00
SetMode(mode)
2014-10-08 23:40:42 +00:00
}
}
func SetMode(value string) {
switch value {
case DebugMode:
2015-04-07 10:27:23 +00:00
ginMode = debugCode
case ReleaseMode:
2015-04-07 10:27:23 +00:00
ginMode = releaseCode
2014-08-20 23:01:05 +00:00
case TestMode:
2015-04-07 10:27:23 +00:00
ginMode = testCode
default:
2015-04-08 00:58:35 +00:00
panic("gin mode unknown: " + value)
}
2015-04-07 10:27:23 +00:00
modeName = value
}
func DisableBindValidation() {
binding.Validator = nil
}
func EnableJsonDecoderUseNumber() {
binding.EnableDecoderUseNumber = true
}
func Mode() string {
2015-04-07 10:27:23 +00:00
return modeName
}