mirror of https://github.com/Sonarr/Sonarr
125 lines
3.0 KiB
C#
125 lines
3.0 KiB
C#
using System.Collections.Generic;
|
|
using System.Linq;
|
|
|
|
namespace NzbDrone.Core.Datastore
|
|
{
|
|
public interface IBasicRepository<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);
|
|
IList<TModel> InsertMany(IList<TModel> model);
|
|
IList<TModel> UpdateMany(IList<TModel> model);
|
|
void DeleteMany(List<TModel> model);
|
|
void Purge();
|
|
bool HasItems();
|
|
}
|
|
|
|
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 Single()
|
|
{
|
|
return Queryable.Single();
|
|
}
|
|
|
|
public TModel SingleOrDefault()
|
|
{
|
|
return Queryable.SingleOrDefault();
|
|
}
|
|
|
|
public TModel Insert(TModel model)
|
|
{
|
|
return ObjectDatabase.Insert(model);
|
|
}
|
|
|
|
public TModel Update(TModel model)
|
|
{
|
|
return ObjectDatabase.Update(model);
|
|
}
|
|
|
|
public void Delete(TModel model)
|
|
{
|
|
ObjectDatabase.Delete(model);
|
|
}
|
|
|
|
public IList<TModel> InsertMany(IList<TModel> model)
|
|
{
|
|
return ObjectDatabase.InsertMany(model);
|
|
}
|
|
|
|
public IList<TModel> UpdateMany(IList<TModel> model)
|
|
{
|
|
return ObjectDatabase.UpdateMany(model);
|
|
}
|
|
|
|
public void DeleteMany(List<TModel> model)
|
|
{
|
|
ObjectDatabase.DeleteMany(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));
|
|
}
|
|
|
|
public bool HasItems()
|
|
{
|
|
return Queryable.Any();
|
|
}
|
|
}
|
|
}
|