0
0
Fork 0
mirror of https://github.com/matrix-construct/construct synced 2024-06-16 08:58:20 +02:00

ircd:Ⓜ️:room: Refactor out additional test protocol iterations; minor reorg.

This commit is contained in:
Jason Volk 2018-09-25 16:16:47 -07:00
parent f54e05b7f0
commit cf783d2729
4 changed files with 119 additions and 131 deletions

View file

@ -316,19 +316,18 @@ struct ircd::m::room::members
m::room room;
size_t count() const;
size_t count(const string_view &membership) const;
bool test(const string_view &membership, const event::closure_bool &view) const;
bool test(const event::closure_bool &view) const;
bool for_each(const string_view &membership, const event::closure_bool &) const;
void for_each(const string_view &membership, const event::closure &) const;
bool for_each(const string_view &membership, const closure_bool &) const;
void for_each(const string_view &membership, const closure &) const;
bool for_each(const event::closure_bool &) const;
void for_each(const event::closure &) const;
bool for_each(const closure_bool &) const;
void for_each(const closure &) const;
size_t count(const string_view &membership) const;
size_t count() const;
members(const m::room &room)
:room{room}
{}
@ -348,8 +347,7 @@ struct ircd::m::room::origins
m::room room;
bool _test_(const closure_bool &view) const;
bool test(const closure_bool &view) const;
bool _for_each_(const closure_bool &view) const;
bool for_each(const closure_bool &view) const;
void for_each(const closure &view) const;
bool has(const string_view &origin) const;

View file

@ -2175,7 +2175,7 @@ const
(const m::room &room, const string_view &membership)
{
const m::room::origins origins{room};
return !origins.test([&closure, &seen]
return origins.for_each(origins::closure_bool{[&closure, &seen]
(const string_view &origin)
{
const auto it
@ -2184,11 +2184,11 @@ const
};
if(it != end(seen) && *it == origin)
return false;
return true;
seen.emplace_hint(it, std::string{origin});
return !closure(origin);
});
return closure(origin);
}});
}});
}
@ -2273,7 +2273,7 @@ const
(const m::room &room, const string_view &)
{
const m::room::members members{room};
return !members.test(membership, [&seen, &closure]
return members.for_each(membership, event::closure_bool{[&seen, &closure]
(const m::event &event)
{
const auto &other
@ -2287,11 +2287,11 @@ const
};
if(it != end(seen) && *it == other)
return false;
return true;
seen.emplace_hint(it, std::string{other});
return !closure(m::user{other});
});
return closure(m::user{other});
}});
}});
}

View file

