Lidarr/NzbDrone.Core/Datastore/BasicRepository.cs

75 lines
1.9 KiB
C#

using System.Collections.Generic;
using System.Linq;
namespace NzbDrone.Core.Datastore
{
public interface IBasicRepository<TModel>
{
IEnumerable<TModel> All();
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);
}
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 TModel Get(int id)
{
return Queryable.Single(c => c.OID == 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.OID == 0)
{
return ObjectDatabase.Insert(model);
}
return ObjectDatabase.Update(model);
}
public void Delete(int id)
{
var itemToDelete = Get(id);
ObjectDatabase.Delete(itemToDelete);
}
}
}