aboutsummaryrefslogtreecommitdiff
path: root/MediaBrowser.Server.Implementations/Persistence
diff options
context:
space:
mode:
authorEric Reed <ebr@mediabrowser3.com>2013-06-18 15:28:18 -0400
committerEric Reed <ebr@mediabrowser3.com>2013-06-18 15:28:18 -0400
commitcc728d87c2507d5392686df6e18c7ad2ba5c45bd (patch)
tree1eee3719a4cd8abd78f422900620e62042161766 /MediaBrowser.Server.Implementations/Persistence
parent90155278f8b4465a4b5eaf140c5e6e4905cc8dcf (diff)
parent5d8ed2c16fdeaec1344964778e98cadfaa9571b4 (diff)
Merge branch 'master' of https://github.com/MediaBrowser/MediaBrowser
Diffstat (limited to 'MediaBrowser.Server.Implementations/Persistence')
-rw-r--r--MediaBrowser.Server.Implementations/Persistence/SqliteChapterRepository.cs326
-rw-r--r--MediaBrowser.Server.Implementations/Persistence/SqliteDisplayPreferencesRepository.cs251
-rw-r--r--MediaBrowser.Server.Implementations/Persistence/SqliteExtensions.cs162
-rw-r--r--MediaBrowser.Server.Implementations/Persistence/SqliteItemRepository.cs405
-rw-r--r--MediaBrowser.Server.Implementations/Persistence/SqliteUserDataRepository.cs327
-rw-r--r--MediaBrowser.Server.Implementations/Persistence/SqliteUserRepository.cs313
6 files changed, 1784 insertions, 0 deletions
diff --git a/MediaBrowser.Server.Implementations/Persistence/SqliteChapterRepository.cs b/MediaBrowser.Server.Implementations/Persistence/SqliteChapterRepository.cs
new file mode 100644
index 000000000..dd6343a67
--- /dev/null
+++ b/MediaBrowser.Server.Implementations/Persistence/SqliteChapterRepository.cs
@@ -0,0 +1,326 @@
+using MediaBrowser.Common.Configuration;
+using MediaBrowser.Model.Entities;
+using MediaBrowser.Model.Logging;
+using MediaBrowser.Model.Serialization;
+using System;
+using System.Collections.Generic;
+using System.Data;
+using System.Data.SQLite;
+using System.IO;
+using System.Threading;
+using System.Threading.Tasks;
+
+namespace MediaBrowser.Server.Implementations.Persistence
+{
+ public class SqliteChapterRepository
+ {
+ private SQLiteConnection _connection;
+
+ private readonly ILogger _logger;
+
+ /// <summary>
+ /// The _app paths
+ /// </summary>
+ private readonly IApplicationPaths _appPaths;
+
+ private SQLiteCommand _deleteChaptersCommand;
+ private SQLiteCommand _saveChapterCommand;
+
+ /// <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 SqliteChapterRepository(IApplicationPaths appPaths, ILogManager logManager)
+ {
+ if (appPaths == null)
+ {
+ throw new ArgumentNullException("appPaths");
+ }
+
+ _appPaths = appPaths;
+
+ _logger = logManager.GetLogger(GetType().Name);
+ }
+
+ /// <summary>
+ /// Opens the connection to the database
+ /// </summary>
+ /// <returns>Task.</returns>
+ public async Task Initialize()
+ {
+ var dbFile = Path.Combine(_appPaths.DataPath, "chapters.db");
+
+ _connection = await SqliteExtensions.ConnectToDb(dbFile).ConfigureAwait(false);
+
+ string[] queries = {
+
+ "create table if not exists chapters (ItemId GUID, ChapterIndex INT, StartPositionTicks BIGINT, Name TEXT, ImagePath TEXT, PRIMARY KEY (ItemId, ChapterIndex))",
+ "create index if not exists idx_chapters on chapters(ItemId, ChapterIndex)",
+
+ //pragmas
+ "pragma temp_store = memory"
+ };
+
+ _connection.RunQueries(queries, _logger);
+
+ PrepareStatements();
+ }
+
+ /// <summary>
+ /// The _write lock
+ /// </summary>
+ private readonly SemaphoreSlim _writeLock = new SemaphoreSlim(1, 1);
+
+ /// <summary>
+ /// Prepares the statements.
+ /// </summary>
+ private void PrepareStatements()
+ {
+ _deleteChaptersCommand = new SQLiteCommand
+ {
+ CommandText = "delete from chapters where ItemId=@ItemId"
+ };
+
+ _deleteChaptersCommand.Parameters.Add(new SQLiteParameter("@ItemId"));
+
+ _saveChapterCommand = new SQLiteCommand
+ {
+ CommandText = "replace into chapters (ItemId, ChapterIndex, StartPositionTicks, Name, ImagePath) values (@ItemId, @ChapterIndex, @StartPositionTicks, @Name, @ImagePath)"
+ };
+
+ _saveChapterCommand.Parameters.Add(new SQLiteParameter("@ItemId"));
+ _saveChapterCommand.Parameters.Add(new SQLiteParameter("@ChapterIndex"));
+ _saveChapterCommand.Parameters.Add(new SQLiteParameter("@StartPositionTicks"));
+ _saveChapterCommand.Parameters.Add(new SQLiteParameter("@Name"));
+ _saveChapterCommand.Parameters.Add(new SQLiteParameter("@ImagePath"));
+ }
+
+ /// <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)
+ {
+ if (id == Guid.Empty)
+ {
+ throw new ArgumentNullException("id");
+ }
+
+ using (var cmd = _connection.CreateCommand())
+ {
+ cmd.CommandText = "select StartPositionTicks,Name,ImagePath from Chapters where ItemId = @ItemId order by ChapterIndex asc";
+
+ cmd.Parameters.Add("@ItemId", DbType.Guid).Value = id;
+
+ using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult))
+ {
+ while (reader.Read())
+ {
+ var chapter = new ChapterInfo
+ {
+ StartPositionTicks = reader.GetInt64(0)
+ };
+
+ if (!reader.IsDBNull(1))
+ {
+ chapter.Name = reader.GetString(1);
+ }
+
+ if (!reader.IsDBNull(2))
+ {
+ chapter.ImagePath = reader.GetString(2);
+ }
+
+ yield return chapter;
+ }
+ }
+ }
+ }
+
+ /// <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)
+ {
+ if (id == Guid.Empty)
+ {
+ throw new ArgumentNullException("id");
+ }
+
+ using (var cmd = _connection.CreateCommand())
+ {
+ cmd.CommandText = "select StartPositionTicks,Name,ImagePath from Chapters where ItemId = @ItemId and ChapterIndex=@ChapterIndex";
+
+ cmd.Parameters.Add("@ItemId", DbType.Guid).Value = id;
+ cmd.Parameters.Add("@ChapterIndex", DbType.Int32).Value = index;
+
+ using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult | CommandBehavior.SingleRow))
+ {
+ if (reader.Read())
+ {
+ return new ChapterInfo
+ {
+ StartPositionTicks = reader.GetInt64(0),
+ Name = reader.GetString(1),
+ ImagePath = reader.GetString(2)
+ };
+ }
+ }
+ return null;
+ }
+ }
+
+ /// <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 async Task SaveChapters(Guid id, IEnumerable<ChapterInfo> chapters, CancellationToken cancellationToken)
+ {
+ if (id == Guid.Empty)
+ {
+ throw new ArgumentNullException("id");
+ }
+
+ if (chapters == null)
+ {
+ throw new ArgumentNullException("chapters");
+ }
+
+ if (cancellationToken == null)
+ {
+ throw new ArgumentNullException("cancellationToken");
+ }
+
+ cancellationToken.ThrowIfCancellationRequested();
+
+ await _writeLock.WaitAsync(cancellationToken).ConfigureAwait(false);
+
+ SQLiteTransaction transaction = null;
+
+ try
+ {
+ transaction = _connection.BeginTransaction();
+
+ // First delete chapters
+ _deleteChaptersCommand.Parameters[0].Value = id;
+ _deleteChaptersCommand.Transaction = transaction;
+ await _deleteChaptersCommand.ExecuteNonQueryAsync(cancellationToken);
+
+ var index = 0;
+
+ foreach (var chapter in chapters)
+ {
+ cancellationToken.ThrowIfCancellationRequested();
+
+ _saveChapterCommand.Parameters[0].Value = id;
+ _saveChapterCommand.Parameters[1].Value = index;
+ _saveChapterCommand.Parameters[2].Value = chapter.StartPositionTicks;
+ _saveChapterCommand.Parameters[3].Value = chapter.Name;
+ _saveChapterCommand.Parameters[4].Value = chapter.ImagePath;
+
+ _saveChapterCommand.Transaction = transaction;
+
+ await _saveChapterCommand.ExecuteNonQueryAsync(cancellationToken);
+
+ index++;
+ }
+
+ transaction.Commit();
+ }
+ catch (OperationCanceledException)
+ {
+ if (transaction != null)
+ {
+ transaction.Rollback();
+ }
+
+ throw;
+ }
+ catch (Exception e)
+ {
+ _logger.ErrorException("Failed to save chapters:", e);
+
+ if (transaction != null)
+ {
+ transaction.Rollback();
+ }
+
+ throw;
+ }
+ finally
+ {
+ if (transaction != null)
+ {
+ transaction.Dispose();
+ }
+
+ _writeLock.Release();
+ }
+ }
+
+ /// <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();
+
+ /// <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)
+ {
+ try
+ {
+ lock (_disposeLock)
+ {
+ if (_connection != null)
+ {
+ if (_connection.IsOpen())
+ {
+ _connection.Close();
+ }
+
+ _connection.Dispose();
+ _connection = null;
+ }
+ }
+ }
+ catch (Exception ex)
+ {
+ _logger.ErrorException("Error disposing database", ex);
+ }
+ }
+ }
+ }
+}
diff --git a/MediaBrowser.Server.Implementations/Persistence/SqliteDisplayPreferencesRepository.cs b/MediaBrowser.Server.Implementations/Persistence/SqliteDisplayPreferencesRepository.cs
new file mode 100644
index 000000000..cb965c3f9
--- /dev/null
+++ b/MediaBrowser.Server.Implementations/Persistence/SqliteDisplayPreferencesRepository.cs
@@ -0,0 +1,251 @@
+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.Data.SQLite;
+using System.IO;
+using System.Threading;
+using System.Threading.Tasks;
+
+namespace MediaBrowser.Server.Implementations.Persistence
+{
+ /// <summary>
+ /// Class SQLiteDisplayPreferencesRepository
+ /// </summary>
+ public class SqliteDisplayPreferencesRepository : IDisplayPreferencesRepository
+ {
+ private SQLiteConnection _connection;
+
+ private readonly ILogger _logger;
+
+ /// <summary>
+ /// Gets the name of the repository
+ /// </summary>
+ /// <value>The name.</value>
+ public string Name
+ {
+ get
+ {
+ return "SQLite";
+ }
+ }
+
+ /// <summary>
+ /// The _json serializer
+ /// </summary>
+ private readonly IJsonSerializer _jsonSerializer;
+
+ /// <summary>
+ /// The _app paths
+ /// </summary>
+ private readonly IApplicationPaths _appPaths;
+
+ private readonly SemaphoreSlim _writeLock = new SemaphoreSlim(1, 1);
+
+ /// <summary>
+ /// Initializes a new instance of the <see cref="SqliteDisplayPreferencesRepository" /> 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">
+ /// jsonSerializer
+ /// or
+ /// appPaths
+ /// </exception>
+ public SqliteDisplayPreferencesRepository(IApplicationPaths appPaths, IJsonSerializer jsonSerializer, ILogManager logManager)
+ {
+ if (jsonSerializer == null)
+ {
+ throw new ArgumentNullException("jsonSerializer");
+ }
+ if (appPaths == null)
+ {
+ throw new ArgumentNullException("appPaths");
+ }
+
+ _jsonSerializer = jsonSerializer;
+ _appPaths = appPaths;
+
+ _logger = logManager.GetLogger(GetType().Name);
+ }
+
+ /// <summary>
+ /// Opens the connection to the database
+ /// </summary>
+ /// <returns>Task.</returns>
+ public async Task Initialize()
+ {
+ var dbFile = Path.Combine(_appPaths.DataPath, "displaypreferences.db");
+
+ _connection = await SqliteExtensions.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"
+ };
+
+ _connection.RunQueries(queries, _logger);
+ }
+
+ /// <summary>
+ /// Save the display preferences associated with an item in the repo
+ /// </summary>
+ /// <param name="displayPreferences">The display preferences.</param>
+ /// <param name="cancellationToken">The cancellation token.</param>
+ /// <returns>Task.</returns>
+ /// <exception cref="System.ArgumentNullException">item</exception>
+ public async 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();
+
+ var serialized = _jsonSerializer.SerializeToBytes(displayPreferences);
+
+ await _writeLock.WaitAsync(cancellationToken).ConfigureAwait(false);
+
+ SQLiteTransaction transaction = null;
+
+ try
+ {
+ transaction = _connection.BeginTransaction();
+
+ using (var cmd = _connection.CreateCommand())
+ {
+ cmd.CommandText = "replace into displaypreferences (id, data) values (@1, @2)";
+ cmd.AddParam("@1", displayPreferences.Id);
+ cmd.AddParam("@2", serialized);
+
+ cmd.Transaction = transaction;
+
+ await cmd.ExecuteNonQueryAsync(cancellationToken);
+ }
+
+ transaction.Commit();
+ }
+ catch (OperationCanceledException)
+ {
+ if (transaction != null)
+ {
+ transaction.Rollback();
+ }
+
+ throw;
+ }
+ catch (Exception e)
+ {
+ _logger.ErrorException("Failed to save display preferences:", e);
+
+ if (transaction != null)
+ {
+ transaction.Rollback();
+ }
+
+ throw;
+ }
+ finally
+ {
+ if (transaction != null)
+ {
+ transaction.Dispose();
+ }
+
+ _writeLock.Release();
+ }
+ }
+
+ /// <summary>
+ /// Gets the display preferences.
+ /// </summary>
+ /// <param name="displayPreferencesId">The display preferences id.</param>
+ /// <returns>Task{DisplayPreferences}.</returns>
+ /// <exception cref="System.ArgumentNullException">item</exception>
+ public DisplayPreferences 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 = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult | CommandBehavior.SingleRow))
+ {
+ if (reader.Read())
+ {
+ using (var stream = reader.GetMemoryStream(0))
+ {
+ return _jsonSerializer.DeserializeFromStream<DisplayPreferences>(stream);
+ }
+ }
+ }
+
+ return null;
+ }
+
+ /// <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();
+
+ /// <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)
+ {
+ try
+ {
+ lock (_disposeLock)
+ {
+ if (_connection != null)
+ {
+ if (_connection.IsOpen())
+ {
+ _connection.Close();
+ }
+
+ _connection.Dispose();
+ _connection = null;
+ }
+ }
+ }
+ catch (Exception ex)
+ {
+ _logger.ErrorException("Error disposing database", ex);
+ }
+ }
+ }
+ }
+} \ No newline at end of file
diff --git a/MediaBrowser.Server.Implementations/Persistence/SqliteExtensions.cs b/MediaBrowser.Server.Implementations/Persistence/SqliteExtensions.cs
new file mode 100644
index 000000000..2b14e9b24
--- /dev/null
+++ b/MediaBrowser.Server.Implementations/Persistence/SqliteExtensions.cs
@@ -0,0 +1,162 @@
+using System;
+using System.Data;
+using System.Data.SQLite;
+using System.IO;
+using System.Threading.Tasks;
+using MediaBrowser.Model.Logging;
+
+namespace MediaBrowser.Server.Implementations.Persistence
+{
+ /// <summary>
+ /// Class SQLiteExtensions
+ /// </summary>
+ static class SqliteExtensions
+ {
+ /// <summary>
+ /// Adds the param.
+ /// </summary>
+ /// <param name="cmd">The CMD.</param>
+ /// <param name="param">The param.</param>
+ /// <returns>SQLiteParameter.</returns>
+ /// <exception cref="System.ArgumentNullException"></exception>
+ public static SQLiteParameter AddParam(this SQLiteCommand cmd, string param)
+ {
+ if (string.IsNullOrEmpty(param))
+ {
+ throw new ArgumentNullException();
+ }
+
+ var sqliteParam = new SQLiteParameter(param);
+ cmd.Parameters.Add(sqliteParam);
+ return sqliteParam;
+ }
+
+ /// <summary>
+ /// Adds the param.
+ /// </summary>
+ /// <param name="cmd">The CMD.</param>
+ /// <param name="param">The param.</param>
+ /// <param name="data">The data.</param>
+ /// <returns>SQLiteParameter.</returns>
+ /// <exception cref="System.ArgumentNullException"></exception>
+ public static SQLiteParameter AddParam(this SQLiteCommand cmd, string param, object data)
+ {
+ if (string.IsNullOrEmpty(param))
+ {
+ throw new ArgumentNullException();
+ }
+
+ var sqliteParam = AddParam(cmd, param);
+ sqliteParam.Value = data;
+ return sqliteParam;
+ }
+
+ /// <summary>
+ /// Determines whether the specified conn is open.
+ /// </summary>
+ /// <param name="conn">The conn.</param>
+ /// <returns><c>true</c> if the specified conn is open; otherwise, <c>false</c>.</returns>
+ public static bool IsOpen(this SQLiteConnection conn)
+ {
+ return conn.State == ConnectionState.Open;
+ }
+
+ /// <summary>
+ /// Gets a stream from a DataReader at a given ordinal
+ /// </summary>
+ /// <param name="reader">The reader.</param>
+ /// <param name="ordinal">The ordinal.</param>
+ /// <returns>Stream.</returns>
+ /// <exception cref="System.ArgumentNullException">reader</exception>
+ public static Stream GetMemoryStream(this IDataReader reader, int ordinal)
+ {
+ if (reader == null)
+ {
+ throw new ArgumentNullException("reader");
+ }
+
+ var memoryStream = new MemoryStream();
+ var num = 0L;
+ var array = new byte[4096];
+ long bytes;
+ do
+ {
+ bytes = reader.GetBytes(ordinal, num, array, 0, array.Length);
+ memoryStream.Write(array, 0, (int)bytes);
+ num += bytes;
+ }
+ while (bytes > 0L);
+ memoryStream.Position = 0;
+ return memoryStream;
+ }
+
+ /// <summary>
+ /// Runs the queries.
+ /// </summary>
+ /// <param name="connection">The connection.</param>
+ /// <param name="queries">The queries.</param>
+ /// <param name="logger">The logger.</param>
+ /// <returns><c>true</c> if XXXX, <c>false</c> otherwise</returns>
+ /// <exception cref="System.ArgumentNullException">queries</exception>
+ public static void RunQueries(this IDbConnection connection, string[] queries, ILogger logger)
+ {
+ if (queries == null)
+ {
+ throw new ArgumentNullException("queries");
+ }
+
+ using (var tran = connection.BeginTransaction())
+ {
+ try
+ {
+ using (var cmd = connection.CreateCommand())
+ {
+ foreach (var query in queries)
+ {
+ cmd.Transaction = tran;
+ cmd.CommandText = query;
+ cmd.ExecuteNonQuery();
+ }
+ }
+
+ tran.Commit();
+ }
+ catch (Exception e)
+ {
+ logger.ErrorException("Error running queries", e);
+ tran.Rollback();
+ throw;
+ }
+ }
+ }
+
+ /// <summary>
+ /// Connects to db.
+ /// </summary>
+ /// <param name="dbPath">The db path.</param>
+ /// <returns>Task{IDbConnection}.</returns>
+ /// <exception cref="System.ArgumentNullException">dbPath</exception>
+ public static async Task<SQLiteConnection> ConnectToDb(string dbPath)
+ {
+ if (string.IsNullOrEmpty(dbPath))
+ {
+ throw new ArgumentNullException("dbPath");
+ }
+
+ var connectionstr = new SQLiteConnectionStringBuilder
+ {
+ PageSize = 4096,
+ CacheSize = 4096,
+ SyncMode = SynchronizationModes.Off,
+ DataSource = dbPath,
+ JournalMode = SQLiteJournalModeEnum.Wal
+ };
+
+ var connection = new SQLiteConnection(connectionstr.ConnectionString);
+
+ await connection.OpenAsync().ConfigureAwait(false);
+
+ return connection;
+ }
+ }
+} \ No newline at end of file
diff --git a/MediaBrowser.Server.Implementations/Persistence/SqliteItemRepository.cs b/MediaBrowser.Server.Implementations/Persistence/SqliteItemRepository.cs
new file mode 100644
index 000000000..b3251ddb9
--- /dev/null
+++ b/MediaBrowser.Server.Implementations/Persistence/SqliteItemRepository.cs
@@ -0,0 +1,405 @@
+using MediaBrowser.Common.Configuration;
+using MediaBrowser.Controller.Entities;
+using MediaBrowser.Controller.Persistence;
+using MediaBrowser.Model.Entities;
+using MediaBrowser.Model.Logging;
+using MediaBrowser.Model.Serialization;
+using System;
+using System.Collections.Generic;
+using System.Data;
+using System.Data.SQLite;
+using System.IO;
+using System.Linq;
+using System.Threading;
+using System.Threading.Tasks;
+
+namespace MediaBrowser.Server.Implementations.Persistence
+{
+ /// <summary>
+ /// Class SQLiteItemRepository
+ /// </summary>
+ public class SqliteItemRepository : IItemRepository
+ {
+ private SQLiteConnection _connection;
+
+ private readonly ILogger _logger;
+
+ /// <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 SQLiteCommand _saveItemCommand;
+
+ private readonly string _criticReviewsPath;
+
+ private SqliteChapterRepository _chapterRepository;
+
+ /// <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);
+
+ _chapterRepository = new SqliteChapterRepository(appPaths, 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).ConfigureAwait(false);
+
+ string[] queries = {
+
+ "create table if not exists baseitems (guid GUID primary key, data BLOB)",
+ "create index if not exists idx_baseitems on baseitems(guid)",
+
+ //pragmas
+ "pragma temp_store = memory"
+ };
+
+ _connection.RunQueries(queries, _logger);
+
+ PrepareStatements();
+
+ await _chapterRepository.Initialize().ConfigureAwait(false);
+ }
+
+ /// <summary>
+ /// The _write lock
+ /// </summary>
+ private readonly SemaphoreSlim _writeLock = new SemaphoreSlim(1, 1);
+
+ /// <summary>
+ /// Prepares the statements.
+ /// </summary>
+ private void PrepareStatements()
+ {
+ _saveItemCommand = new SQLiteCommand
+ {
+ CommandText = "replace into baseitems (guid, data) values (@1, @2)"
+ };
+
+ _saveItemCommand.Parameters.Add(new SQLiteParameter("@1"));
+ _saveItemCommand.Parameters.Add(new SQLiteParameter("@2"));
+ }
+
+ /// <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");
+ }
+
+ if (cancellationToken == null)
+ {
+ throw new ArgumentNullException("cancellationToken");
+ }
+
+ cancellationToken.ThrowIfCancellationRequested();
+
+ await _writeLock.WaitAsync(cancellationToken).ConfigureAwait(false);
+
+ SQLiteTransaction transaction = null;
+
+ try
+ {
+ transaction = _connection.BeginTransaction();
+
+ foreach (var item in items)
+ {
+ cancellationToken.ThrowIfCancellationRequested();
+
+ _saveItemCommand.Parameters[0].Value = item.Id;
+ _saveItemCommand.Parameters[1].Value = _jsonSerializer.SerializeToBytes(item);
+
+ _saveItemCommand.Transaction = transaction;
+
+ await _saveItemCommand.ExecuteNonQueryAsync(cancellationToken);
+ }
+
+ 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>
+ /// <param name="type">The type.</param>
+ /// <returns>BaseItem.</returns>
+ /// <exception cref="System.ArgumentNullException">id</exception>
+ /// <exception cref="System.ArgumentException"></exception>
+ public BaseItem RetrieveItem(Guid id, Type type)
+ {
+ if (id == Guid.Empty)
+ {
+ throw new ArgumentNullException("id");
+ }
+
+ using (var cmd = _connection.CreateCommand())
+ {
+ cmd.CommandText = "select data from baseitems where guid = @guid";
+ var guidParam = cmd.Parameters.Add("@guid", DbType.Guid);
+ guidParam.Value = id;
+
+ using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult | CommandBehavior.SingleRow))
+ {
+ if (reader.Read())
+ {
+ using (var stream = reader.GetMemoryStream(0))
+ {
+ return _jsonSerializer.DeserializeFromStream(stream, type) as BaseItem;
+ }
+ }
+ }
+ return null;
+ }
+ }
+
+ /// <summary>
+ /// Gets the critic reviews.
+ /// </summary>
+ /// <param name="itemId">The item id.</param>
+ /// <returns>Task{IEnumerable{ItemReview}}.</returns>
+ public Task<IEnumerable<ItemReview>> GetCriticReviews(Guid itemId)
+ {
+ return Task.Run<IEnumerable<ItemReview>>(() =>
+ {
+
+ 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>();
+ }
+
+ });
+ }
+
+ /// <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)
+ {
+ return Task.Run(() =>
+ {
+ if (!Directory.Exists(_criticReviewsPath))
+ {
+ Directory.CreateDirectory(_criticReviewsPath);
+ }
+
+ var path = Path.Combine(_criticReviewsPath, itemId + ".json");
+
+ _jsonSerializer.SerializeToFile(criticReviews.ToList(), path);
+ });
+ }
+
+ /// <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)
+ {
+ 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)
+ {
+ 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)
+ {
+ 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();
+
+ /// <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)
+ {
+ try
+ {
+ lock (_disposeLock)
+ {
+ if (_connection != null)
+ {
+ if (_connection.IsOpen())
+ {
+ _connection.Close();
+ }
+
+ _connection.Dispose();
+ _connection = null;
+ }
+ }
+ }
+ catch (Exception ex)
+ {
+ _logger.ErrorException("Error disposing database", ex);
+ }
+
+ if (_chapterRepository != null)
+ {
+ _chapterRepository.Dispose();
+ _chapterRepository = null;
+ }
+ }
+ }
+ }
+} \ No newline at end of file
diff --git a/MediaBrowser.Server.Implementations/Persistence/SqliteUserDataRepository.cs b/MediaBrowser.Server.Implementations/Persistence/SqliteUserDataRepository.cs
new file mode 100644
index 000000000..1d127ae96
--- /dev/null
+++ b/MediaBrowser.Server.Implementations/Persistence/SqliteUserDataRepository.cs
@@ -0,0 +1,327 @@
+using MediaBrowser.Common.Configuration;
+using MediaBrowser.Controller.Entities;
+using MediaBrowser.Controller.Persistence;
+using MediaBrowser.Model.Logging;
+using MediaBrowser.Model.Serialization;
+using System;
+using System.Collections.Concurrent;
+using System.Data;
+using System.Data.SQLite;
+using System.IO;
+using System.Threading;
+using System.Threading.Tasks;
+
+namespace MediaBrowser.Server.Implementations.Persistence
+{
+ public class SqliteUserDataRepository : IUserDataRepository
+ {
+ private readonly ILogger _logger;
+
+ private readonly ConcurrentDictionary<string, UserItemData> _userData = new ConcurrentDictionary<string, UserItemData>();
+
+ private readonly SemaphoreSlim _writeLock = new SemaphoreSlim(1, 1);
+
+ private SQLiteConnection _connection;
+
+ /// <summary>
+ /// Gets the name of the repository
+ /// </summary>
+ /// <value>The name.</value>
+ public string Name
+ {
+ get
+ {
+ return "SQLite";
+ }
+ }
+
+ private readonly IJsonSerializer _jsonSerializer;
+
+ /// <summary>
+ /// The _app paths
+ /// </summary>
+ private readonly IApplicationPaths _appPaths;
+
+ /// <summary>
+ /// Initializes a new instance of the <see cref="SqliteUserDataRepository"/> 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">
+ /// jsonSerializer
+ /// or
+ /// appPaths
+ /// </exception>
+ public SqliteUserDataRepository(IApplicationPaths appPaths, IJsonSerializer jsonSerializer, ILogManager logManager)
+ {
+ if (jsonSerializer == null)
+ {
+ throw new ArgumentNullException("jsonSerializer");
+ }
+ if (appPaths == null)
+ {
+ throw new ArgumentNullException("appPaths");
+ }
+
+ _jsonSerializer = jsonSerializer;
+ _appPaths = appPaths;
+ _logger = logManager.GetLogger(GetType().Name);
+ }
+
+ /// <summary>
+ /// Opens the connection to the database
+ /// </summary>
+ /// <returns>Task.</returns>
+ public async Task Initialize()
+ {
+ var dbFile = Path.Combine(_appPaths.DataPath, "userdata.db");
+
+ _connection = await SqliteExtensions.ConnectToDb(dbFile).ConfigureAwait(false);
+
+ string[] queries = {
+
+ "create table if not exists userdata (key nvarchar, userId GUID, data BLOB)",
+ "create unique index if not exists userdataindex on userdata (key, userId)",
+ "create table if not exists schema_version (table_name primary key, version)",
+ //pragmas
+ "pragma temp_store = memory"
+ };
+
+ _connection.RunQueries(queries, _logger);
+ }
+
+ /// <summary>
+ /// Saves the user data.
+ /// </summary>
+ /// <param name="userId">The user id.</param>
+ /// <param name="key">The key.</param>
+ /// <param name="userData">The user data.</param>
+ /// <param name="cancellationToken">The cancellation token.</param>
+ /// <returns>Task.</returns>
+ /// <exception cref="System.ArgumentNullException">userData
+ /// or
+ /// cancellationToken
+ /// or
+ /// userId
+ /// or
+ /// userDataId</exception>
+ public async Task SaveUserData(Guid userId, string key, UserItemData userData, CancellationToken cancellationToken)
+ {
+ if (userData == null)
+ {
+ throw new ArgumentNullException("userData");
+ }
+ if (cancellationToken == null)
+ {
+ throw new ArgumentNullException("cancellationToken");
+ }
+ if (userId == Guid.Empty)
+ {
+ throw new ArgumentNullException("userId");
+ }
+ if (string.IsNullOrEmpty(key))
+ {
+ throw new ArgumentNullException("key");
+ }
+
+ cancellationToken.ThrowIfCancellationRequested();
+
+ try
+ {
+ await PersistUserData(userId, key, userData, cancellationToken).ConfigureAwait(false);
+
+ var newValue = userData;
+
+ // Once it succeeds, put it into the dictionary to make it available to everyone else
+ _userData.AddOrUpdate(GetInternalKey(userId, key), newValue, delegate { return newValue; });
+ }
+ catch (Exception ex)
+ {
+ _logger.ErrorException("Error saving user data", ex);
+
+ throw;
+ }
+ }
+
+ /// <summary>
+ /// Gets the internal key.
+ /// </summary>
+ /// <param name="userId">The user id.</param>
+ /// <param name="key">The key.</param>
+ /// <returns>System.String.</returns>
+ private string GetInternalKey(Guid userId, string key)
+ {
+ return userId + key;
+ }
+
+ /// <summary>
+ /// Persists the user data.
+ /// </summary>
+ /// <param name="userId">The user id.</param>
+ /// <param name="key">The key.</param>
+ /// <param name="userData">The user data.</param>
+ /// <param name="cancellationToken">The cancellation token.</param>
+ /// <returns>Task.</returns>
+ public async Task PersistUserData(Guid userId, string key, UserItemData userData, CancellationToken cancellationToken)
+ {
+ cancellationToken.ThrowIfCancellationRequested();
+
+ var serialized = _jsonSerializer.SerializeToBytes(userData);
+
+ cancellationToken.ThrowIfCancellationRequested();
+
+ await _writeLock.WaitAsync(cancellationToken).ConfigureAwait(false);
+
+ SQLiteTransaction transaction = null;
+
+ try
+ {
+ transaction = _connection.BeginTransaction();
+
+ using (var cmd = _connection.CreateCommand())
+ {
+ cmd.CommandText = "replace into userdata (key, userId, data) values (@1, @2, @3)";
+ cmd.AddParam("@1", key);
+ cmd.AddParam("@2", userId);
+ cmd.AddParam("@3", serialized);
+
+ cmd.Transaction = transaction;
+
+ await cmd.ExecuteNonQueryAsync(cancellationToken);
+ }
+
+ transaction.Commit();
+ }
+ catch (OperationCanceledException)
+ {
+ if (transaction != null)
+ {
+ transaction.Rollback();
+ }
+
+ throw;
+ }
+ catch (Exception e)
+ {
+ _logger.ErrorException("Failed to save user data:", e);
+
+ if (transaction != null)
+ {
+ transaction.Rollback();
+ }
+
+ throw;
+ }
+ finally
+ {
+ if (transaction != null)
+ {
+ transaction.Dispose();
+ }
+
+ _writeLock.Release();
+ }
+ }
+
+ /// <summary>
+ /// Gets the user data.
+ /// </summary>
+ /// <param name="userId">The user id.</param>
+ /// <param name="key">The key.</param>
+ /// <returns>Task{UserItemData}.</returns>
+ /// <exception cref="System.ArgumentNullException">
+ /// userId
+ /// or
+ /// key
+ /// </exception>
+ public UserItemData GetUserData(Guid userId, string key)
+ {
+ if (userId == Guid.Empty)
+ {
+ throw new ArgumentNullException("userId");
+ }
+ if (string.IsNullOrEmpty(key))
+ {
+ throw new ArgumentNullException("key");
+ }
+
+ return _userData.GetOrAdd(GetInternalKey(userId, key), keyName => RetrieveUserData(userId, key));
+ }
+
+ /// <summary>
+ /// Retrieves the user data.
+ /// </summary>
+ /// <param name="userId">The user id.</param>
+ /// <param name="key">The key.</param>
+ /// <returns>Task{UserItemData}.</returns>
+ private UserItemData RetrieveUserData(Guid userId, string key)
+ {
+ using (var cmd = _connection.CreateCommand())
+ {
+ cmd.CommandText = "select data from userdata where key = @key and userId=@userId";
+
+ var idParam = cmd.Parameters.Add("@key", DbType.String);
+ idParam.Value = key;
+
+ var userIdParam = cmd.Parameters.Add("@userId", DbType.Guid);
+ userIdParam.Value = userId;
+
+ using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult | CommandBehavior.SingleRow))
+ {
+ if (reader.Read())
+ {
+ using (var stream = reader.GetMemoryStream(0))
+ {
+ return _jsonSerializer.DeserializeFromStream<UserItemData>(stream);
+ }
+ }
+ }
+
+ return new UserItemData();
+ }
+ }
+
+ /// <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();
+
+ /// <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)
+ {
+ try
+ {
+ lock (_disposeLock)
+ {
+ if (_connection != null)
+ {
+ if (_connection.IsOpen())
+ {
+ _connection.Close();
+ }
+
+ _connection.Dispose();
+ _connection = null;
+ }
+ }
+ }
+ catch (Exception ex)
+ {
+ _logger.ErrorException("Error disposing database", ex);
+ }
+ }
+ }
+ }
+} \ No newline at end of file
diff --git a/MediaBrowser.Server.Implementations/Persistence/SqliteUserRepository.cs b/MediaBrowser.Server.Implementations/Persistence/SqliteUserRepository.cs
new file mode 100644
index 000000000..09e34cf08
--- /dev/null
+++ b/MediaBrowser.Server.Implementations/Persistence/SqliteUserRepository.cs
@@ -0,0 +1,313 @@
+using MediaBrowser.Common.Configuration;
+using MediaBrowser.Controller.Entities;
+using MediaBrowser.Controller.Persistence;
+using MediaBrowser.Model.Logging;
+using MediaBrowser.Model.Serialization;
+using System;
+using System.Collections.Generic;
+using System.Data;
+using System.Data.SQLite;
+using System.IO;
+using System.Threading;
+using System.Threading.Tasks;
+
+namespace MediaBrowser.Server.Implementations.Persistence
+{
+ /// <summary>
+ /// Class SQLiteUserRepository
+ /// </summary>
+ public class SqliteUserRepository : IUserRepository
+ {
+ private readonly ILogger _logger;
+
+ private readonly SemaphoreSlim _writeLock = new SemaphoreSlim(1, 1);
+
+ private SQLiteConnection _connection;
+
+ /// <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>
+ /// Initializes a new instance of the <see cref="SqliteUserRepository" /> 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</exception>
+ public SqliteUserRepository(IApplicationPaths appPaths, IJsonSerializer jsonSerializer, ILogManager logManager)
+ {
+ if (appPaths == null)
+ {
+ throw new ArgumentNullException("appPaths");
+ }
+ if (jsonSerializer == null)
+ {
+ throw new ArgumentNullException("jsonSerializer");
+ }
+
+ _appPaths = appPaths;
+ _jsonSerializer = jsonSerializer;
+
+ _logger = logManager.GetLogger(GetType().Name);
+ }
+
+ /// <summary>
+ /// Opens the connection to the database
+ /// </summary>
+ /// <returns>Task.</returns>
+ public async Task Initialize()
+ {
+ var dbFile = Path.Combine(_appPaths.DataPath, "users.db");
+
+ _connection = await SqliteExtensions.ConnectToDb(dbFile).ConfigureAwait(false);
+
+ string[] queries = {
+
+ "create table if not exists users (guid GUID primary key, data BLOB)",
+ "create index if not exists idx_users on users(guid)",
+ "create table if not exists schema_version (table_name primary key, version)",
+ //pragmas
+ "pragma temp_store = memory"
+ };
+
+ _connection.RunQueries(queries, _logger);
+ }
+
+ /// <summary>
+ /// Save a user in the repo
+ /// </summary>
+ /// <param name="user">The user.</param>
+ /// <param name="cancellationToken">The cancellation token.</param>
+ /// <returns>Task.</returns>
+ /// <exception cref="System.ArgumentNullException">user</exception>
+ public async Task SaveUser(User user, CancellationToken cancellationToken)
+ {
+ if (user == null)
+ {
+ throw new ArgumentNullException("user");
+ }
+
+ if (cancellationToken == null)
+ {
+ throw new ArgumentNullException("cancellationToken");
+ }
+
+ cancellationToken.ThrowIfCancellationRequested();
+
+ var serialized = _jsonSerializer.SerializeToBytes(user);
+
+ cancellationToken.ThrowIfCancellationRequested();
+
+ await _writeLock.WaitAsync(cancellationToken).ConfigureAwait(false);
+
+ SQLiteTransaction transaction = null;
+
+ try
+ {
+ transaction = _connection.BeginTransaction();
+
+ using (var cmd = _connection.CreateCommand())
+ {
+ cmd.CommandText = "replace into users (guid, data) values (@1, @2)";
+ cmd.AddParam("@1", user.Id);
+ cmd.AddParam("@2", serialized);
+
+ cmd.Transaction = transaction;
+
+ await cmd.ExecuteNonQueryAsync(cancellationToken);
+ }
+
+ transaction.Commit();
+ }
+ catch (OperationCanceledException)
+ {
+ if (transaction != null)
+ {
+ transaction.Rollback();
+ }
+
+ throw;
+ }
+ catch (Exception e)
+ {
+ _logger.ErrorException("Failed to save user:", e);
+
+ if (transaction != null)
+ {
+ transaction.Rollback();
+ }
+
+ throw;
+ }
+ finally
+ {
+ if (transaction != null)
+ {
+ transaction.Dispose();
+ }
+
+ _writeLock.Release();
+ }
+ }
+
+ /// <summary>
+ /// Retrieve all users from the database
+ /// </summary>
+ /// <returns>IEnumerable{User}.</returns>
+ public IEnumerable<User> RetrieveAllUsers()
+ {
+ using (var cmd = _connection.CreateCommand())
+ {
+ cmd.CommandText = "select data from users";
+
+ using (var reader = cmd.ExecuteReader(CommandBehavior.SequentialAccess | CommandBehavior.SingleResult))
+ {
+ while (reader.Read())
+ {
+ using (var stream = reader.GetMemoryStream(0))
+ {
+ var user = _jsonSerializer.DeserializeFromStream<User>(stream);
+ yield return user;
+ }
+ }
+ }
+ }
+ }
+
+ /// <summary>
+ /// Deletes the user.
+ /// </summary>
+ /// <param name="user">The user.</param>
+ /// <param name="cancellationToken">The cancellation token.</param>
+ /// <returns>Task.</returns>
+ /// <exception cref="System.ArgumentNullException">user</exception>
+ public async Task DeleteUser(User user, CancellationToken cancellationToken)
+ {
+ if (user == null)
+ {
+ throw new ArgumentNullException("user");
+ }
+
+ if (cancellationToken == null)
+ {
+ throw new ArgumentNullException("cancellationToken");
+ }
+
+ cancellationToken.ThrowIfCancellationRequested();
+
+ await _writeLock.WaitAsync(cancellationToken).ConfigureAwait(false);
+
+ SQLiteTransaction transaction = null;
+
+ try
+ {
+ transaction = _connection.BeginTransaction();
+
+ using (var cmd = _connection.CreateCommand())
+ {
+ cmd.CommandText = "delete from users where guid=@guid";
+
+ var guidParam = cmd.Parameters.Add("@guid", DbType.Guid);
+ guidParam.Value = user.Id;
+
+ cmd.Transaction = transaction;
+
+ await cmd.ExecuteNonQueryAsync(cancellationToken).ConfigureAwait(false);
+ }
+
+ transaction.Commit();
+ }
+ catch (OperationCanceledException)
+ {
+ if (transaction != null)
+ {
+ transaction.Rollback();
+ }
+
+ throw;
+ }
+ catch (Exception e)
+ {
+ _logger.ErrorException("Failed to delete user:", e);
+
+ if (transaction != null)
+ {
+ transaction.Rollback();
+ }
+
+ throw;
+ }
+ finally
+ {
+ if (transaction != null)
+ {
+ transaction.Dispose();
+ }
+
+ _writeLock.Release();
+ }
+ }
+
+ /// <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();
+
+ /// <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)
+ {
+ try
+ {
+ lock (_disposeLock)
+ {
+ if (_connection != null)
+ {
+ if (_connection.IsOpen())
+ {
+ _connection.Close();
+ }
+
+ _connection.Dispose();
+ _connection = null;
+ }
+ }
+ }
+ catch (Exception ex)
+ {
+ _logger.ErrorException("Error disposing database", ex);
+ }
+ }
+ }
+ }
+} \ No newline at end of file