Sonarr/NzbDrone.Core/Datastore/BasicRepository.cs

165 lines
4.3 KiB
C#
Raw Normal View History

2013-03-24 04:16:00 +00:00
using System;
using System.Collections.Generic;
using System.Data;
2013-02-05 04:07:07 +00:00
using System.Linq;
2013-03-24 04:16:00 +00:00
using System.Linq.Expressions;
using ServiceStack.OrmLite;
2013-02-05 04:07:07 +00:00
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();
bool Any(Expression<Func<TModel, bool>> predicate);
TModel Get(int id);
2013-03-24 04:16:00 +00:00
TModel Single(Expression<Func<TModel, bool>> predicate);
TModel SingleOrDefault();
2013-03-24 04:16:00 +00:00
TModel SingleOrDefault(Expression<Func<TModel, bool>> predicate);
List<TModel> Where(Expression<Func<TModel, bool>> predicate);
2013-02-19 06:56:02 +00:00
TModel Insert(TModel model);
TModel Update(TModel model);
2013-03-24 04:16:00 +00:00
TModel Upsert(TModel model);
void Delete(int id);
void Delete(TModel model);
2013-03-24 04:16:00 +00:00
void InsertMany(IList<TModel> model);
void UpdateMany(IList<TModel> model);
void DeleteMany(List<TModel> model);
2013-02-23 19:38:25 +00:00
void Purge();
2013-03-02 18:25:39 +00:00
bool HasItems();
2013-03-24 04:16:00 +00:00
void DeleteMany(IEnumerable<int> ids);
void UpdateFields<TKey>(TModel model, Expression<Func<TModel, TKey>> onlyFields);
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
{
private readonly IDbConnection _database;
2013-03-24 04:16:00 +00:00
public BasicRepository(IDbConnection database)
2013-02-05 04:07:07 +00:00
{
_database = database;
2013-02-05 04:07:07 +00:00
}
2013-02-19 06:56:02 +00:00
public IEnumerable<TModel> All()
2013-02-05 04:07:07 +00:00
{
return _database.Select<TModel>();
2013-02-05 04:07:07 +00:00
}
2013-02-23 19:38:25 +00:00
public int Count()
{
return (int)_database.Count<TModel>();
}
public bool Any(Expression<Func<TModel, bool>> predicate)
{
return _database.Exists<TModel>(predicate);
2013-02-23 19:38:25 +00:00
}
public TModel Get(int id)
2013-02-05 04:07:07 +00:00
{
return _database.GetById<TModel>(id);
2013-02-05 04:07:07 +00:00
}
2013-03-24 04:16:00 +00:00
public TModel Single(Expression<Func<TModel, bool>> predicate)
{
return _database.Select(predicate).Single();
}
public TModel SingleOrDefault()
{
2013-03-24 04:16:00 +00:00
return All().Single();
}
public TModel Single()
{
throw new System.NotImplementedException();
}
public TModel SingleOrDefault(Expression<Func<TModel, bool>> predicate)
{
return _database.Select(predicate).SingleOrDefault();
2013-03-24 04:16:00 +00:00
}
public List<TModel> Where(Expression<Func<TModel, bool>> predicate)
{
return _database.Select(predicate);
}
2013-02-19 06:56:02 +00:00
public TModel Insert(TModel model)
2013-02-05 04:07:07 +00:00
{
_database.Insert(model);
model.Id = (int)_database.GetLastInsertId();
2013-03-24 04:16:00 +00:00
return model;
2013-02-05 04:07:07 +00:00
}
public TModel Update(TModel model)
{
_database.Update(model);
2013-03-24 04:16:00 +00:00
return model;
}
2013-03-24 04:16:00 +00:00
public void Delete(TModel model)
{
_database.Delete(model);
}
2013-03-24 04:16:00 +00:00
public void InsertMany(IList<TModel> models)
{
_database.InsertAll(models);
}
2013-03-24 04:16:00 +00:00
public void UpdateMany(IList<TModel> models)
{
_database.UpdateAll(models);
}
2013-03-24 04:16:00 +00:00
public void DeleteMany(List<TModel> models)
{
_database.DeleteAll(models);
}
2013-03-24 04:16:00 +00:00
public TModel Upsert(TModel model)
{
if (model.Id == 0)
2013-02-23 19:38:25 +00:00
{
_database.Insert(model);
model.Id = (int)_database.GetLastInsertId();
2013-03-24 04:16:00 +00:00
return model;
2013-02-23 19:38:25 +00:00
}
_database.Update(model);
2013-03-24 04:16:00 +00:00
return model;
}
public void Delete(int id)
2013-02-05 04:07:07 +00:00
{
_database.DeleteById<TModel>(id);
2013-02-05 04:07:07 +00:00
}
2013-02-23 19:38:25 +00:00
public void DeleteMany(IEnumerable<int> ids)
{
_database.DeleteByIds<TModel>(ids);
2013-02-23 19:38:25 +00:00
}
public void Purge()
{
_database.DeleteAll<TModel>();
2013-02-23 19:38:25 +00:00
}
2013-03-02 18:25:39 +00:00
public bool HasItems()
{
2013-03-24 04:16:00 +00:00
return Count() > 0;
}
public void UpdateFields<TKey>(TModel model, Expression<Func<TModel, TKey>> onlyFields)
2013-03-24 04:16:00 +00:00
{
if (model.Id == 0)
{
throw new InvalidOperationException("Attempted to updated model without ID");
}
_database.UpdateOnly(model, onlyFields, m => m.Id == model.Id);
2013-03-02 18:25:39 +00:00
}
2013-02-05 04:07:07 +00:00
}
}