nostrdb

commit 966b23f008faff35f31528db73e4147562782724
parent 25494ff7712d6caa74b4a8cdf14b86cd83773d31
Author: William Casarin <jb55@jb55.com>
Date:   Fri, 13 Dec 2024 17:21:58 -0800

introduce ndb_id_u64_ts

This will be the key used by our note_profile_kind indee

Signed-off-by: William Casarin <jb55@jb55.com>

Diffstat:
M
src/nostrdb.c
 | 
44
++++++++++++++++++++++++++++++++++++++++++++

1 file changed, 44 insertions(+), 0 deletions(-)
diff --git a/src/nostrdb.c b/src/nostrdb.c
@@ -229,6 +229,13 @@ enum ndb_query_plan {
 	NDB_PLAN_TAGS,
 };
 
+// A id + u64 + timestamp
+struct ndb_id_u64_ts {
+	unsigned char id[32]; // pubkey, id, etc
+	uint64_t u64; // kind, etc
+	uint64_t timestamp;
+};
+
 // A clustered key with an id and a timestamp
 struct ndb_tsid {
 	unsigned char id[32];
@@ -1402,6 +1409,14 @@ static int ndb_begin_rw_query(struct ndb *ndb, struct ndb_txn *txn)
 	return _ndb_begin_query(ndb, txn, 0);
 }
 
+static inline void ndb_id_u64_ts_init(struct ndb_id_u64_ts *key,
+				      unsigned char *id, uint64_t iu64,
+				      uint64_t timestamp)
+{
+	memcpy(key->id, id, 32);
+	key->u64 = iu64;
+	key->timestamp = timestamp;
+}
 
 // Migrations
 //
@@ -1516,6 +1531,35 @@ int ndb_db_version(struct ndb *ndb)
 	return version;
 }
 
+// custom pubkey+kind+timestamp comparison function. This is used by lmdb to
+// perform b+ tree searches over the pubkey+kind+timestamp index
+static int ndb_id_u64_ts_compare(const MDB_val *a, const MDB_val *b)
+{
+	struct ndb_id_u64_ts *tsa, *tsb;
+	MDB_val a2 = *a, b2 = *b;
+
+	a2.mv_size = sizeof(tsa->id);
+	b2.mv_size = sizeof(tsb->id);
+
+	int cmp = mdb_cmp_memn(&a2, &b2);
+	if (cmp) return cmp;
+
+	tsa = a->mv_data;
+	tsb = b->mv_data;
+
+	if (tsa->u64 < tsb->u64)
+		return -1;
+	else if (tsa->u64 > tsb->u64)
+		return 1;
+
+	if (tsa->timestamp < tsb->timestamp)
+		return -1;
+	else if (tsa->timestamp > tsb->timestamp)
+		return 1;
+
+	return 0;
+}
+
 // custom kind+timestamp comparison function. This is used by lmdb to perform
 // b+ tree searches over the kind+timestamp index
 static int ndb_u64_ts_compare(const MDB_val *a, const MDB_val *b)