123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836 |
- using MediaBrowser.Common.Configuration;
- using MediaBrowser.Controller.Entities;
- using MediaBrowser.Controller.Entities.Audio;
- using MediaBrowser.Controller.Entities.Movies;
- using MediaBrowser.Controller.Entities.TV;
- using MediaBrowser.Controller.LiveTv;
- using MediaBrowser.Controller.Persistence;
- using MediaBrowser.Model.Entities;
- using MediaBrowser.Model.Logging;
- using MediaBrowser.Model.Querying;
- using MediaBrowser.Model.Serialization;
- using System;
- using System.Collections.Generic;
- using System.Data;
- using System.Globalization;
- using System.IO;
- using System.Linq;
- using System.Runtime.Serialization;
- using System.Threading;
- using System.Threading.Tasks;
- using MediaBrowser.Controller.Channels;
- namespace MediaBrowser.Server.Implementations.Persistence
- {
- /// <summary>
- /// Class SQLiteItemRepository
- /// </summary>
- public class SqliteItemRepository : IItemRepository
- {
- private IDbConnection _connection;
- private readonly ILogger _logger;
- private readonly TypeMapper _typeMapper = new TypeMapper();
- /// <summary>
- /// Gets the name of the repository
- /// </summary>
- /// <value>The name.</value>
- public string Name
- {
- get
- {
- return "SQLite";
- }
- }
- /// <summary>
- /// Gets the json serializer.
- /// </summary>
- /// <value>The json serializer.</value>
- private readonly IJsonSerializer _jsonSerializer;
- /// <summary>
- /// The _app paths
- /// </summary>
- private readonly IApplicationPaths _appPaths;
- /// <summary>
- /// The _save item command
- /// </summary>
- private IDbCommand _saveItemCommand;
- private readonly string _criticReviewsPath;
- private SqliteChapterRepository _chapterRepository;
- private SqliteMediaStreamsRepository _mediaStreamsRepository;
- private IDbCommand _deleteChildrenCommand;
- private IDbCommand _saveChildrenCommand;
- private IDbCommand _deleteItemCommand;
- private IDbCommand _deletePeopleCommand;
- private IDbCommand _savePersonCommand;
- private const int LatestSchemaVersion = 13;
- /// <summary>
- /// Initializes a new instance of the <see cref="SqliteItemRepository"/> class.
- /// </summary>
- /// <param name="appPaths">The app paths.</param>
- /// <param name="jsonSerializer">The json serializer.</param>
- /// <param name="logManager">The log manager.</param>
- /// <exception cref="System.ArgumentNullException">
- /// appPaths
- /// or
- /// jsonSerializer
- /// </exception>
- public SqliteItemRepository(IApplicationPaths appPaths, IJsonSerializer jsonSerializer, ILogManager logManager)
- {
- if (appPaths == null)
- {
- throw new ArgumentNullException("appPaths");
- }
- if (jsonSerializer == null)
- {
- throw new ArgumentNullException("jsonSerializer");
- }
- _appPaths = appPaths;
- _jsonSerializer = jsonSerializer;
- _criticReviewsPath = Path.Combine(_appPaths.DataPath, "critic-reviews");
- _logger = logManager.GetLogger(GetType().Name);
- var chapterDbFile = Path.Combine(_appPaths.DataPath, "chapters.db");
- var chapterConnection = SqliteExtensions.ConnectToDb(chapterDbFile, _logger).Result;
- _chapterRepository = new SqliteChapterRepository(chapterConnection, logManager);
- var mediaStreamsDbFile = Path.Combine(_appPaths.DataPath, "mediainfo.db");
- var mediaStreamsConnection = SqliteExtensions.ConnectToDb(mediaStreamsDbFile, _logger).Result;
- _mediaStreamsRepository = new SqliteMediaStreamsRepository(mediaStreamsConnection, logManager);
- }
- /// <summary>
- /// Opens the connection to the database
- /// </summary>
- /// <returns>Task.</returns>
- public async Task Initialize()
- {
- var dbFile = Path.Combine(_appPaths.DataPath, "library.db");
- _connection = await SqliteExtensions.ConnectToDb(dbFile, _logger).ConfigureAwait(false);
- string[] queries = {
- "create table if not exists TypedBaseItems (guid GUID primary key, type TEXT, data BLOB)",
- "create index if not exists idx_TypedBaseItems on TypedBaseItems(guid)",
- "create table if not exists ChildrenIds (ParentId GUID, ItemId GUID, PRIMARY KEY (ParentId, ItemId))",
- "create index if not exists idx_ChildrenIds on ChildrenIds(ParentId,ItemId)",
- "create table if not exists People (ItemId GUID, Name TEXT NOT NULL, Role TEXT, PersonType TEXT, SortOrder int, ListOrder int)",
- //pragmas
- "pragma temp_store = memory",
- "pragma shrink_memory"
- };
- _connection.RunQueries(queries, _logger);
- _connection.AddColumn(_logger, "TypedBaseItems", "Path", "Text");
- _connection.AddColumn(_logger, "TypedBaseItems", "StartDate", "DATETIME");
- _connection.AddColumn(_logger, "TypedBaseItems", "EndDate", "DATETIME");
- _connection.AddColumn(_logger, "TypedBaseItems", "ChannelId", "Text");
- _connection.AddColumn(_logger, "TypedBaseItems", "IsMovie", "BIT");
- _connection.AddColumn(_logger, "TypedBaseItems", "IsSports", "BIT");
- _connection.AddColumn(_logger, "TypedBaseItems", "IsKids", "BIT");
- _connection.AddColumn(_logger, "TypedBaseItems", "CommunityRating", "Float");
- _connection.AddColumn(_logger, "TypedBaseItems", "CustomRating", "Text");
- _connection.AddColumn(_logger, "TypedBaseItems", "IndexNumber", "INT");
- _connection.AddColumn(_logger, "TypedBaseItems", "IsLocked", "BIT");
- _connection.AddColumn(_logger, "TypedBaseItems", "Name", "Text");
- _connection.AddColumn(_logger, "TypedBaseItems", "OfficialRating", "Text");
- _connection.AddColumn(_logger, "TypedBaseItems", "MediaType", "Text");
- _connection.AddColumn(_logger, "TypedBaseItems", "Overview", "Text");
- _connection.AddColumn(_logger, "TypedBaseItems", "ParentIndexNumber", "INT");
- _connection.AddColumn(_logger, "TypedBaseItems", "PremiereDate", "DATETIME");
- _connection.AddColumn(_logger, "TypedBaseItems", "ProductionYear", "INT");
- _connection.AddColumn(_logger, "TypedBaseItems", "ParentId", "GUID");
- _connection.AddColumn(_logger, "TypedBaseItems", "Genres", "Text");
- _connection.AddColumn(_logger, "TypedBaseItems", "ParentalRatingValue", "INT");
- _connection.AddColumn(_logger, "TypedBaseItems", "SchemaVersion", "INT");
- _connection.AddColumn(_logger, "TypedBaseItems", "SortName", "Text");
- _connection.AddColumn(_logger, "TypedBaseItems", "RunTimeTicks", "BIGINT");
- _connection.AddColumn(_logger, "TypedBaseItems", "OfficialRatingDescription", "Text");
- _connection.AddColumn(_logger, "TypedBaseItems", "HomePageUrl", "Text");
- _connection.AddColumn(_logger, "TypedBaseItems", "VoteCount", "INT");
- _connection.AddColumn(_logger, "TypedBaseItems", "DisplayMediaType", "Text");
- _connection.AddColumn(_logger, "TypedBaseItems", "DateCreated", "DATETIME");
- _connection.AddColumn(_logger, "TypedBaseItems", "DateModified", "DATETIME");
- _connection.AddColumn(_logger, "TypedBaseItems", "ForcedSortName", "Text");
- _connection.AddColumn(_logger, "TypedBaseItems", "IsOffline", "BIT");
- _connection.AddColumn(_logger, "TypedBaseItems", "LocationType", "Text");
- _connection.AddColumn(_logger, "TypedBaseItems", "IsSeries", "BIT");
- _connection.AddColumn(_logger, "TypedBaseItems", "IsLive", "BIT");
- _connection.AddColumn(_logger, "TypedBaseItems", "IsNews", "BIT");
- _connection.AddColumn(_logger, "TypedBaseItems", "IsPremiere", "BIT");
- _connection.AddColumn(_logger, "TypedBaseItems", "EpisodeTitle", "Text");
- _connection.AddColumn(_logger, "TypedBaseItems", "IsRepeat", "BIT");
- _connection.AddColumn(_logger, "TypedBaseItems", "PreferredMetadataLanguage", "Text");
- _connection.AddColumn(_logger, "TypedBaseItems", "PreferredMetadataCountryCode", "Text");
- _connection.AddColumn(_logger, "TypedBaseItems", "IsHD", "BIT");
- _connection.AddColumn(_logger, "TypedBaseItems", "ExternalEtag", "Text");
- _connection.AddColumn(_logger, "TypedBaseItems", "ExternalImagePath", "Text");
- PrepareStatements();
- _mediaStreamsRepository.Initialize();
- _chapterRepository.Initialize();
- }
- /// <summary>
- /// The _write lock
- /// </summary>
- private readonly SemaphoreSlim _writeLock = new SemaphoreSlim(1, 1);
- private readonly string[] _retriveItemColumns =
- {
- "type",
- "data",
- "StartDate",
- "EndDate",
- "IsOffline",
- "ChannelId",
- "IsMovie",
- "IsSports",
- "IsKids",
- "IsSeries",
- "IsLive",
- "IsNews",
- "IsPremiere",
- "EpisodeTitle",
- "IsRepeat",
- "CommunityRating",
- "CustomRating",
- "IndexNumber",
- "IsLocked",
- "PreferredMetadataLanguage",
- "PreferredMetadataCountryCode",
- "IsHD",
- "ExternalEtag",
- "ExternalImagePath"
- };
- /// <summary>
- /// Prepares the statements.
- /// </summary>
- private void PrepareStatements()
- {
- var saveColumns = new List<string>
- {
- "guid",
- "type",
- "data",
- "Path",
- "StartDate",
- "EndDate",
- "ChannelId",
- "IsKids",
- "IsMovie",
- "IsSports",
- "IsSeries",
- "IsLive",
- "IsNews",
- "IsPremiere",
- "EpisodeTitle",
- "IsRepeat",
- "CommunityRating",
- "CustomRating",
- "IndexNumber",
- "IsLocked",
- "Name",
- "OfficialRating",
- "MediaType",
- "Overview",
- "ParentIndexNumber",
- "PremiereDate",
- "ProductionYear",
- "ParentId",
- "Genres",
- "ParentalRatingValue",
- "SchemaVersion",
- "SortName",
- "RunTimeTicks",
- "OfficialRatingDescription",
- "HomePageUrl",
- "VoteCount",
- "DisplayMediaType",
- "DateCreated",
- "DateModified",
- "ForcedSortName",
- "IsOffline",
- "LocationType",
- "PreferredMetadataLanguage",
- "PreferredMetadataCountryCode",
- "IsHD",
- "ExternalEtag",
- "ExternalImagePath"
- };
- _saveItemCommand = _connection.CreateCommand();
- _saveItemCommand.CommandText = "replace into TypedBaseItems (" + string.Join(",", saveColumns.ToArray()) + ") values (";
- for (var i = 1; i <= saveColumns.Count; i++)
- {
- if (i > 1)
- {
- _saveItemCommand.CommandText += ",";
- }
- _saveItemCommand.CommandText += "@" + i.ToString(CultureInfo.InvariantCulture);
- _saveItemCommand.Parameters.Add(_saveItemCommand, "@" + i.ToString(CultureInfo.InvariantCulture));
- }
- _saveItemCommand.CommandText += ")";
- _deleteChildrenCommand = _connection.CreateCommand();
- _deleteChildrenCommand.CommandText = "delete from ChildrenIds where ParentId=@ParentId";
- _deleteChildrenCommand.Parameters.Add(_deleteChildrenCommand, "@ParentId");
- _deleteItemCommand = _connection.CreateCommand();
- _deleteItemCommand.CommandText = "delete from TypedBaseItems where guid=@Id";
- _deleteItemCommand.Parameters.Add(_deleteItemCommand, "@Id");
- _saveChildrenCommand = _connection.CreateCommand();
- _saveChildrenCommand.CommandText = "replace into ChildrenIds (ParentId, ItemId) values (@ParentId, @ItemId)";
- _saveChildrenCommand.Parameters.Add(_saveChildrenCommand, "@ParentId");
- _saveChildrenCommand.Parameters.Add(_saveChildrenCommand, "@ItemId");
- _deletePeopleCommand = _connection.CreateCommand();
- _deletePeopleCommand.CommandText = "delete from People where ItemId=@Id";
- _deletePeopleCommand.Parameters.Add(_deletePeopleCommand, "@Id");
- _savePersonCommand = _connection.CreateCommand();
- _savePersonCommand.CommandText = "insert into People (ItemId, Name, Role, PersonType, SortOrder, ListOrder) values (@ItemId, @Name, @Role, @PersonType, @SortOrder, @ListOrder)";
- _savePersonCommand.Parameters.Add(_savePersonCommand, "@ItemId");
- _savePersonCommand.Parameters.Add(_savePersonCommand, "@Name");
- _savePersonCommand.Parameters.Add(_savePersonCommand, "@Role");
- _savePersonCommand.Parameters.Add(_savePersonCommand, "@PersonType");
- _savePersonCommand.Parameters.Add(_savePersonCommand, "@SortOrder");
- _savePersonCommand.Parameters.Add(_savePersonCommand, "@ListOrder");
- }
- /// <summary>
- /// Save a standard item in the repo
- /// </summary>
- /// <param name="item">The item.</param>
- /// <param name="cancellationToken">The cancellation token.</param>
- /// <returns>Task.</returns>
- /// <exception cref="System.ArgumentNullException">item</exception>
- public Task SaveItem(BaseItem item, CancellationToken cancellationToken)
- {
- if (item == null)
- {
- throw new ArgumentNullException("item");
- }
- return SaveItems(new[] { item }, cancellationToken);
- }
- /// <summary>
- /// Saves the items.
- /// </summary>
- /// <param name="items">The items.</param>
- /// <param name="cancellationToken">The cancellation token.</param>
- /// <returns>Task.</returns>
- /// <exception cref="System.ArgumentNullException">
- /// items
- /// or
- /// cancellationToken
- /// </exception>
- public async Task SaveItems(IEnumerable<BaseItem> items, CancellationToken cancellationToken)
- {
- if (items == null)
- {
- throw new ArgumentNullException("items");
- }
- cancellationToken.ThrowIfCancellationRequested();
- CheckDisposed();
- await _writeLock.WaitAsync(cancellationToken).ConfigureAwait(false);
- IDbTransaction transaction = null;
- try
- {
- transaction = _connection.BeginTransaction();
- foreach (var item in items)
- {
- cancellationToken.ThrowIfCancellationRequested();
- var index = 0;
- _saveItemCommand.GetParameter(index++).Value = item.Id;
- _saveItemCommand.GetParameter(index++).Value = item.GetType().FullName;
- _saveItemCommand.GetParameter(index++).Value = _jsonSerializer.SerializeToBytes(item);
- _saveItemCommand.GetParameter(index++).Value = item.Path;
- var hasStartDate = item as IHasStartDate;
- if (hasStartDate != null)
- {
- _saveItemCommand.GetParameter(index++).Value = hasStartDate.StartDate;
- }
- else
- {
- _saveItemCommand.GetParameter(index++).Value = null;
- }
- _saveItemCommand.GetParameter(index++).Value = item.EndDate;
- _saveItemCommand.GetParameter(index++).Value = item.ChannelId;
- var hasProgramAttributes = item as IHasProgramAttributes;
- if (hasProgramAttributes != null)
- {
- _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.IsKids;
- _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.IsMovie;
- _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.IsSports;
- _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.IsSeries;
- _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.IsLive;
- _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.IsNews;
- _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.IsPremiere;
- _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.EpisodeTitle;
- _saveItemCommand.GetParameter(index++).Value = hasProgramAttributes.IsRepeat;
- }
- else
- {
- _saveItemCommand.GetParameter(index++).Value = null;
- _saveItemCommand.GetParameter(index++).Value = null;
- _saveItemCommand.GetParameter(index++).Value = null;
- _saveItemCommand.GetParameter(index++).Value = null;
- _saveItemCommand.GetParameter(index++).Value = null;
- _saveItemCommand.GetParameter(index++).Value = null;
- _saveItemCommand.GetParameter(index++).Value = null;
- _saveItemCommand.GetParameter(index++).Value = null;
- _saveItemCommand.GetParameter(index++).Value = null;
- }
- _saveItemCommand.GetParameter(index++).Value = item.CommunityRating;
- _saveItemCommand.GetParameter(index++).Value = item.CustomRating;
- _saveItemCommand.GetParameter(index++).Value = item.IndexNumber;
- _saveItemCommand.GetParameter(index++).Value = item.IsLocked;
- _saveItemCommand.GetParameter(index++).Value = item.Name;
- _saveItemCommand.GetParameter(index++).Value = item.OfficialRating;
- _saveItemCommand.GetParameter(index++).Value = item.MediaType;
- _saveItemCommand.GetParameter(index++).Value = item.Overview;
- _saveItemCommand.GetParameter(index++).Value = item.ParentIndexNumber;
- _saveItemCommand.GetParameter(index++).Value = item.PremiereDate;
- _saveItemCommand.GetParameter(index++).Value = item.ProductionYear;
- if (item.ParentId == Guid.Empty)
- {
- _saveItemCommand.GetParameter(index++).Value = null;
- }
- else
- {
- _saveItemCommand.GetParameter(index++).Value = item.ParentId;
- }
- _saveItemCommand.GetParameter(index++).Value = string.Join("|", item.Genres.ToArray());
- _saveItemCommand.GetParameter(index++).Value = item.GetParentalRatingValue();
- _saveItemCommand.GetParameter(index++).Value = LatestSchemaVersion;
- _saveItemCommand.GetParameter(index++).Value = item.SortName;
- _saveItemCommand.GetParameter(index++).Value = item.RunTimeTicks;
- _saveItemCommand.GetParameter(index++).Value = item.OfficialRatingDescription;
- _saveItemCommand.GetParameter(index++).Value = item.HomePageUrl;
- _saveItemCommand.GetParameter(index++).Value = item.VoteCount;
- _saveItemCommand.GetParameter(index++).Value = item.DisplayMediaType;
- _saveItemCommand.GetParameter(index++).Value = item.DateCreated;
- _saveItemCommand.GetParameter(index++).Value = item.DateModified;
- _saveItemCommand.GetParameter(index++).Value = item.ForcedSortName;
- _saveItemCommand.GetParameter(index++).Value = item.IsOffline;
- _saveItemCommand.GetParameter(index++).Value = item.LocationType.ToString();
- _saveItemCommand.GetParameter(index++).Value = item.PreferredMetadataLanguage;
- _saveItemCommand.GetParameter(index++).Value = item.PreferredMetadataCountryCode;
- _saveItemCommand.GetParameter(index++).Value = item.IsHD;
- _saveItemCommand.GetParameter(index++).Value = item.ExternalEtag;
- _saveItemCommand.GetParameter(index++).Value = item.ExternalImagePath;
- _saveItemCommand.Transaction = transaction;
- _saveItemCommand.ExecuteNonQuery();
- }
- transaction.Commit();
- }
- catch (OperationCanceledException)
- {
- if (transaction != null)
- {
- transaction.Rollback();
- }
- throw;
- }
- catch (Exception e)
- {
- _logger.ErrorException("Failed to save items:", e);
- if (transaction != null)
- {
- transaction.Rollback();
- }
- throw;
- }
- finally
- {
- if (transaction != null)
- {
- transaction.Dispose();
- }
- _writeLock.Release();
- }
- }
- /// <summary>
- /// Internal retrieve from items or users table
- /// </summary>
- /// <param name="id">The id.</param>
- /// <returns>BaseItem.</returns>
- /// <exception cref="System.ArgumentNullException">id</exception>
- /// <exception cref="System.ArgumentException"></exception>
- public BaseItem RetrieveItem(Guid id)
- {
- if (id == Guid.Empty)
- {
- throw new ArgumentNullException("id");
- }
- CheckDisposed();
- using (var cmd = _connection.CreateCommand())
- {
- cmd.CommandText = "select " + string.Join(",", _retriveItemColumns) + " from TypedBaseItems where guid = @guid";
- cmd.Parameters.Add(cmd, "@guid", DbType.Guid).Value = id;
- using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult | CommandBehavior.SingleRow))
- {
- if (reader.Read())
- {
- return GetItem(reader);
- }
- }
- return null;
- }
- }
- private BaseItem GetItem(IDataReader reader)
- {
- var typeString = reader.GetString(0);
- var type = _typeMapper.GetType(typeString);
- if (type == null)
- {
- _logger.Debug("Unknown type {0}", typeString);
- return null;
- }
- BaseItem item;
- using (var stream = reader.GetMemoryStream(1))
- {
- try
- {
- item = _jsonSerializer.DeserializeFromStream(stream, type) as BaseItem;
- if (item == null)
- {
- return null;
- }
- }
- catch (SerializationException ex)
- {
- _logger.ErrorException("Error deserializing item", ex);
- return null;
- }
- }
- if (!reader.IsDBNull(2))
- {
- var hasStartDate = item as IHasStartDate;
- if (hasStartDate != null)
- {
- hasStartDate.StartDate = reader.GetDateTime(2).ToUniversalTime();
- }
- }
- if (!reader.IsDBNull(3))
- {
- item.EndDate = reader.GetDateTime(3).ToUniversalTime();
- }
- if (!reader.IsDBNull(4))
- {
- item.IsOffline = reader.GetBoolean(4);
- }
- if (!reader.IsDBNull(5))
- {
- item.ChannelId = reader.GetString(5);
- }
- var hasProgramAttributes = item as IHasProgramAttributes;
- if (hasProgramAttributes != null)
- {
- if (!reader.IsDBNull(6))
- {
- hasProgramAttributes.IsMovie = reader.GetBoolean(6);
- }
- if (!reader.IsDBNull(7))
- {
- hasProgramAttributes.IsSports = reader.GetBoolean(7);
- }
- if (!reader.IsDBNull(8))
- {
- hasProgramAttributes.IsKids = reader.GetBoolean(8);
- }
- if (!reader.IsDBNull(9))
- {
- hasProgramAttributes.IsSeries = reader.GetBoolean(9);
- }
- if (!reader.IsDBNull(10))
- {
- hasProgramAttributes.IsLive = reader.GetBoolean(10);
- }
- if (!reader.IsDBNull(11))
- {
- hasProgramAttributes.IsNews = reader.GetBoolean(11);
- }
- if (!reader.IsDBNull(12))
- {
- hasProgramAttributes.IsPremiere = reader.GetBoolean(12);
- }
- if (!reader.IsDBNull(13))
- {
- hasProgramAttributes.EpisodeTitle = reader.GetString(13);
- }
- if (!reader.IsDBNull(14))
- {
- hasProgramAttributes.IsRepeat = reader.GetBoolean(14);
- }
- }
- if (!reader.IsDBNull(15))
- {
- item.CommunityRating = reader.GetFloat(15);
- }
- if (!reader.IsDBNull(16))
- {
- item.CustomRating = reader.GetString(16);
- }
- if (!reader.IsDBNull(17))
- {
- item.IndexNumber = reader.GetInt32(17);
- }
- if (!reader.IsDBNull(18))
- {
- item.IsLocked = reader.GetBoolean(18);
- }
- if (!reader.IsDBNull(19))
- {
- item.PreferredMetadataLanguage = reader.GetString(19);
- }
- if (!reader.IsDBNull(20))
- {
- item.PreferredMetadataCountryCode = reader.GetString(20);
- }
- if (!reader.IsDBNull(21))
- {
- item.IsHD = reader.GetBoolean(21);
- }
- if (!reader.IsDBNull(22))
- {
- item.ExternalEtag = reader.GetString(22);
- }
- if (!reader.IsDBNull(23))
- {
- item.ExternalImagePath = reader.GetString(23);
- }
- return item;
- }
- /// <summary>
- /// Gets the critic reviews.
- /// </summary>
- /// <param name="itemId">The item id.</param>
- /// <returns>Task{IEnumerable{ItemReview}}.</returns>
- public IEnumerable<ItemReview> GetCriticReviews(Guid itemId)
- {
- try
- {
- var path = Path.Combine(_criticReviewsPath, itemId + ".json");
- return _jsonSerializer.DeserializeFromFile<List<ItemReview>>(path);
- }
- catch (DirectoryNotFoundException)
- {
- return new List<ItemReview>();
- }
- catch (FileNotFoundException)
- {
- return new List<ItemReview>();
- }
- }
- private readonly Task _cachedTask = Task.FromResult(true);
- /// <summary>
- /// Saves the critic reviews.
- /// </summary>
- /// <param name="itemId">The item id.</param>
- /// <param name="criticReviews">The critic reviews.</param>
- /// <returns>Task.</returns>
- public Task SaveCriticReviews(Guid itemId, IEnumerable<ItemReview> criticReviews)
- {
- Directory.CreateDirectory(_criticReviewsPath);
- var path = Path.Combine(_criticReviewsPath, itemId + ".json");
- _jsonSerializer.SerializeToFile(criticReviews.ToList(), path);
- return _cachedTask;
- }
- /// <summary>
- /// Gets chapters for an item
- /// </summary>
- /// <param name="id">The id.</param>
- /// <returns>IEnumerable{ChapterInfo}.</returns>
- /// <exception cref="System.ArgumentNullException">id</exception>
- public IEnumerable<ChapterInfo> GetChapters(Guid id)
- {
- CheckDisposed();
- return _chapterRepository.GetChapters(id);
- }
- /// <summary>
- /// Gets a single chapter for an item
- /// </summary>
- /// <param name="id">The id.</param>
- /// <param name="index">The index.</param>
- /// <returns>ChapterInfo.</returns>
- /// <exception cref="System.ArgumentNullException">id</exception>
- public ChapterInfo GetChapter(Guid id, int index)
- {
- CheckDisposed();
- return _chapterRepository.GetChapter(id, index);
- }
- /// <summary>
- /// Saves the chapters.
- /// </summary>
- /// <param name="id">The id.</param>
- /// <param name="chapters">The chapters.</param>
- /// <param name="cancellationToken">The cancellation token.</param>
- /// <returns>Task.</returns>
- /// <exception cref="System.ArgumentNullException">
- /// id
- /// or
- /// chapters
- /// or
- /// cancellationToken
- /// </exception>
- public Task SaveChapters(Guid id, IEnumerable<ChapterInfo> chapters, CancellationToken cancellationToken)
- {
- CheckDisposed();
- return _chapterRepository.SaveChapters(id, chapters, cancellationToken);
- }
- /// <summary>
- /// Performs application-defined tasks associated with freeing, releasing, or resetting unmanaged resources.
- /// </summary>
- public void Dispose()
- {
- Dispose(true);
- GC.SuppressFinalize(this);
- }
- private readonly object _disposeLock = new object();
- private bool _disposed;
- private void CheckDisposed()
- {
- if (_disposed)
- {
- throw new ObjectDisposedException(GetType().Name + " has been disposed and cannot be accessed.");
- }
- }
- /// <summary>
- /// Releases unmanaged and - optionally - managed resources.
- /// </summary>
- /// <param name="dispose"><c>true</c> to release both managed and unmanaged resources; <c>false</c> to release only unmanaged resources.</param>
- protected virtual void Dispose(bool dispose)
- {
- if (dispose)
- {
- _disposed = true;
- try
- {
- lock (_disposeLock)
- {
- _writeLock.Wait();
- if (_connection != null)
- {
- if (_connection.IsOpen())
- {
- _connection.Close();
- }
- _connection.Dispose();
- _connection = null;
- }
- if (_chapterRepository != null)
- {
- _chapterRepository.Dispose();
- _chapterRepository = null;
- }
- if (_mediaStreamsRepository != null)
- {
- _mediaStreamsRepository.Dispose();
- _mediaStreamsRepository = null;
- }
- }
- }
- catch (Exception ex)
- {
- _logger.ErrorException("Error disposing database", ex);
- }
- }
- }
- public IEnumerable<Guid> GetChildren(Guid parentId)
- {
- if (parentId == Guid.Empty)
- {
- throw new ArgumentNullException("parentId");
- }
- CheckDisposed();
- using (var cmd = _connection.CreateCommand())
- {
- cmd.CommandText = "select ItemId from ChildrenIds where ParentId = @ParentId";
- cmd.Parameters.Add(cmd, "@ParentId", DbType.Guid).Value = parentId;
- using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult))
- {
- while (reader.Read())
- {
- yield return reader.GetGuid(0);
- }
- }
- }
- }
- public IEnumerable<BaseItem> GetChildrenItems(Guid parentId)
- {
- if (parentId == Guid.Empty)
- {
- throw new ArgumentNullException("parentId");
- }
- CheckDisposed();
- using (var cmd = _connection.CreateCommand())
- {
- cmd.CommandText = "select " + string.Join(",", _retriveItemColumns) + " from TypedBaseItems where guid in (select ItemId from ChildrenIds where ParentId = @ParentId)";
- cmd.Parameters.Add(cmd, "@ParentId", DbType.Guid).Value = parentId;
- using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult))
- {
- while (reader.Read())
- {
- var item = GetItem(reader);
- if (item != null)
- {
- yield return item;
- }
- }
- }
- }
- }
- public IEnumerable<BaseItem> GetItemsOfType(Type type)
- {
- if (type == null)
- {
- throw new ArgumentNullException("type");
- }
- CheckDisposed();
- using (var cmd = _connection.CreateCommand())
- {
- cmd.CommandText = "select " + string.Join(",", _retriveItemColumns) + " from TypedBaseItems where type = @type";
- cmd.Parameters.Add(cmd, "@type", DbType.String).Value = type.FullName;
- using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult))
- {
- while (reader.Read())
- {
- var item = GetItem(reader);
- if (item != null)
- {
- yield return item;
- }
- }
- }
- }
- }
- public QueryResult<BaseItem> GetItems(InternalItemsQuery query)
- {
- if (query == null)
- {
- throw new ArgumentNullException("query");
- }
- CheckDisposed();
- using (var cmd = _connection.CreateCommand())
- {
- cmd.CommandText = "select " + string.Join(",", _retriveItemColumns) + " from TypedBaseItems";
- var whereClauses = GetWhereClauses(query, cmd, false);
- var whereTextWithoutPaging = whereClauses.Count == 0 ?
- string.Empty :
- " where " + string.Join(" AND ", whereClauses.ToArray());
- whereClauses = GetWhereClauses(query, cmd, true);
- var whereText = whereClauses.Count == 0 ?
- string.Empty :
- " where " + string.Join(" AND ", whereClauses.ToArray());
- cmd.CommandText += whereText;
- cmd.CommandText += GetOrderByText(query);
- if (query.Limit.HasValue)
- {
- cmd.CommandText += " LIMIT " + query.Limit.Value.ToString(CultureInfo.InvariantCulture);
- }
- cmd.CommandText += "; select count (guid) from TypedBaseItems" + whereTextWithoutPaging;
- _logger.Debug(cmd.CommandText);
- var list = new List<BaseItem>();
- var count = 0;
- using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess))
- {
- while (reader.Read())
- {
- var item = GetItem(reader);
- if (item != null)
- {
- list.Add(item);
- }
- }
- if (reader.NextResult() && reader.Read())
- {
- count = reader.GetInt32(0);
- }
- }
- return new QueryResult<BaseItem>()
- {
- Items = list.ToArray(),
- TotalRecordCount = count
- };
- }
- }
- private string GetOrderByText(InternalItemsQuery query)
- {
- if (query.SortBy == null || query.SortBy.Length == 0)
- {
- return string.Empty;
- }
- var sortOrder = query.SortOrder == SortOrder.Descending ? "DESC" : "ASC";
- return " ORDER BY " + string.Join(",", query.SortBy.Select(i => MapOrderByField(i) + " " + sortOrder).ToArray());
- }
- private string MapOrderByField(string name)
- {
- return name;
- }
- public List<Guid> GetItemIdsList(InternalItemsQuery query)
- {
- if (query == null)
- {
- throw new ArgumentNullException("query");
- }
- CheckDisposed();
- using (var cmd = _connection.CreateCommand())
- {
- cmd.CommandText = "select guid from TypedBaseItems";
- var whereClauses = GetWhereClauses(query, cmd, true);
- var whereText = whereClauses.Count == 0 ?
- string.Empty :
- " where " + string.Join(" AND ", whereClauses.ToArray());
- cmd.CommandText += whereText;
- cmd.CommandText += GetOrderByText(query);
- if (query.Limit.HasValue)
- {
- cmd.CommandText += " LIMIT " + query.Limit.Value.ToString(CultureInfo.InvariantCulture);
- }
- var list = new List<Guid>();
- _logger.Debug(cmd.CommandText);
- using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess))
- {
- while (reader.Read())
- {
- list.Add(reader.GetGuid(0));
- }
- }
- return list;
- }
- }
- public QueryResult<Tuple<Guid, string>> GetItemIdsWithPath(InternalItemsQuery query)
- {
- if (query == null)
- {
- throw new ArgumentNullException("query");
- }
- CheckDisposed();
- using (var cmd = _connection.CreateCommand())
- {
- cmd.CommandText = "select guid,path from TypedBaseItems";
- var whereClauses = GetWhereClauses(query, cmd, false);
- var whereTextWithoutPaging = whereClauses.Count == 0 ?
- string.Empty :
- " where " + string.Join(" AND ", whereClauses.ToArray());
- whereClauses = GetWhereClauses(query, cmd, true);
- var whereText = whereClauses.Count == 0 ?
- string.Empty :
- " where " + string.Join(" AND ", whereClauses.ToArray());
- cmd.CommandText += whereText;
- cmd.CommandText += GetOrderByText(query);
- if (query.Limit.HasValue)
- {
- cmd.CommandText += " LIMIT " + query.Limit.Value.ToString(CultureInfo.InvariantCulture);
- }
- cmd.CommandText += "; select count (guid) from TypedBaseItems" + whereTextWithoutPaging;
- var list = new List<Tuple<Guid, string>>();
- var count = 0;
- _logger.Debug(cmd.CommandText);
- using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess))
- {
- while (reader.Read())
- {
- var id = reader.GetGuid(0);
- string path = null;
- if (!reader.IsDBNull(1))
- {
- path = reader.GetString(1);
- }
- list.Add(new Tuple<Guid, string>(id, path));
- }
- if (reader.NextResult() && reader.Read())
- {
- count = reader.GetInt32(0);
- }
- }
- return new QueryResult<Tuple<Guid, string>>()
- {
- Items = list.ToArray(),
- TotalRecordCount = count
- };
- }
- }
- public QueryResult<Guid> GetItemIds(InternalItemsQuery query)
- {
- if (query == null)
- {
- throw new ArgumentNullException("query");
- }
- CheckDisposed();
- using (var cmd = _connection.CreateCommand())
- {
- cmd.CommandText = "select guid from TypedBaseItems";
- var whereClauses = GetWhereClauses(query, cmd, false);
- var whereTextWithoutPaging = whereClauses.Count == 0 ?
- string.Empty :
- " where " + string.Join(" AND ", whereClauses.ToArray());
- whereClauses = GetWhereClauses(query, cmd, true);
- var whereText = whereClauses.Count == 0 ?
- string.Empty :
- " where " + string.Join(" AND ", whereClauses.ToArray());
- cmd.CommandText += whereText;
- cmd.CommandText += GetOrderByText(query);
- if (query.Limit.HasValue)
- {
- cmd.CommandText += " LIMIT " + query.Limit.Value.ToString(CultureInfo.InvariantCulture);
- }
- cmd.CommandText += "; select count (guid) from TypedBaseItems" + whereTextWithoutPaging;
- var list = new List<Guid>();
- var count = 0;
- _logger.Debug(cmd.CommandText);
- using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess))
- {
- while (reader.Read())
- {
- list.Add(reader.GetGuid(0));
- }
- if (reader.NextResult() && reader.Read())
- {
- count = reader.GetInt32(0);
- }
- }
- return new QueryResult<Guid>()
- {
- Items = list.ToArray(),
- TotalRecordCount = count
- };
- }
- }
- private List<string> GetWhereClauses(InternalItemsQuery query, IDbCommand cmd, bool addPaging)
- {
- var whereClauses = new List<string>();
- if (query.IsCurrentSchema.HasValue)
- {
- if (query.IsCurrentSchema.Value)
- {
- whereClauses.Add("(SchemaVersion not null AND SchemaVersion=@SchemaVersion)");
- }
- else
- {
- whereClauses.Add("(SchemaVersion is null or SchemaVersion<>@SchemaVersion)");
- }
- cmd.Parameters.Add(cmd, "@SchemaVersion", DbType.Int32).Value = LatestSchemaVersion;
- }
- if (query.IsOffline.HasValue)
- {
- whereClauses.Add("IsOffline=@IsOffline");
- cmd.Parameters.Add(cmd, "@IsOffline", DbType.Boolean).Value = query.IsOffline;
- }
- if (query.LocationType.HasValue)
- {
- whereClauses.Add("LocationType=@LocationType");
- cmd.Parameters.Add(cmd, "@LocationType", DbType.String).Value = query.LocationType.Value;
- }
- if (query.IsMovie.HasValue)
- {
- whereClauses.Add("IsMovie=@IsMovie");
- cmd.Parameters.Add(cmd, "@IsMovie", DbType.Boolean).Value = query.IsMovie;
- }
- if (query.IsKids.HasValue)
- {
- whereClauses.Add("IsKids=@IsKids");
- cmd.Parameters.Add(cmd, "@IsKids", DbType.Boolean).Value = query.IsKids;
- }
- if (query.IsSports.HasValue)
- {
- whereClauses.Add("IsSports=@IsSports");
- cmd.Parameters.Add(cmd, "@IsSports", DbType.Boolean).Value = query.IsSports;
- }
- var includeTypes = query.IncludeItemTypes.SelectMany(MapIncludeItemTypes).ToArray();
- if (includeTypes.Length == 1)
- {
- whereClauses.Add("type=@type");
- cmd.Parameters.Add(cmd, "@type", DbType.String).Value = includeTypes[0];
- }
- else if (includeTypes.Length > 1)
- {
- var inClause = string.Join(",", includeTypes.Select(i => "'" + i + "'").ToArray());
- whereClauses.Add(string.Format("type in ({0})", inClause));
- }
- var excludeTypes = query.ExcludeItemTypes.SelectMany(MapIncludeItemTypes).ToArray();
- if (excludeTypes.Length == 1)
- {
- whereClauses.Add("type<>@type");
- cmd.Parameters.Add(cmd, "@type", DbType.String).Value = excludeTypes[0];
- }
- else if (excludeTypes.Length > 1)
- {
- var inClause = string.Join(",", excludeTypes.Select(i => "'" + i + "'").ToArray());
- whereClauses.Add(string.Format("type not in ({0})", inClause));
- }
- if (query.ChannelIds.Length == 1)
- {
- whereClauses.Add("ChannelId=@ChannelId");
- cmd.Parameters.Add(cmd, "@ChannelId", DbType.String).Value = query.ChannelIds[0];
- }
- if (query.ChannelIds.Length > 1)
- {
- var inClause = string.Join(",", query.ChannelIds.Select(i => "'" + i + "'").ToArray());
- whereClauses.Add(string.Format("ChannelId in ({0})", inClause));
- }
- if (query.ParentId.HasValue)
- {
- whereClauses.Add("ParentId=@ParentId");
- cmd.Parameters.Add(cmd, "@ParentId", DbType.Guid).Value = query.ParentId.Value;
- }
- if (query.MinEndDate.HasValue)
- {
- whereClauses.Add("EndDate>=@MinEndDate");
- cmd.Parameters.Add(cmd, "@MinEndDate", DbType.Date).Value = query.MinEndDate.Value;
- }
- if (query.MaxEndDate.HasValue)
- {
- whereClauses.Add("EndDate<=@MaxEndDate");
- cmd.Parameters.Add(cmd, "@MaxEndDate", DbType.Date).Value = query.MaxEndDate.Value;
- }
- if (query.MinStartDate.HasValue)
- {
- whereClauses.Add("StartDate>=@MinStartDate");
- cmd.Parameters.Add(cmd, "@MinStartDate", DbType.Date).Value = query.MinStartDate.Value;
- }
- if (query.MaxStartDate.HasValue)
- {
- whereClauses.Add("StartDate<=@MaxStartDate");
- cmd.Parameters.Add(cmd, "@MaxStartDate", DbType.Date).Value = query.MaxStartDate.Value;
- }
- if (query.IsAiring.HasValue)
- {
- if (query.IsAiring.Value)
- {
- whereClauses.Add("StartDate<=@MaxStartDate");
- cmd.Parameters.Add(cmd, "@MaxStartDate", DbType.Date).Value = DateTime.UtcNow;
- whereClauses.Add("EndDate>=@MinEndDate");
- cmd.Parameters.Add(cmd, "@MinEndDate", DbType.Date).Value = DateTime.UtcNow;
- }
- else
- {
- whereClauses.Add("(StartDate>@IsAiringDate OR EndDate < @IsAiringDate)");
- cmd.Parameters.Add(cmd, "@IsAiringDate", DbType.Date).Value = DateTime.UtcNow;
- }
- }
- if (!string.IsNullOrWhiteSpace(query.Person))
- {
- whereClauses.Add("Guid in (select ItemId from People where Name=@PersonName)");
- cmd.Parameters.Add(cmd, "@PersonName", DbType.String).Value = query.Person;
- }
- if (!string.IsNullOrWhiteSpace(query.NameContains))
- {
- whereClauses.Add("Name like @NameContains");
- cmd.Parameters.Add(cmd, "@NameContains", DbType.String).Value = "%" + query.NameContains + "%";
- }
- if (query.Genres.Length > 0)
- {
- var genres = new List<string>();
- var index = 0;
- foreach (var genre in query.Genres)
- {
- genres.Add("Genres like @Genres" + index);
- cmd.Parameters.Add(cmd, "@Genres" + index, DbType.String).Value = "%" + genre + "%";
- index++;
- }
- var genreCaluse = "(" + string.Join(" OR ", genres.ToArray()) + ")";
- whereClauses.Add(genreCaluse);
- }
- if (query.MaxParentalRating.HasValue)
- {
- whereClauses.Add("(ParentalRatingValue is NULL OR ParentalRatingValue<=@MaxParentalRating)");
- cmd.Parameters.Add(cmd, "@MaxParentalRating", DbType.Int32).Value = query.MaxParentalRating.Value;
- }
- if (query.HasParentalRating.HasValue)
- {
- if (query.HasParentalRating.Value)
- {
- whereClauses.Add("ParentalRatingValue NOT NULL");
- }
- else
- {
- whereClauses.Add("ParentalRatingValue IS NULL");
- }
- }
- if (query.HasDeadParentId.HasValue)
- {
- if (query.HasDeadParentId.Value)
- {
- whereClauses.Add("ParentId NOT NULL AND ParentId NOT IN (select guid from TypedBaseItems)");
- }
- }
- if (addPaging)
- {
- if (query.StartIndex.HasValue && query.StartIndex.Value > 0)
- {
- var pagingWhereText = whereClauses.Count == 0 ?
- string.Empty :
- " where " + string.Join(" AND ", whereClauses.ToArray());
- var orderBy = GetOrderByText(query);
- whereClauses.Add(string.Format("guid NOT IN (SELECT guid FROM TypedBaseItems {0}" + orderBy + " LIMIT {1})",
- pagingWhereText,
- query.StartIndex.Value.ToString(CultureInfo.InvariantCulture)));
- }
- }
- return whereClauses;
- }
- private static readonly Type[] KnownTypes =
- {
- typeof(LiveTvProgram),
- typeof(LiveTvChannel),
- typeof(LiveTvVideoRecording),
- typeof(LiveTvAudioRecording),
- typeof(Series),
- typeof(Audio),
- typeof(MusicAlbum),
- typeof(MusicArtist),
- typeof(MusicGenre),
- typeof(MusicVideo),
- typeof(Movie),
- typeof(BoxSet),
- typeof(Episode),
- typeof(Season),
- typeof(Series),
- typeof(Book),
- typeof(CollectionFolder),
- typeof(Folder),
- typeof(Game),
- typeof(GameGenre),
- typeof(GameSystem),
- typeof(Genre),
- typeof(Person),
- typeof(Photo),
- typeof(PhotoAlbum),
- typeof(Studio),
- typeof(UserRootFolder),
- typeof(UserView),
- typeof(Video),
- typeof(Year),
- typeof(Channel)
- };
- private static Dictionary<string, string[]> GetTypeMapDictionary()
- {
- var dict = new Dictionary<string, string[]>();
- foreach (var t in KnownTypes)
- {
- dict[t.Name] = new[] { t.FullName };
- }
- dict["Recording"] = new[] { typeof(LiveTvAudioRecording).FullName, typeof(LiveTvVideoRecording).FullName };
- dict["Program"] = new[] { typeof(LiveTvProgram).FullName };
- dict["TvChannel"] = new[] { typeof(LiveTvChannel).FullName };
- return dict;
- }
- // Not crazy about having this all the way down here, but at least it's in one place
- readonly Dictionary<string, string[]> _types = GetTypeMapDictionary();
- private IEnumerable<string> MapIncludeItemTypes(string value)
- {
- string[] result;
- if (_types.TryGetValue(value, out result))
- {
- return result;
- }
- return new[] { value };
- }
- public async Task DeleteItem(Guid id, CancellationToken cancellationToken)
- {
- if (id == Guid.Empty)
- {
- throw new ArgumentNullException("id");
- }
- CheckDisposed();
- await _writeLock.WaitAsync(cancellationToken).ConfigureAwait(false);
- IDbTransaction transaction = null;
- try
- {
- transaction = _connection.BeginTransaction();
- // First delete children
- _deleteChildrenCommand.GetParameter(0).Value = id;
- _deleteChildrenCommand.Transaction = transaction;
- _deleteChildrenCommand.ExecuteNonQuery();
- // Delete people
- _deletePeopleCommand.GetParameter(0).Value = id;
- _deletePeopleCommand.Transaction = transaction;
- _deletePeopleCommand.ExecuteNonQuery();
- // Delete the item
- _deleteItemCommand.GetParameter(0).Value = id;
- _deleteItemCommand.Transaction = transaction;
- _deleteItemCommand.ExecuteNonQuery();
- transaction.Commit();
- }
- catch (OperationCanceledException)
- {
- if (transaction != null)
- {
- transaction.Rollback();
- }
- throw;
- }
- catch (Exception e)
- {
- _logger.ErrorException("Failed to save children:", e);
- if (transaction != null)
- {
- transaction.Rollback();
- }
- throw;
- }
- finally
- {
- if (transaction != null)
- {
- transaction.Dispose();
- }
- _writeLock.Release();
- }
- }
- public async Task SaveChildren(Guid parentId, IEnumerable<Guid> children, CancellationToken cancellationToken)
- {
- if (parentId == Guid.Empty)
- {
- throw new ArgumentNullException("parentId");
- }
- if (children == null)
- {
- throw new ArgumentNullException("children");
- }
- CheckDisposed();
- await _writeLock.WaitAsync(cancellationToken).ConfigureAwait(false);
- IDbTransaction transaction = null;
- try
- {
- transaction = _connection.BeginTransaction();
- // First delete
- _deleteChildrenCommand.GetParameter(0).Value = parentId;
- _deleteChildrenCommand.Transaction = transaction;
- _deleteChildrenCommand.ExecuteNonQuery();
- foreach (var id in children)
- {
- cancellationToken.ThrowIfCancellationRequested();
- _saveChildrenCommand.GetParameter(0).Value = parentId;
- _saveChildrenCommand.GetParameter(1).Value = id;
- _saveChildrenCommand.Transaction = transaction;
- _saveChildrenCommand.ExecuteNonQuery();
- }
- transaction.Commit();
- }
- catch (OperationCanceledException)
- {
- if (transaction != null)
- {
- transaction.Rollback();
- }
- throw;
- }
- catch (Exception e)
- {
- _logger.ErrorException("Failed to save children:", e);
- if (transaction != null)
- {
- transaction.Rollback();
- }
- throw;
- }
- finally
- {
- if (transaction != null)
- {
- transaction.Dispose();
- }
- _writeLock.Release();
- }
- }
- public IEnumerable<MediaStream> GetMediaStreams(MediaStreamQuery query)
- {
- CheckDisposed();
- return _mediaStreamsRepository.GetMediaStreams(query);
- }
- public Task SaveMediaStreams(Guid id, IEnumerable<MediaStream> streams, CancellationToken cancellationToken)
- {
- CheckDisposed();
- return _mediaStreamsRepository.SaveMediaStreams(id, streams, cancellationToken);
- }
- public List<string> GetPeopleNames(InternalPeopleQuery query)
- {
- if (query == null)
- {
- throw new ArgumentNullException("query");
- }
- CheckDisposed();
- using (var cmd = _connection.CreateCommand())
- {
- cmd.CommandText = "select Distinct Name from People";
- var whereClauses = GetPeopleWhereClauses(query, cmd);
- if (whereClauses.Count > 0)
- {
- cmd.CommandText += " where " + string.Join(" AND ", whereClauses.ToArray());
- }
- cmd.CommandText += " order by ListOrder";
- var list = new List<string>();
- using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult))
- {
- while (reader.Read())
- {
- list.Add(reader.GetString(0));
- }
- }
- return list;
- }
- }
- public List<PersonInfo> GetPeople(InternalPeopleQuery query)
- {
- if (query == null)
- {
- throw new ArgumentNullException("query");
- }
- CheckDisposed();
- using (var cmd = _connection.CreateCommand())
- {
- cmd.CommandText = "select ItemId, Name, Role, PersonType, SortOrder from People";
- var whereClauses = GetPeopleWhereClauses(query, cmd);
- if (whereClauses.Count > 0)
- {
- cmd.CommandText += " where " + string.Join(" AND ", whereClauses.ToArray());
- }
- cmd.CommandText += " order by ListOrder";
- var list = new List<PersonInfo>();
- using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult))
- {
- while (reader.Read())
- {
- list.Add(GetPerson(reader));
- }
- }
- return list;
- }
- }
- private List<string> GetPeopleWhereClauses(InternalPeopleQuery query, IDbCommand cmd)
- {
- var whereClauses = new List<string>();
- if (query.ItemId != Guid.Empty)
- {
- whereClauses.Add("ItemId=@ItemId");
- cmd.Parameters.Add(cmd, "@ItemId", DbType.Guid).Value = query.ItemId;
- }
- if (query.AppearsInItemId != Guid.Empty)
- {
- whereClauses.Add("Name in (Select Name from People where ItemId=@AppearsInItemId)");
- cmd.Parameters.Add(cmd, "@AppearsInItemId", DbType.Guid).Value = query.AppearsInItemId;
- }
- if (query.PersonTypes.Count == 1)
- {
- whereClauses.Add("PersonType=@PersonType");
- cmd.Parameters.Add(cmd, "@PersonType", DbType.String).Value = query.PersonTypes[0];
- }
- if (query.PersonTypes.Count > 1)
- {
- var val = string.Join(",", query.PersonTypes.Select(i => "'" + i + "'").ToArray());
- whereClauses.Add("PersonType in (" + val + ")");
- }
- if (query.ExcludePersonTypes.Count == 1)
- {
- whereClauses.Add("PersonType<>@PersonType");
- cmd.Parameters.Add(cmd, "@PersonType", DbType.String).Value = query.ExcludePersonTypes[0];
- }
- if (query.ExcludePersonTypes.Count > 1)
- {
- var val = string.Join(",", query.ExcludePersonTypes.Select(i => "'" + i + "'").ToArray());
- whereClauses.Add("PersonType not in (" + val + ")");
- }
- if (query.MaxListOrder.HasValue)
- {
- whereClauses.Add("ListOrder<=@MaxListOrder");
- cmd.Parameters.Add(cmd, "@MaxListOrder", DbType.Int32).Value = query.MaxListOrder.Value;
- }
- if (!string.IsNullOrWhiteSpace(query.NameContains))
- {
- whereClauses.Add("Name like @NameContains");
- cmd.Parameters.Add(cmd, "@NameContains", DbType.String).Value = "%" + query.NameContains + "%";
- }
- return whereClauses;
- }
- public async Task UpdatePeople(Guid itemId, List<PersonInfo> people)
- {
- if (itemId == Guid.Empty)
- {
- throw new ArgumentNullException("itemId");
- }
- if (people == null)
- {
- throw new ArgumentNullException("people");
- }
- CheckDisposed();
- var cancellationToken = CancellationToken.None;
- await _writeLock.WaitAsync(cancellationToken).ConfigureAwait(false);
- IDbTransaction transaction = null;
- try
- {
- transaction = _connection.BeginTransaction();
- // First delete
- _deletePeopleCommand.GetParameter(0).Value = itemId;
- _deletePeopleCommand.Transaction = transaction;
- _deletePeopleCommand.ExecuteNonQuery();
- var listIndex = 0;
- foreach (var person in people)
- {
- cancellationToken.ThrowIfCancellationRequested();
- _savePersonCommand.GetParameter(0).Value = itemId;
- _savePersonCommand.GetParameter(1).Value = person.Name;
- _savePersonCommand.GetParameter(2).Value = person.Role;
- _savePersonCommand.GetParameter(3).Value = person.Type;
- _savePersonCommand.GetParameter(4).Value = person.SortOrder;
- _savePersonCommand.GetParameter(5).Value = listIndex;
- _savePersonCommand.Transaction = transaction;
- _savePersonCommand.ExecuteNonQuery();
- listIndex++;
- }
- transaction.Commit();
- }
- catch (OperationCanceledException)
- {
- if (transaction != null)
- {
- transaction.Rollback();
- }
- throw;
- }
- catch (Exception e)
- {
- _logger.ErrorException("Failed to save people:", e);
- if (transaction != null)
- {
- transaction.Rollback();
- }
- throw;
- }
- finally
- {
- if (transaction != null)
- {
- transaction.Dispose();
- }
- _writeLock.Release();
- }
- }
- private PersonInfo GetPerson(IDataReader reader)
- {
- var item = new PersonInfo();
- item.ItemId = reader.GetGuid(0);
- item.Name = reader.GetString(1);
- if (!reader.IsDBNull(2))
- {
- item.Role = reader.GetString(2);
- }
- if (!reader.IsDBNull(3))
- {
- item.Type = reader.GetString(3);
- }
- if (!reader.IsDBNull(4))
- {
- item.SortOrder = reader.GetInt32(4);
- }
- return item;
- }
- }
- }
|