diff --git a/examples/map/offheap-eviction/src/main/kotlin/com/target/nativememoryallocator/examples/map/offheap.eviction/OffHeapEviction.kt b/examples/map/offheap-eviction/src/main/kotlin/com/target/nativememoryallocator/examples/map/offheap.eviction/OffHeapEviction.kt index 2fac626..900e892 100644 --- a/examples/map/offheap-eviction/src/main/kotlin/com/target/nativememoryallocator/examples/map/offheap.eviction/OffHeapEviction.kt +++ b/examples/map/offheap-eviction/src/main/kotlin/com/target/nativememoryallocator/examples/map/offheap.eviction/OffHeapEviction.kt @@ -68,6 +68,8 @@ private class OffHeapEviction { logger.info { "nativeMemoryMap.size = ${nativeMemoryMap.size}" } logger.info { "nativeMemoryAllocator.nativeMemoryAllocatorMetadata = ${nativeMemoryAllocator.nativeMemoryAllocatorMetadata}" } + logger.info { "nativeMemoryMap.hottestKeys(10) = ${nativeMemoryMap.hottestKeys(numKeys = 10)}" } + logger.info { "nativeMemoryMap.coldestKeys(10) = ${nativeMemoryMap.coldestKeys(numKeys = 10)}" } val randomIndexValue = nativeMemoryMap.get(key = randomIndex) randomIndexValue?.let { diff --git a/src/main/kotlin/com/target/nativememoryallocator/map/NativeMemoryMap.kt b/src/main/kotlin/com/target/nativememoryallocator/map/NativeMemoryMap.kt index 27df6e2..fdae1e9 100644 --- a/src/main/kotlin/com/target/nativememoryallocator/map/NativeMemoryMap.kt +++ b/src/main/kotlin/com/target/nativememoryallocator/map/NativeMemoryMap.kt @@ -99,6 +99,28 @@ interface NativeMemoryMap : BaseNativeMemoryMa * @return [Set] of [Map.Entry] for the map */ val entries: Set> + + /** + * [Set] of most-used keys in the map. + * + * This will only work if [NativeMemoryMapBackend.CAFFEINE] is used, + * and caffeine is configured with maximumSize eviction policy. + * + * @param numKeys number of keys to return + * @return [Set] of hottest [KEY_TYPE] keys + */ + fun hottestKeys(numKeys: Int): Set + + /** + * [Set] of least-used keys in the map. + * + * This will only work if [NativeMemoryMapBackend.CAFFEINE] is used, + * and caffeine is configured with maximumSize eviction policy. + * + * @param numKeys number of keys to return + * @return [Set] of coldest [KEY_TYPE] keys + */ + fun coldestKeys(numKeys: Int): Set } /** diff --git a/src/main/kotlin/com/target/nativememoryallocator/map/impl/CaffeineNativeMemoryMapImpl.kt b/src/main/kotlin/com/target/nativememoryallocator/map/impl/CaffeineNativeMemoryMapImpl.kt index 6a721b7..32a9bfc 100644 --- a/src/main/kotlin/com/target/nativememoryallocator/map/impl/CaffeineNativeMemoryMapImpl.kt +++ b/src/main/kotlin/com/target/nativememoryallocator/map/impl/CaffeineNativeMemoryMapImpl.kt @@ -115,4 +115,18 @@ internal class CaffeineNativeMemoryMapImpl( caffeineStats = caffeineCache.stats(), ) + /** + * Override hottestKeys to use caffeine eviction policy. + */ + override fun hottestKeys(numKeys: Int): Set { + return caffeineCache.policy().eviction().map { it.hottest(numKeys).keys }.orElse(mutableSetOf()) + } + + /** + * Override coldestKeys to use caffeine eviction policy. + */ + override fun coldestKeys(numKeys: Int): Set { + return caffeineCache.policy().eviction().map { it.coldest(numKeys).keys }.orElse(mutableSetOf()) + } + } \ No newline at end of file diff --git a/src/main/kotlin/com/target/nativememoryallocator/map/impl/NativeMemoryMapImpl.kt b/src/main/kotlin/com/target/nativememoryallocator/map/impl/NativeMemoryMapImpl.kt index fb98b1d..476b697 100644 --- a/src/main/kotlin/com/target/nativememoryallocator/map/impl/NativeMemoryMapImpl.kt +++ b/src/main/kotlin/com/target/nativememoryallocator/map/impl/NativeMemoryMapImpl.kt @@ -140,4 +140,8 @@ internal class NativeMemoryMapImpl( override val operationCounters: NativeMemoryMapOperationCounters? = null + override fun hottestKeys(numKeys: Int): Set = emptySet() + + override fun coldestKeys(numKeys: Int): Set = emptySet() + } \ No newline at end of file