1 use crate::fx::FxHashMap;
2 use crate::undo_log::{Rollback, Snapshots, UndoLogs, VecLog};
3 use std::borrow::{Borrow, BorrowMut};
5 use std::marker::PhantomData;
8 pub use crate::undo_log::Snapshot;
13 pub type SnapshotMapStorage<K, V> = SnapshotMap<K, V, FxHashMap<K, V>, ()>;
14 pub type SnapshotMapRef<'a, K, V, L> = SnapshotMap<K, V, &'a mut FxHashMap<K, V>, &'a mut L>;
16 pub struct SnapshotMap<K, V, M = FxHashMap<K, V>, L = VecLog<UndoLog<K, V>>> {
19 _marker: PhantomData<(K, V)>,
22 // HACK(eddyb) manual impl avoids `Default` bounds on `K` and `V`.
23 impl<K, V, M, L> Default for SnapshotMap<K, V, M, L>
28 fn default() -> Self {
29 SnapshotMap { map: Default::default(), undo_log: Default::default(), _marker: PhantomData }
33 pub enum UndoLog<K, V> {
39 impl<K, V, M, L> SnapshotMap<K, V, M, L> {
40 pub fn with_log<L2>(&mut self, undo_log: L2) -> SnapshotMap<K, V, &mut M, L2> {
41 SnapshotMap { map: &mut self.map, undo_log, _marker: PhantomData }
45 impl<K, V, M, L> SnapshotMap<K, V, M, L>
48 M: BorrowMut<FxHashMap<K, V>> + Borrow<FxHashMap<K, V>>,
49 L: UndoLogs<UndoLog<K, V>>,
51 pub fn clear(&mut self) {
52 self.map.borrow_mut().clear();
53 self.undo_log.clear();
56 pub fn insert(&mut self, key: K, value: V) -> bool {
57 match self.map.borrow_mut().insert(key.clone(), value) {
59 self.undo_log.push(UndoLog::Inserted(key));
63 self.undo_log.push(UndoLog::Overwrite(key, old_value));
69 pub fn remove(&mut self, key: K) -> bool {
70 match self.map.borrow_mut().remove(&key) {
72 self.undo_log.push(UndoLog::Overwrite(key, old_value));
79 pub fn get(&self, key: &K) -> Option<&V> {
80 self.map.borrow().get(key)
84 impl<K, V> SnapshotMap<K, V>
88 pub fn snapshot(&mut self) -> Snapshot {
89 self.undo_log.start_snapshot()
92 pub fn commit(&mut self, snapshot: Snapshot) {
93 self.undo_log.commit(snapshot)
96 pub fn rollback_to(&mut self, snapshot: Snapshot) {
97 let map = &mut self.map;
98 self.undo_log.rollback_to(|| map, snapshot)
102 impl<'k, K, V, M, L> ops::Index<&'k K> for SnapshotMap<K, V, M, L>
104 K: Hash + Clone + Eq,
105 M: Borrow<FxHashMap<K, V>>,
108 fn index(&self, key: &'k K) -> &V {
109 &self.map.borrow()[key]
113 impl<K, V, M, L> Rollback<UndoLog<K, V>> for SnapshotMap<K, V, M, L>
116 M: Rollback<UndoLog<K, V>>,
118 fn reverse(&mut self, undo: UndoLog<K, V>) {
119 self.map.reverse(undo)
123 impl<K, V> Rollback<UndoLog<K, V>> for FxHashMap<K, V>
127 fn reverse(&mut self, undo: UndoLog<K, V>) {
129 UndoLog::Inserted(key) => {
133 UndoLog::Overwrite(key, old_value) => {
134 self.insert(key, old_value);
137 UndoLog::Purged => {}