Initial commit

tlog/clog.go

@@ -0,0 +1,81 @@
+package tlog
+
+import (
+	"context"
+	"fmt"
+
+	"google.golang.org/grpc/metadata"
+)
+
+func Cdebug(ctx context.Context, args ...interface{}) {
+	id := fmt.Sprintf("trace_id=%s", getTraceIdFromCtx(ctx))
+	data := append([]interface{}{id}, args...)
+	l.p(DEBUG, data...)
+}
+
+func Cdebugf(ctx context.Context, format string, args ...interface{}) {
+	f := "trace_id=%s||" + format
+	data := append([]interface{}{getTraceIdFromCtx(ctx)}, args...)
+	l.pf(DEBUG, f, data...)
+}
+
+func Cinfo(ctx context.Context, args ...interface{}) {
+	id := fmt.Sprintf("trace_id=%s", getTraceIdFromCtx(ctx))
+	data := append([]interface{}{id}, args...)
+	l.p(INFO, data...)
+}
+
+func Cinfof(ctx context.Context, format string, args ...interface{}) {
+	f := "trace_id=%s||" + format
+	data := append([]interface{}{getTraceIdFromCtx(ctx)}, args...)
+	l.pf(INFO, f, data...)
+}
+
+func Cerror(ctx context.Context, args ...interface{}) {
+	id := fmt.Sprintf("trace_id=%s", getTraceIdFromCtx(ctx))
+	data := append([]interface{}{id}, args...)
+	l.p(ERROR, data...)
+}
+
+func Cerrorf(ctx context.Context, format string, args ...interface{}) {
+	f := "trace_id=%s||" + format
+	data := append([]interface{}{getTraceIdFromCtx(ctx)}, args...)
+	l.pf(ERROR, f, data...)
+}
+
+func Cwarning(ctx context.Context, args ...interface{}) {
+	id := fmt.Sprintf("trace_id=%s", getTraceIdFromCtx(ctx))
+	data := append([]interface{}{id}, args...)
+	l.p(WARNING, data...)
+}
+
+func Cwarningf(ctx context.Context, format string, args ...interface{}) {
+	f := "trace_id=%s||" + format
+	data := append([]interface{}{getTraceIdFromCtx(ctx)}, args...)
+	l.pf(WARNING, f, data...)
+}
+
+func Cfatal(ctx context.Context, args ...interface{}) {
+	id := fmt.Sprintf("trace_id=%s", getTraceIdFromCtx(ctx))
+	data := append([]interface{}{id}, args...)
+	l.p(FATAL, data...)
+}
+
+func Cfatalf(ctx context.Context, format string, args ...interface{}) {
+	f := "trace_id=%s||" + format
+	data := append([]interface{}{getTraceIdFromCtx(ctx)}, args...)
+	l.pf(FATAL, f, data...)
+}
+
+func getTraceIdFromCtx(ctx context.Context) string {
+	md, ok := metadata.FromIncomingContext(ctx)
+	if !ok {
+		return "0"
+	}
+	v, ok := md["trace_id"]
+	if !ok || len(v) < 1 {
+		return "0"
+	}
+
+	return v[0]
+}