Check status of sent messages.

This commit is contained in:
Nicolás Ortega Froysa 2017-09-25 17:19:16 +02:00
parent 763277dd9b
commit 59d59549e1
No known key found for this signature in database
GPG Key ID: FEC70E3BAE2E69BF
2 changed files with 50 additions and 10 deletions

View File

@ -47,6 +47,15 @@ struct message {
struct user from; ///< Info on the peer who sent the message. struct user from; ///< Info on the peer who sent the message.
}; };
/**
* @brief Status of a message.
*/
enum {
NC_PENDING = 1, ///< Status of message is pending.
NC_GOOD = 2, ///< Message was sent properly.
NC_BAD = 3, ///< Message failed to send.
};
/** /**
* @brief Initialize a local DHT node. If port is 0 then the default port * @brief Initialize a local DHT node. If port is 0 then the default port
* number will be used. * number will be used.
@ -137,10 +146,23 @@ void NeoComm_free_message(struct message *msg);
* @param channel_name Name of the channel to send the message to. * @param channel_name Name of the channel to send the message to.
* @param message A message to send to the channel. * @param message A message to send to the channel.
* *
* @return 1 upon success, 0 upon failure. Use NeoComm_get_last_error for a * @return Upon success it returns the time that the message was sent which
* text description of the error. * can be used as a token to see if the message was sent properly using
* NeoComm_check_message, if the function failed then 0 will be returned.
*/ */
int NeoComm_send_message(const char *channel_name, const char *message); time_t NeoComm_send_message(const char *channel_name, const char *message);
/**
* @brief Check the status of a sent message.
*
* @param token The token of a message to check its status.
*
* @return The status of the message (NC_GOOD, NC_BAD, or NC_PENDING). If the
* token does not exist then 0 is returned.
*
* @warning Once checked the token becomes invalid.
*/
int NeoComm_check_message(const time_t token);
/** /**
* @brief Get the last error that occurred in text. * @brief Get the last error that occurred in text.

View File

@ -23,9 +23,11 @@
std::map<std::string, struct channel_info> channels; std::map<std::string, struct channel_info> channels;
static std::mt19937_64 rand_dev {dht::crypto::random_device{}()}; static std::mt19937_64 rand_dev { dht::crypto::random_device{}() };
static std::uniform_int_distribution<dht::Value::Id> rand_id; static std::uniform_int_distribution<dht::Value::Id> rand_id;
static std::map<time_t, int> sent_messages;
int NeoComm_join_channel(const char *channel_name) { int NeoComm_join_channel(const char *channel_name) {
if(not node.isRunning()) if(not node.isRunning())
{ {
@ -54,7 +56,7 @@ int NeoComm_join_channel(const char *channel_name) {
channels[s_chan_name] = { channels[s_chan_name] = {
/*.hash =*/ chan_hash, /*.hash =*/ chan_hash,
/*.token =*/ node.listen<dht::ImMessage>(chan_hash, /*.token =*/ node.listen<dht::ImMessage>(chan_hash,
[=](dht::ImMessage &&msg) { [&](dht::ImMessage &&msg) {
channels[s_chan_name].msgs.push_back(msg); channels[s_chan_name].msgs.push_back(msg);
return true; return true;
}), }),
@ -98,7 +100,12 @@ void NeoComm_free_message(struct message *msg) {
free(msg); free(msg);
} }
int NeoComm_send_message(const char *channel_name, const char *message) { time_t NeoComm_send_message(const char *channel_name, const char *message) {
if(not node.isRunning())
{
add_error("NeoComm must be initialized.");
return 0;
}
const dht::InfoHash chan_hash = channels[channel_name].hash; const dht::InfoHash chan_hash = channels[channel_name].hash;
const time_t now = const time_t now =
std::chrono::system_clock::to_time_t(std::chrono::system_clock::now()); std::chrono::system_clock::to_time_t(std::chrono::system_clock::now());
@ -108,10 +115,21 @@ int NeoComm_send_message(const char *channel_name, const char *message) {
add_error("Not connected to channel"); add_error("Not connected to channel");
return 0; return 0;
} }
sent_messages[now] = NC_PENDING;
node.putSigned(chan_hash, node.putSigned(chan_hash,
dht::ImMessage(rand_id(rand_dev), message, now), [](bool sent) { dht::ImMessage(rand_id(rand_dev), message, now), [&](bool sent) {
// TODO: Figure out what to do here for failed messages. sent_messages[now] = sent ? NC_GOOD : NC_BAD;
}); });
return now;
return 1; }
int NeoComm_check_message(const time_t token) {
if(sent_messages.find(token) == sent_messages.end())
{
add_error("No message found with that token.");
return 0;
}
const int status = sent_messages[token];
sent_messages.erase(token);
return status;
} }