nhyatt/bitwarden
1
0
Fork 0
mirror of https://github.com/bitwarden/server.git synced 2025-04-07 05:58:13 -05:00
Code

calculate storage updates

Browse Source
This commit is contained in:
Kyle Spearrin 2020-01-10 20:05:58 -05:00
parent 5bb440563f
commit bc0901348b
2 changed files with 41 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
src/Core/Repositories/EntityFramework/OrganizationRepository.cs
View File

@ -49,10 +49,28 @@ namespace Bit.Core.Repositories.EntityFramework
} }
} }
public Task UpdateStorageAsync(Guid id) public async Task UpdateStorageAsync(Guid id)
{ {
// TODO using(var scope = ServiceScopeFactory.CreateScope())
return Task.FromResult(0); {
var dbContext = GetDatabaseContext(scope);
var ciphers = await dbContext.Ciphers
.Where(e => e.UserId == null && e.OrganizationId == id).ToListAsync();
var storage = ciphers.Sum(e => e.AttachmentsJson?.RootElement.EnumerateArray()
.Sum(p => p.GetProperty("Size").GetInt64()) ?? 0);
var organization = new EFModel.Organization
{
Id = id,
RevisionDate = DateTime.UtcNow,
Storage = storage,
};
var set = GetDbSet(dbContext);
set.Attach(organization);
var entry = dbContext.Entry(organization);
entry.Property(e => e.RevisionDate).IsModified = true;
entry.Property(e => e.Storage).IsModified = true;
await dbContext.SaveChangesAsync();
}
} }
public async Task<ICollection<DataModel.OrganizationAbility>> GetManyAbilitiesAsync() public async Task<ICollection<DataModel.OrganizationAbility>> GetManyAbilitiesAsync()

23
src/Core/Repositories/EntityFramework/UserRepository.cs
View File

@ -83,10 +83,27 @@ namespace Bit.Core.Repositories.EntityFramework
} }
} }
public Task UpdateStorageAsync(Guid id) public async Task UpdateStorageAsync(Guid id)
{ {
// TODO using(var scope = ServiceScopeFactory.CreateScope())
return Task.FromResult(0); {
var dbContext = GetDatabaseContext(scope);
var ciphers = await dbContext.Ciphers.Where(e => e.UserId == id).ToListAsync();
var storage = ciphers.Sum(e => e.AttachmentsJson?.RootElement.EnumerateArray()
.Sum(p => p.GetProperty("Size").GetInt64()) ?? 0);
var user = new EFModel.User
{
Id = id,
RevisionDate = DateTime.UtcNow,
Storage = storage,
};
var set = GetDbSet(dbContext);
set.Attach(user);
var entry = dbContext.Entry(user);
entry.Property(e => e.RevisionDate).IsModified = true;
entry.Property(e => e.Storage).IsModified = true;
await dbContext.SaveChangesAsync();
}
} }
public async Task UpdateRenewalReminderDateAsync(Guid id, DateTime renewalReminderDate) public async Task UpdateRenewalReminderDateAsync(Guid id, DateTime renewalReminderDate)