sonarr-repo-only/NzbDrone.Core/Datastore/BasicRepository.cs

using System.Collections.Generic;
using System.Linq;

namespace NzbDrone.Core.Datastore
{
    public interface IBasicRepository<TModel>
    {
        IEnumerable<TModel> All();
        int Count();
        TModel Get(int id);
        TModel Insert(TModel model);
        TModel Update(TModel model);
        TModel Upsert(TModel model);
        void Delete(int id);
        IList<TModel> InsertMany(IList<TModel> model);
        IList<TModel> UpdateMany(IList<TModel> model);
        void Purge();
    }

    public class BasicRepository<TModel> : IBasicRepository<TModel> where TModel : ModelBase, new()
    {
        public BasicRepository(IObjectDatabase objectDatabase)
        {
            ObjectDatabase = objectDatabase;
        }

        public IObjectDatabase ObjectDatabase { get; private set; }

        protected IEnumerable<TModel> Queryable { get { return ObjectDatabase.AsQueryable<TModel>(); } }

        public IEnumerable<TModel> All()
        {
            return Queryable.ToList();
        }

        public int Count()
        {
            return Queryable.Count();
        }

        public TModel Get(int id)
        {
            return Queryable.Single(c => c.Id == id);
        }

        public TModel Insert(TModel model)
        {
            return ObjectDatabase.Insert(model);
        }

        public TModel Update(TModel model)
        {
            return ObjectDatabase.Update(model);
        }

        public IList<TModel> InsertMany(IList<TModel> model)
        {
            return ObjectDatabase.InsertMany(model);
        }

        public IList<TModel> UpdateMany(IList<TModel> model)
        {
            return ObjectDatabase.UpdateMany(model);
        }

        public TModel Upsert(TModel model)
        {
            if (model.Id == 0)
            {
                return ObjectDatabase.Insert(model);
            }
            return ObjectDatabase.Update(model);
        }

        public void Delete(int id)
        {
            var itemToDelete = Get(id);
            ObjectDatabase.Delete(itemToDelete);
        }

        public void DeleteMany(IEnumerable<int> ids)
        {
            foreach (var id in ids)
            {
                Delete(id);
            }
        }

        public void Purge()
        {
            DeleteMany(Queryable.Select(c => c.Id));
        }
    }
}