84a43f9a3c
a function to dump state at the block, but we haven't decided exactly what it should dump yet.
114 lines
4.4 KiB
C++
114 lines
4.4 KiB
C++
/*
|
|
* Copyright (c) 2015 Cryptonomex, Inc., and contributors.
|
|
*
|
|
* The MIT License
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
* of this software and associated documentation files (the "Software"), to deal
|
|
* in the Software without restriction, including without limitation the rights
|
|
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
* copies of the Software, and to permit persons to whom the Software is
|
|
* furnished to do so, subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be included in
|
|
* all copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
* THE SOFTWARE.
|
|
*/
|
|
#include <graphene/generate_genesis/generate_genesis.hpp>
|
|
|
|
#include <graphene/chain/database.hpp>
|
|
#include <graphene/chain/genesis_state.hpp>
|
|
#include <graphene/chain/witness_object.hpp>
|
|
#include <graphene/time/time.hpp>
|
|
|
|
#include <graphene/utilities/key_conversion.hpp>
|
|
|
|
#include <fc/smart_ref_impl.hpp>
|
|
#include <fc/thread/thread.hpp>
|
|
|
|
#include <iostream>
|
|
|
|
using namespace graphene::generate_genesis_plugin;
|
|
using std::string;
|
|
using std::vector;
|
|
|
|
namespace bpo = boost::program_options;
|
|
|
|
void generate_genesis_plugin::plugin_set_program_options(
|
|
boost::program_options::options_description& command_line_options,
|
|
boost::program_options::options_description& config_file_options)
|
|
{
|
|
command_line_options.add_options()
|
|
("output-genesis-file,o", bpo::value<std::string>()->default_value("genesis.json"), "Genesis file to create")
|
|
("snapshot-block-number", bpo::value<uint32_t>()->default_value(1), "Block number at which to snapshot balances")
|
|
;
|
|
config_file_options.add(command_line_options);
|
|
}
|
|
|
|
std::string generate_genesis_plugin::plugin_name()const
|
|
{
|
|
return "generate_genesis";
|
|
}
|
|
|
|
void generate_genesis_plugin::plugin_initialize(const boost::program_options::variables_map& options)
|
|
{ try {
|
|
ilog("generate genesis plugin: plugin_initialize() begin");
|
|
_options = &options;
|
|
|
|
_genesis_filename = options["output-genesis-file"].as<std::string>();
|
|
_block_to_snapshot = options["snapshot-block-number"].as<uint32_t>();
|
|
database().applied_block.connect([this](const graphene::chain::signed_block& b){ block_applied(b); });
|
|
ilog("generate genesis plugin: plugin_initialize() end");
|
|
} FC_LOG_AND_RETHROW() }
|
|
|
|
void generate_genesis_plugin::plugin_startup()
|
|
{ try {
|
|
ilog("generate genesis plugin: plugin_startup() begin");
|
|
chain::database& d = database();
|
|
if (d.head_block_num() == _block_to_snapshot)
|
|
{
|
|
ilog("generate genesis plugin: already at snapshot block");
|
|
generate_snapshot();
|
|
}
|
|
else if (d.head_block_num() > _block_to_snapshot)
|
|
elog("generate genesis plugin: already passed snapshot block, you must reindex to return to the snapshot state");
|
|
else
|
|
elog("generate genesis plugin: waiting for block ${snapshot_block} to generate snapshot, current head is ${head}",
|
|
("snapshot_block", _block_to_snapshot)("head", d.head_block_num()));
|
|
|
|
ilog("generate genesis plugin: plugin_startup() end");
|
|
} FC_CAPTURE_AND_RETHROW() }
|
|
|
|
void generate_genesis_plugin::block_applied(const graphene::chain::signed_block& b)
|
|
{
|
|
if (b.block_num() == _block_to_snapshot)
|
|
{
|
|
ilog("generate genesis plugin: snapshot block has arrived");
|
|
generate_snapshot();
|
|
}
|
|
}
|
|
|
|
void generate_genesis_plugin::generate_snapshot()
|
|
{
|
|
ilog("generate genesis plugin: generating snapshot now");
|
|
graphene::chain::genesis_state_type new_genesis_state;
|
|
chain::database& d = database();
|
|
auto& account_index = d.get_index_type<graphene::chain::account_index>();
|
|
auto& account_by_id_index = account_index.indices().get<graphene::chain::by_id>();
|
|
for (const graphene::chain::account_object& account_obj : account_by_id_index)
|
|
{
|
|
//new_genesis_state.initial_accounts.emplace_back(genesis_state_type::initial_account_type(account_obj.name, account
|
|
}
|
|
}
|
|
|
|
void generate_genesis_plugin::plugin_shutdown()
|
|
{
|
|
}
|
|
|