TechEmpower.FrameworkBenchmarks/frameworks/CSharp/genhttp/Benchmarks/Tests/UpdateResource.cs

using Benchmarks.Model;
using GenHTTP.Modules.Webservices;
using Microsoft.EntityFrameworkCore;

namespace Benchmarks.Tests;

public sealed class UpdateResource
{
    private static readonly Random Random = new();

    [ResourceMethod(":queries")]
    public ValueTask<List<World>> UpdateWorldsFromPath(string queries) => UpdateWorlds(queries);

    [ResourceMethod]
    public async ValueTask<List<World>> UpdateWorlds(string queries)
    {
        var count = 1;

        int.TryParse(queries, out count);

        if (count < 1)
        {
            count = 1;
        }
        else if (count > 500)
        {
            count = 500;
        }

        var result = new List<World>(count);

        var ids = Enumerable.Range(1, 10000).Select(x => Random.Next(1, 10001)).Distinct().Take(count).ToArray();

        var context = Database.TrackingPool.Rent();

        try
        {
            foreach (var id in ids)
            {
                var record = await context.World.FirstOrDefaultAsync(w => w.Id == id).ConfigureAwait(false);

                var old = record.RandomNumber;

                var current = old;

                for (var i = 0; i < 5; i++)
                {
                    current = Random.Next(1, 10001);

                    if (current != old)
                    {
                        break;
                    }
                }

                record.RandomNumber = current;

                result.Add(record);

                await context.SaveChangesAsync();
            }
        }
        finally
        {
            Database.TrackingPool.Return(context);
        }

        return result;
    }
}