worker.h

#pragma once
 
#include <functional>
#include <memory>
#include <ucs/type/status.h>
#include <unordered_map>
 
#include <ucp/api/ucp.h>
 
#include "ucxpp/address.h"
#include "ucxpp/awaitable.h"
#include "ucxpp/context.h"
 
namespace ucxpp {
 
class worker : public std::enable_shared_from_this<worker> {
  friend class local_address;
  friend class endpoint;
  ucp_worker_h worker_;
  std::shared_ptr<context> ctx_;
  int event_fd_;
 
public:
  worker(std::shared_ptr<context> ctx);
 
  int event_fd() const;
 
  std::shared_ptr<context> context_ptr() const;
 
  local_address get_address() const;
 
  ucp_worker_h handle() const;
 
  bool progress() const;
 
  void wait() const;
 
  bool arm() const;
 
  tag_recv_awaitable tag_recv(void *buffer, size_t length, ucp_tag_t tag,
                              ucp_tag_t tag_mask = 0xFFFFFFFFFFFFFFFF) const;
 
  void fence();
 
  worker_flush_awaitable flush();
 
  ~worker();
};
 
} // namespace ucxpp