wobject 0.1.0

wobject
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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121

use std::io::{self, Read, Write};

use whasher::{GxPapayaSet, new_papaya_set};

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
#[repr(u8)]
pub enum SetOperation {
  Sadd = 0,
  Srem = 1,
  Spop = 2,
  Smove = 3,
  Srandmember = 4,
  Smembers = 5,
  Sinter = 6,
  Sunion = 7,
  Sdiff = 8,
  Sinterstore = 9,
  Sunionstore = 10,
  Sdiffstore = 11,
  Sismember = 12,
  Scard = 13,
}

/// 在 garnet 中的相对路径:libs/server/Objects/Set/SetObject.cs:SetObject
pub struct SetObject {
  pub set: GxPapayaSet<Vec<u8>>,
}

impl SetObject {
  pub fn new() -> Self {
    Self {
      set: new_papaya_set(),
    }
  }

  /// 在 garnet 中的相对路径:libs/server/Objects/Set/SetObject.cs:SetObject(BinaryReader)
  pub fn deserialize<R: Read>(reader: &mut R) -> io::Result<Self> {
    let mut buf = Vec::new();
    reader.read_to_end(&mut buf)?;
    let items: Vec<Vec<u8>> =
      bitcode::decode(&buf).map_err(|e| io::Error::new(io::ErrorKind::InvalidData, e))?;

    let set = new_papaya_set();
    let pin = set.pin();
    for item in items {
      pin.insert(item);
    }

    drop(pin);
    Ok(Self { set })
  }

  /// 在 garnet 中的相对路径:libs/server/Objects/Set/SetObject.cs:Serialize
  pub fn serialize<W: Write>(&self, writer: &mut W) -> io::Result<()> {
    let pin = self.set.pin();
    let mut items = Vec::with_capacity(pin.len());
    for item in pin.iter() {
      items.push(item.clone());
    }
    let bytes = bitcode::encode(&items);
    writer.write_all(&bytes)
  }

  /// 在 garnet 中的相对路径:libs/server/Objects/Set/SetObject.cs:Operate
  pub fn operate(&self, op: SetOperation, key: &[u8]) -> bool {
    let pin = self.set.pin();
    match op {
      SetOperation::Sadd => pin.insert(key.to_vec()),
      SetOperation::Srem => pin.remove(key),
      SetOperation::Sismember => pin.contains(key),
      _ => false,
    }
  }

  /// 在 garnet 中的相对路径:libs/server/Objects/Set/SetObject.cs:Count
  pub fn count(&self) -> usize {
    self.set.pin().len()
  }
}

impl Default for SetObject {
  fn default() -> Self {
    Self::new()
  }
}

impl SetObject {
  /// 在 garnet 中的相对路径:libs/server/Objects/Set/SetObject.cs:GetMembers（SMEMBERS 输出）
  pub fn get_keys(&self) -> Vec<Vec<u8>> {
    let pin = self.set.pin();
    pin.iter().cloned().collect()
  }

  /// 在 garnet 中的相对路径:libs/server/Objects/Set/SetObjectImpl.cs:SetPopImpl（SPOP）
  ///
  /// 刻意差异修正：原实现恒取首元素（确定性弹出，语义偏差）；对齐 C# 的
  /// `RandomNumberGenerator.GetInt32(0, Set.Count)` 随机下标弹出
  /// （`nth(index)` 与 C# `Set.ElementAt(index)` 同为 O(index) 链上遍历，复杂度持平）
  pub fn pop(&self) -> Option<Vec<u8>> {
    let pin = self.set.pin();
    let len = pin.len();
    if len == 0 {
      return None;
    }
    let item = pin.iter().nth(fastrand::usize(..len)).cloned()?;
    pin.remove(&item);
    Some(item)
  }

  /// 在 garnet 中的相对路径:libs/server/Objects/Set/SetObjectImpl.cs:SetRandomMember（SRANDMEMBER）
  ///
  /// 刻意差异修正：原实现恒取首元素；对齐 C# `RandomUtils.PickRandomIndex` 随机采样
  pub fn random_member(&self) -> Option<Vec<u8>> {
    let pin = self.set.pin();
    let len = pin.len();
    if len == 0 {
      return None;
    }
    pin.iter().nth(fastrand::usize(..len)).cloned()
  }
}