my_logger.go

package log

import (
	"fmt"
	"io"
	"os"
	"runtime"
	"strings"
	"time"
)

var (
	DefaultTimeFormat = time.RFC3339
)

type Fields map[string]interface{}

type myLogger struct {
	Writer    io.Writer
	ErrWriter io.Writer
	prefix    string
	begin     time.Time
	level     uint8
	data      Fields
}

func logPrefix(level uint8) string {
	return fmt.Sprintf("time=%q level=%s function=%s", time.Now().Format(DefaultTimeFormat), unmarshalLevel(level), funcName())
}

func (fields Fields) String() string {
	if len(fields) == 0 {
		return ""
	}
	var buffer strings.Builder
	for k, v := range fields {
		switch v := v.(type) {
		case string:
			buffer.WriteString(fmt.Sprintf(" %s=%q", k, v))
		default:
			buffer.WriteString(fmt.Sprintf(" %s=%v", k, v))
		}
	}
	return buffer.String()
}

func (l myLogger) log(flag uint8, s ...interface{}) {
	var w io.Writer
	{
		if flag == errorFlag {
			w = l.ErrWriter
		} else {
			w = l.Writer
		}
	}
	f := fmt.Sprintf("%s%s msg=%q%s", logPrefix(flag), l.prefix, fmt.Sprint(s...), l.data)
	fmt.Fprintln(w, f)
}

func (l myLogger) logf(flag uint8, format string, s ...interface{}) {
	var w io.Writer
	{
		if flag == errorFlag {
			w = l.ErrWriter
		} else {
			w = l.Writer
		}
	}
	format = fmt.Sprintf("%%s%%s msg=\"%s\"%s\n", format, l.data)
	s = append([]interface{}{logPrefix(flag), l.prefix}, s...)
	fmt.Fprintf(w, format, s...)
}

func funcName() string {
	pc := make([]uintptr, 10)
	runtime.Callers(2, pc)
	for _, p := range pc {
		fn := runtime.FuncForPC(p)
		if !strings.HasPrefix(fn.Name(), "github.com/rjbasitali/go-log") {
			fname := fn.Name()
			i := strings.LastIndexByte(fname, byte('/'))
			if i == -1 {
				return fname
			}
			return fname[i+1:]
		}
	}
	return ""
}

func NewLogger(w, errw io.Writer) Logger {
	if w == nil {
		w = os.Stdout
	}
	if errw == nil {
		errw = os.Stderr
	}
	return myLogger{
		Writer:    w,
		ErrWriter: errw,
		begin:     time.Now(),
		level:     traceFlag,
	}
}