65 lines
1.8 KiB
C++
65 lines
1.8 KiB
C++
#ifndef __TCP_CLIENT_H__
|
|
#define __TCP_CLIENT_H__
|
|
/*
|
|
*******************************************************************************
|
|
* INCLUDE
|
|
*******************************************************************************
|
|
*/
|
|
#include <Poco/Net/StreamSocket.h>
|
|
#include <Poco/Task.h>
|
|
#include <Poco/Timespan.h>
|
|
#include <functional>
|
|
|
|
|
|
/*
|
|
*******************************************************************************
|
|
* CLASS DECLARE
|
|
*******************************************************************************
|
|
*/
|
|
namespace Network
|
|
{
|
|
|
|
class TCPClient: public Poco::Task
|
|
{
|
|
public:
|
|
enum class State
|
|
{
|
|
DISCONNECTED,
|
|
CONNECT_TIMEOUT,
|
|
CONNECT_REFUSED,
|
|
CONNECTING,
|
|
CONNECTED,
|
|
};
|
|
|
|
using StateChangedCallback = std::function<void(State, const std::string&)>;
|
|
using DataIncomingCallback = std::function<void(const uint8_t*, uint32_t)>;
|
|
|
|
public:
|
|
TCPClient(std::string host, int port, Poco::Timespan timeout, uint32_t sz = 10240);
|
|
~TCPClient();
|
|
|
|
int send(const uint8_t* data, uint32_t len);
|
|
void start();
|
|
void stop();
|
|
void setStateChangedCallback(const StateChangedCallback& callback);
|
|
void setDataIncomingCallback(const DataIncomingCallback& callback);
|
|
|
|
protected:
|
|
bool connect();
|
|
void receive(const uint8_t* data, uint32_t len);
|
|
virtual void runTask() override;
|
|
void setState(State state, const std::string& desc);
|
|
|
|
private:
|
|
Poco::Thread _thread;
|
|
Poco::Net::StreamSocket _socket;
|
|
Poco::Net::SocketAddress _address;
|
|
Poco::Timespan _timeout;
|
|
uint32_t _bufsz;
|
|
Poco::SharedPtr<uint8_t> _pBuffer;
|
|
StateChangedCallback _stateChanged;
|
|
DataIncomingCallback _dataIncoming;
|
|
};
|
|
|
|
}
|
|
#endif
|