From a2ba58bcc8c7cfa08c0aa3db94d1d845e5468f15 Mon Sep 17 00:00:00 2001
From: Nico Fricke <NiFNi@users.noreply.github.com>
Date: Tue, 15 May 2018 01:42:04 +0200
Subject: [PATCH] optional use_peers param to generate_work rpc (#862)

---
 rai/node/rpc.cpp | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/rai/node/rpc.cpp b/rai/node/rpc.cpp
index ef092567..4675c797 100644
--- a/rai/node/rpc.cpp
+++ b/rai/node/rpc.cpp
@@ -4228,12 +4228,13 @@ void rai::rpc_handler::work_generate ()
 	if (rpc.config.enable_control)
 	{
 		std::string hash_text (request.get<std::string> ("hash"));
+		bool use_peers (request.get_optional<bool> ("use_peers") == true);
 		rai::block_hash hash;
 		auto error (hash.decode_hex (hash_text));
 		if (!error)
 		{
 			auto rpc_l (shared_from_this ());
-			node.work.generate (hash, [rpc_l](boost::optional<uint64_t> const & work_a) {
+			auto callback = [rpc_l](boost::optional<uint64_t> const & work_a) {
 				if (work_a)
 				{
 					boost::property_tree::ptree response_l;
@@ -4244,7 +4245,15 @@ void rai::rpc_handler::work_generate ()
 				{
 					error_response (rpc_l->response, "Cancelled");
 				}
-			});
+			};
+			if (!use_peers)
+			{
+				node.work.generate (hash, callback);
+			}
+			else
+			{
+				node.generate_work (hash, callback);
+			}
 		}
 		else
 		{