DatabaseCollection.cs

Go to the documentation of this file.

using System;
using System.Collections;
using System.Collections.Generic;
using System.Linq;
using System.Linq.Expressions;
using System.Threading;
 
using Microsoft.EntityFrameworkCore;
 
namespace Tgstation.Server.Host.Database
{
    sealed class DatabaseCollection<TModel> : IDatabaseCollection<TModel>
        where TModel : class
    {
        readonly DbSet<TModel> dbSet;
 
        public DatabaseCollection(DbSet<TModel> dbSet)
        {
            this.dbSet = dbSet ?? throw new ArgumentNullException(nameof(dbSet));
        }
 
        public IEnumerable<TModel> Local => dbSet.Local;
 
        public Type ElementType => dbSet.AsQueryable().ElementType;
 
        public Expression Expression => dbSet.AsQueryable().Expression;
 
        public IQueryProvider Provider => dbSet.AsQueryable().Provider;
 
        public void Add(TModel model) => dbSet.Add(model);
 
        public void AddRange(IEnumerable<TModel> models) => dbSet.AddRange(models);
 
        public void Attach(TModel model) => dbSet.Attach(model);
 
        public IAsyncEnumerator<TModel> GetAsyncEnumerator(CancellationToken cancellationToken = default) => dbSet.AsAsyncEnumerable().GetAsyncEnumerator(cancellationToken);
 
        public IEnumerator<TModel> GetEnumerator() => dbSet.AsQueryable().GetEnumerator();
 
        public void Remove(TModel model) => dbSet.Remove(model);
 
        public void RemoveRange(IEnumerable<TModel> models) => dbSet.RemoveRange(models);
 
        IEnumerator IEnumerable.GetEnumerator() => dbSet.AsQueryable().GetEnumerator();
    }
}