Add database path (#43)

planetarium · May 8, 2024 · 1d4c28c · 1d4c28c
1 parent 6f08fe5
commit 1d4c28c
Show file tree

Hide file tree

Showing 11 changed files with 152 additions and 64 deletions.
diff --git a/Mimir.Worker/appsettings.json b/Mimir.Worker/appsettings.json
@@ -7,7 +7,7 @@
   },
   "Configuration": {
     "MongoDbConnectionString": "mongodb://rootuser:rootpass@localhost:27017",
-    "DatabaseName": "mimir",
+    "DatabaseName": "odin",
     "HeadlessEndpoint": "https://9c-main-full-state.nine-chronicles.com/graphql"
   }
 }
diff --git a/Mimir/Controllers/AgentController.cs b/Mimir/Controllers/AgentController.cs
@@ -1,17 +1,21 @@
+using Libplanet.Crypto;
 using Microsoft.AspNetCore.Mvc;
 using Mimir.Models.Agent;
-using Libplanet.Crypto;
 using Mimir.Services;
 using Mimir.Util;
 
 namespace Mimir.Controllers;
 
 [ApiController]
-[Route("agent")]
+[Route("{network}/agent")]
 public class AccountController : ControllerBase
 {
     [HttpGet("{agentAddress}/avatars")]
-    public async Task<AvatarsResponse> GetAvatars(string agentAddress, IStateService stateService)
+    public async Task<AvatarsResponse> GetAvatars(
+        string network,
+        string agentAddress,
+        IStateService stateService
+    )
     {
         var stateGetter = new StateGetter(stateService);
         var avatars = await stateGetter.GetAvatarStatesAsync(new Address(agentAddress));
@@ -21,8 +25,8 @@ public async Task<AvatarsResponse> GetAvatars(string agentAddress, IStateService
             return new AvatarsResponse([]);
         }
 
-        return new AvatarsResponse(avatars
-            .Select(e => new Avatar(e.address.ToString(), e.name, e.level))
-            .ToList());
+        return new AvatarsResponse(
+            avatars.Select(e => new Avatar(e.address.ToString(), e.name, e.level)).ToList()
+        );
     }
 }
diff --git a/Mimir/Controllers/ArenaController.cs b/Mimir/Controllers/ArenaController.cs
@@ -1,22 +1,22 @@
 using Libplanet.Crypto;
 using Microsoft.AspNetCore.Mvc;
-using Nekoyume.TableData;
 using Mimir.Arena;
 using Mimir.Models.Arena;
 using Mimir.Repositories;
 using Mimir.Services;
 using Mimir.Util;
+using Nekoyume.TableData;
 
 namespace Mimir.Controllers;
 
 [ApiController]
