Question

我有boost bimap

#include <iostream>
#include <utility>
#include <boost/bimap.hpp>
#include <boost/bimap/set_of.hpp>
#include <boost/bimap/multiset_of.hpp>

namespace bimaps = boost::bimaps;
typedef boost::bimap<bimaps::set_of<unsigned long long int>,
        bimaps::multiset_of<unsigned long long int > > bimap_reference;
typedef bimap_reference::value_type position;
bimap_reference numbers;

int main()
{
    numbers.insert(position(12345689, 1000000000));
    numbers.insert(position(23456789, 8000000000));   
    return 0;
}

我有大约18,000,000个条目。从理论上讲，它应占用~2.7GB的空间（180000000 * 8 * 2 = 2880000000字节= 2880000000/1024 * 1024 * 1024 = ~2.7GB）。现在我想找到boost bimap占用的实际空间，我该怎么做。

Answer 1

就像你提到的问题中的评论一样，你可以重载new和delete运算符来跟踪内存分配和解除分配。 this文章的全局替换部分下的示例显示了一个非常简单的示例：

void* operator new(std::size_t sz) {
    std::printf("global op new called, size = %zu\n", sz);
    return std::malloc(sz);
}
void operator delete(void* ptr) noexcept {
    std::puts("global op delete called");
    std::free(ptr);
}

此示例的唯一问题是您无法确定释放多少内存。要解决此问题，请查看accepted answer问题的How to track memory allocations in C++ (especially new/delete)。

上述答案中的示例使用std::map和自定义分配器来存储已分配内存块的地址和大小。在delete运算符重载内部，它会删除具有指定地址的元素。几乎没有任何修改，它可以用于您的要求：

#include <map>

template<typename T>
struct MemoryMapAllocator : std::allocator<T> {
    typedef typename std::allocator<T>::pointer pointer;
    typedef typename std::allocator<T>::size_type size_type;
    template<typename U> struct rebind { typedef MemoryMapAllocator<U> other; };

    MemoryMapAllocator() {}

    template<typename U>
    MemoryMapAllocator(const MemoryMapAllocator<U>& u) : std::allocator<T>(u) {}

    pointer allocate(size_type size, std::allocator<void>::const_pointer = 0) {
        void* p = std::malloc(size * sizeof(T));
        if(p == 0)
            throw std::bad_alloc();
        return static_cast<pointer>(p);
    }
    void deallocate(pointer p, size_type) {
        std::free(p);
    }
};

typedef std::map<void*, std::size_t, std::less<void*>,
            MemoryMapAllocator<std::pair<void* const, std::size_t>>> MemoryMap;

MemoryMap& getMemoryMap() {
    static MemoryMap memMap;
    return memMap;
}

std::size_t totalAllocatedMemory() {
    std::size_t sum = 0;
    for(auto& e : getMemoryMap())
        sum += e.second;
    return sum;
}

void* operator new(std::size_t size) {
    void* mem = std::malloc(size == 0 ? 1 : size);

    if(mem == 0)
        throw std::bad_alloc();

    getMemoryMap()[mem] = size;
    return mem;
}

void operator delete(void* mem) {
    getMemoryMap().erase(mem);
    std::free(mem);
}

Live Demo

如何通过boost bimap找到内存

1 个答案: