aboutsummaryrefslogtreecommitdiff
path: root/src/main/scala/xyz/driver/pdsuicommon/synchronization/db/SlickDbAction.scala
blob: 57cc3d45afd36c3a4cc6de99e822bb021fd7c9f1 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
package xyz.driver.pdsuicommon.synchronization.db

import slick.dbio.DBIO
import xyz.driver.pdsuicommon.logging._
import xyz.driver.pdsuicommon.synchronization.utils.{FakeIdGen, FakeIdMap}

import scala.concurrent.ExecutionContext
import scalaz.Monad

trait SlickDbAction[+T] {
  def entity: T
}

object SlickDbAction {

  final case class Create[T](entity: T) extends SlickDbAction[T]
  final case class Update[T](entity: T) extends SlickDbAction[T]
  final case class Delete[T](entity: T) extends SlickDbAction[T]

  // Use it only inside of a transaction!
  def unsafeRun[T](actions: List[SlickDbAction[T]], dataSource: SlickDataSource[T])(
          implicit core: FakeIdGen[T],
          executionContext: ExecutionContext,
          dbioMonad: Monad[DBIO]): DBIO[FakeIdMap[T]] = {
    unsafeRun(DBIO.successful(FakeIdMap.empty))(actions, dataSource)
  }

  // Use it only inside of a transaction!
  def unsafeRun[T](initial: DBIO[FakeIdMap[T]])(actions: List[SlickDbAction[T]], dataSource: SlickDataSource[T])(
          implicit core: FakeIdGen[T],
          executionContext: ExecutionContext,
          dbioMonad: Monad[DBIO]): DBIO[FakeIdMap[T]] = {
    // TODO Squash Updates and Delete to one operation, when bugs in repositories will be fixed
    actions.foldLeft(initial) {
      case (previousActions, Create(x)) =>
        for {
          r      <- previousActions
          newArm <- dataSource.create(x)
        } yield {
          r + (core(newArm) -> newArm)
        }

      case (previousActions, Update(x)) =>
        for {
          r          <- previousActions
          updatedArm <- dataSource.update(x).getOrElse(x)
        } yield {
          r - core(updatedArm) + (core(updatedArm) -> updatedArm)
        }

      case (previousActions, Delete(_)) if dataSource.isDictionary =>
        previousActions // We don't delete entities from dictionaries

      case (previousActions, Delete(x)) =>
        for {
          r <- previousActions
          _ <- dataSource.delete(x).run
        } yield {
          r - core(x)
        }
    }
  }

  implicit def toPhiString[T](input: SlickDbAction[T])(implicit inner: T => PhiString): PhiString = input match {
    case Create(x) => phi"Create($x)"
    case Update(x) => phi"Update($x)"
    case Delete(x) => phi"Delete($x)"
  }

}