Unification calculate ranking by arena

planetarium · Oct 16, 2024 · 81452a1 · 81452a1
1 parent 172e2fd
commit 81452a1
Show file tree

Hide file tree

Showing 2 changed files with 75 additions and 10 deletions.
diff --git a/NineChronicles.DataProvider.Tests/WorldBossRankingQueryTest.cs b/NineChronicles.DataProvider.Tests/WorldBossRankingQueryTest.cs
@@ -189,10 +189,10 @@ public async Task WorldBossRanking_Sort()
         {
             [0] = 2,
             [1] = 2,
-            [2] = 3,
-            [3] = 3,
-            [4] = 4,
-            [5] = 5,
+            [2] = 4,
+            [3] = 4,
+            [4] = 5,
+            [5] = 6,
         };
         for (int j = 0; j < 6; j++)
         {

diff --git a/NineChronicles.DataProvider/Store/MySqlStore.cs b/NineChronicles.DataProvider/Store/MySqlStore.cs
@@ -2310,21 +2310,86 @@ public void StoreWorldBossMigration(int raidId)
         public List<WorldBossRankingModel> GetWorldBossRanking(int raidId, int? queryOffset, int? queryLimit)
         {
             using NineChroniclesContext? ctx = _dbContextFactory.CreateDbContext();
+            List<WorldBossRankingModel> query = ctx
+                .Raiders
+                .Where(i => i.RaidId == raidId)
+                .OrderByDescending(i => i.TotalScore)
+                .Select(i => new WorldBossRankingModel
+                {
+                    Ranking = 0,
+                    AvatarName = i.AvatarName,
+                    HighScore = (int)i.HighScore,
+                    TotalScore = (int)i.TotalScore,
+                    Cp = i.Cp,
+                    Level = i.Level,
+                    Address = i.Address,
+                    IconId = i.IconId,
+                })
+                .ToList();
+            int? currentScore = null;
+            var currentRank = 1;
+            var trunk = new List<WorldBossRankingModel>();
+            var result = new List<WorldBossRankingModel>();
+            for (int i = 0; i < query.Count; i++)
+            {
+                var model = query[i];
+                if (!currentScore.HasValue)
+                {
+                    currentScore = model.TotalScore;
+                    trunk.Add(model);
+                    continue;
+                }
+
+                if (currentScore.Value == model.TotalScore)
+                {
+                    trunk.Add(model);
+                    currentRank++;
+                    if (i < query.Count - 1)
+                    {
+                        continue;
+                    }
+
+                    foreach (var modelInTrunk in trunk)
+                    {
+                        modelInTrunk.Ranking = currentRank;
+                        result.Add(modelInTrunk);
+                    }
+
+                    trunk.Clear();
+
+                    continue;
+                }
+
+                foreach (var modelInTrunk in trunk)
+                {
+                    modelInTrunk.Ranking = currentRank;
+                    result.Add(modelInTrunk);
+                }
+
+                trunk.Clear();
+                if (i < query.Count - 1)
+                {
+                    trunk.Add(model);
+                    currentScore = model.TotalScore;
+                    currentRank++;
+                    continue;
+                }
+
+                model.Ranking = currentRank + 1;
+                result.Add(model);
+            }
 
-            // Call ToList for convert query result from IQueryable
-            IEnumerable<WorldBossRankingModel> query = ctx.Set<WorldBossRankingModel>()
-                .FromSqlRaw("SET @prev_score = (SELECT max(`TotalScore`) FROM `Raiders` WHERE `RaidId` = {0}); SET @rank = (SELECT count(*) FROM `Raiders` WHERE `RaidId` = {0} AND `TotalScore` = @prev_score); SELECT `AvatarName`, `HighScore`, `TotalScore`, `Cp`, `Level`, `Address`, `IconId`, @rank := IF(@prev_score = TotalScore, @rank, @rank + 1) as `Ranking`, @prev_score := `TotalScore` FROM `Raiders` WHERE `RaidId` = {0} ORDER BY `TotalScore` DESC", raidId).ToList();
             if (queryOffset.HasValue)
             {
-                query = query.Skip(queryOffset.Value);
+                result = result.Skip(queryOffset.Value).ToList();
             }
 
             if (queryLimit.HasValue)
             {
-                query = query.Take(queryLimit.Value);
+                result = result.Take(queryLimit.Value).ToList();
             }
 
-            return query.OrderBy(i => i.Ranking).ToList();
+            return result.OrderBy(i => i.Ranking).ToList();
         }
 
         public int GetTotalRaiders(int raidId)