Function BenchmarkLogRotation

base/logging_test.go:112–161

Benchmark the time it takes to write x bytes of data to a logger, and optionally rotate and compress it.

(b *testing.B)

// Benchmark the time it takes to write x bytes of data to a logger, and optionally rotate and compress it.
func BenchmarkLogRotation(b *testing.B) {

	tests := []struct {
		rotate   bool
		compress bool
		numBytes int
	}{
		{rotate: false, compress: false, numBytes: 0},
		{rotate: false, compress: false, numBytes: 1024 * 1000},   // 1MB
		{rotate: false, compress: false, numBytes: 1024 * 100000}, // 100MB
		{rotate: true, compress: false, numBytes: 0},
		{rotate: true, compress: false, numBytes: 1024 * 1000},   // 1MB
		{rotate: true, compress: false, numBytes: 1024 * 100000}, // 100MB
		{rotate: true, compress: true, numBytes: 0},
		{rotate: true, compress: true, numBytes: 1024 * 1000},   // 1MB
		{rotate: true, compress: true, numBytes: 1024 * 100000}, // 100MB
	}

	for _, test := range tests {
		b.Run(fmt.Sprintf("rotate:%t-compress:%t-bytes:%v", test.rotate, test.compress, test.numBytes), func(bm *testing.B) {
			data := FastRandBytes(bm, test.numBytes)

			logPath := b.TempDir()
			logger := lumberjack.Logger{Filename: filepath.Join(logPath, "output.log"), Compress: test.compress}

			bm.ResetTimer()
			for i := 0; i < bm.N; i++ {
				_, _ = logger.Write(data)
				if test.rotate {
					_ = logger.Rotate()
				}
			}
			bm.StopTimer()

			// Tidy up temp log files in a retry loop because
			// we can't remove temp dir while the async compression is still writing log files
			assert.NoError(bm, logger.Close())
			ctx := TestCtx(bm)
			err, _ := RetryLoop(ctx, "benchmark-logrotate-teardown",
				func() (shouldRetry bool, err error, value interface{}) {
					err = os.RemoveAll(logPath)
					return err != nil, err, nil
				},
				CreateDoublingSleeperFunc(3, 250),
			)
			assert.NoError(bm, err)
		})
	}

}

func TestLogColor(t *testing.T) {
	origColor := consoleLogger.Load().ColorEnabled