monorepo/LCU.API.Internal.DocumentIngestor/LCU.API.Internal.DocumentIngestorLib/Repositories/PdfPartitionerRunInfoRepository.cs

using System;
using System.Threading.Tasks;
using Microsoft.EntityFrameworkCore;
using LCU.API.Internal.DocumentIngestorLib.Persistence;
using LCU.Library.Internal.DocumentIngestorModels.Entities;

namespace LCU.API.Internal.DocumentIngestorLib.Repositories
{
    public class PdfPartitionerRunInfoRepository : BaseRepository, IPdfPartitionerRunInfoRepository
    {
        public PdfPartitionerRunInfoRepository(AppDbContext context) : base(context) { }

        public async Task<PdfPartitionerRunInfo> FindByIdAsync(Guid id)
        {
            return await _context.PdfPartitionerRunInfos.FirstOrDefaultAsync(pdfPartitionerRunInfo => pdfPartitionerRunInfo.PdfPartitionerRunInfoId == id);
        }

        public async Task<PdfPartitionerRunInfo> AddAsync(PdfPartitionerRunInfo pdfPartitionerRunInfo)
        {
            var result = await _context.PdfPartitionerRunInfos.AddAsync(pdfPartitionerRunInfo);
            return result.Entity;
        }

        public void Update(PdfPartitionerRunInfo pdfPartitionerRunInfo)
        {
            _context.PdfPartitionerRunInfos.Update(pdfPartitionerRunInfo);
        }

        public void Remove(PdfPartitionerRunInfo pdfPartitionerRunInfo)
        {
            _context.PdfPartitionerRunInfos.Remove(pdfPartitionerRunInfo);
        }
    }
}