-[Route("arena")]
+[Route("{network}/arena")]
 public class ArenaController(ArenaRankingRepository arenaRankingRepository) : ControllerBase
 {
     [HttpGet("ranking/{avatarAddress}/rank")]
-    public async Task<long> GetRankByAvatarAddress(string avatarAddress)
+    public async Task<long> GetRankByAvatarAddress(string network, string avatarAddress)
     {
-        var rank = await arenaRankingRepository.GetRankByAvatarAddress(avatarAddress);
+        var rank = await arenaRankingRepository.GetRankByAvatarAddress(network, avatarAddress);
         if (rank == 0)
         {
             Response.StatusCode = StatusCodes.Status404NotFound;
@@ -27,15 +27,17 @@ public async Task<long> GetRankByAvatarAddress(string avatarAddress)
     }
 
     [HttpGet("ranking")]
-    public async Task<List<ArenaRanking>> GetRanking(int limit, int offset)
+    public async Task<List<ArenaRanking>> GetRanking(string network, int limit, int offset)
     {
-        return await arenaRankingRepository.GetRanking(limit, offset);
+        return await arenaRankingRepository.GetRanking(network, limit, offset);
     }
 
     [HttpPost("simulate")]
     public async Task<ArenaSimulateResponse> Simulate(
+        string network,
         [FromBody] ArenaSimulateRequest arenaSimulateRequest,
-        IStateService stateService)
+        IStateService stateService
+    )
     {
         var stateGetter = new StateGetter(stateService);
 

diff --git a/Mimir/Controllers/AvatarController.cs b/Mimir/Controllers/AvatarController.cs
@@ -5,13 +5,13 @@
 namespace Mimir.Controllers;
 
 [ApiController]
-[Route("avatars")]
+[Route("{network}/avatars")]
 public class AvatarController(AvatarRepository avatarRepository) : ControllerBase
 {
     [HttpGet("{avatarAddress}/inventory")]
-    public Inventory? GetInventory(string avatarAddress)
+    public Inventory? GetInventory(string network, string avatarAddress)
     {
-        var inventory = avatarRepository.GetInventory(avatarAddress);
+        var inventory = avatarRepository.GetInventory(network, avatarAddress);
         if (inventory is null)
         {
             Response.StatusCode = StatusCodes.Status404NotFound;

diff --git a/Mimir/Options/DatabaseOption.cs b/Mimir/Options/DatabaseOption.cs
@@ -3,5 +3,4 @@ namespace Mimir.Options;
 public class DatabaseOption
 {
     public string ConnectionString { get; set; }
-    public string DatabaseName { get; set; }
 }
diff --git a/Mimir/Repositories/ArenaRankingRespository.cs b/Mimir/Repositories/ArenaRankingRespository.cs
@@ -1,19 +1,33 @@
-using MongoDB.Bson;
-using MongoDB.Driver;
 using Mimir.Models.Agent;
 using Mimir.Models.Arena;
 using Mimir.Services;
+using MongoDB.Bson;
+using MongoDB.Driver;
 
 namespace Mimir.Repositories;
 
-public class ArenaRankingRepository(MongoDBCollectionService mongoDBCollectionService, IStateService stateService)
+public class ArenaRankingRepository : BaseRepository<BsonDocument>
 {
-    private readonly IMongoCollection<dynamic> _arenaCollection =
-        mongoDBCollectionService.GetCollection<dynamic>("arena");
-    private readonly CpRepository _cpRepository = new(stateService);
+    private readonly CpRepository _cpRepository;
+
+    public ArenaRankingRepository(
+        MongoDBCollectionService mongoDBCollectionService,
+        IStateService stateService
+    )
+        : base(mongoDBCollectionService)
+    {
+        _cpRepository = new CpRepository(stateService);
+    }
+
+    protected override string GetCollectionName()
+    {
+        return "arena";
+    }
 
-    public async Task<long> GetRankByAvatarAddress(string avatarAddress)
+    public async Task<long> GetRankByAvatarAddress(string network, string avatarAddress)
     {
+        var collection = GetCollection(network);
+
         var pipelines = new BsonDocument[]
         {
             new("$sort", new BsonDocument("Score.Score", -1)),
@@ -22,35 +36,41 @@ public async Task<long> GetRankByAvatarAddress(string avatarAddress)
                 new BsonDocument
                 {
                     { "_id", BsonNull.Value },
-                    { "docs", new BsonDocument("$push", "$$ROOT") },
+                    { "docs", new BsonDocument("$push", "$$ROOT") }
                 }
             ),
             new(
                 "$unwind",
-                new BsonDocument { { "path", "$docs" }, { "includeArrayIndex", "Rank" }, }
+                new BsonDocument { { "path", "$docs" }, { "includeArrayIndex", "Rank" } }
             ),
-            new("$match", new BsonDocument("docs.AvatarAddress", avatarAddress)),
+            new("$match", new BsonDocument("docs.AvatarAddress", avatarAddress))
         };
-        var aggregation = await _arenaCollection.Aggregate<dynamic>(pipelines).ToListAsync();
-        return aggregation.Count == 0
-            ? 0
-            : (long)aggregation.First().Rank;
+
+        var aggregation = await collection.Aggregate<dynamic>(pipelines).ToListAsync();
+        return aggregation.Count == 0 ? 0 : (long)aggregation.First().Rank;
     }
 
-    public async Task<List<ArenaRanking>> GetRanking(long limit, long offset)
+    public async Task<List<ArenaRanking>> GetRanking(string network, long limit, long offset)
     {
+        var collection = GetCollection(network);
+
         var pipelines = new[]
         {
             @"{ $setWindowFields: { partitionBy: '', sortBy: { 'Score.Score': -1 }, output: { Rank: { $rank: {} } } } }",
             $@"{{ $skip: {offset} }}",
             $@"{{ $limit: {limit} }}",
             @"{ $lookup: { from: 'avatars', localField: 'AvatarAddress', foreignField: 'Avatar.address', as: 'Avatar' } }",
             @"{ $unwind: { path: '$Avatar', preserveNullAndEmptyArrays: true } }",
-            @"{ $unset: ['Avatar.Avatar.inventory', 'Avatar.Avatar.mailBox', 'Avatar.Avatar.stageMap', 'Avatar.Avatar.monsterMap', 'Avatar.Avatar.itemMap', 'Avatar.Avatar.eventMap'] }",
-        }.Select(BsonDocument.Parse).ToArray();
-        var aggregation = _arenaCollection.Aggregate<BsonDocument>(pipelines).ToList();
-        var arenaRankings = await Task.WhenAll(aggregation.OfType<BsonDocument>().Select(BuildArenaRankingFromDocument));
-        return [.. arenaRankings];
+            @"{ $unset: ['Avatar.Avatar.inventory', 'Avatar.Avatar.mailBox', 'Avatar.Avatar.stageMap', 'Avatar.Avatar.monsterMap', 'Avatar.Avatar.itemMap', 'Avatar.Avatar.eventMap'] }"
+        }
+            .Select(BsonDocument.Parse)
+            .ToArray();
+
+        var aggregation = collection.Aggregate<BsonDocument>(pipelines).ToList();
+        var arenaRankings = await Task.WhenAll(
+            aggregation.OfType<BsonDocument>().Select(BuildArenaRankingFromDocument)
+        );
+        return arenaRankings.ToList();
     }
 
     private async Task<ArenaRanking> BuildArenaRankingFromDocument(BsonDocument document)
@@ -80,13 +100,23 @@ private async Task<ArenaRanking> BuildArenaRankingFromDocument(BsonDocument docu
         arenaRanking.Avatar = avatar;
 
         var characterId = document["Avatar"]["Avatar"]["characterId"].AsInt32;
-        var equipmentIds = document["Avatar"]["ItemSlot"]["Equipments"].AsBsonArray.Select(x => x.AsString);
-        var costumeIds = document["Avatar"]["ItemSlot"]["Costumes"].AsBsonArray.Select(x => x.AsString);
-        var runeSlots = document["Avatar"]["RuneSlot"].AsBsonArray.Select(rune =>
-            (rune["RuneId"].AsInt32, rune["Level"].AsInt32)
-        );
+        var equipmentIds = document["Avatar"]
+            ["ItemSlot"]["Equipments"]
+            .AsBsonArray.Select(x => x.AsString);
+        var costumeIds = document["Avatar"]
+            ["ItemSlot"]["Costumes"]
+            .AsBsonArray.Select(x => x.AsString);
+        var runeSlots = document["Avatar"]
+            ["RuneSlot"]
+            .AsBsonArray.Select(rune => (rune["RuneId"].AsInt32, rune["Level"].AsInt32));
 
-        var cp = await _cpRepository.CalculateCp(avatar, characterId, equipmentIds, costumeIds, runeSlots);
+        var cp = await _cpRepository.CalculateCp(
+            avatar,
+            characterId,
+            equipmentIds,
+            costumeIds,
+            runeSlots
+        );
         arenaRanking.CP = cp;
         Console.WriteLine($"CP Calculate {arenaRanking.ArenaAddress}");
 

diff --git a/Mimir/Repositories/AvatarRepository.cs b/Mimir/Repositories/AvatarRepository.cs
@@ -1,25 +1,27 @@
-using MongoDB.Bson;
-using MongoDB.Driver;
 using Mimir.Models.Avatar;
 using Mimir.Services;
+using MongoDB.Bson;
+using MongoDB.Driver;
 
 namespace Mimir.Repositories;
 
-public class AvatarRepository(MongoDBCollectionService mongoDBCollectionService)
+public class AvatarRepository : BaseRepository<BsonDocument>
 {
-    private readonly IMongoCollection<BsonDocument> _avatarsCollection =
-        mongoDBCollectionService.GetCollection<BsonDocument>("avatars");
+    public AvatarRepository(MongoDBCollectionService mongoDBCollectionService)
+        : base(mongoDBCollectionService) { }
 
-    public Inventory? GetInventory(string avatarAddress)
+    protected override string GetCollectionName()
     {
-        var filter = Builders<BsonDocument>.Filter.Eq(f => f["Avatar"]["address"], avatarAddress);
-        var projection = Builders<BsonDocument>.Projection.Include(f => f["Avatar"]["inventory"]["Equipments"]);
-        var document = _avatarsCollection.Find(filter).Project(projection).FirstOrDefault();
-        if (document is null)
-        {
-            return null;
-        }
-
-        return new Inventory(document["Avatar"]["inventory"]);
+        return "avatars";
+    }
+
+    public Inventory? GetInventory(string network, string avatarAddress)
+    {
+        var collection = GetCollection(network);
+        var filter = Builders<BsonDocument>.Filter.Eq("Avatar.address", avatarAddress);
+        var projection = Builders<BsonDocument>.Projection.Include("Avatar.inventory.Equipments");
+        var document = collection.Find(filter).Project(projection).FirstOrDefault();
+
+        return document is null ? null : new Inventory(document["Avatar"]["inventory"]);
     }
 }
diff --git a/Mimir/Repositories/BaseRepositories.cs b/Mimir/Repositories/BaseRepositories.cs
@@ -0,0 +1,48 @@
+using System.Collections.Generic;
+using Mimir.Services;
+using MongoDB.Bson;
+using MongoDB.Driver;
+
+namespace Mimir.Repositories;
+
+public abstract class BaseRepository<T>
+{
+    private readonly MongoDBCollectionService _mongoDBCollectionService;
+    private readonly Dictionary<string, IMongoCollection<T>> _collections = new();
+
+    protected BaseRepository(MongoDBCollectionService mongoDBCollectionService)
+    {
+        _mongoDBCollectionService = mongoDBCollectionService;
+
+        var networks = new[] { "heimdall", "odin" };
+        var collectionName = GetCollectionName();
+
+        foreach (var network in networks)
+        {
+            string databaseName = network switch
+            {
+                "heimdall" => "heimdall",
+                "odin" => "odin",
+                _ => throw new ArgumentException("Invalid network name", nameof(network))
+            };
+
+            var collection = _mongoDBCollectionService.GetCollection<T>(
+                collectionName,
+                databaseName
+            );
+            _collections[network] = collection;
+        }
+    }
+
+    protected abstract string GetCollectionName();
+
+    protected IMongoCollection<T> GetCollection(string network)
+    {
+        if (_collections.TryGetValue(network, out var collection))
+        {
+            return collection;
+        }
+
+        throw new ArgumentException("Invalid network name", nameof(network));
+    }
+}
diff --git a/Mimir/Services/MongoDBCollectionService.cs b/Mimir/Services/MongoDBCollectionService.cs
@@ -8,10 +8,10 @@ public class MongoDBCollectionService(IOptions<DatabaseOption> databaseOption)
 {
     private readonly IOptions<DatabaseOption> _databaseOption = databaseOption;
 
-    public IMongoCollection<T> GetCollection<T>(string collectionName)
+    public IMongoCollection<T> GetCollection<T>(string collectionName, string databaseName)
     {
         var client = new MongoClient(_databaseOption.Value.ConnectionString);
-        var database = client.GetDatabase(_databaseOption.Value.DatabaseName);
+        var database = client.GetDatabase(databaseName);
         return database.GetCollection<T>(collectionName);
     }
 }
diff --git a/Mimir/appsettings.Development.json b/Mimir/appsettings.Development.json
@@ -5,6 +5,9 @@
       "Microsoft.AspNetCore": "Warning"
     }
   },
+  "Database": {
+    "ConnectionString": "mongodb://rootuser:rootpass@localhost:27017"
+  },
   "StateService": {
     "HeadlessEndpoint": "https://9c-main-full-state.nine-chronicles.com/graphql"
   }

diff --git a/Mimir/appsettings.json b/Mimir/appsettings.json
@@ -6,10 +6,10 @@
     }
   },
   "AllowedHosts": "*",
+  "Database": {
+    "ConnectionString": "mongodb://rootuser:rootpass@localhost:27017"
+  },
   "StateService": {
     "HeadlessEndpoint": "https://9c-main-full-state.nine-chronicles.com/graphql"
-  },
-  "DataProvider": {
-    "Endpoint": "http://heimdall-dp.9c.gg/graphql"
   }
 }