public void TestAddGetMany(
[Values(LightningDbSyncMode.Fsync, LightningDbSyncMode.Async, LightningDbSyncMode.NoSync)] LightningDbSyncMode asyncStore,
[Values(typeof(LocalClientFactory), typeof(GrpcClientFactory))] Type clientFactory,
[Values(1, 10, 100, 1000, 10000, 100000)] int iterations)
{
var config = Config;
config.SyncMode = asyncStore;
using (var client = CreateClient(clientFactory, config))
{
var valueLength = 1024;
var sw = new Stopwatch();
sw.Start();
var kvs = BuildKeyValuePairs(iterations, valueLength);
sw.Stop();
Console.WriteLine($"Preparing data elapsed: {sw.Elapsed}");
sw = new Stopwatch();
sw.Start();
var addRet = client.TryAdd(kvs.Keys, (k, s) =>
{
var buffer = kvs[k].Item2;
s.Write(buffer, 0, buffer.Length);
}, DateTimeOffset.Now.AddSeconds(500));
sw.Stop();
Console.WriteLine($"Write elapsed: {sw.Elapsed}");
Assert.AreEqual(kvs.Count, addRet.Count);
Assert.IsTrue(addRet.All(k => kvs.ContainsKey(k)));
var getDict = new Dictionary <string, byte[]>();
sw = new Stopwatch();
sw.Start();
var getRet = client.TryGet(kvs.Keys, (k, s) =>
{
var readBuffer = new byte[valueLength];
var readBytes = s.Read(readBuffer, 0, readBuffer.Length);
getDict[k] = readBuffer.Take(readBytes).ToArray();
});
sw.Stop();
Console.WriteLine($"Read elapsed: {sw.Elapsed}");
Assert.AreEqual(kvs.Keys.Count, getRet.Count);
Assert.IsTrue(getRet.All(k => kvs.ContainsKey(k)));
Assert.AreEqual(kvs.Keys.Count, getDict.Count);
foreach (var kv in getDict)
{
Assert.IsTrue(kvs.ContainsKey(kv.Key));
CollectionAssert.AreEqual(kvs[kv.Key].Item2, kv.Value);
}
}
}
public void TestAddGet(
[Values(LightningDbSyncMode.Fsync, LightningDbSyncMode.Async, LightningDbSyncMode.NoSync)] LightningDbSyncMode asyncStore,
[Values(typeof(LocalClientFactory), typeof(GrpcClientFactory))] Type clientFactory,
[Values(1, 10, 100, 1000, 10000)] int iterations)
{
var config = Config;
config.SyncMode = asyncStore;
using (var client = CreateClient(clientFactory, config))
{
var valueLength = 1024;
var totalWrite = new TimeSpan();
var totalRead = new TimeSpan();
var swTotal = new Stopwatch();
swTotal.Start();
var kvs = BuildKeyValuePairs(iterations, valueLength);
swTotal.Stop();
Console.WriteLine($"Preparing data elapsed: {swTotal.Elapsed}");
swTotal = new Stopwatch();
swTotal.Start();
foreach (var kv in kvs)
{
var testKey = kv.Key;
var testValue = kv.Value.Item2;
var sw = new Stopwatch();
sw.Start();
var addRet = client.TryAdd(new[] { testKey }, (k, s) => s.Write(testValue, 0, testValue.Length), DateTimeOffset.UtcNow.AddSeconds(500));
sw.Stop();
Console.WriteLine($"Write {kv.Value.Item1} elapsed: {sw.Elapsed}");
totalWrite += sw.Elapsed;
Assert.AreEqual(1, addRet.Count);
Assert.AreEqual(testKey, addRet.First());
var readBuffer = new byte[valueLength];
int readBytes = -1;
sw = new Stopwatch();
sw.Start();
var getRet = client.TryGet(new[] { testKey }, (k, s) => readBytes = s.Read(readBuffer, 0, readBuffer.Length));
sw.Stop();
Console.WriteLine($"Read {kv.Value.Item1} elapsed: {sw.Elapsed}");
totalRead += sw.Elapsed;
Assert.AreEqual(1, getRet.Count);
Assert.AreEqual(testKey, getRet.First());
Assert.AreEqual(testValue.Length, readBytes);
CollectionAssert.AreEqual(testValue, readBuffer.Take(readBytes).ToArray());
}
swTotal.Stop();
Console.WriteLine($"Elapsed total: {swTotal.Elapsed}");
Console.WriteLine($"Elapsed total Write: {totalWrite}");
Console.WriteLine($"Elapsed total Read: {totalRead}");
}
}
public void TestAddGetManyParallel(
[Values(1, 2, 3, 4)] int writeStreams,
[Values(2, 3, 4, 10)] int readStreams,
[Values(LightningDbSyncMode.Fsync, LightningDbSyncMode.Async, LightningDbSyncMode.NoSync)] LightningDbSyncMode asyncStore,
[Values(1, 10, 100, 1000, 10000, 100000)] int iterations)
{
var config = Config;
config.SyncMode = asyncStore;
using (var client = new GrpcTestClient(config))
{
var valueLength = 1024;
var sw = new Stopwatch();
sw.Start();
var kvsAll = BuildKeyValuePairs(iterations * writeStreams, valueLength);
sw.Stop();
Console.WriteLine($"Preparing data elapsed: {sw.Elapsed}");
var swAll = new Stopwatch();
swAll.Start();
Parallel.For(0, writeStreams, new ParallelOptions {
MaxDegreeOfParallelism = writeStreams
}, i =>
{
var kvs = kvsAll.Skip(i * iterations).Take(iterations).ToDictionary(kv => kv.Key, kv => kv.Value);
using (var writeClient = new LightClient(new Channel($"127.0.0.1:{client.ReplicaConfig.Port}", ChannelCredentials.Insecure), new ClientConfig
{
UseStreaming = true,
Compression = None,
HashedWith = Md5
}))
{
sw = new Stopwatch();
sw.Start();
var addRet = writeClient.TryAdd(kvs.Keys, (k, s) =>
{
var buffer = kvs[k].Item2;
s.Write(buffer, 0, buffer.Length);
}, DateTimeOffset.Now.AddSeconds(500));
sw.Stop();
Console.WriteLine($"Write elapsed: {sw.Elapsed}");
Assert.AreEqual(kvs.Count, addRet.Count);
Assert.IsTrue(addRet.All(k => kvs.ContainsKey(k)));
}
});
swAll.Stop();
Console.WriteLine($"Write Elapsed total: {swAll.Elapsed}");
Console.WriteLine($"Write Elapsed per thread: {swAll.Elapsed / writeStreams}");
swAll = new Stopwatch();
swAll.Start();
Parallel.For(0, readStreams, new ParallelOptions {
MaxDegreeOfParallelism = readStreams
}, i =>
{
var kvs = kvsAll.Skip((i % writeStreams) * iterations).Take(iterations).ToDictionary(kv => kv.Key, kv => kv.Value);
using (var readClient = new LightClient(new Channel($"127.0.0.1:{client.ReplicaConfig.Port}", ChannelCredentials.Insecure), new ClientConfig
{
UseStreaming = true,
Compression = None,
HashedWith = Md5
}))
{
var getDict = new Dictionary <string, byte[]>();
sw = new Stopwatch();
sw.Start();
var getRet = readClient.TryGet(kvs.Keys, (k, s) =>
{
var readBuffer = new byte[valueLength];
var readBytes = s.Read(readBuffer, 0, readBuffer.Length);
getDict[k] = readBuffer.Take(readBytes).ToArray();
});
sw.Stop();
Console.WriteLine($"Read elapsed: {sw.Elapsed}");
Assert.AreEqual(kvs.Keys.Count, getRet.Count);
Assert.IsTrue(getRet.All(k => kvs.ContainsKey(k)));
Assert.AreEqual(kvs.Keys.Count, getDict.Count);
foreach (var kv in getDict)
{
Assert.IsTrue(kvs.ContainsKey(kv.Key));
CollectionAssert.AreEqual(kvs[kv.Key].Item2, kv.Value);
}
}
});
swAll.Stop();
Console.WriteLine($"Read Elapsed total: {swAll.Elapsed}");
Console.WriteLine($"Read Elapsed per thread: {swAll.Elapsed / readStreams}");
}
}