MirrorHub/construct
0
0
Fork 0
mirror of https://github.com/matrix-construct/construct synced 2024-12-29 08:54:02 +01:00
Code Issues Releases Wiki Activity

modules/m_room: More sensible head reset operation.

Browse source
This commit is contained in:
Jason Volk 2018-05-22 18:19:53 -07:00
parent 956c67beca
commit 725e339405
1 changed files with 35 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

51
modules/m_room.cc
View file

@ -185,42 +185,61 @@ extern "C" size_t
head__reset(const m::room &room) head__reset(const m::room &room)
{ {
size_t ret{0}; size_t ret{0};
const m::room::state state{room};
const auto create_id
{
state.get("m.room.create")
};
m::room::messages it m::room::messages it
{ {
room, create_id room
}; };
if(!it) if(!it)
return ret; return ret;
// Replacement will be the single new head
const m::event replacement
{
*it
};
db::txn txn db::txn txn
{ {
*m::dbs::events *m::dbs::events
}; };
// Iterate all of the existing heads with a delete operation
m::dbs::write_opts opts; m::dbs::write_opts opts;
opts.op = db::op::DELETE; opts.op = db::op::DELETE;
opts.head = true; opts.head = true;
for(; it; ++it) m::room::head{room}.for_each([&room, &opts, &txn, &ret]
(const m::event::idx &event_idx, const m::event::id &event_id)
{ {
const m::event &event{*it}; const m::event::fetch event
opts.event_idx = it.event_idx(); {
event_idx, std::nothrow
};
if(!event.valid)
{
log::derror
{
"Invalid event '%s' idx %lu in head for %s",
string_view{event_id},
event_idx,
string_view{room.room_id}
};
return;
}
opts.event_idx = event_idx;
m::dbs::_index__room_head(txn, event, opts); m::dbs::_index__room_head(txn, event, opts);
++ret; ++ret;
} });
{ // Finally add the replacement to the txn
opts.op = db::op::SET; opts.op = db::op::SET;
const m::event::fetch event{opts.event_idx}; opts.event_idx = it.event_idx();
m::dbs::_index__room_head(txn, event, opts); m::dbs::_index__room_head(txn, replacement, opts);
}
// Commit txn
txn(); txn();
return ret; return ret;
} }