feat: set all child

master
Evan Chen 2021-11-02 11:08:23 +08:00
parent a870c866e0
commit a41e1855ca
4 changed files with 109 additions and 7 deletions

41
log.go
View File

@ -86,10 +86,12 @@ type Logger struct {
debug_tmpl *template.Template
warn_tmpl *template.Template
info_tmpl *template.Template
subs []*Logger
}
func Default() *Logger {
func New(name string) *Logger {
l := &Logger{
system: name,
err: os.Stderr,
out: os.Stdout,
color: true,
@ -112,6 +114,10 @@ func (l *Logger) Sub(sys string) *Logger {
warn_tmpl: l.warn_tmpl,
info_tmpl: l.info_tmpl,
}
if l.subs == nil {
l.subs = make([]*Logger, 0)
}
l.subs = append(l.subs, ret)
return ret
}
@ -134,15 +140,36 @@ func (l *Logger) SetErrOutput(err io.Writer) {
l.err = err
}
func (l *Logger) SetErrOutputAll(err io.Writer) {
for _, v := range getAllOffsprings(l) {
v.err = err
}
}
func (l *Logger) SetOutput(out io.Writer) {
l.out = out
}
func (l *Logger) SetOutputAll(out io.Writer) {
for _, v := range getAllOffsprings(l) {
v.out = out
}
}
func (l *Logger) Reload() error {
l.guessColor()
return l.ParseTmpl()
}
func (l *Logger) ReloadAll() error {
for _, v := range getAllOffsprings(l) {
if err := v.Reload(); err != nil {
return err
}
}
return nil
}
func NewLogFormater() *LogFormater {
return &LogFormater{
ErrTmplStr: DEFAULT_ERR,
@ -156,7 +183,6 @@ func (l *Logger) DefaultFuncMap() template.FuncMap {
funcMap := template.FuncMap{
"Time": func() string { return time.Now().Format("2006/01/02 15:04:05") },
}
setColorMap(funcMap, l.color)
return funcMap
}
@ -164,22 +190,25 @@ func (l *Logger) ParseTmpl() error {
if l.funcMap == nil {
l.funcMap = l.DefaultFuncMap()
}
err_tmpl, err := template.New("err_tmpl").Funcs(l.funcMap).Parse(l.formatter.ErrTmplStr)
funcMap := copyFuncMap(l.funcMap)
setColorMap(funcMap, l.color)
err_tmpl, err := template.New("err_tmpl").Funcs(funcMap).Parse(l.formatter.ErrTmplStr)
if err != nil {
return err
}
l.err_tmpl = err_tmpl
warn_tmpl, err := template.New("warn_tmpl").Funcs(l.funcMap).Parse(l.formatter.WarnTmplStr)
warn_tmpl, err := template.New("warn_tmpl").Funcs(funcMap).Parse(l.formatter.WarnTmplStr)
if err != nil {
return err
}
l.warn_tmpl = warn_tmpl
info_tmpl, err := template.New("info_tmpl").Funcs(l.funcMap).Parse(l.formatter.InfoTmplStr)
info_tmpl, err := template.New("info_tmpl").Funcs(funcMap).Parse(l.formatter.InfoTmplStr)
if err != nil {
return err
}
l.info_tmpl = info_tmpl
debug_tmpl, err := template.New("debug_tmpl").Funcs(l.funcMap).Parse(l.formatter.DebugTmplStr)
debug_tmpl, err := template.New("debug_tmpl").Funcs(funcMap).Parse(l.formatter.DebugTmplStr)
if err != nil {
return err
}

View File

@ -1,11 +1,13 @@
package log
import (
"io"
"os"
"testing"
)
func TestDev(t *testing.T) {
LEVEL = Lerror | Ldebug | Lwarn | Linfo
PROD = false
// log.system = "dev"
Error("d", "sdf", "sdfsdf")
@ -97,3 +99,24 @@ func TestToFile(t *testing.T) {
l.Info("d", "sdf", "sdfsdf")
l.InfoF(H{"Test": "set"}, "d", "sdf", "sdfsdf")
}
func TestApplyChild(t *testing.T) {
PROD = true
l := Sub("parent")
l1 := l.Sub("child1")
l2 := l.Sub("child2")
l3 := l.Sub("child3")
l1.Info("child1")
l2.Info("child2")
l3.Info("child3")
l.SetOutputAll(io.Discard)
l.ReloadAll()
l1.Info("child1-discard")
l2.Info("child2-discard")
l3.Info("child3-discard")
SetOutputAll(os.Stdout)
ReloadAll()
l1.Info("child1-stdout")
l2.Info("child2-stdout")
l3.Info("child3-stdout")
}

32
std.go
View File

@ -1,11 +1,41 @@
package log
var std = Default()
import "io"
var std = New("")
func Sub(sys string) *Logger {
return std.Sub(sys)
}
func SetColor(c bool) {
std.SetColor(c)
}
func SetErrOutput(err io.Writer) {
std.SetErrOutput(err)
}
func SetErrOutputAll(err io.Writer) {
std.SetErrOutputAll(err)
}
func SetOutput(out io.Writer) {
std.SetOutput(out)
}
func SetOutputAll(out io.Writer) {
std.SetOutputAll(out)
}
func Reload() error {
return std.Reload()
}
func ReloadAll() error {
return std.ReloadAll()
}
func ErrorF(fields H, v ...interface{}) {
if PROD {
std.output(terror, 3, "", fields, v...)

20
util.go
View File

@ -3,6 +3,7 @@ package log
import (
"fmt"
"runtime"
"text/template"
)
func caller(depth int) string {
@ -27,3 +28,22 @@ func stack() string {
buf = make([]byte, 2*len(buf))
}
}
func getAllOffsprings(l *Logger) []*Logger {
if l.subs == nil || len(l.subs) == 0 {
return []*Logger{l}
}
loggers := []*Logger{}
for i := range l.subs {
loggers = append(loggers, getAllOffsprings(l.subs[i])...)
}
return loggers
}
func copyFuncMap(source template.FuncMap) template.FuncMap {
dest := template.FuncMap{}
for k, v := range source {
dest[k] = v
}
return dest
}