wei cheng
6 years ago
parent
3f2c51b56e
commit
c243e54e2b
@ -0,0 +1,50 @@ |
|||||||
|
package log |
||||||
|
|
||||||
|
import ( |
||||||
|
"bytes" |
||||||
|
"fmt" |
||||||
|
"strconv" |
||||||
|
"strings" |
||||||
|
) |
||||||
|
|
||||||
|
type verboseModule map[string]int32 |
||||||
|
|
||||||
|
type logFilter []string |
||||||
|
|
||||||
|
func (f *logFilter) String() string { |
||||||
|
return fmt.Sprint(*f) |
||||||
|
} |
||||||
|
|
||||||
|
// Set sets the value of the named command-line flag.
|
||||||
|
// format: -log.filter key1,key2
|
||||||
|
func (f *logFilter) Set(value string) error { |
||||||
|
for _, i := range strings.Split(value, ",") { |
||||||
|
*f = append(*f, strings.TrimSpace(i)) |
||||||
|
} |
||||||
|
return nil |
||||||
|
} |
||||||
|
|
||||||
|
func (m verboseModule) String() string { |
||||||
|
// FIXME strings.Builder
|
||||||
|
var buf bytes.Buffer |
||||||
|
for k, v := range m { |
||||||
|
buf.WriteString(k) |
||||||
|
buf.WriteString(strconv.FormatInt(int64(v), 10)) |
||||||
|
buf.WriteString(",") |
||||||
|
} |
||||||
|
return buf.String() |
||||||
|
} |
||||||
|
|
||||||
|
// Set sets the value of the named command-line flag.
|
||||||
|
// format: -log.module file=1,file2=2
|
||||||
|
func (m verboseModule) Set(value string) error { |
||||||
|
for _, i := range strings.Split(value, ",") { |
||||||
|
kv := strings.Split(i, "=") |
||||||
|
if len(kv) == 2 { |
||||||
|
if v, err := strconv.ParseInt(kv[1], 10, 64); err == nil { |
||||||
|
m[strings.TrimSpace(kv[0])] = int32(v) |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
return nil |
||||||
|
} |
||||||
@ -0,0 +1,61 @@ |
|||||||
|
package log |
||||||
|
|
||||||
|
import ( |
||||||
|
"context" |
||||||
|
"io/ioutil" |
||||||
|
"os" |
||||||
|
|
||||||
|
"github.com/sirupsen/logrus" |
||||||
|
) |
||||||
|
|
||||||
|
func init() { |
||||||
|
redirectLogrus() |
||||||
|
} |
||||||
|
|
||||||
|
func redirectLogrus() { |
||||||
|
// FIXME: because of different stack depth call runtime.Caller will get error function name.
|
||||||
|
logrus.AddHook(redirectHook{}) |
||||||
|
if os.Getenv("LOGRUS_STDOUT") == "" { |
||||||
|
logrus.SetOutput(ioutil.Discard) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
type redirectHook struct{} |
||||||
|
|
||||||
|
func (redirectHook) Levels() []logrus.Level { |
||||||
|
return logrus.AllLevels |
||||||
|
} |
||||||
|
|
||||||
|
func (redirectHook) Fire(entry *logrus.Entry) error { |
||||||
|
lv := _infoLevel |
||||||
|
var logrusLv string |
||||||
|
var verbose int32 |
||||||
|
switch entry.Level { |
||||||
|
case logrus.FatalLevel, logrus.PanicLevel: |
||||||
|
logrusLv = entry.Level.String() |
||||||
|
fallthrough |
||||||
|
case logrus.ErrorLevel: |
||||||
|
lv = _errorLevel |
||||||
|
case logrus.WarnLevel: |
||||||
|
lv = _warnLevel |
||||||
|
case logrus.InfoLevel: |
||||||
|
lv = _infoLevel |
||||||
|
case logrus.DebugLevel: |
||||||
|
// use verbose log replace of debuglevel
|
||||||
|
verbose = 10 |
||||||
|
} |
||||||
|
args := make([]D, 0, len(entry.Data)+1) |
||||||
|
args = append(args, D{Key: _log, Value: entry.Message}) |
||||||
|
for k, v := range entry.Data { |
||||||
|
args = append(args, D{Key: k, Value: v}) |
||||||
|
} |
||||||
|
if logrusLv != "" { |
||||||
|
args = append(args, D{Key: "logrus_lv", Value: logrusLv}) |
||||||
|
} |
||||||
|
if verbose != 0 { |
||||||
|
V(verbose).Infov(context.Background(), args...) |
||||||
|
} else { |
||||||
|
h.Log(context.Background(), lv, args...) |
||||||
|
} |
||||||
|
return nil |
||||||
|
} |
||||||
@ -0,0 +1,54 @@ |
|||||||
|
package log |
||||||
|
|
||||||
|
import ( |
||||||
|
"reflect" |
||||||
|
"strings" |
||||||
|
"testing" |
||||||
|
"time" |
||||||
|
) |
||||||
|
|
||||||
|
func TestFuncName(t *testing.T) { |
||||||
|
name := funcName(1) |
||||||
|
if !strings.Contains(name, "util_test.go:11") { |
||||||
|
t.Errorf("expect contains util_test.go:11 got %s", name) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
func Test_toMap(t *testing.T) { |
||||||
|
type args struct { |
||||||
|
args []D |
||||||
|
} |
||||||
|
tests := []struct { |
||||||
|
name string |
||||||
|
args args |
||||||
|
want map[string]interface{} |
||||||
|
}{ |
||||||
|
{ |
||||||
|
args: args{[]D{KVString("test", "hello")}}, |
||||||
|
want: map[string]interface{}{"test": "hello"}, |
||||||
|
}, |
||||||
|
{ |
||||||
|
args: args{[]D{KVInt64("test", 123)}}, |
||||||
|
want: map[string]interface{}{"test": int64(123)}, |
||||||
|
}, |
||||||
|
{ |
||||||
|
args: args{[]D{KVFloat32("test", float32(1.01))}}, |
||||||
|
want: map[string]interface{}{"test": float32(1.01)}, |
||||||
|
}, |
||||||
|
{ |
||||||
|
args: args{[]D{KVFloat32("test", float32(1.01))}}, |
||||||
|
want: map[string]interface{}{"test": float32(1.01)}, |
||||||
|
}, |
||||||
|
{ |
||||||
|
args: args{[]D{KVDuration("test", time.Second)}}, |
||||||
|
want: map[string]interface{}{"test": time.Second}, |
||||||
|
}, |
||||||
|
} |
||||||
|
for _, tt := range tests { |
||||||
|
t.Run(tt.name, func(t *testing.T) { |
||||||
|
if got := toMap(tt.args.args...); !reflect.DeepEqual(got, tt.want) { |
||||||
|
t.Errorf("toMap() = %v, want %v", got, tt.want) |
||||||
|
} |
||||||
|
}) |
||||||
|
} |
||||||
|
} |
||||||
Loading…
Reference in new issue