accounting-allocator.h 2.95 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
// Copyright 2016 the V8 project authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef V8_ZONE_ACCOUNTING_ALLOCATOR_H_
#define V8_ZONE_ACCOUNTING_ALLOCATOR_H_

#include "include/v8-platform.h"
#include "src/base/atomic-utils.h"
#include "src/base/atomicops.h"
#include "src/base/macros.h"
#include "src/base/platform/mutex.h"
#include "src/base/platform/semaphore.h"
#include "src/base/platform/time.h"
#include "src/zone/zone-segment.h"
16
#include "testing/gtest/include/gtest/gtest_prod.h"  // nogncheck
17 18 19 20

namespace v8 {
namespace internal {

21
class V8_EXPORT_PRIVATE AccountingAllocator {
22
 public:
23
  static const size_t kMaxPoolSize = 8ul * KB;
24

25 26
  AccountingAllocator();
  virtual ~AccountingAllocator();
27

28 29 30 31 32
  // Gets an empty segment from the pool or creates a new one.
  virtual Segment* GetSegment(size_t bytes);
  // Return unneeded segments to either insert them into the pool or release
  // them if the pool is already full or memory pressure is high.
  virtual void ReturnSegment(Segment* memory);
33 34 35 36

  size_t GetCurrentMemoryUsage() const;
  size_t GetMaxMemoryUsage() const;

37 38 39
  size_t GetCurrentPoolSize() const;

  void MemoryPressureNotification(MemoryPressureLevel level);
40 41 42 43 44
  // Configures the zone segment pool size limits so the pool does not
  // grow bigger than max_pool_size.
  // TODO(heimbuef): Do not accept segments to pool that are larger than
  // their size class requires. Sometimes the zones generate weird segments.
  void ConfigureSegmentPool(const size_t max_pool_size);
45

heimbuef's avatar
heimbuef committed
46 47 48
  virtual void ZoneCreation(const Zone* zone) {}
  virtual void ZoneDestruction(const Zone* zone) {}

49
 private:
50 51 52 53
  FRIEND_TEST(Zone, SegmentPoolConstraints);

  static const size_t kMinSegmentSizePower = 13;
  static const size_t kMaxSegmentSizePower = 18;
54 55 56

  STATIC_ASSERT(kMinSegmentSizePower <= kMaxSegmentSizePower);

57 58 59
  static const size_t kNumberBuckets =
      1 + kMaxSegmentSizePower - kMinSegmentSizePower;

60 61 62 63 64 65 66 67 68 69 70 71
  // Allocates a new segment. Returns nullptr on failed allocation.
  Segment* AllocateSegment(size_t bytes);
  void FreeSegment(Segment* memory);

  // Returns a segment from the pool of at least the requested size.
  Segment* GetSegmentFromPool(size_t requested_size);
  // Trys to add a segment to the pool. Returns false if the pool is full.
  bool AddSegmentToPool(Segment* segment);

  // Empties the pool and puts all its contents onto the garbage stack.
  void ClearPool();

72
  Segment* unused_segments_heads_[kNumberBuckets];
73

74 75
  size_t unused_segments_sizes_[kNumberBuckets];
  size_t unused_segments_max_sizes_[kNumberBuckets];
76 77 78

  base::Mutex unused_segments_mutex_;

79 80
  base::AtomicWord current_memory_usage_ = 0;
  base::AtomicWord max_memory_usage_ = 0;
81
  base::AtomicWord current_pool_size_ = 0;
82

83 84
  base::AtomicValue<MemoryPressureLevel> memory_pressure_level_;

85 86 87 88 89 90 91
  DISALLOW_COPY_AND_ASSIGN(AccountingAllocator);
};

}  // namespace internal
}  // namespace v8

#endif  // V8_ZONE_ACCOUNTING_ALLOCATOR_H_