notfire/Iceshrimp.NET
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[backend/masto-client] Fix concurrent DbContext access

Browse source
This commit is contained in:
Laura Hausmann 2024-02-03 04:05:48 +01:00
parent 17884c4975
commit d3da11f827
No known key found for this signature in database
GPG key ID: D044E84C5BE01605
5 changed files with 18 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
Iceshrimp.Backend/Controllers/Mastodon/Renderers/NoteRenderer.cs
View file

@ -1,7 +1,10 @@
using System.Collections;
using Iceshrimp.Backend.Controllers.Mastodon.Schemas.Entities;
using Iceshrimp.Backend.Core.Configuration;
using Iceshrimp.Backend.Core.Database.Tables;
using Iceshrimp.Backend.Core.Extensions;
using Iceshrimp.Backend.Core.Helpers.LibMfm.Conversion;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Options;
namespace Iceshrimp.Backend.Controllers.Mastodon.Renderers;
@ -42,4 +45,8 @@ public class NoteRenderer(IOptions<Config.InstanceSection> config, UserRenderer
return res;
}
public async Task<IEnumerable<Status>> RenderManyAsync(IEnumerable<Note> notes) {
return await notes.Select(RenderAsync).AwaitAllAsync();
}
}

3
Iceshrimp.Backend/Controllers/Mastodon/Renderers/UserRenderer.cs
View file

@ -44,7 +44,6 @@ public class UserRenderer(IOptions<Config.InstanceSection> config, DatabaseConte
}
public async Task<Account> RenderAsync(User user) {
var profile = await db.UserProfiles.FirstOrDefaultAsync(p => p.User == user);
return await RenderAsync(user, profile);
return await RenderAsync(user, user.UserProfile);
}
}

9
Iceshrimp.Backend/Core/Extensions/QueryableExtensions.cs
View file

@ -10,6 +10,7 @@ namespace Iceshrimp.Backend.Core.Extensions;
public static class NoteQueryableExtensions {
public static IQueryable<Note> WithIncludes(this IQueryable<Note> query) {
return query.Include(p => p.User)
.ThenInclude(p => p.UserProfile)
.Include(p => p.Renote)
.ThenInclude(p => p != null ? p.User : null)
.Include(p => p.Reply)
@ -34,7 +35,7 @@ public static class NoteQueryableExtensions {
.OrderByDescending(note => note.Id),
{ MinId: not null } => query.Where(note => note.Id.IsGreaterThan(p.MinId)).OrderBy(note => note.Id),
{ MaxId: not null } => query.Where(note => note.Id.IsLessThan(p.MaxId)).OrderByDescending(note => note.Id),
_ => query
_ => query.OrderByDescending(note => note.Id)
};
return query.Take(Math.Min(p.Limit ?? defaultLimit, maxLimit));
@ -48,13 +49,9 @@ public static class NoteQueryableExtensions {
return query.Where(note => note.User == user || note.User.IsFollowedBy(user));
}
public static IQueryable<Note> OrderByIdDesc(this IQueryable<Note> query) {
return query.OrderByDescending(note => note.Id);
}
public static async Task<IEnumerable<Status>> RenderAllForMastodonAsync(
this IQueryable<Note> notes, NoteRenderer renderer) {
var list = await notes.ToListAsync();
return await list.Select(renderer.RenderAsync).AwaitAllAsync();
return await renderer.RenderManyAsync(list);
}
}

8
Iceshrimp.Backend/Core/Middleware/AuthenticationMiddleware.cs
View file

@ -18,8 +18,12 @@ public class AuthenticationMiddleware(DatabaseContext db) : IMiddleware {
return;
}
var token = header[7..];
var session = await db.Sessions.Include(p => p.User).FirstOrDefaultAsync(p => p.Token == token && p.Active);
var token = header[7..];
var session = await db.Sessions
.Include(p => p.User)
.ThenInclude(p => p.UserProfile)
.FirstOrDefaultAsync(p => p.Token == token && p.Active);
if (session == null) {
await next(ctx);
return;

1
Iceshrimp.Backend/Core/Middleware/OauthAuthenticationMiddleware.cs
View file

@ -22,6 +22,7 @@ public class OauthAuthenticationMiddleware(DatabaseContext db) : IMiddleware {
header = header[7..];
var token = await db.OauthTokens
.Include(p => p.User)
.ThenInclude(p => p.UserProfile)
.Include(p => p.App)
.FirstOrDefaultAsync(p => p.Token == header && p.Active);