cc-metric-collector/internal/ccLogger/cclogger.go

package cclogger

import (
	"fmt"
	"runtime"
	"os"
	"log"
)


var (
    globalDebug = false
    stdout = os.Stdout
    stderr = os.Stderr
    debugLog *log.Logger = nil
    infoLog *log.Logger = nil
    errorLog *log.Logger = nil
    warnLog *log.Logger = nil
    defaultLog *log.Logger = nil
)

func initLogger() {
    if debugLog == nil {
        debugLog = log.New(stderr, "DEBUG ", log.LstdFlags)
    }
    if infoLog == nil {
        infoLog = log.New(stdout, "INFO ", log.LstdFlags)
    }
    if errorLog == nil {
        errorLog = log.New(stderr, "ERROR ", log.LstdFlags)
    }
    if warnLog == nil {
        warnLog = log.New(stderr, "WARN ", log.LstdFlags)
    }
    if defaultLog == nil {
        defaultLog = log.New(stdout, "", log.LstdFlags)
    }
}

func Print(e ... interface{}) {
    initLogger()
    defaultLog.Print(e)
}

func ComponentPrint(component string, e ... interface{}) {
    initLogger()
    defaultLog.Print(fmt.Sprintf("[%s] ", component), e)
}

func Info(e ... interface{}) {
    initLogger()
    infoLog.Print(e)
}

func ComponentInfo(component string, e ... interface{}) {
    initLogger()
    infoLog.Print(fmt.Sprintf("[%s] ", component), e)
}

func Debug(e ... interface{}) {
    initLogger()
    if globalDebug == true {
        debugLog.Print(e)
    }
}

func ComponentDebug(component string, e ... interface{}) {
    initLogger()
    if globalDebug == true && debugLog != nil {
        //CCComponentPrint(debugLog, component,  e)
        debugLog.Print(fmt.Sprintf("[%s] ", component), e)
    }
}

func Error(e ... interface{}) {
    initLogger()
    _, fn, line, _ := runtime.Caller(1)
    errorLog.Print(fmt.Sprintf("[%s:%d] ", fn, line), e)
}

func ComponentError(component string, e ... interface{}) {
    initLogger()
    _, fn, line, _ := runtime.Caller(1)
    errorLog.Print(fmt.Sprintf("[%s|%s:%d] ", component, fn, line), e)
}

func SetDebug() {
    globalDebug = true
    initLogger()
}


func SetOutput(filename string) {
    if filename == "stderr" {
        if stderr != os.Stderr && stderr != os.Stdout {
            stderr.Close()
        }
        stderr = os.Stderr
    } else if filename == "stdout" {
        if stderr != os.Stderr && stderr != os.Stdout {
            stderr.Close()
        }
        stderr = os.Stdout
    } else {
        file, err := os.OpenFile(filename, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0600)
        if err == nil {
            defer file.Close()
            stderr = file
        }
    }
    debugLog = nil
    errorLog = nil
    warnLog = nil
    initLogger()
}