minster586/nadekobot
1
0
Fork 0
mirror of https://gitlab.com/Kwoth/nadekobot.git synced 2025-09-11 01:38:27 -04:00
Code Issues Packages Projects Releases Wiki Activity

small wallet change

Browse Source
This commit is contained in:
Kwoth
2022-03-25 01:07:36 +01:00
parent 9bf9e06dbf
commit 95bde7021a
1 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/NadekoBot/Services/Currency/DefaultWallet.cs
View File

@@ -19,9 +19,10 @@ public class DefaultWallet : IWallet
public async Task<long> GetBalance() public async Task<long> GetBalance()
{ {
await using var ctx = _db.GetDbContext(); await using var ctx = _db.GetDbContext();
var userId = UserId;
return await ctx.DiscordUser return await ctx.DiscordUser
.ToLinqToDBTable() .ToLinqToDBTable()
.Where(x => x.UserId == UserId) .Where(x => x.UserId == userId)
.Select(x => x.CurrencyAmount) .Select(x => x.CurrencyAmount)
.FirstOrDefaultAsync(); .FirstOrDefaultAsync();
} }
@@ -33,8 +34,9 @@ public class DefaultWallet : IWallet
await using var ctx = _db.GetDbContext(); await using var ctx = _db.GetDbContext();
var userId = UserId;
var changed = await ctx.DiscordUser var changed = await ctx.DiscordUser
.Where(x => x.UserId == UserId && x.CurrencyAmount >= amount) .Where(x => x.UserId == userId && x.CurrencyAmount >= amount)
.UpdateAsync(x => new() .UpdateAsync(x => new()
{ {
CurrencyAmount = x.CurrencyAmount - amount CurrencyAmount = x.CurrencyAmount - amount
@@ -42,14 +44,14 @@ public class DefaultWallet : IWallet
if (changed == 0) if (changed == 0)
return false; return false;
await ctx await ctx
.GetTable<CurrencyTransaction>() .GetTable<CurrencyTransaction>()
.InsertAsync(() => new() .InsertAsync(() => new()
{ {
Amount = -amount, Amount = -amount,
Note = txData.Note, Note = txData.Note,
UserId = UserId, UserId = userId,
Type = txData.Type, Type = txData.Type,
Extra = txData.Extra, Extra = txData.Extra,
OtherId = txData.OtherId, OtherId = txData.OtherId,
@@ -65,11 +67,12 @@ public class DefaultWallet : IWallet
throw new ArgumentOutOfRangeException(nameof(amount), "Amount must be greater than 0."); throw new ArgumentOutOfRangeException(nameof(amount), "Amount must be greater than 0.");
await using var ctx = _db.GetDbContext(); await using var ctx = _db.GetDbContext();
var userId = UserId;
await using (var tran = await ctx.Database.BeginTransactionAsync()) await using (var tran = await ctx.Database.BeginTransactionAsync())
{ {
var changed = await ctx.DiscordUser var changed = await ctx.DiscordUser
.Where(x => x.UserId == UserId) .Where(x => x.UserId == userId)
.UpdateAsync(x => new() .UpdateAsync(x => new()
{ {
CurrencyAmount = x.CurrencyAmount + amount CurrencyAmount = x.CurrencyAmount + amount
@@ -79,7 +82,7 @@ public class DefaultWallet : IWallet
{ {
await ctx.DiscordUser await ctx.DiscordUser
.ToLinqToDBTable() .ToLinqToDBTable()
.Value(x => x.UserId, UserId) .Value(x => x.UserId, userId)
.Value(x => x.Username, "Unknown") .Value(x => x.Username, "Unknown")
.Value(x => x.Discriminator, "????") .Value(x => x.Discriminator, "????")
.Value(x => x.CurrencyAmount, amount) .Value(x => x.CurrencyAmount, amount)
@@ -93,7 +96,7 @@ public class DefaultWallet : IWallet
.InsertAsync(() => new() .InsertAsync(() => new()
{ {
Amount = amount, Amount = amount,
UserId = UserId, UserId = userId,
Note = txData.Note, Note = txData.Note,
Type = txData.Type, Type = txData.Type,
Extra = txData.Extra, Extra = txData.Extra,