@ -1472,6 +1472,72 @@ const
// room::members
//
size_t
ircd::m::room::members::count()
const
{
const room::state state
{
room
};
return state.count("m.room.member");
}
size_t
ircd::m::room::members::count(const string_view &membership)
const
{
// Allow empty membership string to count all memberships
if(!membership)
return count();
// joined members optimization. Only possible when seeking
// membership="join" on the present state of the room.
if(!room.event_id && membership == "join")
{
size_t ret{0};
const room::origins origins{room};
origins._for_each_([&ret](const string_view &)
{
++ret;
return true;
});
return ret;
}
// The list of event fields to fetch for the closure
static const event::keys keys
{
event::keys::include
{
"event_id", // Added for any upstack usage (but may be unnecessary).
"membership", // Required for membership test.
"content", // Required because synapse events randomly have no event.membership
}
};
const m::event::fetch::opts fopts
{
keys, room.fopts? room.fopts->gopts : db::gopts{}
};
const room::state state
{
room, &fopts
};
size_t ret{0};
state.for_each("m.room.member", event::closure{[&ret, &membership]
(const m::event &event)
{
ret += m::membership(event) == membership;
}});
return ret;
}
void
ircd::m::room::members::for_each(const closure &closure)
const
@ -1494,14 +1560,14 @@ const
}
bool
ircd::m::room::members::test(const event::closure_bool &closure)
ircd::m::room::members::for_each(const event::closure_bool &closure)
const
{
const room::state state{room};
return !state.for_each("m.room.member", event::closure_bool{[&closure]
return state.for_each("m.room.member", event::closure_bool{[&closure]
(const m::event &event)
{
return !closure(event);
return closure(event);
}});
}
@ -1557,7 +1623,7 @@ const
room.fopts = theirs;
}};
return !test(membership, event::closure_bool{[&closure]
return for_each(membership, event::closure_bool{[&closure]
(const event &event)
{
const user::id &user_id
@ -1565,7 +1631,7 @@ const
at<"state_key"_>(event)
};
return !closure(user_id);
return closure(user_id);
}});
}
@ -1574,28 +1640,28 @@ ircd::m::room::members::for_each(const string_view &membership,
const event::closure &closure)
const
{
test(membership, [&closure]
(const event &event)
for_each(membership, event::closure_bool{[&closure]
(const m::event &event)
{
closure(event);
return false;
});
}});
}
bool
ircd::m::room::members::test(const string_view &membership,
const event::closure_bool &closure)
ircd::m::room::members::for_each(const string_view &membership,
const event::closure_bool &closure)
const
{
if(empty(membership))
return test(closure);
return for_each(closure);
// joined members optimization. Only possible when seeking
// membership="join" on the present state of the room.
if(!room.event_id && membership == "join")
{
const room::origins origins{room};
return origins._test_([&closure, this]
return origins._for_each_([&closure, this]
(const string_view &key)
{
const string_view &member
@ -1603,7 +1669,7 @@ const
std::get<1>(dbs::room_joined_key(key))
};
bool ret{false};
bool ret{true};
room.get(std::nothrow, "m.room.member", member, event::closure{[&closure, &ret]
(const event &event)
{
@ -1614,83 +1680,17 @@ const
});
}
return test(event::closure_bool{[&membership, &closure]
return for_each(event::closure_bool{[&membership, &closure]
(const event &event)
{
if(m::membership(event) == membership)
if(closure(event))
return true;
if(!closure(event))
return false;
return false;
return true;
}});
}
size_t
ircd::m::room::members::count(const string_view &membership)
const
{
// Allow empty membership string to count all memberships
if(!membership)
return count();
// joined members optimization. Only possible when seeking
// membership="join" on the present state of the room.
if(!room.event_id && membership == "join")
{
size_t ret{0};
const room::origins origins{room};
origins._test_([&ret](const string_view &)
{
++ret;
return false;
});
return ret;
}
// The list of event fields to fetch for the closure
static const event::keys keys
{
event::keys::include
{
"event_id", // Added for any upstack usage (but may be unnecessary).
"membership", // Required for membership test.
"content", // Required because synapse events randomly have no event.membership
}
};
const m::event::fetch::opts fopts
{
keys, room.fopts? room.fopts->gopts : db::gopts{}
};
const room::state state
{
room, &fopts
};
size_t ret{0};
state.for_each("m.room.member", event::closure{[&ret, &membership]
(const m::event &event)
{
ret += m::membership(event) == membership;
}});
return ret;
}
size_t
ircd::m::room::members::count()
const
{
const room::state state
{
room
};
return state.count("m.room.member");
}
//
// room::origins
//
@ -1748,19 +1748,19 @@ bool
ircd::m::room::origins::only(const string_view &origin)
const
{
bool ret{false};
test([&ret, &origin]
(const string_view &origin_)
ushort ret{2};
for_each(closure_bool{[&ret, &origin]
(const string_view &origin_) -> bool
{
if(origin == origin_)
ret = true;
else if(ret)
ret = false;
ret = 1;
else
ret = 0;
return ret;
});
}});
return ret;
return ret == 1;
}
bool
@ -1803,30 +1803,21 @@ void
ircd::m::room::origins::for_each(const closure &view)
const
{
test([&view](const string_view &origin)
for_each(closure_bool{[&view]
(const string_view &origin)
{
view(origin);
return false;
});
return true;
}});
}
bool
ircd::m::room::origins::for_each(const closure_bool &view)
const
{
return !test([&view](const string_view &origin)
{
return !view(origin);
});
}
bool
ircd::m::room::origins::test(const closure_bool &view)
const
{
string_view last;
char lastbuf[256];
return _test_([&last, &lastbuf, &view]
return _for_each_([&last, &lastbuf, &view]
(const string_view &key)
{
const string_view &origin
@ -1835,18 +1826,18 @@ const
};
if(origin == last)
return false;
if(view(origin))
return true;
if(!view(origin))
return false;
last = { lastbuf, copy(lastbuf, origin) };
return false;
return true;
});
}
bool
ircd::m::room::origins::_test_(const closure_bool &view)
ircd::m::room::origins::_for_each_(const closure_bool &view)
const
{
db::index &index
@ -1866,11 +1857,11 @@ const
lstrip(it->first, "\0"_sv)
};
if(view(key))
return true;
if(!view(key))
return false;
}
return false;
return true;
}
//

View file

@ -5424,10 +5424,9 @@ console_cmd__room__origins(opt &out, const string_view &line)
room
};
origins.test([&out](const string_view &origin)
origins.for_each([&out](const string_view &origin)
{
out << origin << std::endl;
return false;
});
return true;