diff options
Diffstat (limited to 'src/mongo/gotools/vendor/src/github.com/spacemonkeygo/spacelog/text.go')
-rw-r--r-- | src/mongo/gotools/vendor/src/github.com/spacemonkeygo/spacelog/text.go | 80 |
1 files changed, 80 insertions, 0 deletions
diff --git a/src/mongo/gotools/vendor/src/github.com/spacemonkeygo/spacelog/text.go b/src/mongo/gotools/vendor/src/github.com/spacemonkeygo/spacelog/text.go new file mode 100644 index 00000000000..8b36ce99f50 --- /dev/null +++ b/src/mongo/gotools/vendor/src/github.com/spacemonkeygo/spacelog/text.go @@ -0,0 +1,80 @@ +// 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 +} |