org.hnau.base.data.AutoKeyCache.kt Maven / Gradle / Ivy

Go to download
package org.hnau.base.data


open class AutoKeyCache(
        private val keysGenerator: () -> K,
        private val tryCount: Int = 1000
) {

    private val container = HashMap()

    fun put(data: V): K? {
        val key = generateNewKey() ?: return null
        container[key] = data
        return key
    }

    private fun generateNewKey(): K? {
        repeat(tryCount) {
            keysGenerator()
                    .takeIf { container[it] == null }
                    ?.let { return it }
        }
        return null
    }

    fun get(key: K) = container[key]

    fun remove(key: K) = container.remove(key)

}