Method create_tracker

tiledb/common/memory_tracker.cc:271–313

}

shared_ptr<MemoryTracker> MemoryTrackerManager::create_tracker(
    uint64_t memory_budget, std::function<void()> on_budget_exceeded) {
  /*
   * The MemoryTracker class has a protected constructor to hopefully help
   * self-document that instances should almost never be created directly
   * except in test code. There exists a
   * `tiledb::test::create_test_memory_tracker()` API that can be used in
   * tests to create untracked instances of MemoryTracker.
   *
   * This just uses the standard private derived class to enable the use of
   * `make_shared` to create instances in specific bits of code.
   */
  class MemoryTrackerCreator : public MemoryTracker {
   public:
    /**
     * Pass through to the protected MemoryTracker constructor for
     * make_shared.
     */
    MemoryTrackerCreator(
        uint64_t memory_budget, std::function<void()> on_budget_exceeded)
        : MemoryTracker(memory_budget, on_budget_exceeded) {
    }
  };

  std::lock_guard<std::mutex> lg(mutex_);

  // Delete any expired weak_ptr instances
  size_t idx = 0;
  while (idx < trackers_.size()) {
    if (trackers_[idx].expired()) {
      trackers_.erase(trackers_.begin() + idx);
    } else {
      idx++;
    }
  }

  // Create a new tracker
  auto ret = make_shared<MemoryTrackerCreator>(
      HERE(), memory_budget, on_budget_exceeded);
  trackers_.emplace(trackers_.begin(), ret);

  return ret;
}

std::string MemoryTrackerManager::to_json() {
  std::lock_guard<std::mutex> lg(mutex_);