mirror of https://github.com/lidarr/Lidarr
added repo base
This commit is contained in:
parent
9e4bb278ef
commit
34038245eb
|
@ -1,26 +1,14 @@
|
|||
using System.Collections.Generic;
|
||||
using NzbDrone.Core.RootFolders;
|
||||
using PetaPoco;
|
||||
|
||||
namespace NzbDrone.Core.Repository
|
||||
{
|
||||
public interface IRootDir
|
||||
{
|
||||
int Id { get; set; }
|
||||
string Path { get; set; }
|
||||
|
||||
[ResultColumn]
|
||||
ulong FreeSpace { get; set; }
|
||||
|
||||
[Ignore]
|
||||
List<string> UnmappedFolders { get; set; }
|
||||
}
|
||||
|
||||
[TableName("RootDirs")]
|
||||
[PrimaryKey("Id", autoIncrement = true)]
|
||||
public class RootDir : IRootDir
|
||||
public class RootDir : BaseModel
|
||||
{
|
||||
public virtual int Id { get; set; }
|
||||
|
||||
public string Path { get; set; }
|
||||
|
||||
[ResultColumn]
|
||||
|
|
|
@ -1,45 +1,73 @@
|
|||
using System.Collections.Generic;
|
||||
using Eloquera.Client;
|
||||
using NzbDrone.Core.Datastore;
|
||||
using NzbDrone.Core.Repository;
|
||||
using System.Linq;
|
||||
|
||||
namespace NzbDrone.Core.RootFolders
|
||||
{
|
||||
public interface IRootFolderRepository
|
||||
|
||||
public abstract class BaseModel
|
||||
{
|
||||
List<RootDir> All();
|
||||
RootDir Get(int rootFolderId);
|
||||
RootDir Add(RootDir rootFolder);
|
||||
[ID]
|
||||
public int Id;
|
||||
}
|
||||
|
||||
public interface IBasicRepository<TModel>
|
||||
{
|
||||
List<TModel> All();
|
||||
TModel Get(int rootFolderId);
|
||||
TModel Add(TModel rootFolder);
|
||||
void Delete(int rootFolderId);
|
||||
}
|
||||
|
||||
public class RootFolderRepository : IRootFolderRepository
|
||||
|
||||
public abstract class BasicRepository<TModel> : IBasicRepository<TModel> where TModel : BaseModel, new()
|
||||
{
|
||||
private readonly EloqueraDb _db;
|
||||
|
||||
public RootFolderRepository(EloqueraDb db)
|
||||
|
||||
public BasicRepository(EloqueraDb eloqueraDb)
|
||||
{
|
||||
_db = db;
|
||||
EloqueraDb = eloqueraDb;
|
||||
}
|
||||
|
||||
public List<RootDir> All()
|
||||
protected EloqueraDb EloqueraDb { get; private set; }
|
||||
|
||||
|
||||
public List<TModel> All()
|
||||
{
|
||||
return _db.AsQueryable<RootDir>().ToList();
|
||||
return EloqueraDb.AsQueryable<TModel>().ToList();
|
||||
}
|
||||
|
||||
public RootDir Get(int rootFolderId)
|
||||
public TModel Get(int rootFolderId)
|
||||
{
|
||||
return _db.AsQueryable<RootDir>().Single(c => c.Id == rootFolderId);
|
||||
return EloqueraDb.AsQueryable<TModel>().Single(c => c.Id == rootFolderId);
|
||||
}
|
||||
|
||||
public RootDir Add(RootDir rootFolder)
|
||||
public TModel Add(TModel rootFolder)
|
||||
{
|
||||
return _db.Insert(rootFolder);
|
||||
return EloqueraDb.Insert(rootFolder);
|
||||
}
|
||||
|
||||
public void Delete(int rootFolderId)
|
||||
{
|
||||
_db.Delete(Get(rootFolderId));
|
||||
var itemToDelete = Get(rootFolderId);
|
||||
EloqueraDb.Delete(itemToDelete);
|
||||
}
|
||||
}
|
||||
|
||||
public interface IRootFolderRepository : IBasicRepository<RootDir>
|
||||
{
|
||||
|
||||
}
|
||||
|
||||
|
||||
//This way we only need to implement none_custom methods for repos, like custom queries etc... rest is done automagically.
|
||||
public class RootFolderRepository : BasicRepository<RootDir>, IRootFolderRepository
|
||||
{
|
||||
public RootFolderRepository(EloqueraDb eloqueraDb)
|
||||
: base(eloqueraDb)
|
||||
{
|
||||
}
|
||||
}
|
||||
|
||||
|
|
Loading…
Reference in New Issue