aboutsummaryrefslogtreecommitdiff
path: root/aoc2017-gleam/src/helpers/state.gleam
blob: cbbad81e47e36639f35dff89ff7ed16c7b1caf5a (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
import gleam/erlang/process.{type Subject, Normal}
import gleam/option.{None}
import gleam/otp/actor.{type Next, Continue, Stop}
import gleam/set.{type Set}

const timeout = 1000

pub type Message(k) {
  Shutdown
  Check(key: k, client: Subject(Bool))
  Add(key: k)
  Drop(key: k)
  Pop(client: Subject(Result(k, Nil)))
}

fn handle_message(message: Message(k), set: Set(k)) -> Next(Message(k), Set(k)) {
  case message {
    Shutdown -> Stop(Normal)
    Check(key, client) -> {
      process.send(client, set.contains(set, key))
      Continue(set, None)
    }
    Add(key) -> Continue(set.insert(set, key), None)
    Drop(key) -> Continue(set.delete(set, key), None)
    Pop(client) -> {
      case set.to_list(set) {
        [next, ..] -> {
          process.send(client, Ok(next))
          Continue(set.delete(set, next), None)
        }
        [] -> {
          process.send(client, Error(Nil))
          Stop(Normal)
        }
      }
    }
  }
}

pub fn start_actor(with: Set(a)) {
  let assert Ok(actor) = actor.start(with, handle_message)
  actor
}

pub fn pop(actor) {
  process.call(actor, Pop, timeout)
}

pub fn check(actor, value) {
  process.call(actor, Check(value, _), timeout)
}

pub fn drop(actor, value) {
  process.send(actor, Drop(value))
}