blob: a698a87e1d1455065df8c13170cf84b87777520a (
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
|
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)))
let no_such_branch branch_name =
condition "no-such-branch" "A branch with the specified name was not found"
("branch" |=| Concepts.Value.String branch_name)
let branch_head_mismatch branch_name reference head =
condition "branch-head-mismatch" "The HEAD of the branch did not match what was expected"
("branch" |=| Concepts.Value.String branch_name &
"reference" |=| Concepts.Value.String (Concepts.Hash.to_hum_string reference) &
"head" |=| 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)
let root_for tx label =
let open Utilities.Result in
let* label = S.get tx (S.Label label) in
let label = Option.map Concepts.Hash.hash_of_blob label in
Ok label
class manager storage label head = object (self)
inherit Lifecycle.null
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 update branch_name reference new_branch =
let open Utilities.Result in
SI.with_transaction storage (fun tx ->
Utilities.Atomic.mswap head (fun head ->
let* branch = M.lookup tx branch_name head
|> fmap (Option.to_result ~none:(Error.no_such_branch branch_name)) in
if branch = reference then
let* new_head = M.insert tx label new_branch head in
Ok new_head
else
Error (Error.branch_head_mismatch branch_name reference branch)))
|> Result.map ignore
method protocols = Protocols.[Directory.make self]
method list = SI.with_transaction storage (fun tx -> Atomic.get head |> M.keys tx)
method find _ = failwith "TODO"
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 -> failwith "TODO"
| 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
|
