sbt/launch/Cache.scala

20 lines
665 B
Scala
Raw Normal View History

2009-10-18 04:40:02 +02:00
package xsbt.boot
import java.lang.ref.{Reference, SoftReference}
2009-10-18 04:40:02 +02:00
import java.util.HashMap
final class Cache[K,V](create: K => V) extends NotNull
2009-10-18 04:40:02 +02:00
{
private[this] val delegate = new HashMap[K,Reference[V]]
2010-01-28 02:39:07 +01:00
def apply(k: K): V = getFromReference(k, delegate.get(k))
private[this] def getFromReference(k: K, existingRef: Reference[V]) = if(existingRef eq null) newEntry(k) else get(k, existingRef.get)
private[this] def get(k: K, existing: V) = if(existing == null) newEntry(k) else existing
2009-10-18 04:40:02 +02:00
private[this] def newEntry(k: K): V =
{
val v = create(k)
Pre.assert(v != null, "Value for key " + k + " was null")
delegate.put(k, new SoftReference(v))
2009-10-18 04:40:02 +02:00
v
}
}