mirror of
https://github.com/GOSTSec/poolserver
synced 2025-01-14 08:47:53 +00:00
Enable Multithreading
This commit is contained in:
Intel
parent
0bf4950a19
commit
e9c5b31f13
@ -33,7 +33,7 @@ bool InitConfig(int argc, char *argv[])
|
||||
|
||||
// Server
|
||||
descServer.add_options()
|
||||
("MinDiffTime", boost::program_options::value<uint32_t>()->default_value(100), "Minimum server diff time")
|
||||
("ServerThreads", boost::program_options::value<uint32_t>()->default_value(2), "How many threads to use")
|
||||
("MiningAddress", boost::program_options::value<std::string>()->default_value("n1w8gkPXdNGb6edm4vujBn71A72eQFCNCw"), "Address to send coins to")
|
||||
("BitcoinRPC", boost::program_options::value<std::vector<std::string> >()->multitoken(), "Bitcoin RPC login credentials")
|
||||
;
|
||||
@ -106,8 +106,7 @@ int main(int argc, char *argv[])
|
||||
sLog.OpenLogFile(sConfig.Get<std::string>("LogFilePath"));
|
||||
sLog.Info(LOG_GENERAL, "LogFile Started: %s", sLog.logFileLoc.c_str());
|
||||
|
||||
boost::asio::io_service io;
|
||||
Server* server = new Server(io);
|
||||
Server* server = new Server();
|
||||
int exitcode = server->Run();
|
||||
delete server;
|
||||
|
||||
|
||||
@ -15,82 +15,31 @@
|
||||
#include <iostream>
|
||||
#include <algorithm>
|
||||
|
||||
Server::Server(asio::io_service& io) : serverLoops(0), io_service(io)
|
||||
Server::Server()
|
||||
{
|
||||
_io_service = boost::shared_ptr<asio::io_service>(new asio::io_service());
|
||||
}
|
||||
|
||||
Server::~Server()
|
||||
{
|
||||
//delete stratumServer;
|
||||
delete _stratumServer;
|
||||
|
||||
}
|
||||
|
||||
int Server::Run()
|
||||
{
|
||||
sLog.Info(LOG_SERVER, "Server is starting...");
|
||||
|
||||
// Connect to database
|
||||
InitDatabase();
|
||||
|
||||
/*std::vector<Share> shares;
|
||||
|
||||
sLog.Info(LOG_SERVER, "Loading shares...");
|
||||
// Initialize share storage
|
||||
DataMgr::Initialize(*_io_service);
|
||||
|
||||
MySQL::QueryResult result = sDatabase.Query("SELECT MIN(`id`), MAX(`id`) FROM `shares`");
|
||||
MySQL::Field* fields = result->FetchRow();
|
||||
uint32 min = fields[0].Get<uint32>();
|
||||
uint32 max = fields[1].Get<uint32>();
|
||||
sLog.Info(LOG_SERVER, "Min: %u Max: %u", min, max);
|
||||
|
||||
MySQL::PreparedStatement* stmt = sDatabase.GetPreparedStatement(STMT_QUERY_SHARES);
|
||||
for (uint32 i = min; i <= max; i += 500000)
|
||||
{
|
||||
stmt->SetUInt32(0, i);
|
||||
MySQL::QueryResult result2 = sDatabase.Query(stmt);
|
||||
while (MySQL::Field* fields2 = result2->FetchRow()) {
|
||||
shares.push_back(Share(fields2[0].Get<uint64>(), fields2[1].Get<uint32>(), 2981, fields2[2].Get<uint32>()));
|
||||
}
|
||||
sLog.Info(LOG_SERVER, "Shares: %u", shares.size());
|
||||
}
|
||||
|
||||
sLog.Info(LOG_SERVER, "Loaded %u shares", shares.size());*/
|
||||
|
||||
/*std::vector<byte> test = Util::ASCIIToBin("4a5e1e4baab89f3a32518a88c31bc87f618f76673e2cc77ab2127b7afdeda33b");
|
||||
sLog.Info(LOG_SERVER, "Hash: %s", Util::BinToASCII(Crypto::SHA256D(test)).c_str());
|
||||
sLog.Info(LOG_SERVER, "RevHash: %s", Util::BinToASCII(Crypto::SHA256D(Util::Reverse(test))).c_str());
|
||||
|
||||
Bitcoin::Block block;
|
||||
|
||||
ByteBuffer buf(Util::ASCIIToBin("01000000010000000000000000000000000000000000000000000000000000000000000000ffffffff4d04ffff001d0104455468652054696d65732030332f4a616e2f32303039204368616e63656c6c6f72206f6e206272696e6b206f66207365636f6e64206261696c6f757420666f722062616e6b73ffffffff0100f2052a01000000434104678afdb0fe5548271967f1a67130b7105cd6a828e03909a67962e0ea1f61deb649f6bc3f4cef38c4f35504e51ec112de5c384df7ba0b8d578a4c702b6bf11d5fac00000000"));
|
||||
Bitcoin::Transaction trans;
|
||||
buf >> trans;
|
||||
sLog.Info(LOG_SERVER, "Version: %u", trans.version);
|
||||
sLog.Info(LOG_SERVER, "Inputs: %u", trans.in.size());
|
||||
sLog.Info(LOG_SERVER, "PrevOut: %s", Util::BinToASCII(trans.in[0].prevout.hash).c_str());
|
||||
sLog.Info(LOG_SERVER, "PrevOutn: %u", trans.in[0].prevout.n);
|
||||
sLog.Info(LOG_SERVER, "ScriptSig: %s", Util::BinToASCII(trans.in[0].script.script).c_str());
|
||||
sLog.Info(LOG_SERVER, "Inn: %u", trans.in[0].n);
|
||||
sLog.Info(LOG_SERVER, "Outputs: %u", trans.out.size());
|
||||
sLog.Info(LOG_SERVER, "Value: %i", trans.out[0].value);
|
||||
sLog.Info(LOG_SERVER, "PubSig: %s", Util::BinToASCII(trans.out[0].scriptPubKey.script).c_str());
|
||||
sLog.Info(LOG_SERVER, "LockTime: %u", trans.lockTime);
|
||||
block.tx.resize(1);
|
||||
block.tx[0] = trans;
|
||||
block.BuildMerkleTree();
|
||||
sLog.Info(LOG_SERVER, "Hash: %s", Util::BinToASCII(block.merkleRootHash).c_str());*/
|
||||
|
||||
/*ByteBuffer buf(Util::ASCIIToBin("01000000010c432f4fb3e871a8bda638350b3d5c698cf431db8d6031b53e3fb5159e59d4a90000000000ffffffff0100f2052a010000001976a9143744841e13b90b4aca16fe793a7f88da3a23cc7188ac00000000"));
|
||||
|
||||
Bitcoin::Transaction trans;
|
||||
buf >> trans;
|
||||
|
||||
ByteBuffer buf2;
|
||||
buf2 << trans;
|
||||
|
||||
sLog.Info(LOG_SERVER, "Trans: %s", Util::BinToASCII(buf2.vec).c_str());*/
|
||||
|
||||
|
||||
DataMgr::Initialize(io_service);
|
||||
NetworkMgr::Initialize(io_service);
|
||||
// Initialize bitcoin daemons
|
||||
NetworkMgr::Initialize(*_io_service);
|
||||
|
||||
// Connect to bitcoin rpc
|
||||
std::vector<std::string> btcrpc = sConfig.Get<std::vector<std::string> >("BitcoinRPC");
|
||||
for (int i = 0; i < btcrpc.size(); ++i) {
|
||||
std::vector<std::string> params = Util::Explode(btcrpc[i], ";");
|
||||
@ -107,87 +56,24 @@ int Server::Run()
|
||||
NetworkMgr::Instance()->Connect(coninfo);
|
||||
}
|
||||
|
||||
Stratum::Server srv(io_service);
|
||||
// Init stratum
|
||||
_stratumServer = new Stratum::Server(*_io_service);
|
||||
|
||||
// Start stratum server
|
||||
tcp::endpoint endpoint(tcp::v4(), sConfig.Get<uint16>("StratumPort"));
|
||||
srv.Start(endpoint);
|
||||
_stratumServer->Start(endpoint);
|
||||
|
||||
io_service.run();
|
||||
for (uint32 i = 0; i < sConfig.Get<uint32>("ServerThreads"); ++i) {
|
||||
_workerThreads.create_thread(boost::bind(&Server::WorkerThread, this, _io_service));
|
||||
}
|
||||
|
||||
|
||||
//sDatabase.Execute("INSERT INTO `test_table` VALUES ('999', 'sync', '1.1')");
|
||||
//sDatabase.ExecuteAsync("INSERT INTO `test_table` VALUES ('999', 'sync', '1.1')");
|
||||
|
||||
/*MySQL::PreparedStatement* stmt = sDatabase.GetPreparedStatement(STMT_INSERT_SHIT);
|
||||
stmt->SetUInt32(0, 10);
|
||||
stmt->SetString(1, "hello");
|
||||
stmt->SetFloat(2, 5.987);
|
||||
sDatabase.ExecuteAsync(stmt);*/
|
||||
|
||||
//MySQL::PreparedStatement* stmt = sDatabase.GetPreparedStatement(STMT_QUERY_TEST_TABLE);
|
||||
//MySQL::QueryResult result = sDatabase.Query(stmt);
|
||||
|
||||
|
||||
//sDatabase.QueryAsync("SELECT * FROM `test_table`", &AsyncQueryCallback);
|
||||
//MySQL::QueryResult result = sDatabase.Query("SELECT * FROM `test_table`");
|
||||
/*if (result) {
|
||||
sLog.Info(LOG_SERVER, "Metadata: F: %u R: %u", result->GetFieldCount(), result->GetRowCount());
|
||||
while (MySQL::Field* fields = result->FetchRow()) {
|
||||
sLog.Info(LOG_SERVER, "Row: %i %s", fields[0].GetUInt32(), fields[1].GetString().c_str());
|
||||
}
|
||||
} else
|
||||
sLog.Info(LOG_SERVER, "Empty result");*/
|
||||
|
||||
// Start stratum server
|
||||
sLog.Info(LOG_SERVER, "Starting stratum");
|
||||
//stratumServer = new Stratum::Server(Config::GetString("STRATUM_IP"), Config::GetInt("STRATUM_PORT"));
|
||||
|
||||
// Init loop vars
|
||||
uint32_t sleepDuration = 0;
|
||||
int exitcode = 0;
|
||||
running = true;
|
||||
|
||||
// Init diff
|
||||
uint32_t minDiffTime = sConfig.Get<uint32_t>("MinDiffTime");
|
||||
diffStart = boost::chrono::steady_clock::now();
|
||||
|
||||
sLog.Info(LOG_SERVER, "Server is running!");
|
||||
|
||||
while (running)
|
||||
{
|
||||
// Calc time diff
|
||||
boost::chrono::steady_clock::time_point now = boost::chrono::steady_clock::now();
|
||||
uint32_t diff = boost::chrono::duration_cast<boost::chrono::milliseconds>(now - diffStart).count();
|
||||
diffStart = now;
|
||||
|
||||
// Update
|
||||
Update(diff);
|
||||
|
||||
// Mercy for CPU
|
||||
if (diff < minDiffTime+sleepDuration) {
|
||||
sleepDuration = minDiffTime - diff + sleepDuration;
|
||||
boost::this_thread::sleep(boost::posix_time::milliseconds(sleepDuration));
|
||||
} else
|
||||
sleepDuration = 0;
|
||||
|
||||
++serverLoops;
|
||||
|
||||
if (serverLoops > 50)
|
||||
running = false;
|
||||
//std::cout << "Diff: " << diff << ", Loop: " << serverLoops << std::endl;
|
||||
}
|
||||
_workerThreads.join_all();
|
||||
|
||||
sLog.Info(LOG_SERVER, "Server is stopping...");
|
||||
|
||||
sDatabase.Close();
|
||||
|
||||
return exitcode;
|
||||
}
|
||||
|
||||
void Server::Update(uint32_t diff)
|
||||
{
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
bool Server::InitDatabase()
|
||||
@ -206,3 +92,28 @@ bool Server::InitDatabase()
|
||||
}
|
||||
}
|
||||
|
||||
void Server::WorkerThread(boost::shared_ptr<asio::io_service> io_service)
|
||||
{
|
||||
std::stringstream threadid;
|
||||
threadid << boost::this_thread::get_id();
|
||||
|
||||
sLog.Info(LOG_SERVER, "Main server thread #%s started.", threadid.str().c_str());
|
||||
|
||||
while (true)
|
||||
{
|
||||
try
|
||||
{
|
||||
boost::system::error_code ec;
|
||||
_io_service->run(ec);
|
||||
if (ec) {
|
||||
sLog.Error(LOG_SERVER, "IO error caught in main server thread #%s: %s.", threadid.str().c_str(), ec.message().c_str());
|
||||
}
|
||||
break;
|
||||
} catch(std::exception& e) {
|
||||
sLog.Error(LOG_SERVER, "Exception caught in main server thread #%s: %s.", threadid.str().c_str(), e.what());
|
||||
}
|
||||
}
|
||||
|
||||
sLog.Info(LOG_SERVER, "Main server thread #%s stopped.", threadid.str().c_str());
|
||||
}
|
||||
|
||||
|
||||
@ -6,28 +6,30 @@
|
||||
#include <boost/thread.hpp>
|
||||
#include <boost/cstdint.hpp>
|
||||
#include <boost/chrono.hpp>
|
||||
#include <boost/shared_ptr.hpp>
|
||||
|
||||
#define SERVER_MIN_DIFF 100
|
||||
|
||||
class Server
|
||||
{
|
||||
public:
|
||||
Server(asio::io_service& io);
|
||||
Server();
|
||||
~Server();
|
||||
|
||||
Stratum::Server* stratumServer;
|
||||
|
||||
void UploadShares(const boost::system::error_code& e);
|
||||
boost::chrono::steady_clock::time_point diffStart;
|
||||
bool running;
|
||||
uint64_t serverLoops;
|
||||
|
||||
int Run();
|
||||
void Update(uint32_t);
|
||||
|
||||
int Run();
|
||||
bool InitDatabase();
|
||||
|
||||
asio::io_service& io_service;
|
||||
void WorkerThread(boost::shared_ptr<asio::io_service> io_service);
|
||||
|
||||
private:
|
||||
// Protocol servers
|
||||
Stratum::Server* _stratumServer;
|
||||
|
||||
// Main io service
|
||||
boost::shared_ptr<asio::io_service> _io_service;
|
||||
|
||||
// Main thread group
|
||||
boost::thread_group _workerThreads;
|
||||
};
|
||||
|
||||
#endif
|
||||
|
||||
Loading…
Reference in New Issue
Block a user