pending_io.h

Go to the documentation of this file.

// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the Apache 2.0 License.
#pragma once
 
#include <sys/socket.h>
 
template <class T>
struct PendingIO
{
  using free_cb_t = void (*)(T*);
  T* req;
  size_t len;
  sockaddr addr;
  free_cb_t free_cb;
  bool clear;
 
  PendingIO(
    T* req_, size_t len_, sockaddr addr_, free_cb_t free_cb_ = nullptr) :
    req(req_),
    len(len_),
    addr(addr_),
    free_cb(free_cb_),
    clear(false)
  {
    if (free_cb == nullptr)
    {
      // Assume we take ownership
      auto orig = req;
      req = new T[len];
      memcpy(req, orig, len);
    }
  }
 
  PendingIO(PendingIO&& that) :
    req(that.req),
    len(that.len),
    addr(that.addr),
    free_cb(that.free_cb),
    clear(that.clear)
  {
    that.req = nullptr;
  }
 
  PendingIO<T>& operator=(PendingIO&& that)
  {
    req = std::move(that.req);
    len = std::move(that.len);
    addr = std::move(that.addr);
    free_cb = std::move(that.free_cb);
    clear = std::move(that.clear);
    that.req = nullptr;
    return *this;
  }
 
  ~PendingIO()
  {
    if (free_cb)
    {
      free_cb(req);
    }
    else
    {
      delete[] req;
    }
  }
 
  static void clear_empty(std::vector<PendingIO<T>>& list)
  {
    std::remove_if(
      list.begin(), list.end(), [](PendingIO<T>& p) { return p.clear; });
  }
};