mirror of
https://github.com/NoiTheCat/BirthdayBot.git
synced 2024-11-25 01:44:12 +00:00
Change frequency of automatic birthday role checks
Changed up how the background services behave. Removed tick count, greatly increased the interval, and added a manual call to update individual guilds as their information changes. Further changes to this may be made in the near future, going as far as to possibly remvoe this structure altogether.
This commit is contained in:
parent
8ca06279e9
commit
b31813c210
7 changed files with 38 additions and 38 deletions
|
@ -4,24 +4,24 @@
|
|||
''' Handles the execution of periodic background tasks.
|
||||
''' </summary>
|
||||
Class BackgroundServiceRunner
|
||||
Const Interval = 45 ' Tick interval in seconds. Adjust as needed.
|
||||
Const Interval = 8 * 60 ' Tick interval in seconds. Adjust as needed.
|
||||
|
||||
Private ReadOnly Property Workers As List(Of BackgroundService)
|
||||
Private ReadOnly Property WorkerCancel As New CancellationTokenSource
|
||||
Friend ReadOnly Property BirthdayUpdater As BirthdayRoleUpdate
|
||||
|
||||
Private _workerTask As Task
|
||||
Private _tickCount As Integer
|
||||
|
||||
Sub New(instance As BirthdayBot)
|
||||
BirthdayUpdater = New BirthdayRoleUpdate(instance)
|
||||
Workers = New List(Of BackgroundService) From {
|
||||
{New GuildStatistics(instance)},
|
||||
{New Heartbeat(instance)},
|
||||
{New BirthdayRoleUpdate(instance)}
|
||||
{BirthdayUpdater}
|
||||
}
|
||||
End Sub
|
||||
|
||||
Public Sub Start()
|
||||
_tickCount = 0
|
||||
_workerTask = Task.Factory.StartNew(AddressOf WorkerLoop, WorkerCancel.Token,
|
||||
TaskCreationOptions.LongRunning, TaskScheduler.Default)
|
||||
End Sub
|
||||
|
@ -43,7 +43,7 @@ Class BackgroundServiceRunner
|
|||
' Execute background tasks.
|
||||
Dim tasks As New List(Of Task)
|
||||
For Each service In Workers
|
||||
tasks.Add(service.OnTick(_tickCount))
|
||||
tasks.Add(service.OnTick())
|
||||
Next
|
||||
Await Task.WhenAll(tasks)
|
||||
Catch ex As TaskCanceledException
|
||||
|
@ -52,7 +52,6 @@ Class BackgroundServiceRunner
|
|||
Log("Background task", "Unhandled exception in background task thread:")
|
||||
Log("Background task", ex.ToString())
|
||||
End Try
|
||||
_tickCount += 1
|
||||
End While
|
||||
End Function
|
||||
End Class
|
||||
|
|
|
@ -13,5 +13,5 @@ MustInherit Class BackgroundService
|
|||
Program.Log(Me.GetType().Name, message)
|
||||
End Sub
|
||||
|
||||
MustOverride Function OnTick(tick As Integer) As Task
|
||||
MustOverride Function OnTick() As Task
|
||||
End Class
|
||||
|
|
|
@ -4,22 +4,20 @@ Imports Discord.WebSocket
|
|||
Imports NodaTime
|
||||
|
||||
''' <summary>
|
||||
''' Periodically scans all known guilds and adjusts birthday role membership as necessary.
|
||||
''' Also handles birthday announcements.
|
||||
''' Core automatic functionality of the bot. Manages role memberships based on birthday information,
|
||||
''' and optionally sends the announcement message to appropriate guilds.
|
||||
''' </summary>
|
||||
Class BirthdayRoleUpdate
|
||||
Inherits BackgroundService
|
||||
Private ReadOnly Property Clock As IClock
|
||||
|
||||
Public Sub New(instance As BirthdayBot)
|
||||
MyBase.New(instance)
|
||||
Clock = SystemClock.Instance ' can be replaced with FakeClock during testing
|
||||
End Sub
|
||||
|
||||
''' <summary>
|
||||
''' Initial processing: Sets up a task per guild and waits on all.
|
||||
''' Does processing on all available guilds at once.
|
||||
''' </summary>
|
||||
Public Overrides Async Function OnTick(tick As Integer) As Task
|
||||
Public Overrides Async Function OnTick() As Task
|
||||
Dim tasks As New List(Of Task(Of Integer))
|
||||
For Each guild In BotInstance.DiscordClient.Guilds
|
||||
Dim t = ProcessGuildAsync(guild)
|
||||
|
@ -38,19 +36,24 @@ Class BirthdayRoleUpdate
|
|||
Next
|
||||
End Try
|
||||
|
||||
' Usage report: Show how many announcements were done
|
||||
Dim announces = 0
|
||||
Dim guilds = 0
|
||||
For Each task In tasks
|
||||
If task.Result > 0 Then
|
||||
announces += task.Result
|
||||
guilds += 1
|
||||
End If
|
||||
Next
|
||||
If announces > 0 Then Log($"Announcing {announces} birthday(s) in {guilds} guild(s).")
|
||||
' TODO metrics for role sets, unsets, announcements - and how to do that for singles too?
|
||||
End Function
|
||||
|
||||
Async Function ProcessGuildAsync(guild As SocketGuild) As Task(Of Integer)
|
||||
''' <summary>
|
||||
''' Does role and announcement processing for a single specified guild.
|
||||
''' </summary>
|
||||
Public Async Function SingleUpdateFor(guild As SocketGuild) As Task
|
||||
Try
|
||||
Await ProcessGuildAsync(guild)
|
||||
Catch ex As Exception
|
||||
Log("Encountered an error during guild processing:")
|
||||
Log(ex.ToString())
|
||||
End Try
|
||||
|
||||
' TODO metrics for role sets, unsets, announcements - and I mentioned this above too
|
||||
End Function
|
||||
|
||||
Private Async Function ProcessGuildAsync(guild As SocketGuild) As Task(Of Integer)
|
||||
' Gather required information
|
||||
Dim tz As String
|
||||
Dim users As IEnumerable(Of GuildUserSettings)
|
||||
|
@ -156,7 +159,7 @@ Class BirthdayRoleUpdate
|
|||
Dim targetMonth = item.BirthMonth
|
||||
Dim targetDay = item.BirthDay
|
||||
|
||||
Dim checkNow = Clock.GetCurrentInstant().InZone(tz)
|
||||
Dim checkNow = SystemClock.Instance.GetCurrentInstant().InZone(tz)
|
||||
' Special case: If birthday is February 29 and it's not a leap year, recognize it on March 1st
|
||||
If targetMonth = 2 And targetDay = 29 And Not Date.IsLeapYear(checkNow.Year) Then
|
||||
targetMonth = 3
|
||||
|
|
|
@ -10,10 +10,7 @@ Class GuildStatistics
|
|||
DBotsToken = instance.Config.DBotsToken
|
||||
End Sub
|
||||
|
||||
Public Overrides Async Function OnTick(tick As Integer) As Task
|
||||
' Activate roughly every 2 hours (interval: 45)
|
||||
If tick Mod 160 <> 2 Then Return
|
||||
|
||||
Public Overrides Async Function OnTick() As Task
|
||||
Dim count = BotInstance.DiscordClient.Guilds.Count
|
||||
Log($"Currently in {count} guild(s).")
|
||||
|
||||
|
|
|
@ -1,5 +1,5 @@
|
|||
''' <summary>
|
||||
''' Basic heartbeat function - indicates that the background task is still functioning.
|
||||
''' Basic heartbeat function - hints that the background task is still alive.
|
||||
''' </summary>
|
||||
Class Heartbeat
|
||||
Inherits BackgroundService
|
||||
|
@ -8,19 +8,16 @@ Class Heartbeat
|
|||
MyBase.New(instance)
|
||||
End Sub
|
||||
|
||||
Public Overrides Function OnTick(tick As Integer) As Task
|
||||
' Print a message roughly every 15 minutes (assuming 45s per tick).
|
||||
If tick Mod 20 = 0 Then
|
||||
Public Overrides Function OnTick() As Task
|
||||
Dim uptime = DateTimeOffset.UtcNow - Program.BotStartTime
|
||||
Log($"Tick {tick:00000} - Bot uptime: {BotUptime()}")
|
||||
End If
|
||||
Log($"Bot uptime: {BotUptime()}")
|
||||
|
||||
' Disconnection warn
|
||||
For Each shard In BotInstance.DiscordClient.Shards
|
||||
If shard.ConnectionState = Discord.ConnectionState.Disconnected Then
|
||||
Log($"Shard {shard.ShardId} is disconnected! Restart the app if this persists.")
|
||||
' The library alone cannot be restarted as it is in an unknown state. It was not designed to be restarted.
|
||||
' This is the part where we'd signal something to restart us if we were fancy.
|
||||
' TODO This is the part where we'd signal something to restart us if we were fancy.
|
||||
End If
|
||||
Next
|
||||
|
||||
|
|
|
@ -118,6 +118,7 @@ Class BirthdayBot
|
|||
End If
|
||||
End If
|
||||
|
||||
' Execute the command
|
||||
Try
|
||||
Log("Command", $"{channel.Guild.Name}/{author.Username}#{author.Discriminator}: {msg.Content}")
|
||||
Await command(csplit, channel, author)
|
||||
|
@ -130,6 +131,9 @@ Class BirthdayBot
|
|||
' Fail silently.
|
||||
End Try
|
||||
End Try
|
||||
|
||||
' Immediately check for role updates in the invoking guild
|
||||
Await _worker.BirthdayUpdater.SingleUpdateFor(channel.Guild)
|
||||
End If
|
||||
End If
|
||||
End Function
|
||||
|
|
|
@ -4,7 +4,7 @@
|
|||
<OutputType>Exe</OutputType>
|
||||
<RootNamespace>BirthdayBot</RootNamespace>
|
||||
<TargetFramework>netcoreapp2.0</TargetFramework>
|
||||
<Version>1.3.3</Version>
|
||||
<Version>1.3.4</Version>
|
||||
<Authors>Noi</Authors>
|
||||
<Company />
|
||||
<Description>Discord bot for birthday reminders.</Description>
|
||||
|
|
Loading…
Reference in a new issue