Merge pull request #500 from oxarbitrage/elasticsearch-extras

es_objects plugin
This commit is contained in:
oxarbitrage 2018-03-20 17:06:05 -03:00 committed by gladcow
parent 00f14c4729
commit 8d900a5276
9 changed files with 685 additions and 1 deletions

View file

@ -10,3 +10,4 @@ add_subdirectory( generate_genesis )
add_subdirectory( generate_uia_sharedrop_genesis )
add_subdirectory( debug_witness )
add_subdirectory( snapshot )
add_subdirectory( es_objects )

View file

@ -0,0 +1,23 @@
file(GLOB HEADERS "include/graphene/es_objects/*.hpp")
add_library( graphene_es_objects
es_objects.cpp
)
target_link_libraries( graphene_es_objects graphene_chain graphene_app curl )
target_include_directories( graphene_es_objects
PUBLIC "${CMAKE_CURRENT_SOURCE_DIR}/include" )
if(MSVC)
set_source_files_properties(es_objects.cpp PROPERTIES COMPILE_FLAGS "/bigobj" )
endif(MSVC)
install( TARGETS
graphene_es_objects
RUNTIME DESTINATION bin
LIBRARY DESTINATION lib
ARCHIVE DESTINATION lib
)
INSTALL( FILES ${HEADERS} DESTINATION "include/graphene/es_objects" )

View file

