mirror of
https://github.com/Radarr/Radarr
synced 2024-12-25 17:27:59 +00:00
Get by multiple ids added to BasicRepo
This commit is contained in:
parent
ebf82ec09e
commit
c3273b74e8
3 changed files with 14 additions and 12 deletions
|
@ -15,6 +15,7 @@ namespace NzbDrone.Core.Datastore
|
|||
IEnumerable<TModel> All();
|
||||
int Count();
|
||||
TModel Get(int id);
|
||||
IEnumerable<TModel> Get(IEnumerable<int> ids);
|
||||
TModel SingleOrDefault();
|
||||
TModel Insert(TModel model);
|
||||
TModel Update(TModel model);
|
||||
|
@ -64,6 +65,18 @@ public TModel Get(int 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()
|
||||
{
|
||||
return All().Single();
|
||||
|
|
|
@ -13,7 +13,6 @@ public interface ISeriesRepository : IBasicRepository<Series>
|
|||
Series FindByTvdbId(int tvdbId);
|
||||
void SetSeriesType(int seriesId, SeriesTypes seriesTypes);
|
||||
void SetTvRageId(int seriesId, int tvRageId);
|
||||
List<Series> SeriesIsInList(IEnumerable<int> ids);
|
||||
}
|
||||
|
||||
public class SeriesRepository : BasicRepository<Series>, ISeriesRepository
|
||||
|
@ -52,10 +51,5 @@ public void SetTvRageId(int seriesId, int 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 @@ public void UpdateFromSeriesEditor(IList<Series> editedSeries)
|
|||
|
||||
}
|
||||
|
||||
public void SetTvRageId(int seriesId, int tvRageId)
|
||||
{
|
||||
_seriesRepository.SetTvRageId(seriesId, tvRageId);
|
||||
}
|
||||
|
||||
public Series FindByTvdbId(int tvdbId)
|
||||
{
|
||||
return _seriesRepository.FindByTvdbId(tvdbId);
|
||||
|
@ -181,7 +176,7 @@ public bool SeriesPathExists(string folder)
|
|||
|
||||
public List<Series> GetSeriesInList(IEnumerable<int> seriesIds)
|
||||
{
|
||||
return _seriesRepository.SeriesIsInList(seriesIds);
|
||||
return _seriesRepository.Get(seriesIds).ToList();
|
||||
}
|
||||
|
||||
public void HandleAsync(SeriesAddedEvent message)
|
||||
|
|
Loading…
Reference in a new issue