DarthAffe/Artemis
1
0
Fork 0
mirror of https://github.com/Artemis-RGB/Artemis synced 2025-12-12 21:38:38 +00:00
Code Issues Releases Wiki Activity

Storage - Avoid broken LiteDB API

Browse Source
This commit is contained in:
RobertBeekman 2024-02-25 15:32:25 +01:00
parent 5203e95141
commit 24f470a480
2 changed files with 22 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
src/Artemis.Storage/Migrations/Storage/M0024NodeProviders.cs
View File

@ -10,28 +10,33 @@ public class M0024NodeProviders : IStorageMigration
public void Apply(LiteRepository repository) public void Apply(LiteRepository repository)
{ {
List<ProfileCategoryEntity> profileCategories = repository.Query<ProfileCategoryEntity>().ToList(); ILiteCollection<BsonDocument> categoryCollection = repository.Database.GetCollection("ProfileCategoryEntity");
foreach (ProfileCategoryEntity profileCategory in profileCategories) List<BsonDocument> categoriesToUpdate = new();
foreach (BsonDocument profileCategoryBson in categoryCollection.FindAll())
{ {
foreach (ProfileConfigurationEntity profileConfigurationEntity in profileCategory.ProfileConfigurations) BsonArray? profiles = profileCategoryBson["ProfileConfigurations"]?.AsArray;
if (profiles != null)
{ {
profileConfigurationEntity.Version = 1; foreach (BsonValue profile in profiles)
profile["Version"] = 1;
categoriesToUpdate.Add(profileCategoryBson);
} }
repository.Update(profileCategory);
} }
categoryCollection.Update(categoriesToUpdate);
ILiteCollection<BsonDocument> collection = repository.Database.GetCollection("ProfileEntity"); ILiteCollection<BsonDocument> collection = repository.Database.GetCollection("ProfileEntity");
List<BsonDocument> profilesToUpdate = new();
foreach (BsonDocument profileBson in collection.FindAll()) foreach (BsonDocument profileBson in collection.FindAll())
{ {
BsonArray? folders = profileBson["Folders"]?.AsArray; BsonArray? folders = profileBson["Folders"]?.AsArray;
BsonArray? layers = profileBson["Layers"]?.AsArray; BsonArray? layers = profileBson["Layers"]?.AsArray;
if (folders != null) if (folders != null)
{ {
foreach (BsonValue folder in folders) foreach (BsonValue folder in folders)
MigrateProfileElement(folder.AsDocument); MigrateProfileElement(folder.AsDocument);
} }
if (layers != null) if (layers != null)
{ {
foreach (BsonValue layer in layers) foreach (BsonValue layer in layers)
@ -42,9 +47,11 @@ public class M0024NodeProviders : IStorageMigration
MigratePropertyGroup(layer.AsDocument["LayerBrush"]?["PropertyGroup"].AsDocument); MigratePropertyGroup(layer.AsDocument["LayerBrush"]?["PropertyGroup"].AsDocument);
} }
} }
collection.Update(profileBson); profilesToUpdate.Add(profileBson);
} }
collection.Update(profilesToUpdate);
} }
private void MigrateProfileElement(BsonDocument profileElement) private void MigrateProfileElement(BsonDocument profileElement)

8
src/Artemis.Storage/Migrations/Storage/M0025NodeProvidersProfileConfig.cs
View File

@ -1,4 +1,5 @@
using LiteDB; using System.Collections.Generic;
using LiteDB;
namespace Artemis.Storage.Migrations.Storage; namespace Artemis.Storage.Migrations.Storage;
@ -9,6 +10,7 @@ public class M0025NodeProvidersProfileConfig : IStorageMigration
public void Apply(LiteRepository repository) public void Apply(LiteRepository repository)
{ {
ILiteCollection<BsonDocument> categoryCollection = repository.Database.GetCollection("ProfileCategoryEntity"); ILiteCollection<BsonDocument> categoryCollection = repository.Database.GetCollection("ProfileCategoryEntity");
List<BsonDocument> toUpdate = new();
foreach (BsonDocument profileCategoryBson in categoryCollection.FindAll()) foreach (BsonDocument profileCategoryBson in categoryCollection.FindAll())
{ {
BsonArray? profiles = profileCategoryBson["ProfileConfigurations"]?.AsArray; BsonArray? profiles = profileCategoryBson["ProfileConfigurations"]?.AsArray;
@ -19,9 +21,11 @@ public class M0025NodeProvidersProfileConfig : IStorageMigration
profile["Version"] = 2; profile["Version"] = 2;
MigrateNodeScript(profile["ActivationCondition"]?.AsDocument); MigrateNodeScript(profile["ActivationCondition"]?.AsDocument);
} }
toUpdate.Add(profileCategoryBson);
} }
categoryCollection.Update(profileCategoryBson);
} }
categoryCollection.Update(toUpdate);
} }
private void MigrateNodeScript(BsonDocument? nodeScript) private void MigrateNodeScript(BsonDocument? nodeScript)