Source file rpc_implementations.ml

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
open! Core
open! Async
open Memtrace_viewer_common

let init global_state =
  let f user_state () =
    User_state.reset global_state user_state;
    let data = User_state.data user_state in
    let serialized = Data.Serialized.serialize data in
    return serialized
  in
  Rpc.Rpc.implement Protocol.Init.t f
;;

let update global_state =
  let f user_state action =
    User_state.update global_state user_state action;
    let data = User_state.data user_state in
    let serialized = Data.Serialized.serialize data in
    return serialized
  in
  Rpc.Rpc.implement Protocol.Update.t f
;;

let implementations global_state =
  Rpc.Implementations.create_exn
    ~implementations:[ init global_state; update global_state ]
    ~on_unknown_rpc:`Continue
;;