|
- package grlogs
-
- import (
- "fmt"
- "golib.gaore.com/GaoreGo/grlogs/logs"
- "os"
- "sync"
- )
-
- var loggers = sync.Map{}
-
- var envkey = "CENTER_RUNMODE"
-
- type Logger struct {
- Lable string
- *logs.BeeLogger
- }
-
- func (self *Logger) SetAdapter(level Level, adapter Adapter) *Logger {
- if call, ok := adatperMapper[adapter]; ok {
- if err := call(self, level); err != nil {
- fmt.Println(err)
- }
- }
- return self
- }
-
- func New(label string, channelLens ...int64) (l *Logger, loaded bool) {
- var channellens int64
- var tmp interface{}
-
- tmp, loaded = loggers.LoadOrStore(label, new(Logger))
- l = tmp.(*Logger)
- if len(channelLens) > 0 {
- channellens = channelLens[0]
- }
-
- if !loaded {
- l.Lable = label
- l.BeeLogger = logs.NewLogger(channellens)
- l.BeeLogger.Lable = label
- l.Env = os.Getenv(envkey)
- l.SetPrefix(fmt.Sprintf("[env:%s logger:%s]", os.Getenv(envkey), label))
- l.EnableFuncCallDepth(true)
- l.SetLogFuncCallDepth(2)
- }
- return
- }
-
- func Get(label string, channelLens ...int64) (l *Logger) {
- var filelevel Level = LevelInfo
- var loaded bool
- if os.Getenv(envkey) == "dev" {
- filelevel = LevelAll
- }
-
- if l, loaded = New(label, channelLens...); !loaded {
- l.SetAdapter(filelevel, AdapterFile).SetAdapter(LevelAll, AdapterConsole)
- }
- return
- }
-
- func GetEs(label string, channelLens ...int64) (l *Logger) {
- var filelevel Level = LevelInfo
- var loaded bool
- if os.Getenv(envkey) == "dev" {
- filelevel = LevelAll
- }
-
- if l, loaded = New(label, channelLens...); !loaded {
- l.SetAdapter(filelevel, AdapterFile).SetAdapter(LevelAll, AdapterConsole).SetAdapter(filelevel, AdapterElasticSearch)
- }
- return
- }
-
- func GetAli(label string, channelLens ...int64) (l *Logger) {
- var filelevel Level = LevelInfo
- var loaded bool
- if os.Getenv(envkey) == "dev" {
- filelevel = LevelAll
- }
-
- if l, loaded = New(label, channelLens...); !loaded {
- l.SetAdapter(LevelAll, AdapterConsole).SetAdapter(filelevel, AdapterFile).SetAdapter(filelevel, AdapterAliLs)
- }
- return
- }
-
- func Close(lables ...string) {
- wg := &sync.WaitGroup{}
- for _, lable := range lables {
- wg.Add(1)
- go func() {
- if v, ok := loggers.Load(lable); ok {
- if tmp, ok := v.(*Logger); ok {
- loggers.Delete(lable)
- tmp.BeeLogger.Close()
- }
- }
- wg.Done()
- }()
- }
- wg.Wait()
-
- }
-
- func CloseAll() {
- wg := &sync.WaitGroup{}
- loggers.Range(func(key, value interface{}) bool {
- wg.Add(1)
- go func() {
- if tmp, ok := value.(*Logger); ok {
- tmp.BeeLogger.Close()
- loggers.Delete(key)
- }
- wg.Done()
- }()
- return true
- })
- wg.Wait()
- }
|