DatabaseCollection.cs

Go to the documentation of this file.

using Microsoft.EntityFrameworkCore;
using System;
using System.Collections;
using System.Collections.Generic;
using System.Linq;
using System.Linq.Expressions;
using System.Threading;

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();

        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();
    }
}