blob: 016f75a5784ac6aff00cfdaeb4c9ec06e5a61d12 (
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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
module Error = struct
open Concepts.Condition
let invalid_root head =
condition "invalid-root" "The root hash for the database state is missing from the backend. Either your storage is corrupted, or this is a bug in RNT!"
("hash" |=| (Concepts.Value.String (Concepts.Hash.to_hum_string head)))
end
module Make (S : Abstract.Storage.STORAGE) = struct
module M = Merkle.Make (S) (Merkle.StringKey)
module SI = Storage.Make (S)
module B = Branch.Make (S)
let root_for tx label =
let open Utilities.Result in
let* label = S.get tx (S.Label label) in
let data = Option.map (fun blob -> Concepts.Blob.bytes_of_blob blob
|> Bytes.to_string
|> Concepts.Hash.of_raw_string)
label in
Ok data
let persist_root tx label addr =
let data = Concepts.Hash.blob_of_hash addr in
S.put tx (S.Label label) data
class manager storage label head = object (self)
inherit Lifecycle.null
inherit Identity.of_id
val storage : S.connection = storage
val label : string = label
val head : M.node Atomic.t = Atomic.make head
method fetch branch_name =
SI.with_transaction storage (fun tx ->
let head = Atomic.get head in
M.lookup tx branch_name head)
method protocols = Protocols.[Directory.make self; Registry.make self]
method list = SI.with_transaction storage (fun tx -> Atomic.get head |> M.keys tx)
method find key =
let open Utilities.Result in
SI.with_transaction storage (fun tx ->
let* head = Atomic.get head |> M.lookup tx key in
match head with
| None -> Ok None
| Some head -> B.load tx storage head |> Result.map Option.some)
method update key reference value =
let open Utilities.Result in
SI.with_transaction storage (fun tx ->
match
Utilities.Atomic.mswap head (fun head ->
let* branch = M.lookup tx key head |> Result.map_error (fun e -> Some e) in
(* FIXME: we should have an addressable protocol rather than assuming the hash is the content store address *)
let reference = Option.map Protocols.Handle.hash reference in
let value = Option.map Protocols.Handle.hash value in
if branch = reference then
begin
let* new_head = match value with
| Some value -> M.insert tx key value head
| None -> M.remove tx key head
in
let* _ = persist_root tx label (M.hash_of new_head) in
Ok new_head
end
|> Result.map_error (fun e -> Some e)
else
Error None)
with
| Ok _ -> Ok true
| Error None -> Ok false
| Error (Some e) -> Error e)
end
let make storage label =
let open Utilities.Result in
SI.with_transaction storage (fun tx ->
let* addr = root_for tx label in
match addr with
| None ->
Ok (new manager storage label M.empty |> Protocols.Handle.make)
| Some addr ->
let* head = M.find tx addr in
let* head = Option.to_result ~none:(Error.invalid_root addr) head in
Ok (new manager storage label head |> Protocols.Handle.make))
end
|
