transm/net/net_base.h
2024-12-12 23:11:55 +08:00

30 lines
733 B
C++

#pragma once
#include "util.h"
#include <asio.hpp>
#include <mutex>
#include <of_util.h>
using namespace ofen;
class CTcpClient : public std::enable_shared_from_this<CTcpClient>
{
public:
CTcpClient(asio::io_context& io_context, const std::shared_ptr<spdlog::logger>& logger);
~CTcpClient();
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 async_recv();
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_;
};