mattburchett/NadekoBot
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

.divorce cooldown now properly works on multiple shards

Browse Source
This commit is contained in:
Master Kwoth
2017-11-03 12:46:51 +01:00
parent 0dbd60b5ac
commit 1c157ddfc1
4 changed files with 18 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
NadekoBot.Core/Modules/Gambling/Services/WaifuService.cs
View File

@@ -1,7 +1,5 @@
using Discord;
using NadekoBot.Core.Services;
using System;
using System.Collections.Concurrent;
using System.Threading.Tasks;
namespace NadekoBot.Modules.Gambling.Services
@@ -11,9 +9,6 @@ namespace NadekoBot.Modules.Gambling.Services
private readonly DbService _db;
private readonly CurrencyService _cs;
public ConcurrentDictionary<ulong, DateTime> DivorceCooldowns { get; } = new ConcurrentDictionary<ulong, DateTime>();
public ConcurrentDictionary<ulong, DateTime> AffinityCooldowns { get; } = new ConcurrentDictionary<ulong, DateTime>();
public WaifuService(DbService db, CurrencyService cs)
{
_db = db;

13
NadekoBot.Core/Modules/Gambling/WaifuClaimCommands.cs
View File

@@ -213,8 +213,6 @@ namespace NadekoBot.Modules.Gambling
Cooldown
}
private static readonly TimeSpan _divorceLimit = TimeSpan.FromHours(6);
[NadekoCommand, Usage, Description, Aliases]
[RequireContext(ContextType.Guild)]
[Priority(0)]
@@ -229,7 +227,7 @@ namespace NadekoBot.Modules.Gambling
return;
DivorceResult result;
var difference = TimeSpan.Zero;
TimeSpan? remaining = null;
var amount = 0;
WaifuInfo w = null;
using (var uow = _db.UnitOfWork)
@@ -238,9 +236,7 @@ namespace NadekoBot.Modules.Gambling
var now = DateTime.UtcNow;
if (w?.Claimer == null || w.Claimer.UserId != Context.User.Id)
result = DivorceResult.NotYourWife;
else if (_service.DivorceCooldowns.AddOrUpdate(Context.User.Id,
now,
(key, old) => ((difference = now.Subtract(old)) > _divorceLimit) ? now : old) != now)
else if (!_cache.TryAddDivorceCooldown(Context.User.Id, out remaining))
{
result = DivorceResult.Cooldown;
}
@@ -289,10 +285,9 @@ namespace NadekoBot.Modules.Gambling
}
else
{
var remaining = _divorceLimit.Subtract(difference);
await ReplyErrorLocalized("waifu_recent_divorce",
Format.Bold(((int)remaining.TotalHours).ToString()),
Format.Bold(remaining.Minutes.ToString())).ConfigureAwait(false);
Format.Bold(((int)remaining?.TotalHours).ToString()),
Format.Bold(remaining?.Minutes.ToString())).ConfigureAwait(false);
}
}