mirror of
https://github.com/uhIgnacio/EmoteManager.git
synced 2024-08-15 02:23:13 +00:00
cluster startup no longer depend on the first cluster running
This commit is contained in:
parent
0ae7331828
commit
2a73585dfd
1 changed files with 34 additions and 5 deletions
|
@ -18,20 +18,49 @@ from multiprocessing.shared_memory import ShareableList
|
||||||
from discord.ext import commands
|
from discord.ext import commands
|
||||||
from bot_bin.stats import BotBinStats
|
from bot_bin.stats import BotBinStats
|
||||||
|
|
||||||
|
# The resource tracker unlinks shared memory on shutdown and assumes all processes that share memory
|
||||||
|
# share a parent python process. But this is not necessarily the case: the user might decide to run
|
||||||
|
# each cluster as a systemd unit for example. We don't want a crash of a single cluster to unlink
|
||||||
|
# the shared memory segment, so we stub out the resource tracking process here,
|
||||||
|
# as there's no built in way to disable it.
|
||||||
|
# See https://bugs.python.org/issue38119
|
||||||
|
from multiprocessing import resource_tracker
|
||||||
|
resource_tracker.ensure_running = lambda: None
|
||||||
|
resource_tracker.main = lambda: None
|
||||||
|
|
||||||
class Stats(BotBinStats):
|
class Stats(BotBinStats):
|
||||||
def __init__(self, bot):
|
def __init__(self, bot):
|
||||||
super().__init__(bot)
|
super().__init__(bot)
|
||||||
seq = [0] * self.bot.shard_count if self.is_opener() else None
|
|
||||||
# Use our user ID as part of the shm name to allow running multiple instances of the bot on the same machine.
|
# Use our user ID as part of the shm name
|
||||||
self.shlist = ShareableList(seq, name=f'emote-manager-{self.bot.user_id}')
|
# to allow running multiple instances of the bot on the same machine.
|
||||||
|
shm_name = f'emote-manager-{self.bot.user_id}'
|
||||||
|
if self.is_opener():
|
||||||
|
seq = [0] * self.bot.shard_count
|
||||||
|
try:
|
||||||
|
self.shlist = ShareableList(seq, name=shm_name)
|
||||||
|
except FileExistsError:
|
||||||
|
# The file was probably left open from a previous run of the bot,
|
||||||
|
# so let's re-attach and wipe the old data.
|
||||||
|
self.shlist = ShareableList(name=shm_name)
|
||||||
|
# apparently ShareableLists don't support slicing with None values of `stop`
|
||||||
|
self.shlist[:self.bot.shard_count] = seq
|
||||||
|
else:
|
||||||
|
try:
|
||||||
|
self.shlist = ShareableList(name=shm_name)
|
||||||
|
except FileNotFoundError:
|
||||||
|
# looks like we're the first cluster to start up
|
||||||
|
seq = [0] * self.bot.shard_count
|
||||||
|
self.shlist = ShareableList(seq, name=shm_name)
|
||||||
|
|
||||||
self.count()
|
self.count()
|
||||||
|
|
||||||
def is_opener(self):
|
def is_opener(self):
|
||||||
"""return whether this is the process that should open the shared memory"""
|
"""return whether this is the cluster that should open the shared memory"""
|
||||||
return 0 in self.bot.shard_ids
|
return 0 in self.bot.shard_ids
|
||||||
|
|
||||||
def is_reporter(self):
|
def is_reporter(self):
|
||||||
"""return whether we should report stats to the bot lists"""
|
"""return whether this is the cluster that should report stats to the bot lists"""
|
||||||
return self.bot.shard_count - 1 in self.bot.shard_ids
|
return self.bot.shard_count - 1 in self.bot.shard_ids
|
||||||
|
|
||||||
def cog_unload(self):
|
def cog_unload(self):
|
||||||
|
|
Loading…
Reference in a new issue