-
Notifications
You must be signed in to change notification settings - Fork 26
/
BulkUpdateBenchmarks.cs
101 lines (83 loc) · 2.9 KB
/
BulkUpdateBenchmarks.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
using BenchmarkDotNet.Attributes;
using EntityFrameworkCore.SqlServer.SimpleBulks.Benchmarks.Database;
using EntityFrameworkCore.SqlServer.SimpleBulks.BulkInsert;
using EntityFrameworkCore.SqlServer.SimpleBulks.BulkUpdate;
using Microsoft.EntityFrameworkCore;
namespace EntityFrameworkCore.SqlServer.SimpleBulks.Benchmarks
{
[WarmupCount(0)]
[IterationCount(1)]
[InvocationCount(1)]
[MemoryDiagnoser]
public class BulkUpdateBenchmarks
{
private TestDbContext _context;
private List<Customer> _customers;
private List<Guid> _customerIds;
[Params(100, 1000, 10_000, 100_000, 250_000)]
public int RowsCount { get; set; }
//[Params(500_000, 1_000_000)]
//public int RowsCount { get; set; }
[IterationSetup]
public void IterationSetup()
{
_context = new TestDbContext($"Server=127.0.0.1;Database=EntityFrameworkCore.SqlServer.SimpleBulks.Benchmarks.{Guid.NewGuid()};User Id=sa;Password=sqladmin123!@#;Encrypt=False");
_context.Database.EnsureCreated();
_context.Database.SetCommandTimeout(TimeSpan.FromMinutes(2));
_customers = new List<Customer>(RowsCount);
for (int i = 0; i < RowsCount; i++)
{
var customer = new Customer
{
FirstName = "FirstName " + i,
LastName = "LastName " + i,
Index = i,
};
_customers.Add(customer);
}
_context.BulkInsert(_customers);
_customerIds = _customers.Select(x => x.Id).ToList();
}
[IterationCleanup]
public void IterationCleanup()
{
_context.Database.EnsureDeleted();
}
[Benchmark]
public void EFCoreUpdate()
{
var pageSize = 10_000;
var pages = _customerIds.Chunk(pageSize);
var random = new Random(2024);
foreach (var page in pages)
{
var customers = _context.Customers.Where(x => page.Contains(x.Id)).ToList();
foreach (var customer in customers)
{
customer.FirstName = "Updated" + random.Next();
}
}
_context.SaveChanges();
}
[Benchmark]
public void BulkUpdate()
{
var random = new Random(2024);
foreach (var customer in _customers)
{
customer.FirstName = "Updated" + random.Next();
}
var pageSize = 10_000;
var pages = _customers.Chunk(pageSize);
foreach (var page in pages)
{
_context.BulkUpdate(page,
x => new { x.FirstName },
opt =>
{
opt.Timeout = 0;
});
}
}
}
}