319 lines
7.1 KiB
Go
319 lines
7.1 KiB
Go
// Copyright (c) 2016 Couchbase, Inc.
|
|
// 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
|
|
// Unless required by applicable law or agreed to in writing, software distributed under the
|
|
// License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
|
|
// either express or implied. See the License for the specific language governing permissions
|
|
// and limitations under the License.
|
|
|
|
package logging
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
"time"
|
|
)
|
|
|
|
type goLogger struct {
|
|
logger *log.Logger
|
|
level Level
|
|
entryFormatter formatter
|
|
}
|
|
|
|
const (
|
|
_LEVEL = "_level"
|
|
_MSG = "_msg"
|
|
_TIME = "_time"
|
|
_RLEVEL = "_rlevel"
|
|
)
|
|
|
|
func NewLogger(out io.Writer, lvl Level, fmtLogging LogEntryFormatter) *goLogger {
|
|
logger := &goLogger{
|
|
logger: log.New(out, "", 0),
|
|
level: lvl,
|
|
}
|
|
if fmtLogging == JSONFORMATTER {
|
|
logger.entryFormatter = &jsonFormatter{}
|
|
} else if fmtLogging == KVFORMATTER {
|
|
logger.entryFormatter = &keyvalueFormatter{}
|
|
} else {
|
|
logger.entryFormatter = &textFormatter{}
|
|
}
|
|
return logger
|
|
}
|
|
|
|
func (gl *goLogger) Logp(level Level, msg string, kv ...Pair) {
|
|
if gl.logger == nil {
|
|
return
|
|
}
|
|
if level <= gl.level {
|
|
e := newLogEntry(msg, level)
|
|
copyPairs(e, kv)
|
|
gl.log(e)
|
|
}
|
|
}
|
|
|
|
func (gl *goLogger) Debugp(msg string, kv ...Pair) {
|
|
gl.Logp(DEBUG, msg, kv...)
|
|
}
|
|
|
|
func (gl *goLogger) Tracep(msg string, kv ...Pair) {
|
|
gl.Logp(TRACE, msg, kv...)
|
|
}
|
|
|
|
func (gl *goLogger) Requestp(rlevel Level, msg string, kv ...Pair) {
|
|
if gl.logger == nil {
|
|
return
|
|
}
|
|
if REQUEST <= gl.level {
|
|
e := newLogEntry(msg, REQUEST)
|
|
e.Rlevel = rlevel
|
|
copyPairs(e, kv)
|
|
gl.log(e)
|
|
}
|
|
}
|
|
|
|
func (gl *goLogger) Infop(msg string, kv ...Pair) {
|
|
gl.Logp(INFO, msg, kv...)
|
|
}
|
|
|
|
func (gl *goLogger) Warnp(msg string, kv ...Pair) {
|
|
gl.Logp(WARN, msg, kv...)
|
|
}
|
|
|
|
func (gl *goLogger) Errorp(msg string, kv ...Pair) {
|
|
gl.Logp(ERROR, msg, kv...)
|
|
}
|
|
|
|
func (gl *goLogger) Severep(msg string, kv ...Pair) {
|
|
gl.Logp(SEVERE, msg, kv...)
|
|
}
|
|
|
|
func (gl *goLogger) Fatalp(msg string, kv ...Pair) {
|
|
gl.Logp(FATAL, msg, kv...)
|
|
}
|
|
|
|
func (gl *goLogger) Logm(level Level, msg string, kv Map) {
|
|
if gl.logger == nil {
|
|
return
|
|
}
|
|
if level <= gl.level {
|
|
e := newLogEntry(msg, level)
|
|
e.Data = kv
|
|
gl.log(e)
|
|
}
|
|
}
|
|
|
|
func (gl *goLogger) Debugm(msg string, kv Map) {
|
|
gl.Logm(DEBUG, msg, kv)
|
|
}
|
|
|
|
func (gl *goLogger) Tracem(msg string, kv Map) {
|
|
gl.Logm(TRACE, msg, kv)
|
|
}
|
|
|
|
func (gl *goLogger) Requestm(rlevel Level, msg string, kv Map) {
|
|
if gl.logger == nil {
|
|
return
|
|
}
|
|
if REQUEST <= gl.level {
|
|
e := newLogEntry(msg, REQUEST)
|
|
e.Rlevel = rlevel
|
|
e.Data = kv
|
|
gl.log(e)
|
|
}
|
|
}
|
|
|
|
func (gl *goLogger) Infom(msg string, kv Map) {
|
|
gl.Logm(INFO, msg, kv)
|
|
}
|
|
|
|
func (gl *goLogger) Warnm(msg string, kv Map) {
|
|
gl.Logm(WARN, msg, kv)
|
|
}
|
|
|
|
func (gl *goLogger) Errorm(msg string, kv Map) {
|
|
gl.Logm(ERROR, msg, kv)
|
|
}
|
|
|
|
func (gl *goLogger) Severem(msg string, kv Map) {
|
|
gl.Logm(SEVERE, msg, kv)
|
|
}
|
|
|
|
func (gl *goLogger) Fatalm(msg string, kv Map) {
|
|
gl.Logm(FATAL, msg, kv)
|
|
}
|
|
|
|
func (gl *goLogger) Logf(level Level, format string, args ...interface{}) {
|
|
if gl.logger == nil {
|
|
return
|
|
}
|
|
if level <= gl.level {
|
|
e := newLogEntry(fmt.Sprintf(format, args...), level)
|
|
gl.log(e)
|
|
}
|
|
}
|
|
|
|
func (gl *goLogger) Debugf(format string, args ...interface{}) {
|
|
gl.Logf(DEBUG, format, args...)
|
|
}
|
|
|
|
func (gl *goLogger) Tracef(format string, args ...interface{}) {
|
|
gl.Logf(TRACE, format, args...)
|
|
}
|
|
|
|
func (gl *goLogger) Requestf(rlevel Level, format string, args ...interface{}) {
|
|
if gl.logger == nil {
|
|
return
|
|
}
|
|
if REQUEST <= gl.level {
|
|
e := newLogEntry(fmt.Sprintf(format, args...), REQUEST)
|
|
e.Rlevel = rlevel
|
|
gl.log(e)
|
|
}
|
|
}
|
|
|
|
func (gl *goLogger) Infof(format string, args ...interface{}) {
|
|
gl.Logf(INFO, format, args...)
|
|
}
|
|
|
|
func (gl *goLogger) Warnf(format string, args ...interface{}) {
|
|
gl.Logf(WARN, format, args...)
|
|
}
|
|
|
|
func (gl *goLogger) Errorf(format string, args ...interface{}) {
|
|
gl.Logf(ERROR, format, args...)
|
|
}
|
|
|
|
func (gl *goLogger) Severef(format string, args ...interface{}) {
|
|
gl.Logf(SEVERE, format, args...)
|
|
}
|
|
|
|
func (gl *goLogger) Fatalf(format string, args ...interface{}) {
|
|
gl.Logf(FATAL, format, args...)
|
|
}
|
|
|
|
func (gl *goLogger) Level() Level {
|
|
return gl.level
|
|
}
|
|
|
|
func (gl *goLogger) SetLevel(level Level) {
|
|
gl.level = level
|
|
}
|
|
|
|
func (gl *goLogger) log(newEntry *logEntry) {
|
|
s := gl.entryFormatter.format(newEntry)
|
|
gl.logger.Print(s)
|
|
}
|
|
|
|
type logEntry struct {
|
|
Time string
|
|
Level Level
|
|
Rlevel Level
|
|
Message string
|
|
Data Map
|
|
}
|
|
|
|
func newLogEntry(msg string, level Level) *logEntry {
|
|
return &logEntry{
|
|
Time: time.Now().Format("2006-01-02T15:04:05.000-07:00"), // time.RFC3339 with milliseconds
|
|
Level: level,
|
|
Rlevel: NONE,
|
|
Message: msg,
|
|
}
|
|
}
|
|
|
|
func copyPairs(newEntry *logEntry, pairs []Pair) {
|
|
newEntry.Data = make(Map, len(pairs))
|
|
for _, p := range pairs {
|
|
newEntry.Data[p.Name] = p.Value
|
|
}
|
|
}
|
|
|
|
type formatter interface {
|
|
format(*logEntry) string
|
|
}
|
|
|
|
type textFormatter struct {
|
|
}
|
|
|
|
// ex. 2016-02-10T09:15:25.498-08:00 [INFO] This is a message from test in text format
|
|
|
|
func (*textFormatter) format(newEntry *logEntry) string {
|
|
b := &bytes.Buffer{}
|
|
appendValue(b, newEntry.Time)
|
|
if newEntry.Rlevel != NONE {
|
|
fmt.Fprintf(b, "[%s,%s] ", newEntry.Level.String(), newEntry.Rlevel.String())
|
|
} else {
|
|
fmt.Fprintf(b, "[%s] ", newEntry.Level.String())
|
|
}
|
|
appendValue(b, newEntry.Message)
|
|
for key, value := range newEntry.Data {
|
|
appendKeyValue(b, key, value)
|
|
}
|
|
b.WriteByte('\n')
|
|
s := bytes.NewBuffer(b.Bytes())
|
|
return s.String()
|
|
}
|
|
|
|
func appendValue(b *bytes.Buffer, value interface{}) {
|
|
if _, ok := value.(string); ok {
|
|
fmt.Fprintf(b, "%s ", value)
|
|
} else {
|
|
fmt.Fprintf(b, "%v ", value)
|
|
}
|
|
}
|
|
|
|
type keyvalueFormatter struct {
|
|
}
|
|
|
|
// ex. _time=2016-02-10T09:15:25.498-08:00 _level=INFO _msg=This is a message from test in key-value format
|
|
|
|
func (*keyvalueFormatter) format(newEntry *logEntry) string {
|
|
b := &bytes.Buffer{}
|
|
appendKeyValue(b, _TIME, newEntry.Time)
|
|
appendKeyValue(b, _LEVEL, newEntry.Level.String())
|
|
if newEntry.Rlevel != NONE {
|
|
appendKeyValue(b, _RLEVEL, newEntry.Rlevel.String())
|
|
}
|
|
appendKeyValue(b, _MSG, newEntry.Message)
|
|
for key, value := range newEntry.Data {
|
|
appendKeyValue(b, key, value)
|
|
}
|
|
b.WriteByte('\n')
|
|
s := bytes.NewBuffer(b.Bytes())
|
|
return s.String()
|
|
}
|
|
|
|
func appendKeyValue(b *bytes.Buffer, key, value interface{}) {
|
|
if _, ok := value.(string); ok {
|
|
fmt.Fprintf(b, "%v=%s ", key, value)
|
|
} else {
|
|
fmt.Fprintf(b, "%v=%v ", key, value)
|
|
}
|
|
}
|
|
|
|
type jsonFormatter struct {
|
|
}
|
|
|
|
// ex. {"_level":"INFO","_msg":"This is a message from test in json format","_time":"2016-02-10T09:12:59.518-08:00"}
|
|
|
|
func (*jsonFormatter) format(newEntry *logEntry) string {
|
|
if newEntry.Data == nil {
|
|
newEntry.Data = make(Map, 5)
|
|
}
|
|
newEntry.Data[_TIME] = newEntry.Time
|
|
newEntry.Data[_LEVEL] = newEntry.Level.String()
|
|
if newEntry.Rlevel != NONE {
|
|
newEntry.Data[_RLEVEL] = newEntry.Rlevel.String()
|
|
}
|
|
newEntry.Data[_MSG] = newEntry.Message
|
|
serialized, _ := json.Marshal(newEntry.Data)
|
|
s := bytes.NewBuffer(append(serialized, '\n'))
|
|
return s.String()
|
|
}
|