Function BenchmarkRequestCollection

internal/engines/bulk_benchmark_test.go:100–141

BenchmarkRequestCollection tests the performance of request collection

(b *testing.B)

// BenchmarkRequestCollection tests the performance of request collection
func BenchmarkRequestCollection(b *testing.B) {
	ctx := context.Background()
	checker := &MockChecker{}

	config := BulkCheckerConfig{
		ConcurrencyLimit: 10,
		BufferSize:       10000,
	}

	callback := func(id, ct string) {}

	b.ResetTimer()
	b.ReportAllocs()

	for b.Loop() {
		bc, err := NewBulkChecker(ctx, checker, BulkCheckerTypeEntity, callback, config)
		if err != nil {
			b.Fatal(err)
		}

		publisher := NewBulkEntityPublisher(ctx, &base.PermissionLookupEntityRequest{
			TenantId:   "test",
			Metadata:   &base.PermissionLookupEntityRequestMetadata{},
			EntityType: "document",
			Permission: "read",
			Subject:    &base.Subject{Id: "user1"},
		}, bc)

		// Publish requests rapidly
		for j := 0; j < 10000; j++ {
			publisher.Publish(
				&base.Entity{Id: fmt.Sprintf("doc%d", j)},
				&base.PermissionCheckRequestMetadata{},
				&base.Context{},
				base.CheckResult_CHECK_RESULT_UNSPECIFIED,
			)
		}

		bc.StopCollectingRequests()
		bc.Close()
	}
}

// BenchmarkSorting tests the performance of request sorting
func BenchmarkSorting(b *testing.B) {