mirror of
https://github.com/brian8544/turtle-wow.git
synced 2025-01-05 22:34:35 +00:00
88 lines
2.8 KiB
C++
88 lines
2.8 KiB
C++
/************************************************************************************
|
|
*
|
|
* D++, A Lightweight C++ library for Discord
|
|
*
|
|
* Copyright 2021 Craig Edwards and D++ contributors
|
|
* (https://github.com/brainboxdotcc/DPP/graphs/contributors)
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*
|
|
************************************************************************************/
|
|
#include <dpp/export.h>
|
|
#include <mutex>
|
|
#include <iostream>
|
|
#include <variant>
|
|
#include <dpp/cache.h>
|
|
#include <dpp/exception.h>
|
|
|
|
namespace dpp {
|
|
|
|
std::unordered_map<managed*, time_t> deletion_queue;
|
|
std::mutex deletion_mutex;
|
|
|
|
#define cache_helper(type, cache_name, setter, getter, counter) \
|
|
cache<type>* cache_name = nullptr; \
|
|
type * setter (snowflake id) { \
|
|
return cache_name ? ( type * ) cache_name ->find(id) : nullptr; \
|
|
} \
|
|
cache<type>* getter () { \
|
|
if (! cache_name ) { \
|
|
cache_name = new cache<type>(); \
|
|
} \
|
|
return cache_name ; \
|
|
} \
|
|
uint64_t counter () { \
|
|
return ( cache_name ? cache_name ->count() : 0 ); \
|
|
}
|
|
|
|
|
|
/* Because other threads and systems may run for a short while after an event is received, we don't immediately
|
|
* delete pointers when objects are replaced. We put them into a queue, and periodically delete pointers in the
|
|
* queue. This also rehashes unordered_maps to ensure they free their memory.
|
|
*/
|
|
void garbage_collection() {
|
|
time_t now = time(NULL);
|
|
bool repeat = false;
|
|
{
|
|
std::lock_guard<std::mutex> delete_lock(deletion_mutex);
|
|
do {
|
|
repeat = false;
|
|
for (auto g = deletion_queue.begin(); g != deletion_queue.end(); ++g) {
|
|
if (now > g->second + 60) {
|
|
delete g->first;
|
|
deletion_queue.erase(g);
|
|
repeat = true;
|
|
break;
|
|
}
|
|
}
|
|
} while (repeat);
|
|
if (deletion_queue.size() == 0) {
|
|
deletion_queue = {};
|
|
}
|
|
}
|
|
dpp::get_user_cache()->rehash();
|
|
dpp::get_channel_cache()->rehash();
|
|
dpp::get_guild_cache()->rehash();
|
|
dpp::get_role_cache()->rehash();
|
|
dpp::get_emoji_cache()->rehash();
|
|
}
|
|
|
|
|
|
cache_helper(user, user_cache, find_user, get_user_cache, get_user_count);
|
|
cache_helper(channel, channel_cache, find_channel, get_channel_cache, get_channel_count);
|
|
cache_helper(role, role_cache, find_role, get_role_cache, get_role_count);
|
|
cache_helper(guild, guild_cache, find_guild, get_guild_cache, get_guild_count);
|
|
cache_helper(emoji, emoji_cache, find_emoji, get_emoji_cache, get_emoji_count);
|
|
|
|
};
|