sonarr-repo-only/NzbDrone.Core/Datastore/EloqueraDb.cs

76 lines
1.8 KiB
C#

using System;
using System.Collections;
using System.Collections.Generic;
using System.Linq;
using System.Reflection;
using Eloquera.Client;
namespace NzbDrone.Core.Datastore
{
public class EloqueraDb : IDisposable
{
private readonly IdService _idService;
public DB Db { get; private set; }
public EloqueraDb(DB db, IdService idService)
{
_idService = idService;
Db = db;
}
public IEnumerable<T> AsQueryable<T>()
{
return Db.Query<T>();
}
public T Insert<T>(T obj) where T : BaseRepositoryModel
{
_idService.EnsureIds(obj, new HashSet<object>());
Db.Store(obj);
return obj;
}
public IList<T> InsertMany<T>(IList<T> objects) where T : BaseRepositoryModel
{
_idService.EnsureIds(objects, new HashSet<object>());
return DoMany(objects, Insert);
}
public T Update<T>(T obj)
{
Db.Store(obj);
return obj;
}
public IList<T> UpdateMany<T>(IList<T> objects)
{
_idService.EnsureIds(objects, new HashSet<object>());
return DoMany(objects, Update);
}
public void Delete<T>(T obj) where T : new()
{
Db.Delete(obj);
}
public void DeleteMany<T>(IEnumerable<T> objects) where T : new()
{
foreach (var o in objects)
{
Delete(o);
}
}
private IList<T> DoMany<T>(IEnumerable<T> objects, Func<T, T> function)
{
return objects.Select(function).ToList();
}
public void Dispose()
{
Db.Dispose();
}
}
}