mirrors/btcpayserver
1
0
Fork 0
mirror of https://github.com/btcpayserver/btcpayserver.git synced 2024-11-19 01:43:50 +01:00
Code Issues Packages Projects Releases Wiki Activity

Introduce wallet object contagion

Browse Source
This commit is contained in:
Kukks 2022-11-23 08:24:24 +01:00
parent d3e2d5a095
commit 0c144824c8
No known key found for this signature in database
GPG Key ID: 8E5530D9D1C93097
5 changed files with 1640 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
BTCPayServer.Data/Data/WalletObjectLinkData.cs
View File

@ -19,6 +19,7 @@ namespace BTCPayServer.Data
public WalletObjectData A { get; set; }
public WalletObjectData B { get; set; }
public int InfectionRate { get; set; } = 0;
internal static void OnModelCreating(ModelBuilder builder, DatabaseFacade databaseFacade)
{

1565
BTCPayServer.Data/Migrations/20221123072311_WalletObjectInfection.Designer.cs generated Normal file
View File

File diff suppressed because it is too large Load Diff

26
BTCPayServer.Data/Migrations/20221123072311_WalletObjectInfection.cs Normal file
View File

@ -0,0 +1,26 @@
using Microsoft.EntityFrameworkCore.Migrations;
#nullable disable
namespace BTCPayServer.Migrations
{
public partial class WalletObjectInfection : Migration
{
protected override void Up(MigrationBuilder migrationBuilder)
{
migrationBuilder.AddColumn<int>(
name: "InfectionRate",
table: "WalletObjectLinks",
type: "INTEGER",
nullable: false,
defaultValue: 0);
}
protected override void Down(MigrationBuilder migrationBuilder)
{
migrationBuilder.DropColumn(
name: "InfectionRate",
table: "WalletObjectLinks");
}
}
}

6
BTCPayServer.Data/Migrations/ApplicationDbContextModelSnapshot.cs
View File

@ -886,8 +886,12 @@ namespace BTCPayServer.Migrations
b.Property<string>("Data")
.HasColumnType("TEXT");
b.Property<int>("InfectionRate")
b.HasKey("WalletId", "AType", "AId", "BType", "BId");
.HasColumnType("INTEGER");
b.HasKey("WalletId", "AType", "AId", "BType", "BId");
b.HasIndex("WalletId", "BType", "BId");

45
BTCPayServer/Services/WalletRepository.cs
View File

@ -325,7 +325,7 @@ namespace BTCPayServer.Services
}
}
public async Task EnsureWalletObjectLink(WalletObjectId a, WalletObjectId b, JObject? data = null)
public async Task EnsureWalletObjectLink(WalletObjectId a, WalletObjectId b, JObject? data = null, int infection = 0)
{
SortWalletObjectLinks(ref a, ref b);
await using var ctx = _ContextFactory.CreateContext();
@ -336,18 +336,59 @@ namespace BTCPayServer.Services
AId = a.Id,
BType = b.Type,
BId = b.Id,
Data = data?.ToString(Formatting.None)
Data = data?.ToString(Formatting.None),
InfectionRate = infection
};
ctx.WalletObjectLinks.Add(l);
try
{
await ctx.SaveChangesAsync();
await SpreadInfection(a, b, data, infection);
}
catch (DbUpdateException) // already exists
{
}
}
private async Task SpreadInfection(WalletObjectId a, WalletObjectId b, JObject? data = null, int infectionRate = 0 )
{
SortWalletObjectLinks(ref a, ref b);
await using var ctx = _ContextFactory.CreateContext();
var contagion = await ctx.WalletObjectLinks
.Include(data => data.Parent)
.ThenInclude(data => data.ChildLinks)
.Include(data => data.Parent)
.ThenInclude(data => data.ParentLinks)
.Include(data => data.Child)
.ThenInclude(data => data.ChildLinks)
.Include(data => data.Child)
.ThenInclude(data => data.ParentLinks)
.SingleOrDefaultAsync(data =>
data.WalletId == a.WalletId.ToString() &&
data.ParentType == a.Type &&
data.ParentId == a.Id &&
data.ChildType == b.Type &&
data.ChildId == b.Id
);
if (contagion is null)
{
await EnsureWalletObjectLink(a, b, data, infectionRate);
}
if ( contagion.InfectionRate == 0)
{
return;
}
foreach (WalletObjectData parentNeighbour in contagion.Parent.GetNeighbours())
{
await EnsureWalletObjectLink(b, new WalletObjectId(b.WalletId, parentNeighbour.Type, parentNeighbour.Id), data, infectionRate-1);
}
foreach (WalletObjectData childNeighbour in contagion.Child.GetNeighbours())
{
await EnsureWalletObjectLink(a, new WalletObjectId(a.WalletId, childNeighbour.Type, childNeighbour.Id), data, infectionRate-1);
}
}
class WalletObjectIdComparer : IComparer<WalletObjectId>
{
public static readonly WalletObjectIdComparer Instance = new WalletObjectIdComparer();