mode.go 566 B

1234567891011121314151617181920212223242526272829303132333435363738
  1. package gin
  2. import (
  3. "os"
  4. )
  5. const GIN_MODE = "GIN_MODE"
  6. const (
  7. DebugMode string = "debug"
  8. ReleaseMode string = "release"
  9. )
  10. const (
  11. debugCode = iota
  12. releaseCode = iota
  13. )
  14. var gin_mode int = debugCode
  15. func SetMode(value string) {
  16. switch value {
  17. case DebugMode:
  18. gin_mode = debugCode
  19. case ReleaseMode:
  20. gin_mode = releaseCode
  21. default:
  22. panic("gin mode unknown, the allowed modes are: " + DebugMode + " and " + ReleaseMode)
  23. }
  24. }
  25. func init() {
  26. value := os.Getenv(GIN_MODE)
  27. if len(value) == 0 {
  28. SetMode(DebugMode)
  29. } else {
  30. SetMode(value)
  31. }
  32. }