transm/net/net_base.h

38 lines
853 B
C++

#pragma once
#include "util.h"
#include <asio.hpp>
#include <of_util.h>
#include <functional>
#include <mutex>
using namespace ofen;
using ExFun_t = std::function<void(CFrameBuffer* buf)>;
class CServer
{
public:
CServer();
~CServer();
};
class CClient : public std::enable_shared_from_this<CClient>
{
public:
CClient(asio::io_context& io_context, const std::shared_ptr<spdlog::logger>& logger);
~CClient();
public:
bool Connect(const std::string& host, const std::string& port);
void Disconnect();
bool Send(const char* data, int len);
void register_func(ExFun_t& f);
void Receive();
private:
std::shared_ptr<spdlog::logger> logger_;
asio::io_context& io_context_;
asio::ip::tcp::socket socket_;
std::mutex mutex_;
CMutBuffer buffer_;
std::array<char, 1024> tmp_buf_;
ExFun_t fun_;
};