nostr/damus
1
0
Fork 0
mirror of git://jb55.com/damus synced 2024-10-04 19:00:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

nostrdb/refactor: move write id index to its own function

Browse Source 
We will be writing more indices so I'm trying to clean this up a bit
before this function gets too messy

Signed-off-by: William Casarin <jb55@jb55.com>
This commit is contained in:
William Casarin 2023-11-25 15:15:23 -08:00
parent a2af030367
commit bfd78c01ca
1 changed files with 29 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
nostrdb/nostrdb.c
View File

@ -1817,18 +1817,43 @@ static int ndb_write_reaction_stats(struct ndb_txn *txn, struct ndb_note *note)
} }
static int ndb_write_note_id_index(struct ndb_txn *txn, struct ndb_note *note,
uint64_t note_key)
{
struct ndb_tsid tsid;
int rc;
MDB_val key, val;
MDB_dbi id_db;
ndb_tsid_init(&tsid, note->id, note->created_at);
key.mv_data = &tsid;
key.mv_size = sizeof(tsid);
val.mv_data = &note_key;
val.mv_size = sizeof(note_key);
id_db = txn->lmdb->dbs[NDB_DB_NOTE_ID];
if ((rc = mdb_put(txn->mdb_txn, id_db, &key, &val, 0))) {
ndb_debug("write note id index to db failed: %s\n",
mdb_strerror(rc));
return 0;
}
return 1;
}
static uint64_t ndb_write_note(struct ndb_txn *txn, static uint64_t ndb_write_note(struct ndb_txn *txn,
struct ndb_writer_note *note) struct ndb_writer_note *note)
{ {
int rc; int rc;
uint64_t note_key; uint64_t note_key;
struct ndb_tsid tsid; MDB_dbi note_db;
MDB_dbi note_db, id_db;
MDB_val key, val; MDB_val key, val;
// get dbs // get dbs
note_db = txn->lmdb->dbs[NDB_DB_NOTE]; note_db = txn->lmdb->dbs[NDB_DB_NOTE];
id_db = txn->lmdb->dbs[NDB_DB_NOTE_ID];
// get new key // get new key
note_key = ndb_get_last_key(txn->mdb_txn, note_db) + 1; note_key = ndb_get_last_key(txn->mdb_txn, note_db) + 1;
@ -1845,18 +1870,8 @@ static uint64_t ndb_write_note(struct ndb_txn *txn,
} }
// write id index key clustered with created_at // write id index key clustered with created_at
ndb_tsid_init(&tsid, note->note->id, note->note->created_at); if (!ndb_write_note_id_index(txn, note->note, note_key))
key.mv_data = &tsid;
key.mv_size = sizeof(tsid);
val.mv_data = &note_key;
val.mv_size = sizeof(note_key);
if ((rc = mdb_put(txn->mdb_txn, id_db, &key, &val, 0))) {
ndb_debug("write note id index to db failed: %s\n",
mdb_strerror(rc));
return 0; return 0;
}
if (note->note->kind == 7) { if (note->note->kind == 7) {
ndb_write_reaction_stats(txn, note->note); ndb_write_reaction_stats(txn, note->note);