Sweetbread/natsuko
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[perf]: Make db writes only on change, turn per day writes off

Browse Source
This commit is contained in:
Sweetbread 2023-06-18 01:46:54 +03:00
parent 59d6c33a00
commit ee311c9af3
2 changed files with 11 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cogs/economic.py
View File

@ -514,6 +514,8 @@ class Economic(commands.Cog, name="Экономика"):
else: else:
vals = [info[key] for key in info.keys() if int(key) >= ts-period*60**2] vals = [info[key] for key in info.keys() if int(key) >= ts-period*60**2]
if not vals: continue
ax.plot(list(map(int, info.keys()))[-len(vals):], vals, marker=marker, label=self.bot.get_user(user['id']).name) ax.plot(list(map(int, info.keys()))[-len(vals):], vals, marker=marker, label=self.bot.get_user(user['id']).name)
ax.xaxis.set_major_formatter(ticker.FuncFormatter(lambda x, pos: datetime.fromtimestamp(x).strftime('%d.%m'))) ax.xaxis.set_major_formatter(ticker.FuncFormatter(lambda x, pos: datetime.fromtimestamp(x).strftime('%d.%m')))

20
cogs/timeble.py
View File

@ -13,30 +13,20 @@ async def save_activity():
{"type": "server", "id": data['id']}, {"type": "server", "id": data['id']},
{ {
"$push": "$push":
{f"avarage.{i}": data['current'][str(i)] for i in range(24)}, {f"avarage.{i}": data['current'][str(i)] for i in range(24)} | {f"history.{daysoftheweek[day]}.{i}": data['current'][str(i)] for i in range(24)},
"$set": "$set":
{f'yesterday.{i}': data['current'][str(i)] for i in range(24)} {f'yesterday.{i}': data['current'][str(i)] for i in range(24)} | {f"current.{i}": 0 for i in range(24)}
}
)
await db.history.update_one(
{"type": "server", "id": data['id']},
{
"$push":
{f"history.{daysoftheweek[day]}.{i}": data['current'][str(i)] for i in range(24)},
"$set":
{f"current.{i}": 0 for i in range(24)}
} }
) )
async def db_sync_hour(): async def db_sync_hour():
logger.info("Регистрация за час")
async for member in db.members.find(): async for member in db.members.find():
if member['history']['hour'][list(member['history']['hour'].keys())[-1]] != member['exp']:
await db.members.update_one({"_id": member["_id"]}, {"$set": {f"history.hour.{int(datetime.now().timestamp())}": member["exp"]}}) await db.members.update_one({"_id": member["_id"]}, {"$set": {f"history.hour.{int(datetime.now().timestamp())}": member["exp"]}})
if 'guild_stat' in member.keys(): if 'guild_stat' in member.keys():
for guild in member["guild_stat"].keys(): for guild in member["guild_stat"].keys():
if member['guild_stat'][guild]['history']['hour'][list(member['guild_stat'][guild]['history']['hour'].keys())[-1]] != member['guild_stat'][guild]['exp']:
await db.members.update_one({"_id": member["_id"]}, {"$set": {f"guild_stat.{guild}.history.hour.{int(datetime.now().timestamp())}": member["guild_stat"][guild]["exp"]}}) await db.members.update_one({"_id": member["_id"]}, {"$set": {f"guild_stat.{guild}.history.hour.{int(datetime.now().timestamp())}": member["guild_stat"][guild]["exp"]}})
logger.info("Регистрация завершена")
async def db_sync_day(): async def db_sync_day():
logger.info("Регистрация за день") logger.info("Регистрация за день")
@ -53,7 +43,7 @@ async def main(bot):
every().day.at('00:00:00').do(lambda: asyncio.run_coroutine_threadsafe(save_activity(), bot.loop)) every().day.at('00:00:00').do(lambda: asyncio.run_coroutine_threadsafe(save_activity(), bot.loop))
every().hours.at("00:00").do(lambda: asyncio.run_coroutine_threadsafe(db_sync_hour(), bot.loop)) every().hours.at("00:00").do(lambda: asyncio.run_coroutine_threadsafe(db_sync_hour(), bot.loop))
every().day.at("00:00:00").do(lambda: asyncio.run_coroutine_threadsafe(db_sync_day(), bot.loop)) # every().day.at("00:00:00").do(lambda: asyncio.run_coroutine_threadsafe(db_sync_day(), bot.loop))
while True: while True:
run_pending() run_pending()