56 lines
987 B
Go
56 lines
987 B
Go
|
// 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 (
|
||
|
|
"os"
|
||
|
|
|
||
|
|
"github.com/drone/drone/Godeps/_workspace/src/github.com/mattn/go-colorable"
|
||
|
|
)
|
||
|
|
|
||
|
|
const GIN_MODE = "GIN_MODE"
|
||
|
|
|
||
|
|
const (
|
||
|
|
DebugMode string = "debug"
|
||
|
|
ReleaseMode string = "release"
|
||
|
|
TestMode string = "test"
|
||
|
|
)
|
||
|
|
const (
|
||
|
|
debugCode = iota
|
||
|
|
releaseCode = iota
|
||
|
|
testCode = iota
|
||
|
|
)
|
||
|
|
|
||
|
|
var DefaultWriter = colorable.NewColorableStdout()
|
||
|
|
var ginMode int = debugCode
|
||
|
|
var modeName string = DebugMode
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
value := os.Getenv(GIN_MODE)
|
||
|
|
if len(value) == 0 {
|
||
|
|
SetMode(DebugMode)
|
||
|
|
} else {
|
||
|
|
SetMode(value)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func SetMode(value string) {
|
||
|
|
switch value {
|
||
|
|
case DebugMode:
|
||
|
|
ginMode = debugCode
|
||
|
|
case ReleaseMode:
|
||
|
|
ginMode = releaseCode
|
||
|
|
case TestMode:
|
||
|
|
ginMode = testCode
|
||
|
|
default:
|
||
|
|
panic("gin mode unknown: " + value)
|
||
|
|
}
|
||
|
|
modeName = value
|
||
|
|
}
|
||
|
|
|
||
|
|
func Mode() string {
|
||
|
|
return modeName
|
||
|
|
}
|