Update custom craft count GQL

- Now the `itemSubType` parameter is nullable - If no `itemSubType` parameter provided, return all data
planetarium · Aug 14, 2024 · 7e957c8 · 7e957c8
1 parent 263016e
commit 7e957c8
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 5 deletions.
diff --git a/NineChronicles.DataProvider/Queries/NineChroniclesSummaryQuery.cs b/NineChronicles.DataProvider/Queries/NineChroniclesSummaryQuery.cs
@@ -357,15 +357,15 @@ public NineChroniclesSummaryQuery(MySqlStore store, StandaloneContext standalone
             Field<ListGraphType<CustomEquipmentCraftIconCountType>>(
                 "customEquipmentCraftIconCount",
                 arguments: new QueryArguments(
-                    new QueryArgument<NonNullGraphType<StringGraphType>>
+                    new QueryArgument<StringGraphType>
                     {
                         Name = "itemSubType",
                         Description = "ItemSubType to get craft count for icons",
                     }
                     ),
                 resolve: context =>
                 {
-                    var itemSubType = context.GetArgument<string>("itemSubType");
+                    var itemSubType = context.GetArgument<string?>("itemSubType");
                     return Store.GetCustomEquipmentCraftCount(itemSubType);
                 }
             );

diff --git a/NineChronicles.DataProvider/Store/MySql/CustomCraftStore.cs b/NineChronicles.DataProvider/Store/MySql/CustomCraftStore.cs
@@ -76,10 +76,12 @@ await ctx.CustomEquipmentCraftCount.AddAsync(new CustomEquipmentCraftCountModel
             }
         }
 
-        public partial List<CustomEquipmentCraftCountModel> GetCustomEquipmentCraftCount(string itemSubType)
+        public partial List<CustomEquipmentCraftCountModel> GetCustomEquipmentCraftCount(string? itemSubType)
         {
             using var ctx = _dbContextFactory.CreateDbContext();
-            return ctx.CustomEquipmentCraftCount.Where(c => c.ItemSubType == itemSubType).ToList();
+            return itemSubType is null
+                ? ctx.CustomEquipmentCraftCount.ToList()
+                : ctx.CustomEquipmentCraftCount.Where(c => c.ItemSubType == itemSubType).ToList();
         }
     }
 }
diff --git a/NineChronicles.DataProvider/Store/MySqlStore.cs b/NineChronicles.DataProvider/Store/MySqlStore.cs
@@ -1960,7 +1960,7 @@ public void StoreRuneSummonFailList(List<RuneSummonFailModel> runeSummonFailList
         // CustomCraft
         public partial Task StoreCustomEquipmentCraftList(List<CustomEquipmentCraftModel> customEquipmentCraftList);
 
-        public partial List<CustomEquipmentCraftCountModel> GetCustomEquipmentCraftCount(string itemSubType);
+        public partial List<CustomEquipmentCraftCountModel> GetCustomEquipmentCraftCount(string? itemSubType);
         /* CustomCraft */
 
         public List<RaiderModel> GetRaiderList()