81 lines
2.3 KiB
Go
81 lines
2.3 KiB
Go
|
// Copyright (C) 2014 Space Monkey, 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 spacelog
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"fmt"
|
||
|
|
"runtime"
|
||
|
|
"strings"
|
||
|
|
"sync"
|
||
|
|
"text/template"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
// TextHandler is the default implementation of the Handler interface. A
|
||
|
|
// TextHandler, on log events, makes LogEvent structures, passes them to the
|
||
|
|
// configured template, and then passes that output to a configured TextOutput
|
||
|
|
// interface.
|
||
|
|
type TextHandler struct {
|
||
|
|
mtx sync.RWMutex
|
||
|
|
template *template.Template
|
||
|
|
output TextOutput
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewTextHandler creates a Handler that creates LogEvents, passes them to
|
||
|
|
// the given template, and passes the result to output
|
||
|
|
func NewTextHandler(t *template.Template, output TextOutput) *TextHandler {
|
||
|
|
return &TextHandler{template: t, output: output}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Log makes a LogEvent, formats it with the configured template, then passes
|
||
|
|
// the output to configured output sink
|
||
|
|
func (h *TextHandler) Log(logger_name string, level LogLevel, msg string,
|
||
|
|
calldepth int) {
|
||
|
|
h.mtx.RLock()
|
||
|
|
output, template := h.output, h.template
|
||
|
|
h.mtx.RUnlock()
|
||
|
|
event := LogEvent{
|
||
|
|
LoggerName: logger_name,
|
||
|
|
Level: level,
|
||
|
|
Message: strings.TrimRight(msg, "\n\r"),
|
||
|
|
Timestamp: time.Now()}
|
||
|
|
if calldepth >= 0 {
|
||
|
|
_, event.Filepath, event.Line, _ = runtime.Caller(calldepth + 1)
|
||
|
|
}
|
||
|
|
var buf bytes.Buffer
|
||
|
|
err := template.Execute(&buf, &event)
|
||
|
|
if err != nil {
|
||
|
|
output.Output(level, []byte(
|
||
|
|
fmt.Sprintf("log format template failed: %s", err)))
|
||
|
|
return
|
||
|
|
}
|
||
|
|
output.Output(level, buf.Bytes())
|
||
|
|
}
|
||
|
|
|
||
|
|
// SetTextTemplate changes the TextHandler's text formatting template
|
||
|
|
func (h *TextHandler) SetTextTemplate(t *template.Template) {
|
||
|
|
h.mtx.Lock()
|
||
|
|
defer h.mtx.Unlock()
|
||
|
|
h.template = t
|
||
|
|
}
|
||
|
|
|
||
|
|
// SetTextOutput changes the TextHandler's TextOutput sink
|
||
|
|
func (h *TextHandler) SetTextOutput(output TextOutput) {
|
||
|
|
h.mtx.Lock()
|
||
|
|
defer h.mtx.Unlock()
|
||
|
|
h.output = output
|
||
|
|
}
|