Fix profile info distribution

This commit is contained in:
momo5502 2023-04-07 20:25:43 +02:00
parent 4a9c1c8778
commit fb57f0d55f
4 changed files with 25 additions and 21 deletions

View File

@ -143,12 +143,22 @@ namespace auth
return; return;
} }
const auto _ = profile_infos::acquire_profile_lock();
const utils::info_string info_string(params[1]); const utils::info_string info_string(params[1]);
const auto xuid = strtoull(info_string.get("xuid").data(), nullptr, 16); const auto xuid = strtoull(info_string.get("xuid").data(), nullptr, 16);
profile_infos::add_and_distribute_profile_info(target, xuid, info); profile_infos::add_and_distribute_profile_info(target, xuid, info);
game::SV_DirectConnect(target); game::SV_DirectConnect(target);
game::foreach_connected_client([&](game::client_s& client)
{
if (client.address == target)
{
client.xuid = xuid;
}
});
} }
} }

View File

@ -20,7 +20,7 @@ namespace profile_infos
namespace namespace
{ {
using profile_map = std::unordered_map<uint64_t, profile_info>; using profile_map = std::unordered_map<uint64_t, profile_info>;
utils::concurrency::container<profile_map> profile_mapping{}; utils::concurrency::container<profile_map, std::recursive_mutex> profile_mapping{};
std::optional<profile_info> load_profile_info() std::optional<profile_info> load_profile_info()
{ {
@ -68,21 +68,6 @@ namespace profile_infos
}); });
} }
void schedule_pcache_update()
{
static std::atomic_bool update_triggered{false};
if (game::is_server() || update_triggered.exchange(true))
{
return;
}
scheduler::once([]
{
game::PCache_DeleteEntries(game::CONTROLLER_INDEX_FIRST);
update_triggered = false;
}, scheduler::main, 5s);
}
std::unordered_set<uint64_t> get_connected_client_xuids() std::unordered_set<uint64_t> get_connected_client_xuids()
{ {
std::unordered_set<uint64_t> connected_clients{}; std::unordered_set<uint64_t> connected_clients{};
@ -103,10 +88,10 @@ namespace profile_infos
return; return;
} }
profile_mapping.access([](profile_map& profiles)
{
const auto xuids = get_connected_client_xuids(); const auto xuids = get_connected_client_xuids();
profile_mapping.access([&](profile_map& profiles)
{
for (auto i = profiles.begin(); i != profiles.end();) for (auto i = profiles.begin(); i != profiles.end();)
{ {
if (xuids.contains(i->first)) if (xuids.contains(i->first))
@ -145,8 +130,6 @@ namespace profile_infos
{ {
profiles[user_id] = info; profiles[user_id] = info;
}); });
schedule_pcache_update();
} }
void distribute_profile_info_to_user(const game::netadr_t& addr, const uint64_t user_id, const profile_info& info) void distribute_profile_info_to_user(const game::netadr_t& addr, const uint64_t user_id, const profile_info& info)
@ -194,6 +177,11 @@ namespace profile_infos
}); });
} }
std::unique_lock<std::recursive_mutex> acquire_profile_lock()
{
return profile_mapping.acquire_lock();
}
std::optional<profile_info> get_profile_info() std::optional<profile_info> get_profile_info()
{ {
return load_profile_info(); return load_profile_info();

View File

@ -19,6 +19,7 @@ namespace profile_infos
void add_and_distribute_profile_info(const game::netadr_t& addr, uint64_t user_id, const profile_info& info); void add_and_distribute_profile_info(const game::netadr_t& addr, uint64_t user_id, const profile_info& info);
void clear_profile_infos(); void clear_profile_infos();
std::unique_lock<std::recursive_mutex> acquire_profile_lock();
std::optional<profile_info> get_profile_info(); std::optional<profile_info> get_profile_info();
std::optional<profile_info> get_profile_info(uint64_t user_id); std::optional<profile_info> get_profile_info(uint64_t user_id);

View File

@ -45,6 +45,11 @@ namespace utils::concurrency
return object_; return object_;
} }
std::unique_lock<MutexType> acquire_lock()
{
return std::unique_lock<MutexType>{mutex_};
}
private: private:
mutable MutexType mutex_{}; mutable MutexType mutex_{};
T object_{}; T object_{};