Sonarr/NzbDrone.Core/Datastore/BasicDb.cs

using System.Collections.Generic;
using System.Data;
using System.Linq;
using ServiceStack.OrmLite;

namespace NzbDrone.Core.Datastore
{
    public interface IBasicDb<TModel> where TModel : ModelBase, new()
    {
        IEnumerable<TModel> All();
        int Count();
        TModel Get(int id);
        //TModel Single();
        //TModel SingleOrDefault();
        TModel Insert(TModel model);
        TModel Update(TModel model);
        TModel Upsert(TModel model);
        void Delete(int id);
        void Delete(TModel model);
        void InsertMany(IList<TModel> models);
        void UpdateMany(IList<TModel> models);
        void DeleteMany(List<TModel> models);
        void Purge();
        bool HasItems();
    }

    public class BasicDb<TModel> : IBasicDb<TModel> where TModel : ModelBase, new()
    {
        public BasicDb(IDbConnection database)
        {
            Database = database;
        }

        public IDbConnection Database { get; private set; }

        public IEnumerable<TModel> All()
        {
            return Database.Select<TModel>();
        }

        public int Count()
        {
            return (int)Database.Count<TModel>();
        }

        public TModel Get(int id)
        {
            return Database.GetById<TModel>(id);
        }

        //public TModel Single()
        //{
        //    return Queryable.Single();
        //}

        //public TModel SingleOrDefault()
        //{
        //    return Queryable.SingleOrDefault();
        //}

        public TModel Insert(TModel model)
        {
            Database.Insert(model);
            model.Id = (int)Database.GetLastInsertId();
            return model;
        }

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

        public void Delete(TModel model)
        {
            Database.Delete(model);
        }

        public void InsertMany(IList<TModel> models)
        {
            Database.InsertAll(models);
        }

        public void UpdateMany(IList<TModel> models)
        {
            Database.UpdateAll(models);
        }

        public void DeleteMany(List<TModel> models)
        {
            Database.DeleteAll(models);
        }

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

        public void Delete(int id)
        {
            Database.DeleteById<TModel>(id);
        }

        public void DeleteMany(IEnumerable<int> ids)
        {
            Database.DeleteByIds<TModel>(ids);
        }

        public void Purge()
        {
            Database.DeleteAll<TModel>();
        }

        public bool HasItems()
        {
            return Count() > 0;
        }
    }
}