update logging with Init() method

This commit is contained in:
Bradley Cicenas 2017-02-12 05:10:40 +00:00
parent 3513f7485d
commit 4450abf89b
2 changed files with 19 additions and 16 deletions

View File

@ -15,8 +15,9 @@ const (
) )
var ( var (
exited bool Log *CTopLogger
wg sync.WaitGroup wg sync.WaitGroup
exited bool
format = logging.MustStringFormatter( format = logging.MustStringFormatter(
`%{color}%{time:15:04:05.000} %{shortfunc} ▶ %{level:.4s} %{id:03x}%{color:reset} %{message}`, `%{color}%{time:15:04:05.000} %{shortfunc} ▶ %{level:.4s} %{id:03x}%{color:reset} %{message}`,
) )
@ -27,21 +28,22 @@ type CTopLogger struct {
backend *logging.MemoryBackend backend *logging.MemoryBackend
} }
func New(serverEnabled string) *CTopLogger { func Init(serverEnabled string) *CTopLogger {
log := &CTopLogger{ if Log == nil {
Log = &CTopLogger{
logging.MustGetLogger("ctop"), logging.MustGetLogger("ctop"),
logging.NewMemoryBackend(size), logging.NewMemoryBackend(size),
} }
backendFmt := logging.NewBackendFormatter(log.backend, format) backendFmt := logging.NewBackendFormatter(Log.backend, format)
logging.SetBackend(backendFmt) logging.SetBackend(backendFmt)
log.Notice("logger initialized") Log.Notice("logger initialized")
if serverEnabled == "1" { if serverEnabled == "1" {
log.Serve() Log.Serve()
} }
}
return log return Log
} }
func (log *CTopLogger) Exit() { func (log *CTopLogger) Exit() {

View File

@ -6,9 +6,10 @@ import (
ui "github.com/gizak/termui" ui "github.com/gizak/termui"
) )
var log = logging.New(config.Global["loggingEnabled"]) var log *logging.CTopLogger
func main() { func main() {
log = logging.Init(config.Global["loggingEnabled"])
if err := ui.Init(); err != nil { if err := ui.Init(); err != nil {
panic(err) panic(err)
} }