comment Solve

planetarium · Dec 3, 2024 · d49ca81 · d49ca81
1 parent 526ca8d
commit d49ca81
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 3 deletions.
diff --git a/Lib9c.Models/States/CollectionState.cs b/Lib9c.Models/States/CollectionState.cs
@@ -18,6 +18,8 @@ public class CollectionState : IBencodable
     [BsonIgnore, GraphQLIgnore, JsonIgnore]
     public IValue Bencoded => List.Empty.Add(new List(Ids));
 
+    public CollectionState() { }
+
     public CollectionState(IValue bencoded)
     {
         if (bencoded is not List l)

diff --git a/Mimir.MongoDB/Repositories/CollectionRepository.cs b/Mimir.MongoDB/Repositories/CollectionRepository.cs
@@ -6,7 +6,12 @@
 
 namespace Mimir.MongoDB.Repositories;
 
-public class CollectionRepository(IMongoDbService dbService)
+public interface ICollectionRepository
+{
+    Task<CollectionDocument> GetByAddressAsync(Address address);
+}
+
+public class CollectionRepository(IMongoDbService dbService) : ICollectionRepository
 {
     public async Task<CollectionDocument> GetByAddressAsync(Address address)
     {

diff --git a/Mimir/GraphQL/Queries/Query.cs b/Mimir/GraphQL/Queries/Query.cs
@@ -80,7 +80,7 @@ public async Task<string> GetBalanceAsync(
     /// </summary>
     /// <param name="address">The address of the avatar.</param>
     /// <returns>The collection state for the specified avatar address.</returns>
-    public async Task<CollectionState> GetCollectionAsync(Address address, [Service] CollectionRepository repo) =>
+    public async Task<CollectionState> GetCollectionAsync(Address address, [Service] ICollectionRepository repo) =>
         (await repo.GetByAddressAsync(address)).Object;
 
     /// <summary>

diff --git a/Mimir/Program.cs b/Mimir/Program.cs
@@ -43,7 +43,7 @@
 builder.Services.AddSingleton<ArenaParticipantRepository>();
 builder.Services.AddSingleton<IAvatarRepository, AvatarRepository>();
 builder.Services.AddSingleton<BalanceRepository>();
-builder.Services.AddSingleton<CollectionRepository>();
+builder.Services.AddSingleton<ICollectionRepository,CollectionRepository>();
 builder.Services.AddSingleton<IDailyRewardRepository, DailyRewardRepository>();
 builder.Services.AddSingleton<InventoryRepository>();
 builder.Services.AddSingleton<ItemSlotRepository>();