66 lines
1.5 KiB
C++
66 lines
1.5 KiB
C++
#pragma once
|
|
#include "messaging.h"
|
|
#include "msgq.h"
|
|
#include <zmq.h>
|
|
#include <string>
|
|
|
|
#define MAX_POLLERS 128
|
|
|
|
class MSGQContext : public Context {
|
|
private:
|
|
void * context = NULL;
|
|
public:
|
|
MSGQContext();
|
|
void * getRawContext() {return context;}
|
|
~MSGQContext();
|
|
};
|
|
|
|
class MSGQMessage : public Message {
|
|
private:
|
|
char * data;
|
|
size_t size;
|
|
public:
|
|
void init(size_t size);
|
|
void init(char *data, size_t size);
|
|
void takeOwnership(char *data, size_t size);
|
|
size_t getSize(){return size;}
|
|
char * getData(){return data;}
|
|
void close();
|
|
~MSGQMessage();
|
|
};
|
|
|
|
class MSGQSubSocket : public SubSocket {
|
|
private:
|
|
msgq_queue_t * q = NULL;
|
|
int timeout;
|
|
public:
|
|
int connect(Context *context, std::string endpoint, std::string address, bool conflate=false, bool check_endpoint=true);
|
|
void setTimeout(int timeout);
|
|
void * getRawSocket() {return (void*)q;}
|
|
Message *receive(bool non_blocking=false);
|
|
~MSGQSubSocket();
|
|
};
|
|
|
|
class MSGQPubSocket : public PubSocket {
|
|
private:
|
|
msgq_queue_t * q = NULL;
|
|
public:
|
|
int connect(Context *context, std::string endpoint, bool check_endpoint=true);
|
|
int sendMessage(Message *message);
|
|
int send(char *data, size_t size);
|
|
bool all_readers_updated();
|
|
~MSGQPubSocket();
|
|
};
|
|
|
|
class MSGQPoller : public Poller {
|
|
private:
|
|
std::vector<SubSocket*> sockets;
|
|
msgq_pollitem_t polls[MAX_POLLERS];
|
|
size_t num_polls = 0;
|
|
|
|
public:
|
|
void registerSocket(SubSocket *socket);
|
|
std::vector<SubSocket*> poll(int timeout);
|
|
~MSGQPoller(){};
|
|
};
|