message_header.h

// Copyright The SOLA Contributors
//
// Licensed under the MIT License.
// For details on the licensing terms, see the LICENSE file.
// SPDX-License-Identifier: MIT
 
#ifndef MINHTON_MESSAGE_HEADER_H_
#define MINHTON_MESSAGE_HEADER_H_
 
#include <cstdint>
#include <map>
 
#include "minhton/core/constants.h"
#include "minhton/core/node_info.h"
#include "minhton/message/message_logging.h"
#include "minhton/message/types.h"
#include "solanet/serializer/serialize.h"
 
namespace minhton {
class MinhtonMessageHeader {
public:
  MinhtonMessageHeader() = default;
  MinhtonMessageHeader(NodeInfo sender, NodeInfo target, uint64_t ref_event_id = 0);
  uint64_t getEventId() const;
  uint64_t getRefEventId() const;
  NodeInfo getSender() const;
  NodeInfo getTarget() const;
  MessageType getMessageType() const;
 
  void setSender(const NodeInfo &sender);
  void setTarget(const NodeInfo &target);
  void setEventId(uint64_t event_id);
  void setRefEventId(uint64_t ref_event_id);
  void setMessageType(MessageType type);
 
  MessageLoggingInfo getMessageLoggingInfo(bool receive) const;
 
  void setAdditionalLoggingInfo(const MessageLoggingAdditionalInfo &additional_info);
  MessageLoggingAdditionalInfo getAdditionalLoggingInfo() const;
 
  bool validate() const;
 
  SERIALIZE(sender_, target_, message_type_, ref_event_id_, event_id_, additional_logging_info_);
 
private:
  NodeInfo sender_;
  NodeInfo target_;
  MessageType message_type_ = MessageType::kInit;
  uint64_t ref_event_id_ = 0;
  uint64_t event_id_ = 0;
  MessageLoggingAdditionalInfo additional_logging_info_;
};
 
std::string getMessageTypeString(MessageType type);
}  // namespace minhton
#endif