Sonarr/NzbDrone.Core/Datastore/BasicRepository.cs

225 lines
5.7 KiB
C#
Raw Normal View History

2013-03-24 04:16:00 +00:00
using System;
using System.Collections.Generic;
2013-02-05 04:07:07 +00:00
using System.Linq;
2013-03-24 04:16:00 +00:00
using System.Linq.Expressions;
2013-03-25 03:51:32 +00:00
using Marr.Data;
using Marr.Data.QGen;
2013-05-05 21:24:33 +00:00
using NzbDrone.Common.Messaging;
using NzbDrone.Core.Datastore.Events;
using NzbDrone.Common;
2013-03-25 03:51:32 +00:00
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();
TModel Get(int id);
2013-04-18 00:33:38 +00:00
IEnumerable<TModel> Get(IEnumerable<int> ids);
TModel SingleOrDefault();
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 SetFields(TModel model, params Expression<Func<TModel, object>>[] properties);
2013-04-25 04:27:49 +00:00
TModel Single();
2013-02-05 04:07:07 +00:00
}
2013-05-30 23:32:50 +00:00
public class BasicRepository<TModel> : IBasicRepository<TModel> where TModel : ModelBase, new()
2013-02-05 04:07:07 +00:00
{
2013-05-11 20:06:57 +00:00
private readonly IDatabase _database;
2013-05-05 21:24:33 +00:00
private readonly IMessageAggregator _messageAggregator;
2013-04-25 04:27:49 +00:00
//TODO: add assertion to make sure model properly mapped
2013-05-11 20:06:57 +00:00
private IDataMapper DataMapper
{
get { return _database.DataMapper; }
}
2013-05-05 21:24:33 +00:00
public BasicRepository(IDatabase database, IMessageAggregator messageAggregator)
2013-02-05 04:07:07 +00:00
{
2013-05-11 20:06:57 +00:00
_database = database;
2013-05-05 21:24:33 +00:00
_messageAggregator = messageAggregator;
2013-02-05 04:07:07 +00:00
}
2013-03-27 03:44:52 +00:00
protected QueryBuilder<TModel> Query
2013-02-05 04:07:07 +00:00
{
2013-05-11 20:06:57 +00:00
get { return DataMapper.Query<TModel>(); }
2013-03-27 03:44:52 +00:00
}
protected void Delete(Expression<Func<TModel, bool>> filter)
{
2013-05-11 20:06:57 +00:00
DataMapper.Delete(filter);
2013-02-05 04:07:07 +00:00
}
2013-03-25 03:51:32 +00:00
public IEnumerable<TModel> All()
2013-02-23 19:38:25 +00:00
{
2013-05-11 20:06:57 +00:00
return DataMapper.Query<TModel>().ToList();
}
2013-03-25 03:51:32 +00:00
public int Count()
{
2013-05-11 20:06:57 +00:00
return DataMapper.Query<TModel>().GetRowCount();
2013-02-23 19:38:25 +00:00
}
public TModel Get(int id)
2013-02-05 04:07:07 +00:00
{
2013-05-11 20:06:57 +00:00
return DataMapper.Query<TModel>().Single(c => c.Id == id);
2013-02-05 04:07:07 +00:00
}
2013-04-18 00:33:38 +00:00
public IEnumerable<TModel> Get(IEnumerable<int> ids)
{
var query = String.Format("Id IN ({0})", String.Join(",", ids));
2013-04-18 00:33:38 +00:00
var result = Query.Where(query).ToList();
if (result.Count != ids.Count())
{
throw new ApplicationException("Expected query to return {0} rows but returned {1}".Inject(ids.Count(), result.Count));
}
2013-04-18 00:33:38 +00:00
return result;
}
public TModel SingleOrDefault()
2013-04-25 04:27:49 +00:00
{
return All().SingleOrDefault();
}
public TModel Single()
{
2013-03-24 04:16:00 +00:00
return All().Single();
}
2013-02-19 06:56:02 +00:00
public TModel Insert(TModel model)
2013-02-05 04:07:07 +00:00
{
2013-03-24 19:56:51 +00:00
if (model.Id != 0)
{
2013-06-03 03:44:31 +00:00
throw new InvalidOperationException("Can't insert model with existing ID " + model.Id);
2013-03-24 19:56:51 +00:00
}
2013-05-11 20:06:57 +00:00
DataMapper.Insert(model);
PublishModelEvent(model, RepositoryAction.Created);
2013-05-05 21:24:33 +00:00
2013-03-24 04:16:00 +00:00
return model;
2013-02-05 04:07:07 +00:00
}
public TModel Update(TModel model)
{
2013-03-24 19:56:51 +00:00
if (model.Id == 0)
{
throw new InvalidOperationException("Can't update model with ID 0");
}
2013-05-11 20:06:57 +00:00
DataMapper.Update(model, c => c.Id == model.Id);
2013-03-24 04:16:00 +00:00
return model;
}
public void Delete(TModel model)
{
2013-05-11 20:06:57 +00:00
DataMapper.Delete<TModel>(c => c.Id == model.Id);
}
2013-03-24 04:16:00 +00:00
public void InsertMany(IList<TModel> models)
{
2013-03-25 03:51:32 +00:00
foreach (var model in models)
{
Insert(model);
}
}
2013-03-24 04:16:00 +00:00
public void UpdateMany(IList<TModel> models)
{
2013-03-25 03:51:32 +00:00
foreach (var model in models)
{
Update(model);
}
}
2013-03-24 04:16:00 +00:00
public void DeleteMany(List<TModel> models)
{
2013-03-25 03:51:32 +00:00
models.ForEach(Delete);
}
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
{
2013-03-25 03:51:32 +00:00
Insert(model);
2013-03-24 04:16:00 +00:00
return model;
2013-02-23 19:38:25 +00:00
}
2013-03-25 03:51:32 +00:00
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
{
2013-05-11 20:06:57 +00:00
DataMapper.Delete<TModel>(c => c.Id == id);
2013-02-05 04:07:07 +00:00
}
2013-02-23 19:38:25 +00:00
public void DeleteMany(IEnumerable<int> ids)
{
2013-03-25 03:51:32 +00:00
ids.ToList().ForEach(Delete);
2013-02-23 19:38:25 +00:00
}
public void Purge()
{
2013-05-11 20:06:57 +00:00
DataMapper.Delete<TModel>(c => c.Id > -1);
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 SetFields(TModel model, params Expression<Func<TModel, object>>[] properties)
2013-03-24 04:16:00 +00:00
{
if (model.Id == 0)
{
throw new InvalidOperationException("Attempted to updated model without ID");
}
2013-05-11 20:06:57 +00:00
DataMapper.Update<TModel>()
.Where(c => c.Id == model.Id)
.ColumnsIncluding(properties)
.Entity(model)
.Execute();
2013-03-02 18:25:39 +00:00
}
2013-05-10 23:53:50 +00:00
private void PublishModelEvent(TModel model, RepositoryAction action)
{
if (PublishModelEvents)
{
_messageAggregator.PublishEvent(new ModelEvent<TModel>(model, action));
}
}
2013-05-30 23:32:50 +00:00
protected virtual void OnModelChanged(IEnumerable<TModel> models)
{
}
protected virtual void OnModelDeleted(IEnumerable<TModel> models)
{
}
protected virtual bool PublishModelEvents
{
get { return false; }
}
2013-02-05 04:07:07 +00:00
}
}