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);
}
}
) GetBlockChain(
ILogger logger,
string storePath,
Guid?chainId = null,
IKeyValueStore stateKeyValueStore = null
)
{
var policySource = new BlockPolicySource(logger);
IBlockPolicy <NCAction> policy = policySource.GetPolicy();
IStagePolicy <NCAction> stagePolicy = new VolatileStagePolicy <NCAction>();
IStore store = new RocksDBStore(storePath);
if (stateKeyValueStore is null)
{
stateKeyValueStore = new RocksDBKeyValueStore(Path.Combine(storePath, "states"));
}
IStateStore stateStore = new TrieStateStore(stateKeyValueStore);
Guid chainIdValue
= chainId ??
store.GetCanonicalChainId() ??
throw new CommandExitedException(
"No canonical chain ID. Available chain IDs:\n " +
string.Join <Guid>("\n ", store.ListChainIds()),
1);
BlockHash genesisBlockHash;
try
{
genesisBlockHash = store.IterateIndexes(chainIdValue).First();
}
catch (InvalidOperationException)
{
throw new CommandExitedException(
$"The chain {chainIdValue} seems empty; try with another chain ID:\n " +
string.Join <Guid>("\n ", store.ListChainIds()),
1
);
}
Block <NCAction> genesis = store.GetBlock <NCAction>(
policy.GetHashAlgorithm,
genesisBlockHash
);
BlockChain <NCAction> chain = new BlockChain <NCAction>(
policy,
stagePolicy,
store,
stateStore,
genesis
);
return(chain, store, stateKeyValueStore, stateStore);
}
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);
}
}