105 lines
2.5 KiB
Go
105 lines
2.5 KiB
Go
|
package logging
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io/ioutil"
|
||
|
|
"log"
|
||
|
|
"os"
|
||
|
|
"strconv"
|
||
|
|
"sync/atomic"
|
||
|
|
|
||
|
|
"github.com/golang/glog"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
// VerboseFlag enables verbose logging if set to true.
|
||
|
|
VerboseFlag bool
|
||
|
|
// VeryVerboseFlag enables very verbose logging if set to true.
|
||
|
|
VeryVerboseFlag bool
|
||
|
|
// Verbose is this package's verbose Logger.
|
||
|
|
Verbose *log.Logger
|
||
|
|
// VeryVerbose is this package's very verbose Logger.
|
||
|
|
VeryVerbose *log.Logger
|
||
|
|
// Error is this package's error Logger.
|
||
|
|
Error *log.Logger
|
||
|
|
)
|
||
|
|
|
||
|
|
// Counter defines an interface for a monotonically incrementing value.
|
||
|
|
type Counter interface {
|
||
|
|
Inc()
|
||
|
|
}
|
||
|
|
|
||
|
|
// LogCounter implements the Counter interface with a uint64 register.
|
||
|
|
// It's safe for concurrent use.
|
||
|
|
type LogCounter struct {
|
||
|
|
value uint64
|
||
|
|
}
|
||
|
|
|
||
|
|
// Inc increments the counter by one.
|
||
|
|
func (lc *LogCounter) Inc() {
|
||
|
|
atomic.AddUint64(&lc.value, 1)
|
||
|
|
}
|
||
|
|
|
||
|
|
// String returns a string represention of the counter.
|
||
|
|
func (lc *LogCounter) String() string {
|
||
|
|
return strconv.FormatUint(atomic.LoadUint64(&lc.value), 10)
|
||
|
|
}
|
||
|
|
|
||
|
|
// LogOut holds metrics captured in an instrumented runtime.
|
||
|
|
type LogOut struct {
|
||
|
|
MesosRequests Counter
|
||
|
|
MesosSuccess Counter
|
||
|
|
MesosNXDomain Counter
|
||
|
|
MesosFailed Counter
|
||
|
|
NonMesosRequests Counter
|
||
|
|
NonMesosSuccess Counter
|
||
|
|
NonMesosNXDomain Counter
|
||
|
|
NonMesosFailed Counter
|
||
|
|
NonMesosForwarded Counter
|
||
|
|
}
|
||
|
|
|
||
|
|
// CurLog is the default package level LogOut.
|
||
|
|
var CurLog = LogOut{
|
||
|
|
MesosRequests: &LogCounter{},
|
||
|
|
MesosSuccess: &LogCounter{},
|
||
|
|
MesosNXDomain: &LogCounter{},
|
||
|
|
MesosFailed: &LogCounter{},
|
||
|
|
NonMesosRequests: &LogCounter{},
|
||
|
|
NonMesosSuccess: &LogCounter{},
|
||
|
|
NonMesosNXDomain: &LogCounter{},
|
||
|
|
NonMesosFailed: &LogCounter{},
|
||
|
|
NonMesosForwarded: &LogCounter{},
|
||
|
|
}
|
||
|
|
|
||
|
|
// PrintCurLog prints out the current LogOut and then resets
|
||
|
|
func PrintCurLog() {
|
||
|
|
VeryVerbose.Printf("%+v\n", CurLog)
|
||
|
|
}
|
||
|
|
|
||
|
|
// SetupLogs provides the following logs
|
||
|
|
// Verbose = optional verbosity
|
||
|
|
// VeryVerbose = optional verbosity
|
||
|
|
// Error = stderr
|
||
|
|
func SetupLogs() {
|
||
|
|
// initialize logging flags
|
||
|
|
if glog.V(2) {
|
||
|
|
VeryVerboseFlag = true
|
||
|
|
} else if glog.V(1) {
|
||
|
|
VerboseFlag = true
|
||
|
|
}
|
||
|
|
|
||
|
|
logopts := log.Ldate | log.Ltime | log.Lshortfile
|
||
|
|
|
||
|
|
if VerboseFlag {
|
||
|
|
Verbose = log.New(os.Stdout, "VERBOSE: ", logopts)
|
||
|
|
VeryVerbose = log.New(ioutil.Discard, "VERY VERBOSE: ", logopts)
|
||
|
|
} else if VeryVerboseFlag {
|
||
|
|
Verbose = log.New(os.Stdout, "VERY VERBOSE: ", logopts)
|
||
|
|
VeryVerbose = Verbose
|
||
|
|
} else {
|
||
|
|
Verbose = log.New(ioutil.Discard, "VERBOSE: ", logopts)
|
||
|
|
VeryVerbose = log.New(ioutil.Discard, "VERY VERBOSE: ", logopts)
|
||
|
|
}
|
||
|
|
|
||
|
|
Error = log.New(os.Stderr, "ERROR: ", logopts)
|
||
|
|
}
|