ptr.pet
forked from
microcosm.blue/Allegedly
Server tools to backfill, tail, mirror, and verify PLC logs
fjall: only use 6 bytes of CID in keys
+40
-19
src/plc_fjall.rs
+40
-19
src/plc_fjall.rs
···
33
33
}
34
34
35
35
// 59 bytes -> 36 bytes
36
-
fn encode_cid(buf: &mut Vec<u8>, s: &str) -> anyhow::Result<usize> {
37
-
IpldCid::try_from(s)?
38
-
.write_bytes(buf)
39
-
.map_err(|e| anyhow::anyhow!("failed to encode cid {s}: {e}"))
36
+
fn decode_cid_str(s: &str) -> anyhow::Result<Vec<u8>> {
37
+
let cid = IpldCid::try_from(s)?;
38
+
let mut buf = Vec::new();
39
+
cid.write_bytes(&mut buf)
40
+
.map_err(|e| anyhow::anyhow!("failed to encode cid {s}: {e}"))?;
41
+
Ok(buf)
40
42
}
41
43
42
44
fn decode_cid(bytes: &[u8]) -> anyhow::Result<String> {
···
50
52
format!("did:plc:{decoded}")
51
53
}
52
54
53
-
fn op_key(created_at: &Dt, cid: &str) -> anyhow::Result<Vec<u8>> {
55
+
fn op_key(created_at: &Dt, cid_suffix: &[u8]) -> Vec<u8> {
54
56
let micros = created_at.timestamp_micros() as u64;
55
-
let mut key = Vec::with_capacity(8 + 1 + cid.len());
57
+
let mut key = Vec::with_capacity(8 + 1 + cid_suffix.len());
56
58
key.extend_from_slice(µs.to_be_bytes());
57
59
key.push(SEP);
58
-
encode_cid(&mut key, cid)?;
59
-
Ok(key)
60
+
key.extend_from_slice(cid_suffix);
61
+
key
60
62
}
61
63
62
64
fn by_did_prefix(did: &str) -> anyhow::Result<Vec<u8>> {
···
66
68
Ok(p)
67
69
}
68
70
69
-
fn by_did_key(did: &str, created_at: &Dt, cid: &str) -> anyhow::Result<Vec<u8>> {
71
+
fn by_did_key(did: &str, created_at: &Dt, cid_suffix: &[u8]) -> anyhow::Result<Vec<u8>> {
70
72
let mut key = by_did_prefix(did)?;
71
73
let micros = created_at.timestamp_micros() as u64;
72
74
key.extend_from_slice(µs.to_be_bytes());
73
75
key.push(SEP);
74
-
encode_cid(&mut key, cid)?;
76
+
key.extend_from_slice(cid_suffix);
75
77
Ok(key)
76
78
}
77
79
···
785
787
struct DbOp {
786
788
#[serde(with = "serde_bytes")]
787
789
pub did: Vec<u8>,
790
+
#[serde(with = "serde_bytes")]
791
+
pub cid_prefix: Vec<u8>,
788
792
pub nullified: bool,
789
793
pub operation: StoredOp,
790
794
}
···
879
883
}
880
884
881
885
pub fn insert_op(&self, batch: &mut OwnedWriteBatch, op: &CommonOp) -> anyhow::Result<usize> {
882
-
let pk = by_did_key(&op.did, &op.created_at, &op.cid)?;
886
+
let cid_bytes = decode_cid_str(&op.cid)?;
887
+
let cid_prefix = cid_bytes
888
+
.get(..30)
889
+
.ok_or_else(|| anyhow::anyhow!("invalid cid length (prefix): {}", op.cid))?
890
+
.to_vec();
891
+
let cid_suffix = cid_bytes
892
+
.get(30..)
893
+
.ok_or_else(|| anyhow::anyhow!("invalid cid length (suffix): {}", op.cid))?;
894
+
895
+
let pk = by_did_key(&op.did, &op.created_at, cid_suffix)?;
883
896
if self.inner.by_did.get(&pk)?.is_some() {
884
897
return Ok(0);
885
898
}
886
-
let ts_key = op_key(&op.created_at, &op.cid)?;
899
+
let ts_key = op_key(&op.created_at, cid_suffix);
887
900
888
901
let mut encoded_did = Vec::with_capacity(15);
889
902
encode_did(&mut encoded_did, &op.did)?;
···
905
918
906
919
let db_op = DbOp {
907
920
did: encoded_did,
921
+
cid_prefix,
908
922
nullified: op.nullified,
909
923
operation,
910
924
};
···
932
946
let ts_bytes = key_rest
933
947
.get(..8)
934
948
.ok_or_else(|| anyhow::anyhow!("invalid length: {key_rest:?}"))?;
935
-
let cid_bytes = key_rest
949
+
let cid_suffix = key_rest
936
950
.get(9..)
937
951
.ok_or_else(|| anyhow::anyhow!("invalid length: {key_rest:?}"))?;
938
952
939
-
let op_key = [ts_bytes, &[SEP][..], cid_bytes].concat();
953
+
let op_key = [ts_bytes, &[SEP][..], cid_suffix].concat();
940
954
let ts = decode_timestamp(ts_bytes)?;
941
955
942
956
let value = self
···
946
960
.ok_or_else(|| anyhow::anyhow!("op not found: {op_key:?}"))?;
947
961
948
962
let op: DbOp = rmp_serde::from_slice(&value)?;
949
-
let cid = decode_cid(cid_bytes)?;
963
+
let mut full_cid_bytes = op.cid_prefix.clone();
964
+
full_cid_bytes.extend_from_slice(cid_suffix);
965
+
966
+
let cid = decode_cid(&full_cid_bytes)?;
950
967
let did = decode_did(&op.did);
951
968
952
969
Ok(Op {
···
980
997
key.get(..8)
981
998
.ok_or_else(|| anyhow::anyhow!("invalid op key {key:?}"))?,
982
999
)?;
983
-
let cid = decode_cid(
984
-
key.get(9..)
985
-
.ok_or_else(|| anyhow::anyhow!("invalid op key {key:?}"))?,
986
-
)?;
1000
+
let cid_suffix = key
1001
+
.get(9..)
1002
+
.ok_or_else(|| anyhow::anyhow!("invalid op key {key:?}"))?;
1003
+
1004
+
let mut full_cid_bytes = db_op.cid_prefix.clone();
1005
+
full_cid_bytes.extend_from_slice(cid_suffix);
1006
+
1007
+
let cid = decode_cid(&full_cid_bytes)?;
987
1008
let did = decode_did(&db_op.did);
988
1009
989
1010
Ok(Op {