@ -0,0 +1,355 @@
/*
* Copyright (c) 2018 oxarbitrage, 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/es_objects/es_objects.hpp>
#include <fc/smart_ref_impl.hpp>
#include <curl/curl.h>
#include <graphene/chain/proposal_object.hpp>
#include <graphene/chain/balance_object.hpp>
#include <graphene/chain/market_object.hpp>
#include <graphene/utilities/elasticsearch.hpp>
namespace graphene { namespace es_objects {
namespace detail
{
class es_objects_plugin_impl
{
public:
es_objects_plugin_impl(es_objects_plugin& _plugin)
: _self( _plugin )
{ curl = curl_easy_init(); }
virtual ~es_objects_plugin_impl();
void updateDatabase( const vector<object_id_type>& ids , bool isNew);
es_objects_plugin& _self;
std::string _es_objects_elasticsearch_url = "http://localhost:9200/";
uint32_t _es_objects_bulk_replay = 5000;
uint32_t _es_objects_bulk_sync = 10;
bool _es_objects_proposals = true;
bool _es_objects_accounts = true;
bool _es_objects_assets = true;
bool _es_objects_balances = true;
bool _es_objects_limit_orders = true;
bool _es_objects_asset_bitasset = true;
bool _es_objects_logs = true;
CURL *curl; // curl handler
vector <std::string> bulk;
vector<std::string> prepare;
map<object_id_type, std::string> bitassets;
//uint32_t bitasset_seq;
private:
void PrepareProposal(const proposal_object* proposal_object, const fc::time_point_sec block_time, uint32_t block_number);
void PrepareAccount(const account_object* account_object, const fc::time_point_sec block_time, uint32_t block_number);
void PrepareAsset(const asset_object* asset_object, const fc::time_point_sec block_time, uint32_t block_number);
void PrepareBalance(const balance_object* balance_object, const fc::time_point_sec block_time, uint32_t block_number);
void PrepareLimit(const limit_order_object* limit_object, const fc::time_point_sec block_time, uint32_t block_number);
void PrepareBitAsset(const asset_bitasset_data_object* bitasset_object, const fc::time_point_sec block_time, uint32_t block_number);
};
void es_objects_plugin_impl::updateDatabase( const vector<object_id_type>& ids , bool isNew)
{
graphene::chain::database &db = _self.database();
const fc::time_point_sec block_time = db.head_block_time();
const uint32_t block_number = db.head_block_num();
// check if we are in replay or in sync and change number of bulk documents accordingly
uint32_t limit_documents = 0;
if((fc::time_point::now() - block_time) < fc::seconds(30))
limit_documents = _es_objects_bulk_sync;
else
limit_documents = _es_objects_bulk_replay;
if (curl && bulk.size() >= limit_documents) { // we are in bulk time, ready to add data to elasticsearech
if(!graphene::utilities::SendBulk(curl, bulk, _es_objects_elasticsearch_url, _es_objects_logs, "objects_logs"))
elog("Error sending data to database");
bulk.clear();
}
for(auto const& value: ids) {
if(value.is<proposal_object>() && _es_objects_proposals) {
auto obj = db.find_object(value);
auto p = static_cast<const proposal_object*>(obj);
if(p != nullptr)
PrepareProposal(p, block_time, block_number);
}
else if(value.is<account_object>() && _es_objects_accounts) {
auto obj = db.find_object(value);
auto a = static_cast<const account_object*>(obj);
if(a != nullptr)
PrepareAccount(a, block_time, block_number);
}
else if(value.is<asset_object>() && _es_objects_assets) {
auto obj = db.find_object(value);
auto a = static_cast<const asset_object*>(obj);
if(a != nullptr)
PrepareAsset(a, block_time, block_number);
}
else if(value.is<balance_object>() && _es_objects_balances) {
auto obj = db.find_object(value);
auto b = static_cast<const balance_object*>(obj);
if(b != nullptr)
PrepareBalance(b, block_time, block_number);
}
else if(value.is<limit_order_object>() && _es_objects_limit_orders) {
auto obj = db.find_object(value);
auto l = static_cast<const limit_order_object*>(obj);
if(l != nullptr)
PrepareLimit(l, block_time, block_number);
}
else if(value.is<asset_bitasset_data_object>() && _es_objects_asset_bitasset) {
auto obj = db.find_object(value);
auto ba = static_cast<const asset_bitasset_data_object*>(obj);
if(ba != nullptr)
PrepareBitAsset(ba, block_time, block_number);
}
}
}
void es_objects_plugin_impl::PrepareProposal(const proposal_object* proposal_object, const fc::time_point_sec block_time, uint32_t block_number)
{
proposal_struct prop;
prop.object_id = proposal_object->id;
prop.block_time = block_time;
prop.block_number = block_number;
prop.expiration_time = proposal_object->expiration_time;
prop.review_period_time = proposal_object->review_period_time;
prop.proposed_transaction = fc::json::to_string(proposal_object->proposed_transaction);
prop.required_owner_approvals = fc::json::to_string(proposal_object->required_owner_approvals);
prop.available_owner_approvals = fc::json::to_string(proposal_object->available_owner_approvals);
prop.required_active_approvals = fc::json::to_string(proposal_object->required_active_approvals);
prop.available_key_approvals = fc::json::to_string(proposal_object->available_key_approvals);
prop.proposer = proposal_object->proposer;
std::string data = fc::json::to_string(prop);
prepare = graphene::utilities::createBulk("bitshares-proposal", data, "", 1);
bulk.insert(bulk.end(), prepare.begin(), prepare.end());
prepare.clear();
}
void es_objects_plugin_impl::PrepareAccount(const account_object* account_object, const fc::time_point_sec block_time, uint32_t block_number)
{
account_struct acct;
acct.object_id = account_object->id;
acct.block_time = block_time;
acct.block_number = block_number;
acct.membership_expiration_date = account_object->membership_expiration_date;
acct.registrar = account_object->registrar;
acct.referrer = account_object->referrer;
acct.lifetime_referrer = account_object->lifetime_referrer;
acct.network_fee_percentage = account_object->network_fee_percentage;
acct.lifetime_referrer_fee_percentage = account_object->lifetime_referrer_fee_percentage;
acct.referrer_rewards_percentage = account_object->referrer_rewards_percentage;
acct.name = account_object->name;
acct.owner_account_auths = fc::json::to_string(account_object->owner.account_auths);
acct.owner_key_auths = fc::json::to_string(account_object->owner.key_auths);
acct.owner_address_auths = fc::json::to_string(account_object->owner.address_auths);
acct.active_account_auths = fc::json::to_string(account_object->active.account_auths);
acct.active_key_auths = fc::json::to_string(account_object->active.key_auths);
acct.active_address_auths = fc::json::to_string(account_object->active.address_auths);
acct.voting_account = account_object->options.voting_account;
std::string data = fc::json::to_string(acct);
prepare = graphene::utilities::createBulk("bitshares-account", data, "", 1);
bulk.insert(bulk.end(), prepare.begin(), prepare.end());
prepare.clear();
}
void es_objects_plugin_impl::PrepareAsset(const asset_object* asset_object, const fc::time_point_sec block_time, uint32_t block_number)
{
asset_struct _asset;
_asset.object_id = asset_object->id;
_asset.block_time = block_time;
_asset.block_number = block_number;
_asset.symbol = asset_object->symbol;
_asset.issuer = asset_object->issuer;
_asset.is_market_issued = asset_object->is_market_issued();
_asset.dynamic_asset_data_id = asset_object->dynamic_asset_data_id;
_asset.bitasset_data_id = asset_object->bitasset_data_id;
std::string data = fc::json::to_string(_asset);
prepare = graphene::utilities::createBulk("bitshares-asset", data, fc::json::to_string(asset_object->id), 0);
bulk.insert(bulk.end(), prepare.begin(), prepare.end());
prepare.clear();
}
void es_objects_plugin_impl::PrepareBalance(const balance_object* balance_object, const fc::time_point_sec block_time, uint32_t block_number)
{
balance_struct balance;
balance.object_id = balance_object->id;
balance.block_time = block_time;
balance.block_number = block_number;balance.owner = balance_object->owner;
balance.asset_id = balance_object->balance.asset_id;
balance.amount = balance_object->balance.amount;
std::string data = fc::json::to_string(balance);
prepare = graphene::utilities::createBulk("bitshares-balance", data, "", 1);
bulk.insert(bulk.end(), prepare.begin(), prepare.end());
prepare.clear();
}
void es_objects_plugin_impl::PrepareLimit(const limit_order_object* limit_object, const fc::time_point_sec block_time, uint32_t block_number)
{
limit_order_struct limit;
limit.object_id = limit_object->id;
limit.block_time = block_time;
limit.block_number = block_number;
limit.expiration = limit_object->expiration;
limit.seller = limit_object->seller;
limit.for_sale = limit_object->for_sale;
limit.sell_price = limit_object->sell_price;
limit.deferred_fee = limit_object->deferred_fee;
std::string data = fc::json::to_string(limit);
prepare = graphene::utilities::createBulk("bitshares-limitorder", data, "", 1);
bulk.insert(bulk.end(), prepare.begin(), prepare.end());
prepare.clear();
}
void es_objects_plugin_impl::PrepareBitAsset(const asset_bitasset_data_object* bitasset_object, const fc::time_point_sec block_time, uint32_t block_number)
{
if(!bitasset_object->is_prediction_market) {
auto object_id = bitasset_object->id;
auto it = bitassets.find(object_id);
if(it == bitassets.end())
bitassets[object_id] = fc::json::to_string(bitasset_object->current_feed);
else {
if(it->second == fc::json::to_string(bitasset_object->current_feed)) return;
else bitassets[object_id] = fc::json::to_string(bitasset_object->current_feed);
}
bitasset_struct bitasset;
bitasset.object_id = bitasset_object->id;
bitasset.block_time = block_time;
bitasset.block_number = block_number;
bitasset.current_feed = fc::json::to_string(bitasset_object->current_feed);
bitasset.current_feed_publication_time = bitasset_object->current_feed_publication_time;
std::string data = fc::json::to_string(bitasset);
prepare = graphene::utilities::createBulk("bitshares-bitasset", data, "", 1);
bulk.insert(bulk.end(), prepare.begin(), prepare.end());
prepare.clear();
}
}
es_objects_plugin_impl::~es_objects_plugin_impl()
{
return;
}
} // end namespace detail
es_objects_plugin::es_objects_plugin() :
my( new detail::es_objects_plugin_impl(*this) )
{
}
es_objects_plugin::~es_objects_plugin()
{
}
std::string es_objects_plugin::plugin_name()const
{
return "es_objects";
}
std::string es_objects_plugin::plugin_description()const
{
return "Stores blockchain objects in ES database. Experimental.";
}
void es_objects_plugin::plugin_set_program_options(
boost::program_options::options_description& cli,
boost::program_options::options_description& cfg
)
{
cli.add_options()
("es-objects-elasticsearch-url", boost::program_options::value<std::string>(), "Elasticsearch node url")
("es-objects-logs", boost::program_options::value<bool>(), "Log bulk events to database")
("es-objects-bulk-replay", boost::program_options::value<uint32_t>(), "Number of bulk documents to index on replay(5000)")
("es-objects-bulk-sync", boost::program_options::value<uint32_t>(), "Number of bulk documents to index on a syncronied chain(10)")
("es-objects-proposals", boost::program_options::value<bool>(), "Store proposal objects")
("es-objects-accounts", boost::program_options::value<bool>(), "Store account objects")
("es-objects-assets", boost::program_options::value<bool>(), "Store asset objects")
("es-objects-balances", boost::program_options::value<bool>(), "Store balances objects")
("es-objects-limit-orders", boost::program_options::value<bool>(), "Store limit order objects")
("es-objects-asset-bitasset", boost::program_options::value<bool>(), "Store feed data")
;
cfg.add(cli);
}
void es_objects_plugin::plugin_initialize(const boost::program_options::variables_map& options)
{
database().new_objects.connect([&]( const vector<object_id_type>& ids, const flat_set<account_id_type>& impacted_accounts ){ my->updateDatabase(ids, 1); });
database().changed_objects.connect([&]( const vector<object_id_type>& ids, const flat_set<account_id_type>& impacted_accounts ){ my->updateDatabase(ids, 0); });
if (options.count("es-objects-elasticsearch-url")) {
my->_es_objects_elasticsearch_url = options["es-objects-elasticsearch-url"].as<std::string>();
}
if (options.count("es-objects-logs")) {
my->_es_objects_logs = options["es-objects-logs"].as<bool>();
}
if (options.count("es-objects-bulk-replay")) {
my->_es_objects_bulk_replay = options["es-objects-bulk-replay"].as<uint32_t>();
}
if (options.count("es-objects-bulk-sync")) {
my->_es_objects_bulk_sync = options["es-objects-bulk-sync"].as<uint32_t>();
}
if (options.count("es-objects-proposals")) {
my->_es_objects_proposals = options["es-objects-proposals"].as<bool>();
}
if (options.count("es-objects-accounts")) {
my->_es_objects_accounts = options["es-objects-accounts"].as<bool>();
}
if (options.count("es-objects-assets")) {
my->_es_objects_assets = options["es-objects-assets"].as<bool>();
}
if (options.count("es-objects-balances")) {
my->_es_objects_balances = options["es-objects-balances"].as<bool>();
}
if (options.count("es-objects-limit-orders")) {
my->_es_objects_limit_orders = options["es-objects-limit-orders"].as<bool>();
}
if (options.count("es-objects-asset-bitasset")) {
my->_es_objects_asset_bitasset = options["es-objects-asset-bitasset"].as<bool>();
}
}
void es_objects_plugin::plugin_startup()
{
ilog("elasticsearch objects: plugin_startup() begin");
}
} }

View file

@ -0,0 +1,137 @@
/*
* Copyright (c) 2018 oxarbitrage, 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.
*/
#pragma once
#include <graphene/app/plugin.hpp>
#include <graphene/chain/database.hpp>
namespace graphene { namespace es_objects {
using namespace chain;
namespace detail
{
class es_objects_plugin_impl;
}
class es_objects_plugin : public graphene::app::plugin
{
public:
es_objects_plugin();
virtual ~es_objects_plugin();
std::string plugin_name()const override;
std::string plugin_description()const override;
virtual void plugin_set_program_options(
boost::program_options::options_description& cli,
boost::program_options::options_description& cfg) override;
virtual void plugin_initialize(const boost::program_options::variables_map& options) override;
virtual void plugin_startup() override;
friend class detail::es_objects_plugin_impl;
std::unique_ptr<detail::es_objects_plugin_impl> my;
};
struct proposal_struct {
object_id_type object_id;
fc::time_point_sec block_time;
uint32_t block_number;
time_point_sec expiration_time;
optional<time_point_sec> review_period_time;
string proposed_transaction;
string required_active_approvals;
string available_active_approvals;
string required_owner_approvals;
string available_owner_approvals;
string available_key_approvals;
account_id_type proposer;
};
struct account_struct {
object_id_type object_id;
fc::time_point_sec block_time;
uint32_t block_number;
time_point_sec membership_expiration_date;
account_id_type registrar;
account_id_type referrer;
account_id_type lifetime_referrer;
uint16_t network_fee_percentage;
uint16_t lifetime_referrer_fee_percentage;
uint16_t referrer_rewards_percentage;
string name;
string owner_account_auths;
string owner_key_auths;
string owner_address_auths;
string active_account_auths;
string active_key_auths;
string active_address_auths;
account_id_type voting_account;
};
struct asset_struct {
object_id_type object_id;
fc::time_point_sec block_time;
uint32_t block_number;
string symbol;
account_id_type issuer;
bool is_market_issued;
asset_dynamic_data_id_type dynamic_asset_data_id;
optional<asset_bitasset_data_id_type> bitasset_data_id;
};
struct balance_struct {
object_id_type object_id;
fc::time_point_sec block_time;
uint32_t block_number;
address owner;
asset_id_type asset_id;
share_type amount;
};
struct limit_order_struct {
object_id_type object_id;
fc::time_point_sec block_time;
uint32_t block_number;
time_point_sec expiration;
account_id_type seller;
share_type for_sale;
price sell_price;
share_type deferred_fee;
};
struct bitasset_struct {
object_id_type object_id;
fc::time_point_sec block_time;
uint32_t block_number;
string current_feed;
time_point_sec current_feed_publication_time;
time_point_sec feed_expiration_time;
};
} } //graphene::es_objects
FC_REFLECT( graphene::es_objects::proposal_struct, (object_id)(block_time)(block_number)(expiration_time)(review_period_time)(proposed_transaction)(required_active_approvals)(available_active_approvals)(required_owner_approvals)(available_owner_approvals)(available_key_approvals)(proposer) )
FC_REFLECT( graphene::es_objects::account_struct, (object_id)(block_time)(block_number)(membership_expiration_date)(registrar)(referrer)(lifetime_referrer)(network_fee_percentage)(lifetime_referrer_fee_percentage)(referrer_rewards_percentage)(name)(owner_account_auths)(owner_key_auths)(owner_address_auths)(active_account_auths)(active_key_auths)(active_address_auths)(voting_account) )
FC_REFLECT( graphene::es_objects::asset_struct, (object_id)(block_time)(block_number)(symbol)(issuer)(is_market_issued)(dynamic_asset_data_id)(bitasset_data_id) )
FC_REFLECT( graphene::es_objects::balance_struct, (object_id)(block_time)(block_number)(block_time)(owner)(asset_id)(amount) )
FC_REFLECT( graphene::es_objects::limit_order_struct, (object_id)(block_time)(block_number)(expiration)(seller)(for_sale)(sell_price)(deferred_fee) )
FC_REFLECT( graphene::es_objects::bitasset_struct, (object_id)(block_time)(block_number)(current_feed)(current_feed_publication_time) )

