Function Benchmark

riverdriver/riverdrivertest/benchmark.go:17–282

(ctx context.Context, b *testing.B,
	driverWithPool func(ctx context.Context, b *testing.B) (riverdriver.Driver[TTx], string),
	executorWithTx func(ctx context.Context, b *testing.B) riverdriver.Executor,
)

)

func Benchmark[TTx any](ctx context.Context, b *testing.B,
	driverWithPool func(ctx context.Context, b *testing.B) (riverdriver.Driver[TTx], string),
	executorWithTx func(ctx context.Context, b *testing.B) riverdriver.Executor,
) {
	b.Helper()

	setup := func(ctx context.Context, b *testing.B) (riverdriver.Executor, string) {
		b.Helper()
		_, schema := driverWithPool(ctx, b)
		return executorWithTx(ctx, b), schema
	}

	b.Run("JobSetStateIfRunningMany", func(b *testing.B) {
		exec, schema := setup(ctx, b)

		const (
			totalJobs  = 100000
			batchSize  = 2000
			numBatches = totalJobs / batchSize
		)

		// Build a single batch of job parameters to be reused
		insertParams := make([]*riverdriver.JobInsertFullParams, batchSize)
		for i := range batchSize {
			// Create a mix of jobs with different states and metadata
			var (
				metadata []byte
				state    rivertype.JobState
			)

			switch i % 10 {
			case 0, 1, 2, 3, 4, 5:
				// Most jobs are running
				state = rivertype.JobStateRunning
				if i%20 == 0 {
					// Every 20th job has cancel_attempted_at
					metadata = []byte(`{"cancel_attempted_at": "2024-01-01T00:00:00Z"}`)
				}
			case 6, 7:
				// Some jobs are completed (to test no-op path)
				state = rivertype.JobStateCompleted
			case 8, 9:
				// Some jobs are available (to test no-op path)
				state = rivertype.JobStateAvailable
				if i%15 == 0 {
					// Every 15th job has metadata to merge
					metadata = []byte(`{"key": "value"}`)
				}
			}

			insertParams[i] = testfactory.Job_Build(b, &testfactory.JobOpts{
				Metadata: metadata,
				State:    &state,
			})
		}

		// Insert the batch multiple times to reach our total
		var batchJobs []*rivertype.JobRow