refactor database engine/options; add column descriptors

Signed-off-by: Jason Volk <jason@zemos.net>
This commit is contained in:
Jason Volk 2025-01-06 14:07:26 +00:00
parent 02f19cf951
commit 6a0f9add0c
21 changed files with 1358 additions and 965 deletions

View file

@ -2,103 +2,383 @@ use std::{collections::BTreeMap, sync::Arc};
use conduwuit::Result;
use crate::{Engine, Map};
use crate::{
engine::descriptor::{self, CacheDisp, Descriptor},
Engine, Map,
};
pub type Maps = BTreeMap<MapsKey, MapsVal>;
pub(crate) type MapsVal = Arc<Map>;
pub(crate) type MapsKey = String;
pub(super) type Maps = BTreeMap<MapsKey, MapsVal>;
pub(super) type MapsKey = &'static str;
pub(super) type MapsVal = Arc<Map>;
pub(crate) fn open(db: &Arc<Engine>) -> Result<Maps> { open_list(db, MAPS) }
pub(super) fn open(db: &Arc<Engine>) -> Result<Maps> { open_list(db, MAPS) }
#[tracing::instrument(name = "maps", level = "debug", skip_all)]
pub(crate) fn open_list(db: &Arc<Engine>, maps: &[&str]) -> Result<Maps> {
Ok(maps
.iter()
.map(|&name| (name.to_owned(), Map::open(db, name).expect("valid column opened")))
.collect::<Maps>())
pub(super) fn open_list(db: &Arc<Engine>, maps: &[Descriptor]) -> Result<Maps> {
maps.iter()
.map(|desc| Ok((desc.name, Map::open(db, desc.name)?)))
.collect()
}
pub const MAPS: &[&str] = &[
"alias_roomid",
"alias_userid",
"aliasid_alias",
"backupid_algorithm",
"backupid_etag",
"backupkeyid_backup",
"bannedroomids",
"disabledroomids",
"eventid_outlierpdu",
"eventid_pduid",
"eventid_shorteventid",
"global",
"id_appserviceregistrations",
"keychangeid_userid",
"keyid_key",
"lazyloadedids",
"mediaid_file",
"mediaid_user",
"onetimekeyid_onetimekeys",
"pduid_pdu",
"presenceid_presence",
"publicroomids",
"readreceiptid_readreceipt",
"referencedevents",
"roomid_invitedcount",
"roomid_inviteviaservers",
"roomid_joinedcount",
"roomid_pduleaves",
"roomid_shortroomid",
"roomid_shortstatehash",
"roomserverids",
"roomsynctoken_shortstatehash",
"roomuserdataid_accountdata",
"roomuserid_invitecount",
"roomuserid_joined",
"roomuserid_lastprivatereadupdate",
"roomuserid_leftcount",
"roomuserid_privateread",
"roomuseroncejoinedids",
"roomusertype_roomuserdataid",
"senderkey_pusher",
"server_signingkeys",
"servercurrentevent_data",
"servername_educount",
"servernameevent_data",
"serverroomids",
"shorteventid_authchain",
"shorteventid_eventid",
"shorteventid_shortstatehash",
"shortstatehash_statediff",
"shortstatekey_statekey",
"softfailedeventids",
"statehash_shortstatehash",
"statekey_shortstatekey",
"threadid_userids",
"todeviceid_events",
"tofrom_relation",
"token_userdeviceid",
"tokenids",
"url_previews",
"userdeviceid_metadata",
"userdeviceid_token",
"userdevicesessionid_uiaainfo",
"userdevicetxnid_response",
"userfilterid_filter",
"userid_avatarurl",
"userid_blurhash",
"userid_devicelistversion",
"userid_displayname",
"userid_lastonetimekeyupdate",
"userid_masterkeyid",
"userid_password",
"userid_presenceid",
"userid_selfsigningkeyid",
"userid_usersigningkeyid",
"useridprofilekey_value",
"openidtoken_expiresatuserid",
"userroomid_highlightcount",
"userroomid_invitestate",
"userroomid_joined",
"userroomid_leftstate",
"userroomid_notificationcount",
pub(super) static MAPS: &[Descriptor] = &[
Descriptor {
name: "alias_roomid",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "alias_userid",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "aliasid_alias",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "backupid_algorithm",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "backupid_etag",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "backupkeyid_backup",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "bannedroomids",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "disabledroomids",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "eventid_outlierpdu",
cache_disp: CacheDisp::SharedWith("pduid_pdu"),
key_size_hint: Some(48),
val_size_hint: Some(1488),
..descriptor::RANDOM
},
Descriptor {
name: "eventid_pduid",
cache_disp: CacheDisp::Unique,
key_size_hint: Some(48),
val_size_hint: Some(16),
..descriptor::RANDOM
},
Descriptor {
name: "eventid_shorteventid",
cache_disp: CacheDisp::Unique,
key_size_hint: Some(48),
val_size_hint: Some(8),
..descriptor::RANDOM
},
Descriptor {
name: "global",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "id_appserviceregistrations",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "keychangeid_userid",
..descriptor::RANDOM
},
Descriptor {
name: "keyid_key",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "lazyloadedids",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "mediaid_file",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "mediaid_user",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "onetimekeyid_onetimekeys",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "pduid_pdu",
cache_disp: CacheDisp::SharedWith("eventid_outlierpdu"),
key_size_hint: Some(16),
val_size_hint: Some(1520),
..descriptor::SEQUENTIAL
},
Descriptor {
name: "presenceid_presence",
..descriptor::SEQUENTIAL_SMALL
},
Descriptor {
name: "publicroomids",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "readreceiptid_readreceipt",
..descriptor::RANDOM
},
Descriptor {
name: "referencedevents",
..descriptor::RANDOM
},
Descriptor {
name: "roomid_invitedcount",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "roomid_inviteviaservers",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "roomid_joinedcount",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "roomid_pduleaves",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "roomid_shortroomid",
val_size_hint: Some(8),
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "roomid_shortstatehash",
val_size_hint: Some(8),
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "roomserverids",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "roomsynctoken_shortstatehash",
val_size_hint: Some(8),
..descriptor::SEQUENTIAL
},
Descriptor {
name: "roomuserdataid_accountdata",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "roomuserid_invitecount",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "roomuserid_joined",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "roomuserid_lastprivatereadupdate",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "roomuserid_leftcount",
..descriptor::RANDOM
},
Descriptor {
name: "roomuserid_privateread",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "roomuseroncejoinedids",
..descriptor::RANDOM
},
Descriptor {
name: "roomusertype_roomuserdataid",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "senderkey_pusher",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "server_signingkeys",
..descriptor::RANDOM
},
Descriptor {
name: "servercurrentevent_data",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "servername_educount",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "servernameevent_data",
cache_disp: CacheDisp::Unique,
val_size_hint: Some(128),
..descriptor::RANDOM
},
Descriptor {
name: "serverroomids",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "shorteventid_authchain",
cache_disp: CacheDisp::Unique,
key_size_hint: Some(8),
..descriptor::SEQUENTIAL
},
Descriptor {
name: "shorteventid_eventid",
cache_disp: CacheDisp::Unique,
key_size_hint: Some(8),
val_size_hint: Some(48),
..descriptor::SEQUENTIAL_SMALL
},
Descriptor {
name: "shorteventid_shortstatehash",
key_size_hint: Some(8),
val_size_hint: Some(8),
..descriptor::SEQUENTIAL
},
Descriptor {
name: "shortstatehash_statediff",
key_size_hint: Some(8),
..descriptor::SEQUENTIAL_SMALL
},
Descriptor {
name: "shortstatekey_statekey",
cache_disp: CacheDisp::Unique,
key_size_hint: Some(8),
val_size_hint: Some(1016),
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "softfailedeventids",
key_size_hint: Some(48),
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "statehash_shortstatehash",
val_size_hint: Some(8),
..descriptor::RANDOM
},
Descriptor {
name: "statekey_shortstatekey",
cache_disp: CacheDisp::Unique,
key_size_hint: Some(1016),
val_size_hint: Some(8),
..descriptor::RANDOM
},
Descriptor {
name: "threadid_userids",
..descriptor::SEQUENTIAL_SMALL
},
Descriptor {
name: "todeviceid_events",
..descriptor::RANDOM
},
Descriptor {
name: "tofrom_relation",
key_size_hint: Some(8),
val_size_hint: Some(8),
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "token_userdeviceid",
..descriptor::RANDOM_SMALL
},
Descriptor { name: "tokenids", ..descriptor::RANDOM },
Descriptor {
name: "url_previews",
..descriptor::RANDOM
},
Descriptor {
name: "userdeviceid_metadata",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "userdeviceid_token",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "userdevicesessionid_uiaainfo",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "userdevicetxnid_response",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "userfilterid_filter",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "userid_avatarurl",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "userid_blurhash",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "userid_devicelistversion",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "userid_displayname",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "userid_lastonetimekeyupdate",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "userid_masterkeyid",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "userid_password",
..descriptor::RANDOM
},
Descriptor {
name: "userid_presenceid",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "userid_selfsigningkeyid",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "userid_usersigningkeyid",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "useridprofilekey_value",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "openidtoken_expiresatuserid",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "userroomid_highlightcount",
..descriptor::RANDOM
},
Descriptor {
name: "userroomid_invitestate",
..descriptor::RANDOM_SMALL
},
Descriptor {
name: "userroomid_joined",
..descriptor::RANDOM
},
Descriptor {
name: "userroomid_leftstate",
..descriptor::RANDOM
},
Descriptor {
name: "userroomid_notificationcount",
..descriptor::RANDOM
},
];