View file

@ -14,6 +14,7 @@ set(sources
string_escape.cpp
tempdir.cpp
words.cpp
elasticsearch.cpp
${HEADERS})
configure_file("${CMAKE_CURRENT_SOURCE_DIR}/git_revision.cpp.in" "${CMAKE_CURRENT_BINARY_DIR}/git_revision.cpp" @ONLY)

View file

@ -0,0 +1,123 @@
/*
* Copyright (c) 2018 oxarbitrage, 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/utilities/elasticsearch.hpp>
#include <boost/algorithm/string/join.hpp>
#include <fc/log/logger.hpp>
namespace graphene { namespace utilities {
bool SendBulk(CURL *curl, std::vector<std::string>& bulk, std::string elasticsearch_url, bool do_logs, std::string logs_index)
{
// curl buffers to read
std::string readBuffer;
std::string readBuffer_logs;
std::string bulking = "";
bulking = boost::algorithm::join(bulk, "\n");
bulking = bulking + "\n";
bulk.clear();
struct curl_slist *headers = NULL;
headers = curl_slist_append(headers, "Content-Type: application/json");
std::string url = elasticsearch_url + "_bulk";
curl_easy_setopt(curl, CURLOPT_URL, url.c_str());
curl_easy_setopt(curl, CURLOPT_POST, true);
curl_easy_setopt(curl, CURLOPT_HTTPHEADER, headers);
curl_easy_setopt(curl, CURLOPT_POSTFIELDS, bulking.c_str());
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, WriteCallback);
curl_easy_setopt(curl, CURLOPT_WRITEDATA, (void *)&readBuffer);
curl_easy_setopt(curl, CURLOPT_USERAGENT, "libcrp/0.1");
//curl_easy_setopt(curl, CURLOPT_VERBOSE, true);
curl_easy_perform(curl);
long http_code = 0;
curl_easy_getinfo (curl, CURLINFO_RESPONSE_CODE, &http_code);
if(http_code == 200) {
// all good, do nothing
}
else if(http_code == 413) {
elog("413 error: Can be low space disk");
return 0;
}
else {
elog(http_code + "error: Unknown error");
return 0;
}
if(do_logs) {
auto logs = readBuffer;
// do logs
std::string url_logs = elasticsearch_url + logs_index + "/data/";
curl_easy_setopt(curl, CURLOPT_URL, url_logs.c_str());
curl_easy_setopt(curl, CURLOPT_POST, true);
curl_easy_setopt(curl, CURLOPT_HTTPHEADER, headers);
curl_easy_setopt(curl, CURLOPT_POSTFIELDS, logs.c_str());
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, WriteCallback);
curl_easy_setopt(curl, CURLOPT_WRITEDATA, (void *) &readBuffer_logs);
curl_easy_setopt(curl, CURLOPT_USERAGENT, "libcrp/0.1");
curl_easy_perform(curl);
http_code = 0;
curl_easy_getinfo (curl, CURLINFO_RESPONSE_CODE, &http_code);
if(http_code == 200) {
// all good, do nothing
return 1;
}
else if(http_code == 201) {
// 201 is ok
return 1;
}
else if(http_code == 409) {
// 409 for record already exist is ok
return 1;
}
else if(http_code == 413) {
elog("413 error: Can be low space disk");
return 0;
}
else {
elog(http_code + "error: Unknown error");
return 0;
}
}
return 0;
}
std::vector<std::string> createBulk(std::string index_name, std::string data, std::string id, bool onlycreate)
{
std::vector<std::string> bulk;
std::string create_string = "";
if(!onlycreate)
create_string = ",\"_id\" : "+id;
bulk.push_back("{ \"index\" : { \"_index\" : \""+index_name+"\", \"_type\" : \"data\" "+create_string+" } }");
bulk.push_back(data);
return bulk;
}
} } // end namespace graphene::utilities

View file

@ -0,0 +1,42 @@
/*
* Copyright (c) 2018 oxarbitrage, 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.
*/
#pragma once
#include <cstddef>
#include <string>
#include <vector>
#include <curl/curl.h>
static size_t WriteCallback(void *contents, size_t size, size_t nmemb, void *userp)
{
((std::string*)userp)->append((char*)contents, size * nmemb);
return size * nmemb;
}
namespace graphene { namespace utilities {
bool SendBulk(CURL *curl, std::vector <std::string>& bulk, std::string elasticsearch_url, bool do_logs, std::string logs_index);
std::vector<std::string> createBulk(std::string type, std::string data, std::string id, bool onlycreate);
} } // end namespace graphene::utilities

