1
0
Fork 1
mirror of https://github.com/TeamOctolings/Octobot.git synced 2025-04-19 16:33:36 +03:00

feat: profile reminder commands (i haven't tested it lmao)

Signed-off-by: Octol1ttle <l1ttleofficial@outlook.com>
This commit is contained in:
Octol1ttle 2023-12-31 14:38:15 +05:00
parent 5f007ada74
commit 39823b36fc
Signed by: Octol1ttle
GPG key ID: B77C34313AEE1FFF
3 changed files with 85 additions and 23 deletions

View file

@ -5,6 +5,7 @@ using JetBrains.Annotations;
using Octobot.Data; using Octobot.Data;
using Octobot.Extensions; using Octobot.Extensions;
using Octobot.Services; using Octobot.Services;
using Octobot.Services.Profiler;
using Remora.Commands.Attributes; using Remora.Commands.Attributes;
using Remora.Commands.Groups; using Remora.Commands.Groups;
using Remora.Discord.API.Abstractions.Objects; using Remora.Discord.API.Abstractions.Objects;
@ -30,18 +31,20 @@ public class RemindCommandGroup : CommandGroup
private readonly IInteractionCommandContext _context; private readonly IInteractionCommandContext _context;
private readonly IFeedbackService _feedback; private readonly IFeedbackService _feedback;
private readonly GuildDataService _guildData; private readonly GuildDataService _guildData;
private readonly IDiscordRestUserAPI _userApi;
private readonly IDiscordRestInteractionAPI _interactionApi; private readonly IDiscordRestInteractionAPI _interactionApi;
private readonly Profiler _profiler;
private readonly IDiscordRestUserAPI _userApi;
public RemindCommandGroup( public RemindCommandGroup(
IInteractionCommandContext context, GuildDataService guildData, IFeedbackService feedback, IInteractionCommandContext context, GuildDataService guildData, IFeedbackService feedback,
IDiscordRestUserAPI userApi, IDiscordRestInteractionAPI interactionApi) IDiscordRestUserAPI userApi, IDiscordRestInteractionAPI interactionApi, Profiler profiler)
{ {
_context = context; _context = context;
_guildData = guildData; _guildData = guildData;
_feedback = feedback; _feedback = feedback;
_userApi = userApi; _userApi = userApi;
_interactionApi = interactionApi; _interactionApi = interactionApi;
_profiler = profiler;
} }
/// <summary> /// <summary>
@ -53,59 +56,79 @@ public class RemindCommandGroup : CommandGroup
[DiscordDefaultDMPermission(false)] [DiscordDefaultDMPermission(false)]
[RequireContext(ChannelContext.Guild)] [RequireContext(ChannelContext.Guild)]
[UsedImplicitly] [UsedImplicitly]
public async Task<Result> ExecuteListReminderAsync() public async Task<Result> ExecuteListRemindersAsync()
{ {
_profiler.Push("list_reminders_command");
_profiler.Push("preparation");
if (!_context.TryGetContextIDs(out var guildId, out _, out var executorId)) if (!_context.TryGetContextIDs(out var guildId, out _, out var executorId))
{ {
return new ArgumentInvalidError(nameof(_context), "Unable to retrieve necessary IDs from command context"); return _profiler.ReportWithResult(new ArgumentInvalidError(nameof(_context),
"Unable to retrieve necessary IDs from command context"));
} }
_profiler.Push("current_user_get");
var botResult = await _userApi.GetCurrentUserAsync(CancellationToken); var botResult = await _userApi.GetCurrentUserAsync(CancellationToken);
if (!botResult.IsDefined(out var bot)) if (!botResult.IsDefined(out var bot))
{ {
return Result.FromError(botResult); return _profiler.ReportWithResult(Result.FromError(botResult));
} }
_profiler.Pop();
_profiler.Push("executor_get");
var executorResult = await _userApi.GetUserAsync(executorId, CancellationToken); var executorResult = await _userApi.GetUserAsync(executorId, CancellationToken);
if (!executorResult.IsDefined(out var executor)) if (!executorResult.IsDefined(out var executor))
{ {
return Result.FromError(executorResult); return _profiler.ReportWithResult(Result.FromError(executorResult));
} }
_profiler.Pop();
_profiler.Push("guild_data_get");
var data = await _guildData.GetData(guildId, CancellationToken); var data = await _guildData.GetData(guildId, CancellationToken);
Messages.Culture = GuildSettings.Language.Get(data.Settings); Messages.Culture = GuildSettings.Language.Get(data.Settings);
_profiler.Pop();
return await ListRemindersAsync(data.GetOrCreateMemberData(executorId), guildId, executor, bot, CancellationToken); _profiler.Pop();
return _profiler.ReportWithResult(await ListRemindersAsync(data.GetOrCreateMemberData(executorId), guildId,
executor, bot,
CancellationToken));
} }
private Task<Result> ListRemindersAsync(MemberData data, Snowflake guildId, IUser executor, IUser bot, CancellationToken ct) private Task<Result> ListRemindersAsync(MemberData data, Snowflake guildId, IUser executor, IUser bot,
CancellationToken ct)
{ {
_profiler.Push("main");
if (data.Reminders.Count == 0) if (data.Reminders.Count == 0)
{ {
_profiler.Push("no_reminders_send");
var failedEmbed = new EmbedBuilder().WithSmallTitle(Messages.NoRemindersFound, bot) var failedEmbed = new EmbedBuilder().WithSmallTitle(Messages.NoRemindersFound, bot)
.WithColour(ColorsList.Red) .WithColour(ColorsList.Red)
.Build(); .Build();
return _feedback.SendContextualEmbedResultAsync(failedEmbed, ct: ct); return _profiler.PopWithResult(_feedback.SendContextualEmbedResultAsync(failedEmbed, ct: ct));
} }
_profiler.Push("builder_construction");
var builder = new StringBuilder(); var builder = new StringBuilder();
for (var i = 0; i < data.Reminders.Count; i++) for (var i = 0; i < data.Reminders.Count; i++)
{ {
var reminder = data.Reminders[i]; var reminder = data.Reminders[i];
builder.AppendBulletPointLine(string.Format(Messages.ReminderPosition, Markdown.InlineCode((i + 1).ToString()))) builder.AppendBulletPointLine(string.Format(Messages.ReminderPosition,
Markdown.InlineCode((i + 1).ToString())))
.AppendSubBulletPointLine(string.Format(Messages.ReminderText, Markdown.InlineCode(reminder.Text))) .AppendSubBulletPointLine(string.Format(Messages.ReminderText, Markdown.InlineCode(reminder.Text)))
.AppendSubBulletPointLine(string.Format(Messages.ReminderTime, Markdown.Timestamp(reminder.At))) .AppendSubBulletPointLine(string.Format(Messages.ReminderTime, Markdown.Timestamp(reminder.At)))
.AppendSubBulletPointLine(string.Format(Messages.DescriptionActionJumpToMessage, $"https://discord.com/channels/{guildId.Value}/{reminder.ChannelId}/{reminder.MessageId}")); .AppendSubBulletPointLine(string.Format(Messages.DescriptionActionJumpToMessage,
$"https://discord.com/channels/{guildId.Value}/{reminder.ChannelId}/{reminder.MessageId}"));
} }
_profiler.Pop();
_profiler.Push("embed_send");
var embed = new EmbedBuilder().WithSmallTitle( var embed = new EmbedBuilder().WithSmallTitle(
string.Format(Messages.ReminderList, executor.GetTag()), executor) string.Format(Messages.ReminderList, executor.GetTag()), executor)
.WithDescription(builder.ToString()) .WithDescription(builder.ToString())
.WithColour(ColorsList.Cyan) .WithColour(ColorsList.Cyan)
.Build(); .Build();
return _feedback.SendContextualEmbedResultAsync(embed, ct: ct); return _profiler.PopWithResult(_feedback.SendContextualEmbedResultAsync(embed, ct: ct));
} }
/// <summary> /// <summary>
@ -126,9 +149,12 @@ public class RemindCommandGroup : CommandGroup
[Description("Reminder text")] [MaxLength(512)] [Description("Reminder text")] [MaxLength(512)]
string text) string text)
{ {
_profiler.Push("reminder_command");
_profiler.Push("preparation");
if (!_context.TryGetContextIDs(out var guildId, out var channelId, out var executorId)) if (!_context.TryGetContextIDs(out var guildId, out var channelId, out var executorId))
{ {
return new ArgumentInvalidError(nameof(_context), "Unable to retrieve necessary IDs from command context"); return _profiler.ReportWithResult(new ArgumentInvalidError(nameof(_context),
"Unable to retrieve necessary IDs from command context"));
} }
var botResult = await _userApi.GetCurrentUserAsync(CancellationToken); var botResult = await _userApi.GetCurrentUserAsync(CancellationToken);
@ -137,14 +163,18 @@ public class RemindCommandGroup : CommandGroup
return Result.FromError(botResult); return Result.FromError(botResult);
} }
_profiler.Push("executor_get");
var executorResult = await _userApi.GetUserAsync(executorId, CancellationToken); var executorResult = await _userApi.GetUserAsync(executorId, CancellationToken);
if (!executorResult.IsDefined(out var executor)) if (!executorResult.IsDefined(out var executor))
{ {
return Result.FromError(executorResult); return _profiler.ReportWithResult(Result.FromError(executorResult));
} }
_profiler.Pop();
_profiler.Push("guild_data_get");
var data = await _guildData.GetData(guildId, CancellationToken); var data = await _guildData.GetData(guildId, CancellationToken);
Messages.Culture = GuildSettings.Language.Get(data.Settings); Messages.Culture = GuildSettings.Language.Get(data.Settings);
_profiler.Pop();
var parseResult = TimeSpanParser.TryParse(timeSpanString); var parseResult = TimeSpanParser.TryParse(timeSpanString);
if (!parseResult.IsDefined(out var timeSpan)) if (!parseResult.IsDefined(out var timeSpan))
@ -156,6 +186,10 @@ public class RemindCommandGroup : CommandGroup
return await _feedback.SendContextualEmbedResultAsync(failedEmbed, ct: CancellationToken); return await _feedback.SendContextualEmbedResultAsync(failedEmbed, ct: CancellationToken);
} }
_profiler.Pop();
return _profiler.ReportWithResult(await AddReminderAsync(@in, text, data, channelId, executor,
CancellationToken));
}
return await AddReminderAsync(timeSpan, text, data, channelId, executor, CancellationToken); return await AddReminderAsync(timeSpan, text, data, channelId, executor, CancellationToken);
} }
@ -163,14 +197,21 @@ public class RemindCommandGroup : CommandGroup
private async Task<Result> AddReminderAsync(TimeSpan timeSpan, string text, GuildData data, private async Task<Result> AddReminderAsync(TimeSpan timeSpan, string text, GuildData data,
Snowflake channelId, IUser executor, CancellationToken ct = default) Snowflake channelId, IUser executor, CancellationToken ct = default)
{ {
_profiler.Push("main");
var memberData = data.GetOrCreateMemberData(executor.ID); var memberData = data.GetOrCreateMemberData(executor.ID);
var remindAt = DateTimeOffset.UtcNow.Add(timeSpan); var remindAt = DateTimeOffset.UtcNow.Add(@in);
var responseResult = await _interactionApi.GetOriginalInteractionResponseAsync(_context.Interaction.ApplicationID, _context.Interaction.Token, ct);
_profiler.Push("original_response_get");
var responseResult =
await _interactionApi.GetOriginalInteractionResponseAsync(_context.Interaction.ApplicationID,
_context.Interaction.Token, ct);
if (!responseResult.IsDefined(out var response)) if (!responseResult.IsDefined(out var response))
{ {
return (Result)responseResult; return _profiler.PopWithResult(Result.FromError(responseResult));
} }
_profiler.Pop();
memberData.Reminders.Add( memberData.Reminders.Add(
new Reminder new Reminder
{ {
@ -180,6 +221,7 @@ public class RemindCommandGroup : CommandGroup
MessageId = response.ID.Value MessageId = response.ID.Value
}); });
_profiler.Push("embed_send");
var builder = new StringBuilder() var builder = new StringBuilder()
.AppendBulletPointLine(string.Format(Messages.ReminderText, Markdown.InlineCode(text))) .AppendBulletPointLine(string.Format(Messages.ReminderText, Markdown.InlineCode(text)))
.AppendBulletPoint(string.Format(Messages.ReminderTime, Markdown.Timestamp(remindAt))); .AppendBulletPoint(string.Format(Messages.ReminderTime, Markdown.Timestamp(remindAt)));
@ -190,7 +232,7 @@ public class RemindCommandGroup : CommandGroup
.WithFooter(string.Format(Messages.ReminderPosition, memberData.Reminders.Count)) .WithFooter(string.Format(Messages.ReminderPosition, memberData.Reminders.Count))
.Build(); .Build();
return await _feedback.SendContextualEmbedResultAsync(embed, ct: ct); return _profiler.PopWithResult(await _feedback.SendContextualEmbedResultAsync(embed, ct: ct));
} }
/// <summary> /// <summary>
@ -207,33 +249,44 @@ public class RemindCommandGroup : CommandGroup
[Description("Position in list")] [MinValue(1)] [Description("Position in list")] [MinValue(1)]
int position) int position)
{ {
_profiler.Push("delete_reminder_command");
_profiler.Push("preparation");
if (!_context.TryGetContextIDs(out var guildId, out _, out var executorId)) if (!_context.TryGetContextIDs(out var guildId, out _, out var executorId))
{ {
return new ArgumentInvalidError(nameof(_context), "Unable to retrieve necessary IDs from command context"); return _profiler.ReportWithResult(new ArgumentInvalidError(nameof(_context),
"Unable to retrieve necessary IDs from command context"));
} }
_profiler.Push("current_user_get");
var botResult = await _userApi.GetCurrentUserAsync(CancellationToken); var botResult = await _userApi.GetCurrentUserAsync(CancellationToken);
if (!botResult.IsDefined(out var bot)) if (!botResult.IsDefined(out var bot))
{ {
return Result.FromError(botResult); return _profiler.ReportWithResult(Result.FromError(botResult));
} }
_profiler.Pop();
_profiler.Push("guild_data_get");
var data = await _guildData.GetData(guildId, CancellationToken); var data = await _guildData.GetData(guildId, CancellationToken);
Messages.Culture = GuildSettings.Language.Get(data.Settings); Messages.Culture = GuildSettings.Language.Get(data.Settings);
_profiler.Pop();
return await DeleteReminderAsync(data.GetOrCreateMemberData(executorId), position - 1, bot, CancellationToken); _profiler.Pop();
return _profiler.ReportWithResult(await DeleteReminderAsync(data.GetOrCreateMemberData(executorId),
position - 1, bot, CancellationToken));
} }
private Task<Result> DeleteReminderAsync(MemberData data, int index, IUser bot, private Task<Result> DeleteReminderAsync(MemberData data, int index, IUser bot,
CancellationToken ct) CancellationToken ct)
{ {
_profiler.Push("main");
if (index >= data.Reminders.Count) if (index >= data.Reminders.Count)
{ {
_profiler.Push("invalid_position_send");
var failedEmbed = new EmbedBuilder().WithSmallTitle(Messages.InvalidReminderPosition, bot) var failedEmbed = new EmbedBuilder().WithSmallTitle(Messages.InvalidReminderPosition, bot)
.WithColour(ColorsList.Red) .WithColour(ColorsList.Red)
.Build(); .Build();
return _feedback.SendContextualEmbedResultAsync(failedEmbed, ct: ct); return _profiler.PopWithResult(_feedback.SendContextualEmbedResultAsync(failedEmbed, ct: ct));
} }
var reminder = data.Reminders[index]; var reminder = data.Reminders[index];
@ -249,6 +302,6 @@ public class RemindCommandGroup : CommandGroup
.WithColour(ColorsList.Green) .WithColour(ColorsList.Green)
.Build(); .Build();
return _feedback.SendContextualEmbedResultAsync(embed, ct: ct); return _profiler.PopWithResult(_feedback.SendContextualEmbedResultAsync(embed, ct: ct));
} }
} }

View file

@ -63,6 +63,13 @@ public sealed class Profiler
return result; return result;
} }
public Task<Result> PopWithResult(Task<Result> result)
{
LogResultStackTrace(result.GetAwaiter().GetResult());
Pop();
return result;
}
/// <summary> /// <summary>
/// If the profiler took too long to execute, this will log a warning with per-event time usage /// If the profiler took too long to execute, this will log a warning with per-event time usage
/// </summary> /// </summary>

View file

@ -18,6 +18,8 @@ public sealed class ProfilerFactory
/// Creates a new <see cref="Profiler"/>. /// Creates a new <see cref="Profiler"/>.
/// </summary> /// </summary>
/// <returns>A new <see cref="Profiler"/>.</returns> /// <returns>A new <see cref="Profiler"/>.</returns>
// TODO: remove in future profiler PRs
// ReSharper disable once UnusedMember.Global
public Profiler Create() public Profiler Create()
{ {
return _scopeFactory.CreateScope().ServiceProvider.GetRequiredService<Profiler>(); return _scopeFactory.CreateScope().ServiceProvider.GetRequiredService<Profiler>();