-
Notifications
You must be signed in to change notification settings - Fork 5
/
BeelineDb.cs
61 lines (56 loc) · 2.16 KB
/
BeelineDb.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
using System;
using System.Buffers;
using System.Data;
using System.Data.Common;
using System.Diagnostics;
using System.Threading;
using System.Threading.Tasks;
using Beeline.AspNetCoreBenchmarks.Configuration;
using Microsoft.Extensions.Options;
namespace Beeline.AspNetCoreBenchmarks.Data
{
public class BeelineDb
{
private readonly IRandom _random;
private readonly DbProviderFactory _dbProvider;
private readonly string _connectionString;
private ObjectWriter _objectWriter;
public BeelineDb(IRandom random, DbProviderFactory dbProvider, IOptions<AppSettings> appSettings)
{
_random = random;
_dbProvider = dbProvider;
_connectionString = appSettings.Value.ConnectionString;
}
public async Task<int> LoadSingleQueryRow(byte[] buffer, CancellationToken ct = default)
{
using (var db = _dbProvider.CreateConnection())
{
Debug.Assert(db != null);
db.ConnectionString = _connectionString;
using (var cmd = CreateSingleQueryCommand(db, _random.Next(1, 10001)))
{
await db.OpenAsync(ct);
using (var reader = await cmd.ExecuteReaderAsync(ct))
{
if (_objectWriter is null)
{
_objectWriter = new ObjectWriter(RowSerializer.For(reader, true), 1024);
}
return await _objectWriter.WriteSingle(reader, buffer, ct);
}
}
}
}
private static DbCommand CreateSingleQueryCommand(DbConnection db, int id)
{
var cmd = db.CreateCommand();
cmd.CommandText = "SELECT id, randomnumber FROM world WHERE id = @id";
var idParameter = cmd.CreateParameter();
idParameter.ParameterName = "id";
idParameter.DbType = DbType.Int32;
idParameter.Value = id;
cmd.Parameters.Add(idParameter);
return cmd;
}
}
}