swaydb.core.util.HashedMap.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of core_2.13 Show documentation
core
The newest version!
/*
 * Copyright (c) 2020 Simer JS Plaha ([email protected] - @simerplaha)
 *
 * This file is a part of SwayDB.
 *
 * SwayDB is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * SwayDB is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with SwayDB. If not, see .
 *
 * Additional permission under the GNU Affero GPL version 3 section 7:
 * If you modify this Program or any covered work, only by linking or combining
 * it with separate works, the licensors of this Program grant you additional
 * permission to convey the resulting work.
 */

package swaydb.core.util

import java.util.concurrent.ConcurrentHashMap

import scala.collection.mutable
import scala.jdk.CollectionConverters._

private[swaydb] object HashedMap {

  def concurrent[K, OV, V <: OV](nullValue: OV, initialCapacity: Option[Int] = None): Concurrent[K, OV, V] =
    initialCapacity match {
      case Some(capacity) =>
        new Concurrent[K, OV, V](new ConcurrentHashMap(capacity), nullValue)

      case None =>
        new Concurrent[K, OV, V](new ConcurrentHashMap(), nullValue)
    }

  class Concurrent[K, OV, V <: OV](map: ConcurrentHashMap[K, V], nullValue: OV) {
    def get(key: K): OV = {
      val got = map.get(key)
      if (got == null)
        nullValue
      else
        got
    }

    def put(key: K, value: V): Unit =
      map.put(key, value)

    def contains(key: K): Boolean =
      map.containsKey(key)

    def remove(key: K): Unit =
      map.remove(key)

    def head: (K, V) =
      asScala.head

    def last: (K, V) =
      asScala.last

    def asScala: mutable.Map[K, V] =
      map.asScala

    def clear(): Unit =
      map.clear()
  }
}