public void ReopenStoreAfterDispose()
{
var path = Path.Combine(Path.GetTempPath(), $"rocksdb_test_{Guid.NewGuid()}");
try
{
var store = new RocksDBStore(path);
var stateStore =
new TrieStateStore(new MemoryKeyValueStore(), new MemoryKeyValueStore());
var blocks = new BlockChain <DumbAction>(
new NullPolicy <DumbAction>(),
new VolatileStagePolicy <DumbAction>(),
store,
stateStore,
Fx.GenesisBlock
);
store.Dispose();
store = new RocksDBStore(path);
store.Dispose();
}
finally
{
Directory.Delete(path, true);
}
}
public void ParallelGetBlock()
{
var path = Path.Combine(Path.GetTempPath(), $"rocksdb_test_{Guid.NewGuid()}");
var store = new RocksDBStore(path);
try
{
Guid cid = Guid.NewGuid();
store.AppendIndex(cid, Fx.Block1.Hash);
store.AppendIndex(cid, Fx.Block2.Hash);
store.AppendIndex(cid, Fx.Block3.Hash);
store.PutBlock(Fx.Block1);
store.PutBlock(Fx.Block2);
store.PutBlock(Fx.Block3);
store.Dispose();
store = new RocksDBStore(path);
Enumerable.Range(0, 3).AsParallel().ForAll(i =>
{
var bHash = store.IndexBlockHash(cid, i).Value;
var block = store.GetBlock <DumbAction>(Fx.GetHashAlgorithm, bHash);
Assert.NotNull(block);
});
}
finally
{
store.Dispose();
Directory.Delete(path, true);
}
}
public void ParallelGetTransaction()
{
var path = Path.Combine(Path.GetTempPath(), $"rocksdb_test_{Guid.NewGuid()}");
var store = new RocksDBStore(path);
Transaction <DumbAction>[] txs = new[]
{
Fx.Transaction1,
Fx.Transaction2,
Fx.Transaction3,
};
try
{
store.PutTransaction(Fx.Transaction1);
store.PutTransaction(Fx.Transaction2);
store.PutTransaction(Fx.Transaction3);
store.Dispose();
store = new RocksDBStore(path);
Enumerable.Range(0, 3).AsParallel().ForAll(i =>
{
Assert.NotNull(store.GetTransaction <DumbAction>(txs[i].Id));
});
}
finally
{
store.Dispose();
Directory.Delete(path, true);
}
}
public void ListChainIds()
{
var path = Path.Combine(Path.GetTempPath(), $"rocksdb_test_{Guid.NewGuid()}");
var store = new RocksDBStore(path);
try
{
Guid cid = Guid.NewGuid();
store.AppendIndex(cid, Fx.Block1.Hash);
store.AppendIndex(cid, Fx.Block2.Hash);
store.AppendIndex(cid, Fx.Block3.Hash);
store.PutBlock(Fx.Block1);
store.PutBlock(Fx.Block2);
store.PutBlock(Fx.Block3);
Assert.Single(store.ListChainIds());
store.ForkBlockIndexes(cid, Guid.NewGuid(), Fx.Block3.Hash);
Assert.Equal(2, store.ListChainIds().Count());
}
finally
{
store.Dispose();
Directory.Delete(path, true);
}
}
public void Migration(
[Option('o', Description = "Path to migration target root path.")]
string originRootPath,
[Option('d', Description = "Path to migrated dist root path.")]
string distRootPath,
[Option('c', Description = "Skip the copy from target store.")]
bool skipCopy,
[Option('b', Description = "Skip the block from target store.")]
bool skipBlock,
[Option('t', Description = "Skip the transaction from target store.")]
bool skipTransaction
)
{
if (!skipCopy)
{
Console.WriteLine("Copy Start!");
DirectoryCopy(originRootPath, distRootPath, true);
Directory.Delete(Path.Combine(distRootPath, "tx"), true);
Directory.Delete(Path.Combine(distRootPath, "block"), true);
Console.WriteLine("Copy Success!");
}
else
{
Console.WriteLine("Skip copy");
}
var originStore = new MonoRocksDBStore(originRootPath);
var distStore = new RocksDBStore(distRootPath);
var totalLength = originStore.CountBlocks();
if (!skipBlock)
{
Console.WriteLine("Start migrate block.");
foreach (var item in
originStore.IterateBlockHashes().Select((value, i) => new { i, value }))
{
Console.WriteLine($"block progress: {item.i}/{totalLength}");
Block <NCAction> block = originStore.GetBlock <NCAction>(
_ => HashAlgorithmType.Of <SHA256>(), // thunk getter
item.value
);
distStore.PutBlock(block);
}
Console.WriteLine("Finish migrate block.");
}
totalLength = originStore.CountTransactions();
if (!skipTransaction)
{
Console.WriteLine("Start migrate transaction.");
foreach (var item in
originStore.IterateTransactionIds()
.Select((value, i) => new { i, value }))
{
Console.WriteLine($"tx progress: {item.i}/{totalLength}");
distStore.PutTransaction(originStore.GetTransaction <NCAction>(item.value));
}
Console.WriteLine("Finish migrate transaction.");
}
originStore.Dispose();
distStore.Dispose();
Console.WriteLine("Migration Success!");
}
public void PruneOutdatedChainsRocksDb()
{
var path = Path.Combine(Path.GetTempPath(), $"rocksdb_test_{Guid.NewGuid()}");
var store = new RocksDBStore(path);
RocksDb chainDb = null;
int KeysWithChainId(RocksDb db, Guid cid)
{
using (Iterator it = db.NewIterator())
{
byte[] key = cid.ToByteArray();
int count = 0;
for (it.SeekToFirst(); it.Valid(); it.Next())
{
if (!it.Key().Skip(1).ToArray().StartsWith(key))
{
continue;
}
count++;
}
return(count);
}
}
try
{
store.PutBlock(Fx.GenesisBlock);
store.PutBlock(Fx.Block1);
store.PutBlock(Fx.Block2);
store.PutBlock(Fx.Block3);
Guid cid1 = Guid.NewGuid();
int guidLength = cid1.ToByteArray().Length;
store.AppendIndex(cid1, Fx.GenesisBlock.Hash);
store.AppendIndex(cid1, Fx.Block1.Hash);
store.AppendIndex(cid1, Fx.Block2.Hash);
Assert.Single(store.ListChainIds());
Assert.Equal(
new[] { Fx.GenesisBlock.Hash, Fx.Block1.Hash, Fx.Block2.Hash },
store.IterateIndexes(cid1, 0, null));
Guid cid2 = Guid.NewGuid();
store.ForkBlockIndexes(cid1, cid2, Fx.Block1.Hash);
store.AppendIndex(cid2, Fx.Block2.Hash);
store.AppendIndex(cid2, Fx.Block3.Hash);
Assert.Equal(2, store.ListChainIds().Count());
Assert.Equal(
new[] { Fx.GenesisBlock.Hash, Fx.Block1.Hash, Fx.Block2.Hash, Fx.Block3.Hash },
store.IterateIndexes(cid2, 0, null));
Guid cid3 = Guid.NewGuid();
store.ForkBlockIndexes(cid1, cid3, Fx.Block2.Hash);
Assert.Equal(3, store.ListChainIds().Count());
Assert.Equal(
new[] { Fx.GenesisBlock.Hash, Fx.Block1.Hash, Fx.Block2.Hash },
store.IterateIndexes(cid3, 0, null));
Assert.Throws <InvalidOperationException>(() => store.PruneOutdatedChains());
store.PruneOutdatedChains(true);
store.SetCanonicalChainId(cid3);
store.PruneOutdatedChains();
Assert.Single(store.ListChainIds());
Assert.Equal(
new[] { Fx.GenesisBlock.Hash, Fx.Block1.Hash, Fx.Block2.Hash },
store.IterateIndexes(cid3, 0, null));
Assert.Equal(3, store.CountIndex(cid3));
store.Dispose();
store = null;
chainDb = RocksDb.Open(new DbOptions(), Path.Combine(path, "chain"));
Assert.Equal(0, KeysWithChainId(chainDb, cid1));
Assert.Equal(0, KeysWithChainId(chainDb, cid2));
Assert.NotEqual(0, KeysWithChainId(chainDb, cid3));
}
finally
{
store?.Dispose();
chainDb?.Dispose();
Directory.Delete(path, true);
}
}
