Minecon724/dncurrency
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 1 Wiki Activity Actions

Adding account_block_count RPC.

Browse source
This commit is contained in:
clemahieu 2017-04-11 10:18:32 -05:00
commit 6dc7d23676
3 changed files with 49 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

18
rai/core_test/rpc.cpp
View file

@ -97,6 +97,24 @@ TEST (rpc, account_balance)
ASSERT_EQ ("0", pending_text);
}
TEST (rpc, account_block_count)
{
rai::system system (24000, 1);
rai::rpc rpc (system.service, *system.nodes [0], rai::rpc_config (true));
rpc.start ();
boost::property_tree::ptree request;
request.put ("action", "account_block_count");
request.put ("account", rai::test_genesis_key.pub.to_account ());
test_response response (request, rpc, system.service);
while (response.status == 0)
{
system.poll ();
}
ASSERT_EQ (200, response.status);
std::string block_count_text (response.json.get <std::string> ("block_count"));
ASSERT_EQ ("1", block_count_text);
}
TEST (rpc, account_create)
{
rai::system system (24000, 1);

30
rai/node/rpc.cpp
View file

@ -180,6 +180,32 @@ void rai::rpc_handler::account_balance ()
}
}
void rai::rpc_handler::account_block_count ()
{
std::string account_text (request.get <std::string> ("account"));
rai::uint256_union account;
auto error (account.decode_account (account_text));
if (!error)
{
rai::transaction transaction (node.store.environment, nullptr, false);
rai::account_info info;
if (!node.store.account_get (transaction, account, info))
{
boost::property_tree::ptree response_l;
response_l.put ("block_count", std::to_string (info.block_count));
response (response_l);
}
else
{
error_response (response, "Account not found");
}
}
else
{
error_response (response, "Bad account number");
}
}
void rai::rpc_handler::account_create ()
{
if (rpc.config.enable_control)
@ -1794,6 +1820,10 @@ void rai::rpc_handler::process_request ()
{
account_balance ();
}
else if (action == "account_block_count")
{
account_block_count ();
}
else if (action == "account_create")
{
account_create ();

1
rai/node/rpc.hpp
View file

@ -89,6 +89,7 @@ public:
rpc_handler (rai::node &, rai::rpc &, std::string const &, std::function <void (boost::property_tree::ptree const &)> const &);
void process_request ();
void account_balance ();
void account_block_count ();
void account_create ();
void account_get ();
void account_key ();