socket_client.h

#pragma once

#include <string>
#include <shared_mutex>
#include <mutex>

class SocketClient {
public:
    SocketClient(int selectTimeout, int recvTimeout);
    ~SocketClient();

    // 禁止拷贝和赋值
    SocketClient(const SocketClient&) = delete;
    SocketClient& operator=(const SocketClient&) = delete;

    // 连接到服务器
    bool connectToServer(const std::string& host, uint16_t port);

    // 发送数据
    bool sendData(const void* data, size_t length);

    bool isDataAvailable();

    // 接收指定长度的数据
    bool receiveData(void* buffer, size_t length);

    // 关闭连接
    void disconnect();

    // 检查是否已连接
    bool isConnected() const;

private:
    const int selectTimeout;
    const int recvTimeout;
    int m_socket;
    std::shared_mutex sock_mtx;
    std::mutex send_mtx;
};