/*
|
* =====================================================================================
|
*
|
* Filename: socket.h
|
*
|
* Description:
|
*
|
* Version: 1.0
|
* Created: 2021年03月30日 15时49分19秒
|
* Revision: none
|
* Compiler: gcc
|
*
|
* Author: Li Chao (),
|
* Organization:
|
*
|
* =====================================================================================
|
*/
|
|
#ifndef SOCKET_GWTJHBPO
|
#define SOCKET_GWTJHBPO
|
|
#include "shm_queue.h"
|
#include <atomic>
|
#include <condition_variable>
|
#include <functional>
|
#include <memory>
|
#include <mutex>
|
#include <thread>
|
#include <vector>
|
|
class ShmSocket
|
{
|
typedef bhome_shm::ShmMsgQueue Queue;
|
|
public:
|
enum Type {
|
eSockRequest,
|
eSockReply,
|
eSockSubscribe,
|
eSockPublish,
|
};
|
typedef std::function<void(bhome_msg::BHMsg &msg)> RecvCB;
|
|
ShmSocket(Type type);
|
ShmSocket(Type type, bhome_shm::SharedMemory &shm);
|
~ShmSocket();
|
|
// bool Request(const std::string &topic, const void *data, const size_t size, onReply);
|
bool RequestAndWait() { return false; } // call Request, and wait onReply notify cv
|
|
// bool HandleRequest(onData);
|
bool ReadRequest(); // exclude with HandleRequest
|
bool SendReply(); // exclude with HandleRequest
|
|
bool Publish(const std::string &topic, const void *data, const size_t size, const int timeout_ms);
|
bool Subscribe(const std::vector<std::string> &topics, const int timeout_ms);
|
bool RecvSub(std::string &topic, std::string &data, const int timeout_ms);
|
bool SetRecvCallback(const RecvCB &onRecv);
|
|
private:
|
bool HasRecvCB();
|
void Stop();
|
|
bhome_shm::SharedMemory &shm_;
|
Type type_;
|
std::vector<std::thread> workers_;
|
std::mutex mutex_;
|
std::condition_variable cv_recv_cb_;
|
std::atomic<bool> run_;
|
RecvCB onRecv_;
|
|
std::unique_ptr<Queue> mq_;
|
};
|
|
#endif // end of include guard: SOCKET_GWTJHBPO
|
