tech.simter.kv.impl.dao.jpa.KeyValueBlockDaoImpl.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of simter-kv-dao-jpa Show documentation

Dao Implementation By JPA

The newest version!

package tech.simter.kv.impl.dao.jpa

import org.springframework.beans.factory.annotation.Autowired
import org.springframework.stereotype.Repository
import org.springframework.transaction.annotation.Transactional
import tech.simter.kv.impl.dao.jpa.po.KeyValuePo
import java.util.*
import javax.persistence.EntityManager
import javax.persistence.NoResultException
import javax.persistence.PersistenceContext

/**
 * The JPA implementation of [KeyValueBlockDao].
 *
 * @author RJ
 */
@Repository
internal class KeyValueBlockDaoImpl @Autowired constructor(
  @PersistenceContext private val em: EntityManager,
  private val repository: KeyValueJpaRepository
) : KeyValueBlockDao {
  @Transactional(readOnly = true)
  override fun valueOf(key: String): Optional {
    return try {
      Optional.of(
        em.createQuery("select kv.v from KeyValuePo kv where kv.k = :key", String::class.java)
          .setParameter("key", key)
          .singleResult
      )
    } catch (e: NoResultException) {
      Optional.empty()
    }
  }

  @Transactional(readOnly = true)
  override fun find(vararg keys: String): Map {
    return if (keys.isEmpty()) emptyMap()
    else {
      return em.createQuery("select kv from KeyValuePo kv where kv.k in :keys", KeyValuePo::class.java)
        .setParameter("keys", keys.toList())
        .resultList
        .associate { it.k to it.v }
    }
  }

  @Transactional(readOnly = false)
  override fun save(keyValues: Map) {
    repository.saveAll(keyValues.map { KeyValuePo(it.key, it.value) })
  }

  @Transactional(readOnly = false)
  override fun delete(vararg keys: String) {
    if (keys.isNotEmpty()) {
      em.createQuery("delete from KeyValuePo kv where kv.k in :keys")
        .setParameter("keys", keys.toList())
        .executeUpdate()
    }
  }
}