count index

1 files changed, 12 insertions, 2 deletions

diff --git a/database/src/kv/index.rs b/database/src/kv/index.rs
index 27aabd6..522ae18 100644
--- a/database/src/kv/index.rs
+++ b/database/src/kv/index.rs
@@ -19,14 +19,14 @@ use crate::{
 pub fn update_index(
     txn: &mut dyn jellykv::Transaction,
     is: SubtreeNum,
-    ik: IndexKey,
+    ik: &IndexKey,
     row: RowNum,
     ob: Object<'_>,
     remove: bool,
 ) -> Result<()> {
     let mut ks = vec![is.to_be_bytes().to_vec()];
     ik.0.apply(ob, &mut ks);
-    match ik.1 {
+    match &ik.1 {
         SortKey::None => {
             for mut k in ks {
                 k.extend(row.to_be_bytes());
@@ -53,3 +53,13 @@ pub fn update_index(
     }
     Ok(())
 }
+
+pub fn read_count_index(
+    txn: &dyn jellykv::Transaction,
+    is: SubtreeNum,
+    prefix: Vec<u8>,
+) -> Result<u64> {
+    let mut k = is.to_be_bytes().to_vec();
+    k.extend(&prefix);
+    read_counter(txn, &k, 0)
+}