Noi/BirthdayBot
1
0
Fork 0
mirror of https://github.com/NoiTheCat/BirthdayBot.git synced 2024-11-21 21:54:36 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Implement per-shard reporting

Browse source
This commit is contained in:
Noi 2020-05-21 23:53:51 -07:00
parent 2965dd0ea1
commit 2a15fc21c8
1 changed files with 65 additions and 25 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

90
BackgroundServices/GuildStatistics.cs
View file

@ -1,48 +1,88 @@
using System; using System;
using System.Net; using System.Collections.Generic;
using System.Linq;
using System.Net.Http;
using System.Text;
using System.Threading.Tasks; using System.Threading.Tasks;
namespace BirthdayBot.BackgroundServices namespace BirthdayBot.BackgroundServices
{ {
class GuildStatistics : BackgroundService class GuildStatistics : BackgroundService
{ {
private string DBotsToken { get; } private static readonly HttpClient _httpClient = new HttpClient();
public GuildStatistics(BirthdayBot instance) : base(instance) => DBotsToken = instance.Config.DBotsToken; public GuildStatistics(BirthdayBot instance) : base(instance) { }
public async override Task OnTick() public async override Task OnTick()
{ {
var count = BotInstance.DiscordClient.Guilds.Count; var counts = GetGuildCounts();
Log($"Currently in {count} guild(s).");
await SendExternalStatistics(count); // Build this report
int goodCount = 0;
int badCount = 0;
int goodShards = 0;
int badShards = 0;
foreach (var status in counts)
{
if (status.Item2.ShardConnected)
{
goodShards++;
goodCount += status.Item2.GuildCount;
}
else
{
badShards++;
badCount += status.Item2.GuildCount;
}
}
Log($"{goodShards} shard(s) connected, serving {goodCount} guild(s).");
if (badShards != 0) Log($"{badShards} shard(s) unavailable, >={badCount} guild(s) will not count in the next report.");
// Report only connected shards (to not have fluctuating member numbers on initial startup)
await SendExternalStatistics(counts.Where(t => t.Item2.ShardConnected), counts.Count());
}
private struct ShardStatus
{
public bool ShardConnected;
public int GuildCount;
}
private IEnumerable<(int, ShardStatus)> GetGuildCounts()
{
var results = new List<(int, ShardStatus)>();
var shards = BotInstance.DiscordClient.Shards;
foreach (var shard in shards)
{
results.Add((shard.ShardId, new ShardStatus()
{
ShardConnected = shard.ConnectionState == Discord.ConnectionState.Connected,
GuildCount = shard.Guilds.Count
}));
}
return results;
} }
/// <summary> /// <summary>
/// Send statistical information to external services. /// Send statistical information to external services.
/// </summary> /// </summary>
/// <remarks> async Task SendExternalStatistics(IEnumerable<(int, ShardStatus)> shardStats, int totalShards)
/// Only Discord Bots is currently supported. No plans to support others any time soon.
/// </remarks>
async Task SendExternalStatistics(int guildCount)
{ {
var rptToken = BotInstance.Config.DBotsToken; var dbotsToken = BotInstance.Config.DBotsToken;
if (rptToken == null) return; if (dbotsToken != null)
const string apiUrl = "https://discord.bots.gg/api/v1/bots/{0}/stats";
using (var client = new WebClient())
{ {
var uri = new Uri(string.Format(apiUrl, BotInstance.DiscordClient.CurrentUser.Id)); const string dBotsApiUrl = "https://discord.bots.gg/api/v1/bots/{0}/stats";
var data = $"{{ \"guildCount\": {guildCount} }}"; const string Body = "{{ \"shardCount\": {0}, \"shardId\": {1}, \"guildCount\": {2} }}";
client.Headers[HttpRequestHeader.Authorization] = rptToken; var uri = new Uri(string.Format(dBotsApiUrl, BotInstance.DiscordClient.CurrentUser.Id));
client.Headers[HttpRequestHeader.ContentType] = "application/json"; foreach (var shard in shardStats)
try
{ {
await client.UploadStringTaskAsync(uri, data); var post = new HttpRequestMessage(HttpMethod.Post, uri);
Log("Discord Bots: Report sent successfully."); post.Headers.Add("Authorization", dbotsToken);
} catch (WebException ex) var data = string.Format(Body, totalShards, shard.Item1, shard.Item2.GuildCount);
{ post.Content = new StringContent(data, Encoding.UTF8, "application/json");
Log("Discord Bots: Encountered an error. " + ex.Message); await Task.Delay(80); // Discord Bots rate limit for this endpoint is 20 per second
await _httpClient.SendAsync(post);
Log("Discord Bots: Reports sent successfully.");
} }
} }
} }