-
Notifications
You must be signed in to change notification settings - Fork 38
/
log.go
146 lines (127 loc) · 3.28 KB
/
log.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
// Copyright 2017-present Kirill Danshin and Gramework contributors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
package gramework
import (
"os"
"strings"
"sync/atomic"
"github.com/apex/log"
"github.com/apex/log/handlers/cli"
"github.com/valyala/fasthttp"
)
var enableDebug = true
var currentEnvironment *int32
// Environment defines which environment gramework application runs in.
// It may be useful in various cases.
type Environment int32
const (
// DEV is the default environment
DEV Environment = iota
// STAGE envoronment works just like prod environment,
// but with detailed logs
STAGE
// PROD environment itself
PROD
)
func (e Environment) String() string {
switch e {
case DEV:
return "DEV"
case STAGE:
return "STAGE"
case PROD:
return "PROD"
default:
return "<unknown>"
}
}
func init() {
var initEnv int32 = -1
currentEnvironment = &initEnv
environments := map[string]Environment{
"prod": PROD,
"production": PROD,
"stage": STAGE,
"staging": STAGE,
"dev": DEV,
"development": DEV,
}
for envName, envType := range environments {
if isEnvEquals(envName) {
SetEnv(envType)
internalLog.Infof("%s mode", envType)
return
}
}
}
func isEnvEquals(rawEnv string) bool {
grameEnv := strings.ToLower(os.Getenv("GRAMEWORK_ENV"))
env := strings.ToLower(os.Getenv("ENV"))
return strings.HasPrefix(grameEnv, rawEnv) || strings.HasPrefix(env, rawEnv)
}
// SetEnv sets gramework's environment
func SetEnv(e Environment) {
if e != DEV && e != STAGE && e != PROD {
internalLog.Warn("could not set unknown environment value, ignoring")
return
}
if e != GetEnv() {
internalLog.
WithField("prevEnv", GetEnv()).
WithField("newEnv", e).
Warn("Setting a new environment")
}
if e == PROD {
Logger.Level = log.InfoLevel
enableDebug = false
} else {
enableDebug = true
Logger.Level = log.DebugLevel
}
atomic.StoreInt32(currentEnvironment, int32(e))
}
// GetEnv returns current gramework's environment
func GetEnv() Environment {
if currentEnvironment == nil {
return DEV
}
return Environment(atomic.LoadInt32(currentEnvironment))
}
// FastHTTPLoggerAdapter Adapter for passing apex/log used as gramework Logger into fasthttp
type FastHTTPLoggerAdapter struct {
apexLogger log.Interface
fasthttp.Logger
}
// Logger handles default logger
var Logger = &log.Logger{
Level: log.InfoLevel,
Handler: cli.New(os.Stdout),
}
// Errorf logs an error using default logger
func Errorf(msg string, v ...interface{}) {
Logger.Errorf(msg, v...)
}
// NewFastHTTPLoggerAdapter create new *FastHTTPLoggerAdapter
func NewFastHTTPLoggerAdapter(logger *log.Interface) (fasthttplogger *FastHTTPLoggerAdapter) {
fasthttplogger = &FastHTTPLoggerAdapter{
apexLogger: *logger,
}
return fasthttplogger
}
// Printf show message only if set app.Logger.Level = apex/log.DebugLevel
func (l *FastHTTPLoggerAdapter) Printf(msg string, v ...interface{}) {
l.apexLogger.Debugf(msg, v...)
}
var internalLog = func() *log.Entry {
Logger.Level = log.DebugLevel
if !enableDebug {
Logger.Level = log.InfoLevel
}
return Logger.WithField("package", "gramework")
}()