diff options
Diffstat (limited to 'tests/lean/Betree')
| -rw-r--r-- | tests/lean/Betree/Funs.lean | 699 | ||||
| -rw-r--r-- | tests/lean/Betree/FunsExternal.lean | 30 | ||||
| -rw-r--r-- | tests/lean/Betree/FunsExternal_Template.lean | 30 | ||||
| -rw-r--r-- | tests/lean/Betree/Types.lean | 83 | ||||
| -rw-r--r-- | tests/lean/Betree/TypesExternal.lean | 7 | ||||
| -rw-r--r-- | tests/lean/Betree/TypesExternal_Template.lean | 9 |
6 files changed, 858 insertions, 0 deletions
diff --git a/tests/lean/Betree/Funs.lean b/tests/lean/Betree/Funs.lean new file mode 100644 index 00000000..8612ccbc --- /dev/null +++ b/tests/lean/Betree/Funs.lean @@ -0,0 +1,699 @@ +-- THIS FILE WAS AUTOMATICALLY GENERATED BY AENEAS +-- [betree]: function definitions +import Base +import Betree.Types +import Betree.FunsExternal +open Primitives + +namespace betree + +/- [betree::betree::load_internal_node]: + Source: 'src/betree.rs', lines 36:0-36:52 -/ +def betree.load_internal_node + (id : U64) (st : State) : + Result (State × (betree.List (U64 × betree.Message))) + := + betree_utils.load_internal_node id st + +/- [betree::betree::store_internal_node]: + Source: 'src/betree.rs', lines 41:0-41:60 -/ +def betree.store_internal_node + (id : U64) (content : betree.List (U64 × betree.Message)) (st : State) : + Result (State × Unit) + := + betree_utils.store_internal_node id content st + +/- [betree::betree::load_leaf_node]: + Source: 'src/betree.rs', lines 46:0-46:44 -/ +def betree.load_leaf_node + (id : U64) (st : State) : Result (State × (betree.List (U64 × U64))) := + betree_utils.load_leaf_node id st + +/- [betree::betree::store_leaf_node]: + Source: 'src/betree.rs', lines 51:0-51:52 -/ +def betree.store_leaf_node + (id : U64) (content : betree.List (U64 × U64)) (st : State) : + Result (State × Unit) + := + betree_utils.store_leaf_node id content st + +/- [betree::betree::fresh_node_id]: + Source: 'src/betree.rs', lines 55:0-55:48 -/ +def betree.fresh_node_id (counter : U64) : Result (U64 × U64) := + do + let counter1 ← counter + 1#u64 + Result.ok (counter, counter1) + +/- [betree::betree::{betree::betree::NodeIdCounter}::new]: + Source: 'src/betree.rs', lines 206:4-206:20 -/ +def betree.NodeIdCounter.new : Result betree.NodeIdCounter := + Result.ok { next_node_id := 0#u64 } + +/- [betree::betree::{betree::betree::NodeIdCounter}::fresh_id]: + Source: 'src/betree.rs', lines 210:4-210:36 -/ +def betree.NodeIdCounter.fresh_id + (self : betree.NodeIdCounter) : Result (U64 × betree.NodeIdCounter) := + do + let i ← self.next_node_id + 1#u64 + Result.ok (self.next_node_id, { next_node_id := i }) + +/- [betree::betree::upsert_update]: + Source: 'src/betree.rs', lines 234:0-234:70 -/ +def betree.upsert_update + (prev : Option U64) (st : betree.UpsertFunState) : Result U64 := + match prev with + | none => + match st with + | betree.UpsertFunState.Add v => Result.ok v + | betree.UpsertFunState.Sub _ => Result.ok 0#u64 + | some prev1 => + match st with + | betree.UpsertFunState.Add v => + do + let margin ← core_u64_max - prev1 + if margin >= v + then prev1 + v + else Result.ok core_u64_max + | betree.UpsertFunState.Sub v => + if prev1 >= v + then prev1 - v + else Result.ok 0#u64 + +/- [betree::betree::{betree::betree::List<T>#1}::len]: + Source: 'src/betree.rs', lines 276:4-276:24 -/ +divergent def betree.List.len (T : Type) (self : betree.List T) : Result U64 := + match self with + | betree.List.Cons _ tl => do + let i ← betree.List.len T tl + 1#u64 + i + | betree.List.Nil => Result.ok 0#u64 + +/- [betree::betree::{betree::betree::List<T>#1}::split_at]: + Source: 'src/betree.rs', lines 284:4-284:51 -/ +divergent def betree.List.split_at + (T : Type) (self : betree.List T) (n : U64) : + Result ((betree.List T) × (betree.List T)) + := + if n = 0#u64 + then Result.ok (betree.List.Nil, self) + else + match self with + | betree.List.Cons hd tl => + do + let i ← n - 1#u64 + let p ← betree.List.split_at T tl i + let (ls0, ls1) := p + Result.ok (betree.List.Cons hd ls0, ls1) + | betree.List.Nil => Result.fail .panic + +/- [betree::betree::{betree::betree::List<T>#1}::push_front]: + Source: 'src/betree.rs', lines 299:4-299:34 -/ +def betree.List.push_front + (T : Type) (self : betree.List T) (x : T) : Result (betree.List T) := + let (tl, _) := core.mem.replace (betree.List T) self betree.List.Nil + Result.ok (betree.List.Cons x tl) + +/- [betree::betree::{betree::betree::List<T>#1}::pop_front]: + Source: 'src/betree.rs', lines 306:4-306:32 -/ +def betree.List.pop_front + (T : Type) (self : betree.List T) : Result (T × (betree.List T)) := + let (ls, _) := core.mem.replace (betree.List T) self betree.List.Nil + match ls with + | betree.List.Cons x tl => Result.ok (x, tl) + | betree.List.Nil => Result.fail .panic + +/- [betree::betree::{betree::betree::List<T>#1}::hd]: + Source: 'src/betree.rs', lines 318:4-318:22 -/ +def betree.List.hd (T : Type) (self : betree.List T) : Result T := + match self with + | betree.List.Cons hd _ => Result.ok hd + | betree.List.Nil => Result.fail .panic + +/- [betree::betree::{betree::betree::List<(u64, T)>#2}::head_has_key]: + Source: 'src/betree.rs', lines 327:4-327:44 -/ +def betree.ListPairU64T.head_has_key + (T : Type) (self : betree.List (U64 × T)) (key : U64) : Result Bool := + match self with + | betree.List.Cons hd _ => let (i, _) := hd + Result.ok (i = key) + | betree.List.Nil => Result.ok false + +/- [betree::betree::{betree::betree::List<(u64, T)>#2}::partition_at_pivot]: + Source: 'src/betree.rs', lines 339:4-339:73 -/ +divergent def betree.ListPairU64T.partition_at_pivot + (T : Type) (self : betree.List (U64 × T)) (pivot : U64) : + Result ((betree.List (U64 × T)) × (betree.List (U64 × T))) + := + match self with + | betree.List.Cons hd tl => + let (i, t) := hd + if i >= pivot + then Result.ok (betree.List.Nil, betree.List.Cons (i, t) tl) + else + do + let p ← betree.ListPairU64T.partition_at_pivot T tl pivot + let (ls0, ls1) := p + Result.ok (betree.List.Cons (i, t) ls0, ls1) + | betree.List.Nil => Result.ok (betree.List.Nil, betree.List.Nil) + +/- [betree::betree::{betree::betree::Leaf#3}::split]: + Source: 'src/betree.rs', lines 359:4-364:17 -/ +def betree.Leaf.split + (self : betree.Leaf) (content : betree.List (U64 × U64)) + (params : betree.Params) (node_id_cnt : betree.NodeIdCounter) (st : State) : + Result (State × (betree.Internal × betree.NodeIdCounter)) + := + do + let p ← betree.List.split_at (U64 × U64) content params.split_size + let (content0, content1) := p + let p1 ← betree.List.hd (U64 × U64) content1 + let (pivot, _) := p1 + let (id0, node_id_cnt1) ← betree.NodeIdCounter.fresh_id node_id_cnt + let (id1, node_id_cnt2) ← betree.NodeIdCounter.fresh_id node_id_cnt1 + let (st1, _) ← betree.store_leaf_node id0 content0 st + let (st2, _) ← betree.store_leaf_node id1 content1 st1 + let n := betree.Node.Leaf { id := id0, size := params.split_size } + let n1 := betree.Node.Leaf { id := id1, size := params.split_size } + Result.ok (st2, (betree.Internal.mk self.id pivot n n1, node_id_cnt2)) + +/- [betree::betree::{betree::betree::Node#5}::lookup_first_message_for_key]: + Source: 'src/betree.rs', lines 789:4-792:34 -/ +divergent def betree.Node.lookup_first_message_for_key + (key : U64) (msgs : betree.List (U64 × betree.Message)) : + Result ((betree.List (U64 × betree.Message)) × (betree.List (U64 × + betree.Message) → Result (betree.List (U64 × betree.Message)))) + := + match msgs with + | betree.List.Cons x next_msgs => + let (i, m) := x + if i >= key + then Result.ok (betree.List.Cons (i, m) next_msgs, Result.ok) + else + do + let (l, lookup_first_message_for_key_back) ← + betree.Node.lookup_first_message_for_key key next_msgs + let back := + fun ret => + do + let next_msgs1 ← lookup_first_message_for_key_back ret + Result.ok (betree.List.Cons (i, m) next_msgs1) + Result.ok (l, back) + | betree.List.Nil => Result.ok (betree.List.Nil, Result.ok) + +/- [betree::betree::{betree::betree::Node#5}::lookup_in_bindings]: + Source: 'src/betree.rs', lines 636:4-636:80 -/ +divergent def betree.Node.lookup_in_bindings + (key : U64) (bindings : betree.List (U64 × U64)) : Result (Option U64) := + match bindings with + | betree.List.Cons hd tl => + let (i, i1) := hd + if i = key + then Result.ok (some i1) + else + if i > key + then Result.ok none + else betree.Node.lookup_in_bindings key tl + | betree.List.Nil => Result.ok none + +/- [betree::betree::{betree::betree::Node#5}::apply_upserts]: + Source: 'src/betree.rs', lines 819:4-819:90 -/ +divergent def betree.Node.apply_upserts + (msgs : betree.List (U64 × betree.Message)) (prev : Option U64) (key : U64) + : + Result (U64 × (betree.List (U64 × betree.Message))) + := + do + let b ← betree.ListPairU64T.head_has_key betree.Message msgs key + if b + then + do + let (msg, msgs1) ← betree.List.pop_front (U64 × betree.Message) msgs + let (_, m) := msg + match m with + | betree.Message.Insert _ => Result.fail .panic + | betree.Message.Delete => Result.fail .panic + | betree.Message.Upsert s => + do + let v ← betree.upsert_update prev s + betree.Node.apply_upserts msgs1 (some v) key + else + do + let v ← core.option.Option.unwrap U64 prev + let msgs1 ← + betree.List.push_front (U64 × betree.Message) msgs (key, + betree.Message.Insert v) + Result.ok (v, msgs1) + +/- [betree::betree::{betree::betree::Internal#4}::lookup_in_children]: + Source: 'src/betree.rs', lines 395:4-395:63 -/ +mutual divergent def betree.Internal.lookup_in_children + (self : betree.Internal) (key : U64) (st : State) : + Result (State × ((Option U64) × betree.Internal)) + := + if key < self.pivot + then + do + let (st1, (o, n)) ← betree.Node.lookup self.left key st + Result.ok (st1, (o, betree.Internal.mk self.id self.pivot n self.right)) + else + do + let (st1, (o, n)) ← betree.Node.lookup self.right key st + Result.ok (st1, (o, betree.Internal.mk self.id self.pivot self.left n)) + +/- [betree::betree::{betree::betree::Node#5}::lookup]: + Source: 'src/betree.rs', lines 709:4-709:58 -/ +divergent def betree.Node.lookup + (self : betree.Node) (key : U64) (st : State) : + Result (State × ((Option U64) × betree.Node)) + := + match self with + | betree.Node.Internal node => + do + let (st1, msgs) ← betree.load_internal_node node.id st + let (pending, lookup_first_message_for_key_back) ← + betree.Node.lookup_first_message_for_key key msgs + match pending with + | betree.List.Cons p l => + let (k, msg) := p + if k != key + then + do + let (st2, (o, node1)) ← + betree.Internal.lookup_in_children node key st1 + let _ ← + lookup_first_message_for_key_back (betree.List.Cons (k, msg) l) + Result.ok (st2, (o, betree.Node.Internal node1)) + else + match msg with + | betree.Message.Insert v => + do + let _ ← + lookup_first_message_for_key_back (betree.List.Cons (k, + betree.Message.Insert v) l) + Result.ok (st1, (some v, betree.Node.Internal node)) + | betree.Message.Delete => + do + let _ ← + lookup_first_message_for_key_back (betree.List.Cons (k, + betree.Message.Delete) l) + Result.ok (st1, (none, betree.Node.Internal node)) + | betree.Message.Upsert ufs => + do + let (st2, (v, node1)) ← + betree.Internal.lookup_in_children node key st1 + let (v1, pending1) ← + betree.Node.apply_upserts (betree.List.Cons (k, + betree.Message.Upsert ufs) l) v key + let msgs1 ← lookup_first_message_for_key_back pending1 + let (st3, _) ← betree.store_internal_node node1.id msgs1 st2 + Result.ok (st3, (some v1, betree.Node.Internal node1)) + | betree.List.Nil => + do + let (st2, (o, node1)) ← betree.Internal.lookup_in_children node key st1 + let _ ← lookup_first_message_for_key_back betree.List.Nil + Result.ok (st2, (o, betree.Node.Internal node1)) + | betree.Node.Leaf node => + do + let (st1, bindings) ← betree.load_leaf_node node.id st + let o ← betree.Node.lookup_in_bindings key bindings + Result.ok (st1, (o, betree.Node.Leaf node)) + +end + +/- [betree::betree::{betree::betree::Node#5}::filter_messages_for_key]: + Source: 'src/betree.rs', lines 674:4-674:77 -/ +divergent def betree.Node.filter_messages_for_key + (key : U64) (msgs : betree.List (U64 × betree.Message)) : + Result (betree.List (U64 × betree.Message)) + := + match msgs with + | betree.List.Cons p l => + let (k, m) := p + if k = key + then + do + let (_, msgs1) ← + betree.List.pop_front (U64 × betree.Message) (betree.List.Cons (k, m) + l) + betree.Node.filter_messages_for_key key msgs1 + else Result.ok (betree.List.Cons (k, m) l) + | betree.List.Nil => Result.ok betree.List.Nil + +/- [betree::betree::{betree::betree::Node#5}::lookup_first_message_after_key]: + Source: 'src/betree.rs', lines 689:4-692:34 -/ +divergent def betree.Node.lookup_first_message_after_key + (key : U64) (msgs : betree.List (U64 × betree.Message)) : + Result ((betree.List (U64 × betree.Message)) × (betree.List (U64 × + betree.Message) → Result (betree.List (U64 × betree.Message)))) + := + match msgs with + | betree.List.Cons p next_msgs => + let (k, m) := p + if k = key + then + do + let (l, lookup_first_message_after_key_back) ← + betree.Node.lookup_first_message_after_key key next_msgs + let back := + fun ret => + do + let next_msgs1 ← lookup_first_message_after_key_back ret + Result.ok (betree.List.Cons (k, m) next_msgs1) + Result.ok (l, back) + else Result.ok (betree.List.Cons (k, m) next_msgs, Result.ok) + | betree.List.Nil => Result.ok (betree.List.Nil, Result.ok) + +/- [betree::betree::{betree::betree::Node#5}::apply_to_internal]: + Source: 'src/betree.rs', lines 521:4-521:89 -/ +def betree.Node.apply_to_internal + (msgs : betree.List (U64 × betree.Message)) (key : U64) + (new_msg : betree.Message) : + Result (betree.List (U64 × betree.Message)) + := + do + let (msgs1, lookup_first_message_for_key_back) ← + betree.Node.lookup_first_message_for_key key msgs + let b ← betree.ListPairU64T.head_has_key betree.Message msgs1 key + if b + then + match new_msg with + | betree.Message.Insert i => + do + let msgs2 ← betree.Node.filter_messages_for_key key msgs1 + let msgs3 ← + betree.List.push_front (U64 × betree.Message) msgs2 (key, + betree.Message.Insert i) + lookup_first_message_for_key_back msgs3 + | betree.Message.Delete => + do + let msgs2 ← betree.Node.filter_messages_for_key key msgs1 + let msgs3 ← + betree.List.push_front (U64 × betree.Message) msgs2 (key, + betree.Message.Delete) + lookup_first_message_for_key_back msgs3 + | betree.Message.Upsert s => + do + let p ← betree.List.hd (U64 × betree.Message) msgs1 + let (_, m) := p + match m with + | betree.Message.Insert prev => + do + let v ← betree.upsert_update (some prev) s + let (_, msgs2) ← betree.List.pop_front (U64 × betree.Message) msgs1 + let msgs3 ← + betree.List.push_front (U64 × betree.Message) msgs2 (key, + betree.Message.Insert v) + lookup_first_message_for_key_back msgs3 + | betree.Message.Delete => + do + let (_, msgs2) ← betree.List.pop_front (U64 × betree.Message) msgs1 + let v ← betree.upsert_update none s + let msgs3 ← + betree.List.push_front (U64 × betree.Message) msgs2 (key, + betree.Message.Insert v) + lookup_first_message_for_key_back msgs3 + | betree.Message.Upsert _ => + do + let (msgs2, lookup_first_message_after_key_back) ← + betree.Node.lookup_first_message_after_key key msgs1 + let msgs3 ← + betree.List.push_front (U64 × betree.Message) msgs2 (key, + betree.Message.Upsert s) + let msgs4 ← lookup_first_message_after_key_back msgs3 + lookup_first_message_for_key_back msgs4 + else + do + let msgs2 ← + betree.List.push_front (U64 × betree.Message) msgs1 (key, new_msg) + lookup_first_message_for_key_back msgs2 + +/- [betree::betree::{betree::betree::Node#5}::apply_messages_to_internal]: + Source: 'src/betree.rs', lines 502:4-505:5 -/ +divergent def betree.Node.apply_messages_to_internal + (msgs : betree.List (U64 × betree.Message)) + (new_msgs : betree.List (U64 × betree.Message)) : + Result (betree.List (U64 × betree.Message)) + := + match new_msgs with + | betree.List.Cons new_msg new_msgs_tl => + do + let (i, m) := new_msg + let msgs1 ← betree.Node.apply_to_internal msgs i m + betree.Node.apply_messages_to_internal msgs1 new_msgs_tl + | betree.List.Nil => Result.ok msgs + +/- [betree::betree::{betree::betree::Node#5}::lookup_mut_in_bindings]: + Source: 'src/betree.rs', lines 653:4-656:32 -/ +divergent def betree.Node.lookup_mut_in_bindings + (key : U64) (bindings : betree.List (U64 × U64)) : + Result ((betree.List (U64 × U64)) × (betree.List (U64 × U64) → Result + (betree.List (U64 × U64)))) + := + match bindings with + | betree.List.Cons hd tl => + let (i, i1) := hd + if i >= key + then Result.ok (betree.List.Cons (i, i1) tl, Result.ok) + else + do + let (l, lookup_mut_in_bindings_back) ← + betree.Node.lookup_mut_in_bindings key tl + let back := + fun ret => + do + let tl1 ← lookup_mut_in_bindings_back ret + Result.ok (betree.List.Cons (i, i1) tl1) + Result.ok (l, back) + | betree.List.Nil => Result.ok (betree.List.Nil, Result.ok) + +/- [betree::betree::{betree::betree::Node#5}::apply_to_leaf]: + Source: 'src/betree.rs', lines 460:4-460:87 -/ +def betree.Node.apply_to_leaf + (bindings : betree.List (U64 × U64)) (key : U64) (new_msg : betree.Message) + : + Result (betree.List (U64 × U64)) + := + do + let (bindings1, lookup_mut_in_bindings_back) ← + betree.Node.lookup_mut_in_bindings key bindings + let b ← betree.ListPairU64T.head_has_key U64 bindings1 key + if b + then + do + let (hd, bindings2) ← betree.List.pop_front (U64 × U64) bindings1 + match new_msg with + | betree.Message.Insert v => + do + let bindings3 ← betree.List.push_front (U64 × U64) bindings2 (key, v) + lookup_mut_in_bindings_back bindings3 + | betree.Message.Delete => lookup_mut_in_bindings_back bindings2 + | betree.Message.Upsert s => + do + let (_, i) := hd + let v ← betree.upsert_update (some i) s + let bindings3 ← betree.List.push_front (U64 × U64) bindings2 (key, v) + lookup_mut_in_bindings_back bindings3 + else + match new_msg with + | betree.Message.Insert v => + do + let bindings2 ← betree.List.push_front (U64 × U64) bindings1 (key, v) + lookup_mut_in_bindings_back bindings2 + | betree.Message.Delete => lookup_mut_in_bindings_back bindings1 + | betree.Message.Upsert s => + do + let v ← betree.upsert_update none s + let bindings2 ← betree.List.push_front (U64 × U64) bindings1 (key, v) + lookup_mut_in_bindings_back bindings2 + +/- [betree::betree::{betree::betree::Node#5}::apply_messages_to_leaf]: + Source: 'src/betree.rs', lines 444:4-447:5 -/ +divergent def betree.Node.apply_messages_to_leaf + (bindings : betree.List (U64 × U64)) + (new_msgs : betree.List (U64 × betree.Message)) : + Result (betree.List (U64 × U64)) + := + match new_msgs with + | betree.List.Cons new_msg new_msgs_tl => + do + let (i, m) := new_msg + let bindings1 ← betree.Node.apply_to_leaf bindings i m + betree.Node.apply_messages_to_leaf bindings1 new_msgs_tl + | betree.List.Nil => Result.ok bindings + +/- [betree::betree::{betree::betree::Internal#4}::flush]: + Source: 'src/betree.rs', lines 410:4-415:26 -/ +mutual divergent def betree.Internal.flush + (self : betree.Internal) (params : betree.Params) + (node_id_cnt : betree.NodeIdCounter) + (content : betree.List (U64 × betree.Message)) (st : State) : + Result (State × ((betree.List (U64 × betree.Message)) × (betree.Internal + × betree.NodeIdCounter))) + := + do + let p ← + betree.ListPairU64T.partition_at_pivot betree.Message content self.pivot + let (msgs_left, msgs_right) := p + let len_left ← betree.List.len (U64 × betree.Message) msgs_left + if len_left >= params.min_flush_size + then + do + let (st1, p1) ← + betree.Node.apply_messages self.left params node_id_cnt msgs_left st + let (n, node_id_cnt1) := p1 + let len_right ← betree.List.len (U64 × betree.Message) msgs_right + if len_right >= params.min_flush_size + then + do + let (st2, p2) ← + betree.Node.apply_messages self.right params node_id_cnt1 msgs_right + st1 + let (n1, node_id_cnt2) := p2 + Result.ok (st2, (betree.List.Nil, (betree.Internal.mk self.id self.pivot + n n1, node_id_cnt2))) + else + Result.ok (st1, (msgs_right, (betree.Internal.mk self.id self.pivot n + self.right, node_id_cnt1))) + else + do + let (st1, p1) ← + betree.Node.apply_messages self.right params node_id_cnt msgs_right st + let (n, node_id_cnt1) := p1 + Result.ok (st1, (msgs_left, (betree.Internal.mk self.id self.pivot + self.left n, node_id_cnt1))) + +/- [betree::betree::{betree::betree::Node#5}::apply_messages]: + Source: 'src/betree.rs', lines 588:4-593:5 -/ +divergent def betree.Node.apply_messages + (self : betree.Node) (params : betree.Params) + (node_id_cnt : betree.NodeIdCounter) + (msgs : betree.List (U64 × betree.Message)) (st : State) : + Result (State × (betree.Node × betree.NodeIdCounter)) + := + match self with + | betree.Node.Internal node => + do + let (st1, content) ← betree.load_internal_node node.id st + let content1 ← betree.Node.apply_messages_to_internal content msgs + let num_msgs ← betree.List.len (U64 × betree.Message) content1 + if num_msgs >= params.min_flush_size + then + do + let (st2, (content2, p)) ← + betree.Internal.flush node params node_id_cnt content1 st1 + let (node1, node_id_cnt1) := p + let (st3, _) ← betree.store_internal_node node1.id content2 st2 + Result.ok (st3, (betree.Node.Internal node1, node_id_cnt1)) + else + do + let (st2, _) ← betree.store_internal_node node.id content1 st1 + Result.ok (st2, (betree.Node.Internal node, node_id_cnt)) + | betree.Node.Leaf node => + do + let (st1, content) ← betree.load_leaf_node node.id st + let content1 ← betree.Node.apply_messages_to_leaf content msgs + let len ← betree.List.len (U64 × U64) content1 + let i ← 2#u64 * params.split_size + if len >= i + then + do + let (st2, (new_node, node_id_cnt1)) ← + betree.Leaf.split node content1 params node_id_cnt st1 + let (st3, _) ← betree.store_leaf_node node.id betree.List.Nil st2 + Result.ok (st3, (betree.Node.Internal new_node, node_id_cnt1)) + else + do + let (st2, _) ← betree.store_leaf_node node.id content1 st1 + Result.ok (st2, (betree.Node.Leaf { node with size := len }, + node_id_cnt)) + +end + +/- [betree::betree::{betree::betree::Node#5}::apply]: + Source: 'src/betree.rs', lines 576:4-582:5 -/ +def betree.Node.apply + (self : betree.Node) (params : betree.Params) + (node_id_cnt : betree.NodeIdCounter) (key : U64) (new_msg : betree.Message) + (st : State) : + Result (State × (betree.Node × betree.NodeIdCounter)) + := + do + let (st1, p) ← + betree.Node.apply_messages self params node_id_cnt (betree.List.Cons (key, + new_msg) betree.List.Nil) st + let (self1, node_id_cnt1) := p + Result.ok (st1, (self1, node_id_cnt1)) + +/- [betree::betree::{betree::betree::BeTree#6}::new]: + Source: 'src/betree.rs', lines 849:4-849:60 -/ +def betree.BeTree.new + (min_flush_size : U64) (split_size : U64) (st : State) : + Result (State × betree.BeTree) + := + do + let node_id_cnt ← betree.NodeIdCounter.new + let (id, node_id_cnt1) ← betree.NodeIdCounter.fresh_id node_id_cnt + let (st1, _) ← betree.store_leaf_node id betree.List.Nil st + Result.ok (st1, + { + params := { min_flush_size := min_flush_size, split_size := split_size }, + node_id_cnt := node_id_cnt1, + root := (betree.Node.Leaf { id := id, size := 0#u64 }) + }) + +/- [betree::betree::{betree::betree::BeTree#6}::apply]: + Source: 'src/betree.rs', lines 868:4-868:47 -/ +def betree.BeTree.apply + (self : betree.BeTree) (key : U64) (msg : betree.Message) (st : State) : + Result (State × betree.BeTree) + := + do + let (st1, p) ← + betree.Node.apply self.root self.params self.node_id_cnt key msg st + let (n, nic) := p + Result.ok (st1, { self with node_id_cnt := nic, root := n }) + +/- [betree::betree::{betree::betree::BeTree#6}::insert]: + Source: 'src/betree.rs', lines 874:4-874:52 -/ +def betree.BeTree.insert + (self : betree.BeTree) (key : U64) (value : U64) (st : State) : + Result (State × betree.BeTree) + := + betree.BeTree.apply self key (betree.Message.Insert value) st + +/- [betree::betree::{betree::betree::BeTree#6}::delete]: + Source: 'src/betree.rs', lines 880:4-880:38 -/ +def betree.BeTree.delete + (self : betree.BeTree) (key : U64) (st : State) : + Result (State × betree.BeTree) + := + betree.BeTree.apply self key betree.Message.Delete st + +/- [betree::betree::{betree::betree::BeTree#6}::upsert]: + Source: 'src/betree.rs', lines 886:4-886:59 -/ +def betree.BeTree.upsert + (self : betree.BeTree) (key : U64) (upd : betree.UpsertFunState) (st : State) + : + Result (State × betree.BeTree) + := + betree.BeTree.apply self key (betree.Message.Upsert upd) st + +/- [betree::betree::{betree::betree::BeTree#6}::lookup]: + Source: 'src/betree.rs', lines 895:4-895:62 -/ +def betree.BeTree.lookup + (self : betree.BeTree) (key : U64) (st : State) : + Result (State × ((Option U64) × betree.BeTree)) + := + do + let (st1, (o, n)) ← betree.Node.lookup self.root key st + Result.ok (st1, (o, { self with root := n })) + +/- [betree::main]: + Source: 'src/main.rs', lines 4:0-4:9 -/ +def main : Result Unit := + Result.ok () + +/- Unit test for [betree::main] -/ +#assert (main == Result.ok ()) + +end betree diff --git a/tests/lean/Betree/FunsExternal.lean b/tests/lean/Betree/FunsExternal.lean new file mode 100644 index 00000000..859cbd68 --- /dev/null +++ b/tests/lean/Betree/FunsExternal.lean @@ -0,0 +1,30 @@ +-- [betree]: external functions. +import Base +import Betree.Types +open Primitives +open betree + +-- TODO: fill those bodies + +/- [betree::betree_utils::load_internal_node] -/ +def betree_utils.load_internal_node + : + U64 → State → Result (State × (betree.List (U64 × betree.Message))) := + fun _ _ => .fail .panic + +/- [betree::betree_utils::store_internal_node] -/ +def betree_utils.store_internal_node + : + U64 → betree.List (U64 × betree.Message) → State → Result (State + × Unit) := + fun _ _ _ => .fail .panic + +/- [betree::betree_utils::load_leaf_node] -/ +def betree_utils.load_leaf_node + : U64 → State → Result (State × (betree.List (U64 × U64))) := + fun _ _ => .fail .panic + +/- [betree::betree_utils::store_leaf_node] -/ +def betree_utils.store_leaf_node + : U64 → betree.List (U64 × U64) → State → Result (State × Unit) := + fun _ _ _ => .fail .panic diff --git a/tests/lean/Betree/FunsExternal_Template.lean b/tests/lean/Betree/FunsExternal_Template.lean new file mode 100644 index 00000000..014f0d83 --- /dev/null +++ b/tests/lean/Betree/FunsExternal_Template.lean @@ -0,0 +1,30 @@ +-- THIS FILE WAS AUTOMATICALLY GENERATED BY AENEAS +-- [betree]: external functions. +-- This is a template file: rename it to "FunsExternal.lean" and fill the holes. +import Base +import Betree.Types +open Primitives +open betree + +/- [betree::betree_utils::load_internal_node]: + Source: 'src/betree_utils.rs', lines 98:0-98:63 -/ +axiom betree_utils.load_internal_node + : U64 → State → Result (State × (betree.List (U64 × betree.Message))) + +/- [betree::betree_utils::store_internal_node]: + Source: 'src/betree_utils.rs', lines 115:0-115:71 -/ +axiom betree_utils.store_internal_node + : + U64 → betree.List (U64 × betree.Message) → State → Result (State × + Unit) + +/- [betree::betree_utils::load_leaf_node]: + Source: 'src/betree_utils.rs', lines 132:0-132:55 -/ +axiom betree_utils.load_leaf_node + : U64 → State → Result (State × (betree.List (U64 × U64))) + +/- [betree::betree_utils::store_leaf_node]: + Source: 'src/betree_utils.rs', lines 145:0-145:63 -/ +axiom betree_utils.store_leaf_node + : U64 → betree.List (U64 × U64) → State → Result (State × Unit) + diff --git a/tests/lean/Betree/Types.lean b/tests/lean/Betree/Types.lean new file mode 100644 index 00000000..3b46c00c --- /dev/null +++ b/tests/lean/Betree/Types.lean @@ -0,0 +1,83 @@ +-- THIS FILE WAS AUTOMATICALLY GENERATED BY AENEAS +-- [betree]: type definitions +import Base +import Betree.TypesExternal +open Primitives + +namespace betree + +/- [betree::betree::List] + Source: 'src/betree.rs', lines 17:0-17:23 -/ +inductive betree.List (T : Type) := +| Cons : T → betree.List T → betree.List T +| Nil : betree.List T + +/- [betree::betree::UpsertFunState] + Source: 'src/betree.rs', lines 63:0-63:23 -/ +inductive betree.UpsertFunState := +| Add : U64 → betree.UpsertFunState +| Sub : U64 → betree.UpsertFunState + +/- [betree::betree::Message] + Source: 'src/betree.rs', lines 69:0-69:23 -/ +inductive betree.Message := +| Insert : U64 → betree.Message +| Delete : betree.Message +| Upsert : betree.UpsertFunState → betree.Message + +/- [betree::betree::Leaf] + Source: 'src/betree.rs', lines 167:0-167:11 -/ +structure betree.Leaf where + id : U64 + size : U64 + +mutual + +/- [betree::betree::Internal] + Source: 'src/betree.rs', lines 156:0-156:15 -/ +inductive betree.Internal := +| mk : U64 → U64 → betree.Node → betree.Node → betree.Internal + +/- [betree::betree::Node] + Source: 'src/betree.rs', lines 179:0-179:9 -/ +inductive betree.Node := +| Internal : betree.Internal → betree.Node +| Leaf : betree.Leaf → betree.Node + +end + +@[simp, reducible] +def betree.Internal.id (x : betree.Internal) := + match x with | betree.Internal.mk x1 _ _ _ => x1 + +@[simp, reducible] +def betree.Internal.pivot (x : betree.Internal) := + match x with | betree.Internal.mk _ x1 _ _ => x1 + +@[simp, reducible] +def betree.Internal.left (x : betree.Internal) := + match x with | betree.Internal.mk _ _ x1 _ => x1 + +@[simp, reducible] +def betree.Internal.right (x : betree.Internal) := + match x with | betree.Internal.mk _ _ _ x1 => x1 + +/- [betree::betree::Params] + Source: 'src/betree.rs', lines 187:0-187:13 -/ +structure betree.Params where + min_flush_size : U64 + split_size : U64 + +/- [betree::betree::NodeIdCounter] + Source: 'src/betree.rs', lines 201:0-201:20 -/ +structure betree.NodeIdCounter where + next_node_id : U64 + +/- [betree::betree::BeTree] + Source: 'src/betree.rs', lines 218:0-218:17 -/ +structure betree.BeTree where + params : betree.Params + node_id_cnt : betree.NodeIdCounter + root : betree.Node + +end betree diff --git a/tests/lean/Betree/TypesExternal.lean b/tests/lean/Betree/TypesExternal.lean new file mode 100644 index 00000000..34170271 --- /dev/null +++ b/tests/lean/Betree/TypesExternal.lean @@ -0,0 +1,7 @@ +-- [betree]: external types. +import Base +open Primitives + +/- The state type used in the state-error monad -/ +axiom State : Type + diff --git a/tests/lean/Betree/TypesExternal_Template.lean b/tests/lean/Betree/TypesExternal_Template.lean new file mode 100644 index 00000000..12fce657 --- /dev/null +++ b/tests/lean/Betree/TypesExternal_Template.lean @@ -0,0 +1,9 @@ +-- THIS FILE WAS AUTOMATICALLY GENERATED BY AENEAS +-- [betree]: external types. +-- This is a template file: rename it to "TypesExternal.lean" and fill the holes. +import Base +open Primitives + +/- The state type used in the state-error monad -/ +axiom State : Type + |