Sonarr/NzbDrone.Core/Datastore/BasicRepository.cs

101 lines
2.4 KiB
C#
Raw Normal View History

2013-02-13 01:25:32 +00:00
using System.Collections.Generic;
2013-02-05 04:07:07 +00:00
using System.Linq;
namespace NzbDrone.Core.Datastore
{
2013-03-02 18:25:39 +00:00
public interface IBasicRepository<TModel> where TModel : ModelBase, new()
2013-02-05 04:07:07 +00:00
{
2013-02-19 06:56:02 +00:00
IEnumerable<TModel> All();
2013-02-23 19:38:25 +00:00
int Count();
TModel Get(int id);
2013-02-19 06:56:02 +00:00
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);
2013-02-23 19:38:25 +00:00
void Purge();
2013-03-02 18:25:39 +00:00
bool HasItems();
2013-02-05 04:07:07 +00:00
}
public class BasicRepository<TModel> : IBasicRepository<TModel> where TModel : ModelBase, new()
2013-02-05 04:07:07 +00:00
{
2013-02-17 01:52:40 +00:00
public BasicRepository(IObjectDatabase objectDatabase)
2013-02-05 04:07:07 +00:00
{
2013-02-17 01:52:40 +00:00
ObjectDatabase = objectDatabase;
2013-02-05 04:07:07 +00:00
}
public IObjectDatabase ObjectDatabase { get; private set; }
2013-02-05 04:07:07 +00:00
protected IEnumerable<TModel> Queryable { get { return ObjectDatabase.AsQueryable<TModel>(); } }
2013-02-19 06:56:02 +00:00
public IEnumerable<TModel> All()
2013-02-05 04:07:07 +00:00
{
return Queryable.ToList();
2013-02-05 04:07:07 +00:00
}
2013-02-23 19:38:25 +00:00
public int Count()
{
return Queryable.Count();
}
public TModel Get(int id)
2013-02-05 04:07:07 +00:00
{
return Queryable.Single(c => c.Id == id);
2013-02-05 04:07:07 +00:00
}
2013-02-19 06:56:02 +00:00
public TModel Insert(TModel model)
2013-02-05 04:07:07 +00:00
{
2013-02-17 01:52:40 +00:00
return ObjectDatabase.Insert(model);
2013-02-05 04:07:07 +00:00
}
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)
2013-02-23 19:38:25 +00:00
{
return ObjectDatabase.Insert(model);
}
return ObjectDatabase.Update(model);
}
public void Delete(int id)
2013-02-05 04:07:07 +00:00
{
var itemToDelete = Get(id);
2013-02-17 01:52:40 +00:00
ObjectDatabase.Delete(itemToDelete);
2013-02-05 04:07:07 +00:00
}
2013-02-23 19:38:25 +00:00
public void DeleteMany(IEnumerable<int> ids)
{
foreach (var id in ids)
{
Delete(id);
}
}
public void Purge()
{
DeleteMany(Queryable.Select(c => c.Id));
2013-02-23 19:38:25 +00:00
}
2013-03-02 18:25:39 +00:00
public bool HasItems()
{
return Queryable.Any();
}
2013-02-05 04:07:07 +00:00
}
}