mirror of
https://github.com/bitwarden/server
synced 2025-12-18 01:03:17 +00:00
Split out repositories to Infrastructure.Dapper / EntityFramework (#1759)
This commit is contained in:
@@ -0,0 +1,43 @@
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using System.Threading.Tasks;
|
||||
using AutoMapper;
|
||||
using Bit.Core.Repositories;
|
||||
using Bit.Infrastructure.EntityFramework.Models;
|
||||
using Microsoft.EntityFrameworkCore;
|
||||
using Microsoft.Extensions.DependencyInjection;
|
||||
|
||||
namespace Bit.Infrastructure.EntityFramework.Repositories
|
||||
{
|
||||
public class FolderRepository : Repository<Core.Entities.Folder, Folder, Guid>, IFolderRepository
|
||||
{
|
||||
public FolderRepository(IServiceScopeFactory serviceScopeFactory, IMapper mapper)
|
||||
: base(serviceScopeFactory, mapper, (DatabaseContext context) => context.Folders)
|
||||
{ }
|
||||
|
||||
public async Task<Core.Entities.Folder> GetByIdAsync(Guid id, Guid userId)
|
||||
{
|
||||
var folder = await base.GetByIdAsync(id);
|
||||
if (folder == null || folder.UserId != userId)
|
||||
{
|
||||
return null;
|
||||
}
|
||||
|
||||
return folder;
|
||||
}
|
||||
|
||||
public async Task<ICollection<Core.Entities.Folder>> GetManyByUserIdAsync(Guid userId)
|
||||
{
|
||||
using (var scope = ServiceScopeFactory.CreateScope())
|
||||
{
|
||||
var dbContext = GetDatabaseContext(scope);
|
||||
var query = from f in dbContext.Folders
|
||||
where f.UserId == userId
|
||||
select f;
|
||||
var folders = await query.ToListAsync();
|
||||
return Mapper.Map<List<Core.Entities.Folder>>(folders);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user