mirror of
https://github.com/SChernykh/p2pool.git
synced 2024-09-11 16:18:47 -06:00
86 lines
2.3 KiB
C
86 lines
2.3 KiB
C
![]() |
/*
|
||
|
* This file is part of the Monero P2Pool <https://github.com/SChernykh/p2pool>
|
||
|
* Copyright (c) 2021 SChernykh <https://github.com/SChernykh>
|
||
|
*
|
||
|
* This program is free software: you can redistribute it and/or modify
|
||
|
* it under the terms of the GNU General Public License as published by
|
||
|
* the Free Software Foundation, version 3.
|
||
|
*
|
||
|
* This program is distributed in the hope that it will be useful, but
|
||
|
* WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||
|
* General Public License for more details.
|
||
|
*
|
||
|
* You should have received a copy of the GNU General Public License
|
||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||
|
*/
|
||
|
|
||
|
#pragma once
|
||
|
|
||
|
#include "uv_util.h"
|
||
|
#include <unordered_map>
|
||
|
|
||
|
namespace p2pool {
|
||
|
|
||
|
class p2pool_api
|
||
|
{
|
||
|
public:
|
||
|
explicit p2pool_api(const std::string& api_path);
|
||
|
~p2pool_api();
|
||
|
|
||
|
enum class Category {
|
||
|
NETWORK,
|
||
|
};
|
||
|
|
||
|
void on_stop();
|
||
|
|
||
|
template<typename T>
|
||
|
void set(const Category& category, const char* filename, T&& callback) { dump_to_file_async_internal(category, filename, DumpFileCallback<T>(std::move(callback))); }
|
||
|
|
||
|
private:
|
||
|
static void on_dump_to_file(uv_async_t* async) { reinterpret_cast<p2pool_api*>(async->data)->dump_to_file(); }
|
||
|
|
||
|
struct DumpFileWork {
|
||
|
uv_fs_t open_req;
|
||
|
uv_fs_t write_req;
|
||
|
uv_fs_t close_req;
|
||
|
|
||
|
std::string name;
|
||
|
std::vector<char> buf;
|
||
|
};
|
||
|
|
||
|
struct DumpFileCallbackBase
|
||
|
{
|
||
|
virtual ~DumpFileCallbackBase() {}
|
||
|
virtual void operator()(log::Stream&) = 0;
|
||
|
};
|
||
|
|
||
|
template<typename T>
|
||
|
struct DumpFileCallback : public DumpFileCallbackBase
|
||
|
{
|
||
|
explicit FORCEINLINE DumpFileCallback(T&& callback) : m_callback(std::move(callback)) {}
|
||
|
void operator()(log::Stream& s) override { m_callback(s); }
|
||
|
|
||
|
private:
|
||
|
DumpFileCallback& operator=(DumpFileCallback&&) = delete;
|
||
|
|
||
|
T m_callback;
|
||
|
};
|
||
|
|
||
|
void dump_to_file_async_internal(const Category& category, const char* filename, DumpFileCallbackBase&& callback);
|
||
|
void dump_to_file();
|
||
|
static void on_fs_open(uv_fs_t* req);
|
||
|
static void on_fs_write(uv_fs_t* req);
|
||
|
static void on_fs_close(uv_fs_t* req);
|
||
|
|
||
|
std::string m_apiPath;
|
||
|
std::string m_networkPath;
|
||
|
|
||
|
uv_mutex_t m_dumpDataLock;
|
||
|
std::unordered_map<std::string, std::vector<char>> m_dumpData;
|
||
|
|
||
|
uv_async_t m_dumpToFileAsync;
|
||
|
};
|
||
|
|
||
|
} // namespace p2pool
|