im.actor.server.persist.AuthCodeRepo.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of actor-persist Show documentation

actor-persist

There is a newer version: 3.0.0

package im.actor.server.persist

import java.time.LocalDateTime

import im.actor.server.db.ActorPostgresDriver.api._

import im.actor.server.model

final class AuthCodeTable(tag: Tag) extends Table[model.AuthCode](tag, "auth_codes") {
  def transactionHash = column[String]("transaction_hash", O.PrimaryKey)
  def code = column[String]("code")
  def attempts = column[Int]("attempts")
  def createdAt = column[LocalDateTime]("created_at")
  def isDeleted = column[Boolean]("is_deleted")

  def * = (transactionHash, code, attempts, createdAt, isDeleted) <> (model.AuthCode.tupled, model.AuthCode.unapply)
}

object AuthCodeRepo {
  val codes = TableQuery[AuthCodeTable]

  def create(transactionHash: String, code: String) =
    codes += model.AuthCode(transactionHash, code)

  def createOrUpdate(transactionHash: String, code: String) =
    codes.insertOrUpdate(model.AuthCode(transactionHash, code))

  def findByTransactionHash(transactionHash: String) =
    codes.filter(c ⇒ c.transactionHash === transactionHash && c.isDeleted === false).result.headOption

  def deleteByTransactionHash(transactionHash: String) =
    codes.filter(_.transactionHash === transactionHash).map(_.isDeleted).update(true)

  def incrementAttempts(transactionHash: String, currentValue: Int) =
    codes.filter(_.transactionHash === transactionHash).map(_.attempts).update(currentValue + 1)

}