package zapjournald

import (
	"testing"

	"github.com/ssgreg/journald"
	"go.uber.org/zap"
	"go.uber.org/zap/zapcore"
)

func BenchmarkEncoder(b *testing.B) {
	zc := zap.NewProductionConfig()
	zc.EncoderConfig.EncodeTime = zapcore.ISO8601TimeEncoder
	zc.Level = zap.NewAtomicLevelAt(zap.InfoLevel)

	b.Run("console", func(b *testing.B) {
		encoder := zapcore.NewConsoleEncoder(zc.EncoderConfig)
		core := NewCore(zc.Level, encoder, &journald.Journal{}, SyslogFields)
		core.j.TestModeEnabled = true // Disable actual writing to the journal.

		coreWithContext := core.With([]zapcore.Field{
			SyslogFacility(LogDaemon),
			SyslogIdentifier(),
			SyslogPid(),
		})
		l := zap.New(coreWithContext)
		benchmarkEncoder(b, l)
	})
	b.Run("partial", func(b *testing.B) {
		encoder := NewPartialEncoder(zapcore.NewConsoleEncoder(zc.EncoderConfig), SyslogFields)
		core := NewCore(zc.Level, encoder, &journald.Journal{}, SyslogFields)
		core.j.TestModeEnabled = true // Disable actual writing to the journal.

		coreWithContext := core.With([]zapcore.Field{
			SyslogFacility(LogDaemon),
			SyslogIdentifier(),
			SyslogPid(),
		})
		l := zap.New(coreWithContext)
		benchmarkEncoder(b, l)
	})
}

func benchmarkEncoder(b *testing.B, l *zap.Logger) {
	b.Run("no fields", func(b *testing.B) {
		b.ReportAllocs()
		for i := 0; i < b.N; i++ {
			l.Info("Simple log message")
		}
	})
	b.Run("application fields", func(b *testing.B) {
		b.ReportAllocs()
		for i := 0; i < b.N; i++ {
			l.Info("Simple log message", zap.Uint32("count", 123), zap.String("details", "nothing"))
		}
	})
	b.Run("journald fields", func(b *testing.B) {
		b.ReportAllocs()
		for i := 0; i < b.N; i++ {
			l.Info("Simple log message", SyslogIdentifier())
		}
	})
}