1 files changed, 709 insertions, 0 deletions
diff --git a/third_party/rust/rkv/src/store/integermulti.rs b/third_party/rust/rkv/src/store/integermulti.rs
new file mode 100644
index 0000000000..71dd0aae04
--- /dev/null
+++ b/third_party/rust/rkv/src/store/integermulti.rs
@@ -0,0 +1,709 @@
+// Copyright 2018 Mozilla
+//
+// Licensed under the Apache License, Version 2.0 (the "License"); you may not use
+// this file except in compliance with the License. You may obtain a copy of the
+// License at http://www.apache.org/licenses/LICENSE-2.0
+// Unless required by applicable law or agreed to in writing, software distributed
+// under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
+// CONDITIONS OF ANY KIND, either express or implied. See the License for the
+// specific language governing permissions and limitations under the License.
+
+use std::marker::PhantomData;
+
+use crate::{
+    backend::{BackendDatabase, BackendIter, BackendRoCursor, BackendRwTransaction},
+    error::StoreError,
+    readwrite::{Readable, Writer},
+    store::{
+        keys::{Key, PrimitiveInt},
+        multi::{Iter, MultiStore},
+    },
+    value::Value,
+};
+
+type EmptyResult = Result<(), StoreError>;
+
+#[derive(Debug, Eq, PartialEq, Copy, Clone)]
+pub struct MultiIntegerStore<D, K> {
+    inner: MultiStore<D>,
+    phantom: PhantomData<K>,
+}
+
+impl<D, K> MultiIntegerStore<D, K>
+where
+    D: BackendDatabase,
+    K: PrimitiveInt,
+{
+    pub(crate) fn new(db: D) -> MultiIntegerStore<D, K> {
+        MultiIntegerStore {
+            inner: MultiStore::new(db),
+            phantom: PhantomData,
+        }
+    }
+
+    pub fn get<'r, R, I, C>(&self, reader: &'r R, k: K) -> Result<Iter<'r, I>, StoreError>
+    where
+        R: Readable<'r, Database = D, RoCursor = C>,
+        I: BackendIter<'r>,
+        C: BackendRoCursor<'r, Iter = I>,
+        K: 'r,
+    {
+        self.inner.get(reader, Key::new(&k)?)
+    }
+
+    pub fn get_first<'r, R>(&self, reader: &'r R, k: K) -> Result<Option<Value<'r>>, StoreError>
+    where
+        R: Readable<'r, Database = D>,
+    {
+        self.inner.get_first(reader, Key::new(&k)?)
+    }
+
+    pub fn put<T>(&self, writer: &mut Writer<T>, k: K, v: &Value) -> EmptyResult
+    where
+        T: BackendRwTransaction<Database = D>,
+    {
+        self.inner.put(writer, Key::new(&k)?, v)
+    }
+
+    pub fn put_with_flags<T>(
+        &self,
+        writer: &mut Writer<T>,
+        k: K,
+        v: &Value,
+        flags: T::Flags,
+    ) -> EmptyResult
+    where
+        T: BackendRwTransaction<Database = D>,
+    {
+        self.inner.put_with_flags(writer, Key::new(&k)?, v, flags)
+    }
+
+    pub fn delete_all<T>(&self, writer: &mut Writer<T>, k: K) -> EmptyResult
+    where
+        T: BackendRwTransaction<Database = D>,
+    {
+        self.inner.delete_all(writer, Key::new(&k)?)
+    }
+
+    pub fn delete<T>(&self, writer: &mut Writer<T>, k: K, v: &Value) -> EmptyResult
+    where
+        T: BackendRwTransaction<Database = D>,
+    {
+        self.inner.delete(writer, Key::new(&k)?, v)
+    }
+
+    pub fn clear<T>(&self, writer: &mut Writer<T>) -> EmptyResult
+    where
+        T: BackendRwTransaction<Database = D>,
+    {
+        self.inner.clear(writer)
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use crate::*;
+
+    use std::fs;
+
+    use tempfile::Builder;
+
+    #[test]
+    fn test_integer_keys() {
+        let root = Builder::new()
+            .prefix("test_integer_keys")
+            .tempdir()
+            .expect("tempdir");
+        fs::create_dir_all(root.path()).expect("dir created");
+
+        let k = Rkv::new::<backend::SafeMode>(root.path()).expect("new succeeded");
+        let s = k
+            .open_multi_integer("s", StoreOptions::create())
+            .expect("open");
+
+        macro_rules! test_integer_keys {
+            ($type:ty, $key:expr) => {{
+                let mut writer = k.write().expect("writer");
+
+                s.put(&mut writer, $key, &Value::Str("hello!"))
+                    .expect("write");
+                assert_eq!(
+                    s.get_first(&writer, $key).expect("read"),
+                    Some(Value::Str("hello!"))
+                );
+                writer.commit().expect("committed");
+
+                let reader = k.read().expect("reader");
+                assert_eq!(
+                    s.get_first(&reader, $key).expect("read"),
+                    Some(Value::Str("hello!"))
+                );
+            }};
+        }
+
+        test_integer_keys!(u32, std::u32::MIN);
+        test_integer_keys!(u32, std::u32::MAX);
+    }
+
+    #[test]
+    fn test_clear() {
+        let root = Builder::new()
+            .prefix("test_multi_integer_clear")
+            .tempdir()
+            .expect("tempdir");
+        fs::create_dir_all(root.path()).expect("dir created");
+
+        let k = Rkv::new::<backend::SafeMode>(root.path()).expect("new succeeded");
+        let s = k
+            .open_multi_integer("s", StoreOptions::create())
+            .expect("open");
+
+        {
+            let mut writer = k.write().expect("writer");
+            s.put(&mut writer, 1, &Value::Str("hello!")).expect("write");
+            s.put(&mut writer, 1, &Value::Str("hello1!"))
+                .expect("write");
+            s.put(&mut writer, 2, &Value::Str("hello!")).expect("write");
+            assert_eq!(
+                s.get_first(&writer, 1).expect("read"),
+                Some(Value::Str("hello!"))
+            );
+            assert_eq!(
+                s.get_first(&writer, 2).expect("read"),
+                Some(Value::Str("hello!"))
+            );
+            assert_eq!(s.get_first(&writer, 3).expect("read"), None);
+            writer.commit().expect("committed");
+        }
+
+        {
+            let mut writer = k.write().expect("writer");
+            s.clear(&mut writer).expect("cleared");
+            writer.commit().expect("committed");
+
+            let reader = k.read().expect("reader");
+            assert_eq!(s.get_first(&reader, 1).expect("read"), None);
+            assert_eq!(s.get_first(&reader, 2).expect("read"), None);
+            assert_eq!(s.get_first(&reader, 3).expect("read"), None);
+        }
+    }
+
+    #[test]
+    fn test_dup() {
+        let root = Builder::new()
+            .prefix("test_multi_integer_dup")
+            .tempdir()
+            .expect("tempdir");
+        fs::create_dir_all(root.path()).expect("dir created");
+
+        let k = Rkv::new::<backend::SafeMode>(root.path()).expect("new succeeded");
+        let s = k
+            .open_multi_integer("s", StoreOptions::create())
+            .expect("open");
+
+        {
+            let mut writer = k.write().expect("writer");
+            s.put(&mut writer, 1, &Value::Str("hello!")).expect("write");
+            s.put(&mut writer, 1, &Value::Str("hello!")).expect("write");
+            s.put(&mut writer, 1, &Value::Str("hello1!"))
+                .expect("write");
+            assert_eq!(
+                s.get_first(&writer, 1).expect("read"),
+                Some(Value::Str("hello!"))
+            );
+            assert_eq!(s.get_first(&writer, 2).expect("read"), None);
+            assert_eq!(s.get_first(&writer, 3).expect("read"), None);
+            writer.commit().expect("committed");
+        }
+
+        {
+            let mut writer = k.write().expect("writer");
+            s.clear(&mut writer).expect("cleared");
+            writer.commit().expect("committed");
+
+            let reader = k.read().expect("reader");
+            assert_eq!(s.get_first(&reader, 1).expect("read"), None);
+            assert_eq!(s.get_first(&reader, 2).expect("read"), None);
+            assert_eq!(s.get_first(&reader, 3).expect("read"), None);
+        }
+    }
+
+    #[test]
+    fn test_dup_2() {
+        let root = Builder::new()
+            .prefix("test_multi_integer_dup")
+            .tempdir()
+            .expect("tempdir");
+        fs::create_dir_all(root.path()).expect("dir created");
+
+        let k = Rkv::new::<backend::SafeMode>(root.path()).expect("new succeeded");
+        let s = k
+            .open_multi_integer("s", StoreOptions::create())
+            .expect("open");
+
+        {
+            let mut writer = k.write().expect("writer");
+            s.put(&mut writer, 1, &Value::Str("hello!")).expect("write");
+            s.put(&mut writer, 1, &Value::Str("hello!")).expect("write");
+            s.put(&mut writer, 1, &Value::Str("hello1!"))
+                .expect("write");
+
+            let mut iter = s.get(&writer, 1).expect("read");
+            assert_eq!(
+                iter.next().expect("first").expect("ok").1,
+                Value::Str("hello!")
+            );
+            assert_eq!(
+                iter.next().expect("second").expect("ok").1,
+                Value::Str("hello1!")
+            );
+            assert!(iter.next().is_none());
+        }
+    }
+
+    #[test]
+    fn test_del() {
+        let root = Builder::new()
+            .prefix("test_multi_integer_dup")
+            .tempdir()
+            .expect("tempdir");
+        fs::create_dir_all(root.path()).expect("dir created");
+
+        let k = Rkv::new::<backend::SafeMode>(root.path()).expect("new succeeded");
+        let s = k
+            .open_multi_integer("s", StoreOptions::create())
+            .expect("open");
+
+        {
+            let mut writer = k.write().expect("writer");
+            s.put(&mut writer, 1, &Value::Str("hello!")).expect("write");
+            s.put(&mut writer, 1, &Value::Str("hello!")).expect("write");
+            s.put(&mut writer, 1, &Value::Str("hello1!"))
+                .expect("write");
+            {
+                let mut iter = s.get(&writer, 1).expect("read");
+                assert_eq!(
+                    iter.next().expect("first").expect("ok").1,
+                    Value::Str("hello!")
+                );
+                assert_eq!(
+                    iter.next().expect("second").expect("ok").1,
+                    Value::Str("hello1!")
+                );
+                assert!(iter.next().is_none());
+            }
+            writer.commit().expect("committed");
+        }
+
+        {
+            let mut writer = k.write().expect("writer");
+            s.delete(&mut writer, 1, &Value::Str("hello!"))
+                .expect("deleted");
+            writer.commit().expect("committed");
+
+            let reader = k.read().expect("reader");
+            let mut iter = s.get(&reader, 1).expect("read");
+            assert_eq!(
+                iter.next().expect("first").expect("ok").1,
+                Value::Str("hello1!")
+            );
+            assert!(iter.next().is_none());
+        }
+
+        {
+            let mut writer = k.write().expect("writer");
+            s.delete(&mut writer, 1, &Value::Str("hello!"))
+                .expect_err("deleted");
+            writer.commit().expect("committed");
+
+            let reader = k.read().expect("reader");
+            let mut iter = s.get(&reader, 1).expect("read");
+            assert_eq!(
+                iter.next().expect("first").expect("ok").1,
+                Value::Str("hello1!")
+            );
+            assert!(iter.next().is_none());
+        }
+
+        {
+            let mut writer = k.write().expect("writer");
+            s.delete(&mut writer, 1, &Value::Str("hello1!"))
+                .expect("deleted");
+            writer.commit().expect("committed");
+
+            let reader = k.read().expect("reader");
+            let mut iter = s.get(&reader, 1).expect("read");
+            assert!(iter.next().is_none());
+        }
+
+        {
+            let mut writer = k.write().expect("writer");
+            s.delete(&mut writer, 1, &Value::Str("hello1!"))
+                .expect_err("deleted");
+            writer.commit().expect("committed");
+
+            let reader = k.read().expect("reader");
+            let mut iter = s.get(&reader, 1).expect("read");
+            assert!(iter.next().is_none());
+        }
+    }
+
+    #[test]
+    fn test_persist() {
+        let root = Builder::new()
+            .prefix("test_multi_integer_persist")
+            .tempdir()
+            .expect("tempdir");
+        fs::create_dir_all(root.path()).expect("dir created");
+
+        {
+            let k = Rkv::new::<backend::SafeMode>(root.path()).expect("new succeeded");
+            let s = k
+                .open_multi_integer("s", StoreOptions::create())
+                .expect("open");
+
+            let mut writer = k.write().expect("writer");
+            s.put(&mut writer, 1, &Value::Str("hello!")).expect("write");
+            s.put(&mut writer, 1, &Value::Str("hello1!"))
+                .expect("write");
+            s.put(&mut writer, 2, &Value::Str("hello!")).expect("write");
+            {
+                let mut iter = s.get(&writer, 1).expect("read");
+                assert_eq!(
+                    iter.next().expect("first").expect("ok").1,
+                    Value::Str("hello!")
+                );
+                assert_eq!(
+                    iter.next().expect("second").expect("ok").1,
+                    Value::Str("hello1!")
+                );
+                assert!(iter.next().is_none());
+            }
+            writer.commit().expect("committed");
+        }
+
+        {
+            let k = Rkv::new::<backend::SafeMode>(root.path()).expect("new succeeded");
+            let s = k
+                .open_multi_integer("s", StoreOptions::create())
+                .expect("open");
+
+            let reader = k.read().expect("reader");
+            let mut iter = s.get(&reader, 1).expect("read");
+            assert_eq!(
+                iter.next().expect("first").expect("ok").1,
+                Value::Str("hello!")
+            );
+            assert_eq!(
+                iter.next().expect("second").expect("ok").1,
+                Value::Str("hello1!")
+            );
+            assert!(iter.next().is_none());
+        }
+    }
+}
+
+#[cfg(test)]
+mod tests_safe {
+    use super::*;
+    use crate::*;
+
+    use std::fs;
+
+    use tempfile::Builder;
+
+    #[test]
+    fn test_integer_keys() {
+        let root = Builder::new()
+            .prefix("test_integer_keys")
+            .tempdir()
+            .expect("tempdir");
+        fs::create_dir_all(root.path()).expect("dir created");
+
+        let k = Rkv::new::<backend::SafeMode>(root.path()).expect("new succeeded");
+        let s = k
+            .open_multi_integer("s", StoreOptions::create())
+            .expect("open");
+
+        macro_rules! test_integer_keys {
+            ($type:ty, $key:expr) => {{
+                let mut writer = k.write().expect("writer");
+
+                s.put(&mut writer, $key, &Value::Str("hello!"))
+                    .expect("write");
+                assert_eq!(
+                    s.get_first(&writer, $key).expect("read"),
+                    Some(Value::Str("hello!"))
+                );
+                writer.commit().expect("committed");
+
+                let reader = k.read().expect("reader");
+                assert_eq!(
+                    s.get_first(&reader, $key).expect("read"),
+                    Some(Value::Str("hello!"))
+                );
+            }};
+        }
+
+        test_integer_keys!(u32, std::u32::MIN);
+        test_integer_keys!(u32, std::u32::MAX);
+    }
+
+    #[test]
+    fn test_clear() {
+        let root = Builder::new()
+            .prefix("test_multi_integer_clear")
+            .tempdir()
+            .expect("tempdir");
+        fs::create_dir_all(root.path()).expect("dir created");
+
+        let k = Rkv::new::<backend::SafeMode>(root.path()).expect("new succeeded");
+        let s = k
+            .open_multi_integer("s", StoreOptions::create())
+            .expect("open");
+
+        {
+            let mut writer = k.write().expect("writer");
+            s.put(&mut writer, 1, &Value::Str("hello!")).expect("write");
+            s.put(&mut writer, 1, &Value::Str("hello1!"))
+                .expect("write");
+            s.put(&mut writer, 2, &Value::Str("hello!")).expect("write");
+            assert_eq!(
+                s.get_first(&writer, 1).expect("read"),
+                Some(Value::Str("hello!"))
+            );
+            assert_eq!(
+                s.get_first(&writer, 2).expect("read"),
+                Some(Value::Str("hello!"))
+            );
+            assert_eq!(s.get_first(&writer, 3).expect("read"), None);
+            writer.commit().expect("committed");
+        }
+
+        {
+            let mut writer = k.write().expect("writer");
+            s.clear(&mut writer).expect("cleared");
+            writer.commit().expect("committed");
+
+            let reader = k.read().expect("reader");
+            assert_eq!(s.get_first(&reader, 1).expect("read"), None);
+            assert_eq!(s.get_first(&reader, 2).expect("read"), None);
+            assert_eq!(s.get_first(&reader, 3).expect("read"), None);
+        }
+    }
+
+    #[test]
+    fn test_dup() {
+        let root = Builder::new()
+            .prefix("test_multi_integer_dup")
+            .tempdir()
+            .expect("tempdir");
+        fs::create_dir_all(root.path()).expect("dir created");
+
+        let k = Rkv::new::<backend::SafeMode>(root.path()).expect("new succeeded");
+        let s = k
+            .open_multi_integer("s", StoreOptions::create())
+            .expect("open");
+
+        {
+            let mut writer = k.write().expect("writer");
+            s.put(&mut writer, 1, &Value::Str("hello!")).expect("write");
+            s.put(&mut writer, 1, &Value::Str("hello!")).expect("write");
+            s.put(&mut writer, 1, &Value::Str("hello1!"))
+                .expect("write");
+            assert_eq!(
+                s.get_first(&writer, 1).expect("read"),
+                Some(Value::Str("hello!"))
+            );
+            assert_eq!(s.get_first(&writer, 2).expect("read"), None);
+            assert_eq!(s.get_first(&writer, 3).expect("read"), None);
+            writer.commit().expect("committed");
+        }
+
+        {
+            let mut writer = k.write().expect("writer");
+            s.clear(&mut writer).expect("cleared");
+            writer.commit().expect("committed");
+
+            let reader = k.read().expect("reader");
+            assert_eq!(s.get_first(&reader, 1).expect("read"), None);
+            assert_eq!(s.get_first(&reader, 2).expect("read"), None);
+            assert_eq!(s.get_first(&reader, 3).expect("read"), None);
+        }
+    }
+
+    #[test]
+    fn test_dup_2() {
+        let root = Builder::new()
+            .prefix("test_multi_integer_dup")
+            .tempdir()
+            .expect("tempdir");
+        fs::create_dir_all(root.path()).expect("dir created");
+
+        let k = Rkv::new::<backend::SafeMode>(root.path()).expect("new succeeded");
+        let s = k
+            .open_multi_integer("s", StoreOptions::create())
+            .expect("open");
+
+        {
+            let mut writer = k.write().expect("writer");
+            s.put(&mut writer, 1, &Value::Str("hello!")).expect("write");
+            s.put(&mut writer, 1, &Value::Str("hello!")).expect("write");
+            s.put(&mut writer, 1, &Value::Str("hello1!"))
+                .expect("write");
+
+            let mut iter = s.get(&writer, 1).expect("read");
+            assert_eq!(
+                iter.next().expect("first").expect("ok").1,
+                Value::Str("hello!")
+            );
+            assert_eq!(
+                iter.next().expect("second").expect("ok").1,
+                Value::Str("hello1!")
+            );
+            assert!(iter.next().is_none());
+        }
+    }
+
+    #[test]
+    fn test_del() {
+        let root = Builder::new()
+            .prefix("test_multi_integer_dup")
+            .tempdir()
+            .expect("tempdir");
+        fs::create_dir_all(root.path()).expect("dir created");
+
+        let k = Rkv::new::<backend::SafeMode>(root.path()).expect("new succeeded");
+        let s = k
+            .open_multi_integer("s", StoreOptions::create())
+            .expect("open");
+
+        {
+            let mut writer = k.write().expect("writer");
+            s.put(&mut writer, 1, &Value::Str("hello!")).expect("write");
+            s.put(&mut writer, 1, &Value::Str("hello!")).expect("write");
+            s.put(&mut writer, 1, &Value::Str("hello1!"))
+                .expect("write");
+            {
+                let mut iter = s.get(&writer, 1).expect("read");
+                assert_eq!(
+                    iter.next().expect("first").expect("ok").1,
+                    Value::Str("hello!")
+                );
+                assert_eq!(
+                    iter.next().expect("second").expect("ok").1,
+                    Value::Str("hello1!")
+                );
+                assert!(iter.next().is_none());
+            }
+            writer.commit().expect("committed");
+        }
+
+        {
+            let mut writer = k.write().expect("writer");
+            s.delete(&mut writer, 1, &Value::Str("hello!"))
+                .expect("deleted");
+            writer.commit().expect("committed");
+
+            let reader = k.read().expect("reader");
+            let mut iter = s.get(&reader, 1).expect("read");
+            assert_eq!(
+                iter.next().expect("first").expect("ok").1,
+                Value::Str("hello1!")
+            );
+            assert!(iter.next().is_none());
+        }
+
+        {
+            let mut writer = k.write().expect("writer");
+            s.delete(&mut writer, 1, &Value::Str("hello!"))
+                .expect_err("deleted");
+            writer.commit().expect("committed");
+
+            let reader = k.read().expect("reader");
+            let mut iter = s.get(&reader, 1).expect("read");
+            assert_eq!(
+                iter.next().expect("first").expect("ok").1,
+                Value::Str("hello1!")
+            );
+            assert!(iter.next().is_none());
+        }
+
+        {
+            let mut writer = k.write().expect("writer");
+            s.delete(&mut writer, 1, &Value::Str("hello1!"))
+                .expect("deleted");
+            writer.commit().expect("committed");
+
+            let reader = k.read().expect("reader");
+            let mut iter = s.get(&reader, 1).expect("read");
+            assert!(iter.next().is_none());
+        }
+
+        {
+            let mut writer = k.write().expect("writer");
+            s.delete(&mut writer, 1, &Value::Str("hello1!"))
+                .expect_err("deleted");
+            writer.commit().expect("committed");
+
+            let reader = k.read().expect("reader");
+            let mut iter = s.get(&reader, 1).expect("read");
+            assert!(iter.next().is_none());
+        }
+    }
+
+    #[test]
+    fn test_persist() {
+        let root = Builder::new()
+            .prefix("test_multi_integer_persist")
+            .tempdir()
+            .expect("tempdir");
+        fs::create_dir_all(root.path()).expect("dir created");
+
+        {
+            let k = Rkv::new::<backend::SafeMode>(root.path()).expect("new succeeded");
+            let s = k
+                .open_multi_integer("s", StoreOptions::create())
+                .expect("open");
+
+            let mut writer = k.write().expect("writer");
+            s.put(&mut writer, 1, &Value::Str("hello!")).expect("write");
+            s.put(&mut writer, 1, &Value::Str("hello1!"))
+                .expect("write");
+            s.put(&mut writer, 2, &Value::Str("hello!")).expect("write");
+            {
+                let mut iter = s.get(&writer, 1).expect("read");
+                assert_eq!(
+                    iter.next().expect("first").expect("ok").1,
+                    Value::Str("hello!")
+                );
+                assert_eq!(
+                    iter.next().expect("second").expect("ok").1,
+                    Value::Str("hello1!")
+                );
+                assert!(iter.next().is_none());
+            }
+            writer.commit().expect("committed");
+        }
+
+        {
+            let k = Rkv::new::<backend::SafeMode>(root.path()).expect("new succeeded");
+            let s = k
+                .open_multi_integer("s", StoreOptions::create())
+                .expect("open");
+
+            let reader = k.read().expect("reader");
+            let mut iter = s.get(&reader, 1).expect("read");
+            assert_eq!(
+                iter.next().expect("first").expect("ok").1,
+                Value::Str("hello!")
+            );
+            assert_eq!(
+                iter.next().expect("second").expect("ok").1,
+                Value::Str("hello1!")
+            );
+            assert!(iter.next().is_none());
+        }
+    }
+}