CstrHashMap.h

Go to the documentation of this file.

/* -*- c++ -*-
 * Copyright (C) 2007-2015 Hypertable, Inc.
 *
 * This file is part of Hypertable.
 *
 * Hypertable is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 3
 * of the License, or any later version.
 *
 * Hypertable is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Hypertable. If not, see <http://www.gnu.org/licenses/>
 */

#ifndef HYPERTABLE_CSTR_HASHMAP_H
#define HYPERTABLE_CSTR_HASHMAP_H

#include <Common/CstrHashTraits.h>

#include <unordered_map>

namespace Hypertable {

template <typename DataT, class TraitsT = CstrHashTraits<> >
class CstrHashMap : public std::unordered_map<const char *, DataT,
                                              typename TraitsT::hasher,
                                              typename TraitsT::key_equal> {
private:
  typedef std::unordered_map<const char *, DataT, typename TraitsT::hasher,
                             typename TraitsT::key_equal> Base;
public:
  typedef typename Base::iterator iterator;
  typedef typename Base::key_type key_type;
  typedef typename Base::value_type value_type;
  typedef typename TraitsT::key_allocator key_allocator;
  typedef std::pair<iterator, bool> InsRet;

public:
  CstrHashMap() {}

  CstrHashMap(size_t n_buckets) : Base(n_buckets) {}

  InsRet insert(const char *key, const DataT &data) {
    return insert_key(key, Base::insert(value_type(key, data)));
  }

  key_allocator &key_alloc() { return m_alloc; }

  void clear() { Base::clear(); }

private:
  key_allocator m_alloc;

  InsRet insert_key(const char *key, InsRet rv) {
    if (rv.second) {
      char *keycopy = m_alloc.dup(key);
      const_cast<key_type &>(rv.first->first) = keycopy;
    }
    return rv;
  }
};

} // namespace Hypertable

#endif