summaryrefslogtreecommitdiff
path: root/lib/query.ml
blob: 6e359ce4802180bee790c819d5b5bab5ef30e6f7 (plain)
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
type pool = ((module Rapper_helper.CONNECTION), Caqti_error.t) Caqti_eio.Pool.t

module Q = struct
  let transaction =
    [%rapper
      execute
        {sql|
        INSERT INTO transactions (client_id, value, type, description)
        VALUES (%int{client_id}, %int{value}, %Operation.TransactionType{transaction_type}, %string{description})
      |sql}]
  ;;

  let debit =
    [%rapper
      execute
        {sql|
        UPDATE balances SET value = value - %int{value} WHERE client_id = %int{client_id}
      |sql}]
  ;;

  let credit =
    [%rapper
      execute
        {sql|
          UPDATE balances SET value = value + %int{value} WHERE client_id = %int{client_id}
        |sql}]
  ;;

  let client =
    let open Client in
    [%rapper
      get_opt
        {sql|
          SELECT clients.id as @int{id}, @int{mov_limit}, value as @int{balance} FROM clients
          JOIN balances ON balances.client_id = %int{id}
          WHERE clients.id = %int{id}
        |sql}
        record_out]
  ;;

  let balance =
    [%rapper
      get_opt
        {sql|
          SELECT @int{value}, @ptime{now()} as time FROM balances WHERE client_id = %int{client_id}
        |sql}]
  ;;
end

let ( let* ) = Result.bind

(* TODO: Separar em duas funções *)
let execute_transaction ~client_id ~(op : Operation.transaction_op) conn =
  match op with
  | `Credit { value; description } ->
    let* () =
      Q.transaction ~client_id ~value ~description conn ~transaction_type:`Credit
    in
    Q.credit ~value ~client_id conn
  | `Debit { value; description } ->
    let* () =
      Q.transaction ~client_id ~value ~description conn ~transaction_type:`Debit
    in
    Q.debit ~value ~client_id conn
;;

let find_client id conn = Q.client ~id conn
let balance client_id conn = Q.balance ~client_id conn