View file

@ -11,7 +11,7 @@ endif()
# We have to link against graphene_debug_witness because deficiency in our API infrastructure doesn't allow plugins to be fully abstracted #246
target_link_libraries( witness_node
PRIVATE graphene_app graphene_account_history graphene_affiliate_stats graphene_elasticsearch graphene_market_history graphene_witness graphene_chain graphene_debug_witness graphene_bookie graphene_egenesis_full fc ${CMAKE_DL_LIBS} ${PLATFORM_SPECIFIC_LIBS} )
PRIVATE graphene_app graphene_account_history graphene_affiliate_stats graphene_elasticsearch graphene_market_history graphene_witness graphene_chain graphene_debug_witness graphene_bookie graphene_egenesis_full graphene_es_objects fc ${CMAKE_DL_LIBS} ${PLATFORM_SPECIFIC_LIBS} )
# also add dependencies to graphene_generate_genesis graphene_generate_uia_sharedrop_genesis if you want those plugins
install( TARGETS

View file

@ -29,6 +29,7 @@
#include <graphene/account_history/account_history_plugin.hpp>
#include <graphene/accounts_list/accounts_list_plugin.hpp>
#include <graphene/elasticsearch/elasticsearch_plugin.hpp>
#include <graphene/es_objects/es_objects.hpp>
#include <graphene/market_history/market_history_plugin.hpp>
//#include <graphene/generate_genesis/generate_genesis_plugin.hpp>
//#include <graphene/generate_uia_sharedrop_genesis/generate_uia_sharedrop_genesis.hpp>
@ -76,6 +77,7 @@ int main(int argc, char** argv) {
auto debug_witness_plug = node->register_plugin<debug_witness_plugin::debug_witness_plugin>();
auto history_plug = node->register_plugin<account_history::account_history_plugin>();
auto elasticsearch_plug = node->register_plugin<elasticsearch::elasticsearch_plugin>();
auto es_objects_plug = node->register_plugin<es_objects::es_objects_plugin>();
auto market_history_plug = node->register_plugin<market_history::market_history_plugin>();
//auto generate_genesis_plug = node->register_plugin<generate_genesis_plugin::generate_genesis_plugin>();
//auto generate_uia_sharedrop_genesis_plug = node->register_plugin<generate_uia_sharedrop_genesis::generate_uia_sharedrop_genesis_plugin>();