src/kernel/activity/MailboxImpl.hpp

   1 /* Copyright (c) 2007-2023. The SimGrid Team. All rights reserved.          */
   2
   3 /* This program is free software; you can redistribute it and/or modify it
   4  * under the terms of the license (GNU LGPL) which comes with this package. */
   5
   6 #ifndef SIMGRID_KERNEL_ACTIVITY_MAILBOX_HPP
   7 #define SIMGRID_KERNEL_ACTIVITY_MAILBOX_HPP
   8
   9 #include "simgrid/config.h" /* FIXME: KILLME. This makes the ABI config-dependent, but mandatory for the hack below */
  10 #include "simgrid/s4u/Engine.hpp"
  11 #include "simgrid/s4u/Mailbox.hpp"
  12 #include "src/kernel/activity/CommImpl.hpp"
  13 #include "src/kernel/actor/ActorImpl.hpp"
  14
  15 namespace simgrid::kernel::activity {
  16
  17 /** @brief Implementation of the s4u::Mailbox */
  18
  19 class MailboxImpl {
  20   s4u::Mailbox piface_;
  21   std::string name_;
  22   actor::ActorImplPtr permanent_receiver_; // actor to which the mailbox is attached
  23
  24   std::deque<CommImplPtr> comm_queue_;
  25   // messages already received in the permanent receive mode
  26   std::deque<CommImplPtr> done_comm_queue_;
  27
  28   friend s4u::Engine;
  29   friend s4u::Mailbox;
  30   friend s4u::Mailbox* s4u::Engine::mailbox_by_name_or_create(const std::string& name) const;
  31   friend s4u::Mailbox* s4u::Mailbox::by_name(const std::string& name);
  32
  33   static unsigned next_id_; // Next ID to be given
  34   const unsigned id_ = next_id_++;
  35   explicit MailboxImpl(const std::string& name) : piface_(this), name_(name) {}
  36   MailboxImpl(const MailboxImpl&) = delete;
  37   MailboxImpl& operator=(const MailboxImpl&) = delete;
  38
  39 public:
  40   /** @brief Public interface */
  41   unsigned get_id() const { return id_; }
  42
  43   ~MailboxImpl();
  44
  45   const s4u::Mailbox* get_iface() const { return &piface_; }
  46   s4u::Mailbox* get_iface() { return &piface_; }
  47
  48   const std::string& get_name() const { return name_; }
  49   const char* get_cname() const { return name_.c_str(); }
  50   void set_receiver(s4u::ActorPtr actor);
  51   void push(const CommImplPtr& comm);
  52   void push_done(const CommImplPtr& done_comm) { done_comm_queue_.push_back(done_comm); }
  53   void remove(const CommImplPtr& comm);
  54   void clear(bool do_finish);
  55   CommImplPtr iprobe(int type, const std::function<bool(void*, void*, CommImpl*)>& match_fun, void* data);
  56   CommImplPtr find_matching_comm(CommImplType type, const std::function<bool(void*, void*, CommImpl*)>& match_fun,
  57                                  void* this_user_data, const CommImplPtr& my_synchro, bool done, bool remove_matching);
  58   bool is_permanent() const { return permanent_receiver_ != nullptr; }
  59   actor::ActorImplPtr get_permanent_receiver() const { return permanent_receiver_; }
  60   bool empty() const { return comm_queue_.empty(); }
  61   size_t size() const { return comm_queue_.size(); }
  62   const CommImplPtr& front() const { return comm_queue_.front(); }
  63   bool has_some_done_comm() const { return not done_comm_queue_.empty(); }
  64   const CommImplPtr& done_front() const { return done_comm_queue_.front(); }
  65 };
  66 } // namespace simgrid::kernel::activity
  67
  68 #endif