Function BenchmarkMiddlewareWorkAppend

riverlog/river_log_benchmark_test.go:15–61

(b *testing.B)

)

func BenchmarkMiddlewareWorkAppend(b *testing.B) {
	if testing.Short() {
		b.Skip("skipping benchmark in short mode")
	}

	middleware := NewMiddleware(func(w io.Writer) slog.Handler {
		return slog.NewTextHandler(w, nil)
	}, nil)

	cases := []struct {
		name string
		size int
	}{
		{name: "metadata_256kb", size: 256 * 1024},
		{name: "metadata_2mb", size: 2 * 1024 * 1024},
	}

	for _, tc := range cases {
		b.Run(tc.name, func(b *testing.B) {
			metadata := makeMetadataWithLogSize(tc.size)
			appendLogLine := strings.Repeat("x", 2048)

			b.ReportAllocs()
			b.ResetTimer()

			for range b.N {
				metadataUpdates := map[string]any{}
				ctx := context.WithValue(context.Background(), jobexecutor.ContextKeyMetadataUpdates, metadataUpdates)
				job := &rivertype.JobRow{
					Attempt:  1,
					Metadata: metadata,
				}

				if err := middleware.Work(ctx, job, func(ctx context.Context) error {
					Logger(ctx).InfoContext(ctx, appendLogLine)
					return nil
				}); err != nil {
					b.Fatalf("work returned error: %v", err)
				}

				if _, ok := metadataUpdates[metadataKey]; !ok {
					b.Fatal("missing river:log metadata update")
				}
			}
		})
	}
}

func makeMetadataWithLogSize(targetBytes int) []byte {
	if targetBytes <= 0 {