summaryrefslogtreecommitdiffstats
path: root/third_party/rust/rkv/tests/multi-integer-store.rs
blob: 204d52f93d36e15e9774c4d9e7ec5e7f4b36a31f (plain)
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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
// 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.
#![cfg(all(feature = "db-dup-sort", feature = "db-int-key"))]
#![allow(clippy::many_single_char_names)]

use std::fs;

use serde_derive::Serialize;
use tempfile::Builder;

use rkv::{
    backend::Lmdb,
    PrimitiveInt,
    Rkv,
    StoreOptions,
    Value,
};

#[test]
fn test_multi_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::<Lmdb>(root.path()).expect("new succeeded");
    let s = k.open_multi_integer("s", StoreOptions::create()).expect("open");

    macro_rules! test_integer_keys {
        ($store:expr, $key:expr) => {{
            let mut writer = k.write().expect("writer");

            $store.put(&mut writer, $key, &Value::Str("hello1")).expect("write");
            $store.put(&mut writer, $key, &Value::Str("hello2")).expect("write");
            $store.put(&mut writer, $key, &Value::Str("hello3")).expect("write");
            let vals = $store
                .get(&writer, $key)
                .expect("read")
                .map(|result| result.expect("ok"))
                .map(|(_, v)| v)
                .collect::<Vec<Value>>();
            assert_eq!(vals, vec![Value::Str("hello1"), Value::Str("hello2"), Value::Str("hello3")]);
            writer.commit().expect("committed");

            let reader = k.read().expect("reader");
            let vals = $store
                .get(&reader, $key)
                .expect("read")
                .map(|result| result.expect("ok"))
                .map(|(_, v)| v)
                .collect::<Vec<Value>>();
            assert_eq!(vals, vec![Value::Str("hello1"), Value::Str("hello2"), Value::Str("hello3")]);
        }};
    }

    // The integer module provides only the u32 integer key variant
    // of IntegerStore, so we can use it without further ado.
    test_integer_keys!(s, std::u32::MIN);
    test_integer_keys!(s, std::u32::MAX);

    // If you want to use another integer key variant, you need to implement
    // a newtype, implement PrimitiveInt, and implement or derive Serialize
    // for it.  Here we do so for the i32 type.

    // DANGER!  Doing this enables you to open a store with multiple,
    // different integer key types, which may result in unexpected behavior.
    // Make sure you know what you're doing!

    let t = k.open_multi_integer("s", StoreOptions::create()).expect("open");

    #[derive(Serialize)]
    struct I32(i32);
    impl PrimitiveInt for I32 {}
    test_integer_keys!(t, I32(std::i32::MIN));
    test_integer_keys!(t, I32(std::i32::MAX));

    let u = k.open_multi_integer("s", StoreOptions::create()).expect("open");

    #[derive(Serialize)]
    struct U16(u16);
    impl PrimitiveInt for U16 {}
    test_integer_keys!(u, U16(std::u16::MIN));
    test_integer_keys!(u, U16(std::u16::MAX));

    let v = k.open_multi_integer("s", StoreOptions::create()).expect("open");

    #[derive(Serialize)]
    struct U64(u64);
    impl PrimitiveInt for U64 {}
    test_integer_keys!(v, U64(std::u64::MIN));
    test_integer_keys!(v, U64(std::u64::MAX));
}