Reference counting in Scala

RefCount.scala

  case class CountedReference[T <: AutoCloseable](counted: ReferenceCounted[T]) extends AutoCloseable {
    counted.inc()
    override def close() = counted.dec()

    def get = counted.inner
  }

  case class ReferenceCounted[T <: AutoCloseable](inner: T) {
    private val count = new AtomicInteger(0)

    def inc(): Unit = count.incrementAndGet()
    def dec(): Unit = {
      val newCount = count.decrementAndGet()
      if(newCount == 0) inner.close()
    }

    def getReference = new CountedReference(this)
  }