pl.touk.nussknacker.ui.db.timeseries.questdb.ThreadAwareObjectPool.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of nussknacker-designer_2.13 Show documentation

nussknacker-designer

There is a newer version: 1.17.0

Show newest version

package pl.touk.nussknacker.ui.db.timeseries.questdb

import scala.collection.mutable

private[questdb] class ThreadAwareObjectPool[T <: AutoCloseable](objectFactory: () => T) {
  private val pool = new mutable.WeakHashMap[Thread, T]()

  def get(): T = {
    val thread = Thread.currentThread()
    pool.getOrElse(
      thread, {
        val t = objectFactory()
        pool.put(thread, t)
        t
      }
    )
  }

  def clear(): Unit = {
    val values = pool.values.toList
    pool.clear()
    values.foreach(_.close())
  }

}