balsoft/ligo
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Mempool: cache result and operation in ValidatedCache

Browse Source
This commit is contained in:
Pietro Abate 2018-11-19 11:37:43 +01:00 committed by MBourgoin
parent 1420715ab9
commit 0bb2f3d4d1
No known key found for this signature in database
GPG Key ID: 4B3F7008ABB5B2D0
1 changed files with 46 additions and 45 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
src/lib_shell/mempool_worker.ml
View File

@ -259,10 +259,15 @@ module Make(Static: STATIC)(Proto: Registered_protocol.T)
(* validated operations' cache. used for memoization *)
module ValidatedCache = struct
type t = result Operation_hash.Table.t
type t = (result * Operation.t) Operation_hash.Table.t
let encoding =
Operation_hash.Table.encoding result_encoding
let open Data_encoding in
Operation_hash.Table.encoding (
tup2
result_encoding
Operation.encoding
)
let create () = Operation_hash.Table.create 1000
@ -275,7 +280,7 @@ module Make(Static: STATIC)(Proto: Registered_protocol.T)
let iter f t =
Operation_hash.Table.iter f t
let to_mempool t parsed_cache =
let to_mempool t =
let empty = {
Proto_services.Mempool.applied = [] ;
refused = Operation_hash.Map.empty ;
@ -290,23 +295,20 @@ module Make(Static: STATIC)(Proto: Registered_protocol.T)
op.Operation.proto in
{ Proto.shell = op.shell ; protocol_data } in
Operation_hash.Table.fold
(fun hash result acc ->
match ParsedCache.find_hash_opt parsed_cache hash with
(* XXX this invariant should be better enforced *)
| None | Some (Error _) -> assert false
| Some (Ok op) -> begin
(fun hash (result,raw_op) acc ->
let proto_op = map_op raw_op in
match result with
| Applied _ -> {
acc with
Proto_services.Mempool.applied =
(hash, map_op op.raw)::acc.Proto_services.Mempool.applied
(hash, proto_op)::acc.Proto_services.Mempool.applied
}
| Branch_refused err -> {
acc with
Proto_services.Mempool.branch_refused =
Operation_hash.Map.add
hash
(map_op op.raw,err)
(proto_op,err)
acc.Proto_services.Mempool.branch_refused
}
| Branch_delayed err -> {
@ -314,7 +316,7 @@ module Make(Static: STATIC)(Proto: Registered_protocol.T)
Proto_services.Mempool.branch_delayed =
Operation_hash.Map.add
hash
(map_op op.raw,err)
(proto_op,err)
acc.Proto_services.Mempool.branch_delayed
}
| Refused err -> {
@ -322,12 +324,11 @@ module Make(Static: STATIC)(Proto: Registered_protocol.T)
Proto_services.Mempool.refused =
Operation_hash.Map.add
hash
(map_op op.raw,err)
(proto_op,err)
acc.Proto_services.Mempool.refused
}
| _ -> acc
end)
t empty
) t empty
let clear t = Operation_hash.Table.clear t
@ -489,13 +490,13 @@ module Make(Static: STATIC)(Proto: Registered_protocol.T)
let on_validate w parsed_op =
let state = Worker.state w in
match ValidatedCache.find_opt state.cache parsed_op with
| None | Some (Branch_delayed _) ->
| None | Some ((Branch_delayed _),_) ->
validate_helper w parsed_op >>= fun result ->
ValidatedCache.add state.cache parsed_op result;
ValidatedCache.add state.cache parsed_op (result, parsed_op.raw);
(* operations are notified only the first time *)
notify_helper w result parsed_op.raw ;
Lwt.return result
| Some result -> Lwt.return result
| Some (result,_) -> Lwt.return result
(* worker's handlers *)
let on_request :
@ -588,7 +589,7 @@ module Make(Static: STATIC)(Proto: Registered_protocol.T)
(Proto_services.S.Mempool.pending_operations RPC_path.open_root)
(fun w () () ->
let state = Worker.state w in
RPC_answer.return (ValidatedCache.to_mempool state.cache parsed_cache)
RPC_answer.return (ValidatedCache.to_mempool state.cache)
)
let monitor_rpc_directory : t RPC_directory.t =