using MediaBrowser.Common.Configuration; using MediaBrowser.Controller.Persistence; using MediaBrowser.Model.Entities; using MediaBrowser.Model.Logging; using MediaBrowser.Model.Serialization; using System; using System.Data; using System.IO; using System.Threading; using System.Threading.Tasks; namespace MediaBrowser.Server.Implementations.Sqlite { /// /// Class SQLiteDisplayPreferencesRepository /// class SQLiteDisplayPreferencesRepository : SqliteRepository, IDisplayPreferencesRepository { /// /// The repository name /// public const string RepositoryName = "SQLite"; /// /// Gets the name of the repository /// /// The name. public string Name { get { return RepositoryName; } } /// /// The _protobuf serializer /// private readonly IProtobufSerializer _protobufSerializer; /// /// The _app paths /// private readonly IApplicationPaths _appPaths; /// /// Initializes a new instance of the class. /// /// The app paths. /// The protobuf serializer. /// The log manager. /// protobufSerializer public SQLiteDisplayPreferencesRepository(IApplicationPaths appPaths, IProtobufSerializer protobufSerializer, ILogManager logManager) : base(logManager) { if (protobufSerializer == null) { throw new ArgumentNullException("protobufSerializer"); } if (appPaths == null) { throw new ArgumentNullException("appPaths"); } _protobufSerializer = protobufSerializer; _appPaths = appPaths; } /// /// Opens the connection to the database /// /// Task. public async Task Initialize() { var dbFile = Path.Combine(_appPaths.DataPath, "displaypreferences.db"); await ConnectToDB(dbFile).ConfigureAwait(false); string[] queries = { "create table if not exists displaypreferences (id GUID, data BLOB)", "create unique index if not exists displaypreferencesindex on displaypreferences (id)", "create table if not exists schema_version (table_name primary key, version)", //pragmas "pragma temp_store = memory" }; RunQueries(queries); } /// /// Save the display preferences associated with an item in the repo /// /// The display preferences. /// The cancellation token. /// Task. /// item public Task SaveDisplayPreferences(DisplayPreferences displayPreferences, CancellationToken cancellationToken) { if (displayPreferences == null) { throw new ArgumentNullException("displayPreferences"); } if (displayPreferences.Id == Guid.Empty) { throw new ArgumentNullException("displayPreferences.Id"); } if (cancellationToken == null) { throw new ArgumentNullException("cancellationToken"); } cancellationToken.ThrowIfCancellationRequested(); return Task.Run(() => { var serialized = _protobufSerializer.SerializeToBytes(displayPreferences); cancellationToken.ThrowIfCancellationRequested(); var cmd = connection.CreateCommand(); cmd.CommandText = "replace into displaypreferences (id, data) values (@1, @3)"; cmd.AddParam("@1", displayPreferences.Id); cmd.AddParam("@2", serialized); QueueCommand(cmd); }); } /// /// Gets the display preferences. /// /// The display preferences id. /// Task{DisplayPreferences}. /// item public async Task GetDisplayPreferences(Guid displayPreferencesId) { if (displayPreferencesId == Guid.Empty) { throw new ArgumentNullException("displayPreferencesId"); } var cmd = connection.CreateCommand(); cmd.CommandText = "select data from displaypreferences where id = @id"; var idParam = cmd.Parameters.Add("@id", DbType.Guid); idParam.Value = displayPreferencesId; using (var reader = await cmd.ExecuteReaderAsync(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult | CommandBehavior.SingleRow).ConfigureAwait(false)) { if (reader.Read()) { using (var stream = GetStream(reader, 0)) { return _protobufSerializer.DeserializeFromStream(stream); } } } return null; } } }