libcryfs/impl/IdList.h

#pragma once
#ifndef FSPP_FUSE_IDLIST_H_
#define FSPP_FUSE_IDLIST_H_

#include <map>
#include <memory>
#include <mutex>
#include <stdexcept>
#include "messmer/cpp-utils/macros.h"

namespace fspp {

template<class Entry>
class IdList {
public:
  IdList();
  virtual ~IdList();

  int add(std::unique_ptr<Entry> entry);
  Entry *get(int id);
  const Entry *get(int id) const;
  void remove(int id);
private:
  std::map<int, std::unique_ptr<Entry>> _entries;
  int _id_counter;
  mutable std::mutex _mutex;

  DISALLOW_COPY_AND_ASSIGN(IdList<Entry>)
};

template<class Entry>
IdList<Entry>::IdList()
  : _entries(), _id_counter(0), _mutex() {
}

template<class Entry>
IdList<Entry>::~IdList() {
}

template<class Entry>
int IdList<Entry>::add(std::unique_ptr<Entry> entry) {
  std::lock_guard<std::mutex> lock(_mutex);
  //TODO Reuse IDs (ids = descriptors)
  int new_id = ++_id_counter;
  _entries[new_id] = std::move(entry);
  return new_id;
}

template<class Entry>
Entry *IdList<Entry>::get(int id) {
  return const_cast<Entry*>(const_cast<const IdList<Entry>*>(this)->get(id));
}

template<class Entry>
const Entry *IdList<Entry>::get(int id) const {
  std::lock_guard<std::mutex> lock(_mutex);
  const Entry *result = _entries.at(id).get();
  return result;
}

template<class Entry>
void IdList<Entry>::remove(int id) {
  std::lock_guard<std::mutex> lock(_mutex);
  auto found_iter = _entries.find(id);
  if (found_iter == _entries.end()) {
    throw std::out_of_range("Called IdList::remove() with an invalid ID");
  }
  _entries.erase(found_iter);
}

} /* namespace fspp */

#endif /* FSPP_FUSE_IDLIST_H_ */