#include <list>
#include <string>
#include <msg/msg.h>
+#include "hostdata.h"
+
+class message {
+public:
+ enum message_type { INFO, CREDIT, LOAD, CTRL_CLOSE, DATA_CLOSE };
+
+ message(message_type t, double a): type(t), amount(a) { }
+
+ message_type get_type() const { return type; }
+ double get_amount() const { return amount; }
+
+ std::string to_string();
+
+private:
+ message_type type;
+ double amount;
+};
class communicator {
public:
communicator();
~communicator();
- void send(m_task_t task, const char *dest);
- void send(m_task_t task, const std::string& dest);
- m_task_t recv();
- int sent_count();
+ // Start to listen for incoming messages
+ void listen();
+
+ // Send a message to the "dest" mailbox
+ void send(const char* dest, message* msg);
+
+ // Try to receive a message. Returns true on success.
+ // If "wait" is true, blocks until success or error.
+ bool recv(message*& msg, m_host_t& from, bool wait);
+
+ // Try to flush pending sending communications.
+ // If "wait" is true, blocks until success.
+ void flush(bool wait);
+
+ // Advertise that the next "close" message is the last one, and
+ // that we do not await any message after that, either on the
+ // control or the data channel.
+ void next_close_on_ctrl_is_last();
+ void next_close_on_data_is_last();
private:
- typedef std::list<msg_comm_t> comm_list;
- comm_list sent_comm;
- char* recv_mbox;
- msg_comm_t recv_comm;
- m_task_t recv_task;
+ // Myself
+ const hostdata* host;
+
+ // List of pending send communications
+ std::list<msg_comm_t> sent_comm;
+ static const int send_count_before_flush;
+ int send_counter;
+
+ // Control channel for receiving
+ m_task_t ctrl_task; // receive buffer
+ msg_comm_t ctrl_comm; // receive communication
+ bool ctrl_close_is_last; // do not rearm comm after next close
+
+ // Data channel for receiving
+ m_task_t data_task; // receive buffer
+ msg_comm_t data_comm; // receive communication
+ bool data_close_is_last; // do not rearm comm after next close
+
+ const char* get_ctrl_mbox() const { return host->get_ctrl_mbox(); }
+ const char* get_data_mbox() const { return host->get_data_mbox(); }
- void send_acknowledge();
+ // Used to test if a communication is over, and to destroy it if it is.
+ static bool comm_test_n_destroy(msg_comm_t comm);
};
#endif // !COMMUNICATOR_H
