gin/mode.go

57 lines
954 B
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 (
"log"
"os"
2015-04-07 16:37:17 +00:00
"github.com/mattn/go-colorable"
)
const GIN_MODE = "GIN_MODE"
const (
DebugMode string = "debug"
ReleaseMode string = "release"
2014-08-20 23:01:05 +00:00
TestMode string = "test"
)
const (
debugCode = iota
releaseCode = iota
2014-08-20 23:01:05 +00:00
testCode = iota
)
2015-04-07 16:37:17 +00:00
var DefaultLogFile = colorable.NewColorableStdout()
2015-04-07 10:27:23 +00:00
var ginMode int = debugCode
var modeName string = DebugMode
2014-10-08 23:40:42 +00:00
func init() {
value := os.Getenv(GIN_MODE)
if len(value) == 0 {
SetMode(DebugMode)
} else {
SetMode(value)
}
}
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-03-23 04:50:10 +00:00
log.Panic("gin mode unknown: " + value)
}
2015-04-07 10:27:23 +00:00
modeName = value
}
func Mode() string {
2015-04-07 10:27:23 +00:00
return modeName
}