Get by multiple ids added to BasicRepo

pull/23/head
Mark McDowall 12 years ago
parent ebf82ec09e
commit c3273b74e8

@ -15,6 +15,7 @@ namespace NzbDrone.Core.Datastore
IEnumerable<TModel> All(); IEnumerable<TModel> All();
int Count(); int Count();
TModel Get(int id); TModel Get(int id);
IEnumerable<TModel> Get(IEnumerable<int> ids);
TModel SingleOrDefault(); TModel SingleOrDefault();
TModel Insert(TModel model); TModel Insert(TModel model);
TModel Update(TModel model); TModel Update(TModel model);
@ -64,6 +65,18 @@ namespace NzbDrone.Core.Datastore
return _dataMapper.Query<TModel>().Single(c => c.Id == id); return _dataMapper.Query<TModel>().Single(c => c.Id == id);
} }
public IEnumerable<TModel> Get(IEnumerable<int> ids)
{
var idList = ids.ToList();
var result = Query.Where(String.Format("Id IN ({0})", String.Join(",", idList))).ToList();
var resultCount = result.Count;
if (resultCount != idList.Count || result.Select(r => r.Id).Distinct().Count() != resultCount)
throw new InvalidOperationException("Unexpected result from query");
return result;
}
public TModel SingleOrDefault() public TModel SingleOrDefault()
{ {
return All().Single(); return All().Single();

@ -13,7 +13,6 @@ namespace NzbDrone.Core.Tv
Series FindByTvdbId(int tvdbId); Series FindByTvdbId(int tvdbId);
void SetSeriesType(int seriesId, SeriesTypes seriesTypes); void SetSeriesType(int seriesId, SeriesTypes seriesTypes);
void SetTvRageId(int seriesId, int tvRageId); void SetTvRageId(int seriesId, int tvRageId);
List<Series> SeriesIsInList(IEnumerable<int> ids);
} }
public class SeriesRepository : BasicRepository<Series>, ISeriesRepository public class SeriesRepository : BasicRepository<Series>, ISeriesRepository
@ -52,10 +51,5 @@ namespace NzbDrone.Core.Tv
{ {
SetFields(new Series { Id = seriesId, TvRageId = tvRageId }, s => s.TvRageId); SetFields(new Series { Id = seriesId, TvRageId = tvRageId }, s => s.TvRageId);
} }
public List<Series> SeriesIsInList(IEnumerable<int> ids)
{
return Query.Where(String.Format("Id IN ({0})", String.Join(",", ids)));
}
} }
} }

@ -137,11 +137,6 @@ namespace NzbDrone.Core.Tv
} }
public void SetTvRageId(int seriesId, int tvRageId)
{
_seriesRepository.SetTvRageId(seriesId, tvRageId);
}
public Series FindByTvdbId(int tvdbId) public Series FindByTvdbId(int tvdbId)
{ {
return _seriesRepository.FindByTvdbId(tvdbId); return _seriesRepository.FindByTvdbId(tvdbId);
@ -181,7 +176,7 @@ namespace NzbDrone.Core.Tv
public List<Series> GetSeriesInList(IEnumerable<int> seriesIds) public List<Series> GetSeriesInList(IEnumerable<int> seriesIds)
{ {
return _seriesRepository.SeriesIsInList(seriesIds); return _seriesRepository.Get(seriesIds).ToList();
} }
public void HandleAsync(SeriesAddedEvent message) public void HandleAsync(SeriesAddedEvent message)

Loading…
Cancel
Save