status-go/vendor/github.com/spacemonkeygo/spacelog/text.go

81 lines
2.3 KiB
Go
Raw Normal View History

2019-06-09 07:24:20 +00:00
// 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
}