logs_test.go

package logs

import (
	"context"
	"os"
	"runtime"
	"sync/atomic"
	"testing"
	"time"
)

// ---------------------------------------------------------------------------------------------------Parallel
type blackholeStream struct {
	writeCount uint64
}

func (s *blackholeStream) WriteCount() uint64 {
	return atomic.LoadUint64(&s.writeCount)
}

func (s *blackholeStream) Write(p []byte) (int, error) {
	atomic.AddUint64(&s.writeCount, 1)
	return len(p), nil
}
func BenchmarkParallel(b *testing.B) {
	stream := &blackholeStream{}
	logger := New(stream)
	logger.SetText()
	// logger.SetCaller(true)
	// logger.caller = true
	b.ResetTimer()
	b.RunParallel(func(pb *testing.PB) {
		for pb.Next() {
			logger.With().
				Str("str", "str").
				Int("int", 1025).
				Bool("bool", true).
				Int8("int8", 8).
				Int16("int16", 16).
				Int32("int32", 32).
				Int64("int64", 64).
				Uint("uint", 6).
				Uint8("uin8", 8).
				Float32("float32", 3.14).Info()
		}
	})

	if stream.WriteCount() != uint64(b.N) {
		b.Fatalf("Log write count")
	}
}
func BenchmarkLog(b *testing.B) {
	l := New(os.Stdout)
	l.SetFile("./logs/app.log")
	for i := 0; i < b.N; i++ {
		l.Info()
	}
}
func BenchmarkParallelFile(b *testing.B) {
	logger := New(nil)
	logger.SetFile("./logs/app.log")
	// logger.SetCaller(true)
	// logger.caller = true
	b.ResetTimer()
	b.RunParallel(func(pb *testing.PB) {
		for pb.Next() {
			logger.With().
				Str("str", "str").
				Int("int", 1025).
				Bool("bool", true).
				Int("int", 64).
				Int64("int64", 64).Info()
		}
	})
}
func TestLogger(t *testing.T) {
	log.SetCaller(true)
	Debug()
	Debug("debug")
	Debugf("debugf")
	Info()
	Info("info")
	Infof("infof")
	Warn()
	Warn("warn")
	Warnf("warnf")
	Error()
	Error("erro")
	Errorf("errorf")
}

func TestField(t *testing.T) {
	n := New(os.Stdout)
	f := n.With()
	f.Bool("out", false).
		Bool("key", true).
		Int("key", 1).
		Int8("key", 2).
		Int16("key", 3).
		Int32("key", 4).
		Int64("key", 5).
		Uint("key", 6).
		Uint8("key", 7).
		Uint16("key", 8).
		Uint32("key", 9).
		Uint64("key", 10).
		Float32("key", 11.98122).
		Float64("key", 12.987654321).
		Str("key", "a").
		Bytes("key", []byte("b")).
		Time("key", time.Time{}).
		Dur("key", 0).Any("key-any", runtime.BlockProfileRecord{})
	f.Info()
}

func TestLog(t *testing.T) {
	l := New(os.Stdout)
	l.SetCaller(true)
	ctx := TraceCtx(context.Background(), trace())
	l.Ctx(ctx).Info()
	l.Ctx(ctx).Info()
	l.Ctx(ctx).Str("t", "xx").Str("tx", "tt").Info()
	l.Ctx(ctx).Info()
	l.Ctx(ctx).Error()
	s := l.Ctx(ctx)
	s.Bool("b", false)
	s.Info("666")
	s.Info("xx")
}
func TestLog1(t *testing.T) {
	l := New(os.Stdout)
	l.SetCaller(true)
	// l.SetFile("./logs1/app.log")
	defer l.Close()
	ctx := TraceCtx(context.Background(), trace())
	l1 := l.Ctx(ctx).Str("basic", "basic")
	l1.Debug()
	l1.Info()
	l1.Error()
	s := l.Ctx(ctx)
	s.Bool("b", false)
	s.Info("666")
	s.Info("xx")
}
func TestWriter(t *testing.T) {
	SetFile("./logs/app.log")
	// SetText()
	SetText()
	SetCons(true)
	SetCaller(true)
	for i := 0; i < 10; i++ {
		With().Int("idx", i).Debug()
		With().Int("idx", i).Debug("debug")
		With().Int("idx", i).Debugf("debugf")
		With().Int("idx", i).Info()
		With().Int("idx", i).Info("info")
		With().Int("idx", i).Infof("infof")
		With().Int("idx", i).Warn()
		With().Int("idx", i).Warn("warn")
		With().Int("idx", i).Warnf("warnf")
		With().Int("idx", i).Error()
		With().Int("idx", i).Error("erro")
		With().Int("idx", i).Errorf("errorf")
	}
	With().Str("idx", "sp ce").Errorf("omit empty")
	Close()
}

// # 使用benchmark采集3秒的内存维度的数据，并生成文件
// go test run=^$ -bench=^BenchmarkZerologJSONNegative$ github.com/zxysilent/logs -benchmem  -benchtime=3s -memprofile=mem_profile.out
// # 采集CPU维度的数据
// go test -benchmem -benchtime=3s -bench=^BenchmarkZerologJSONNegative1$ -cpuprofile=cpu_profile.out1
// # 查看pprof文件，指定http方式查看
// go tool pprof -http="127.0.0.1:8080" mem_profile.out
// go tool pprof -http="127.0.0.1:8080" cpu_profile.out1
// # 查看pprof文件，直接在命令行查看
// go tool pprof mem_profile.out
// go test -benchmem -run=^$ -bench ^BenchmarkZerologJSONNegative$ github.com/zxysilent/logs -count=1 -v -benchtime=3s -cpuprofile=cpu_profile.out
// go tool pprof -http="127.0.0.1:8080" cpu_profile.out
// go test -benchmem -run=^$ -bench ^BenchmarkZerologJSONPositive1$ github.com/zxysilent/logs -count=1 -v -benchtime=3s -cpuprofile=cpu_profile.out1
// go tool pprof -http="127.0.0.1:8080" cpu_profile.out1