2014-09-30 06:47:30 +02:00
using MediaBrowser.Common.Configuration ;
2013-02-21 05:37:50 +01:00
using MediaBrowser.Controller.Entities ;
using MediaBrowser.Controller.Persistence ;
2013-02-21 21:26:35 +01:00
using MediaBrowser.Model.Logging ;
2013-02-21 02:33:05 +01:00
using System ;
2014-09-30 06:47:30 +02:00
using System.Collections.Generic ;
2013-06-18 11:43:07 +02:00
using System.Data ;
2013-02-21 02:33:05 +01:00
using System.IO ;
using System.Threading ;
using System.Threading.Tasks ;
2013-06-17 22:35:43 +02:00
namespace MediaBrowser.Server.Implementations.Persistence
2013-02-21 02:33:05 +01:00
{
2015-05-03 00:59:01 +02:00
public class SqliteUserDataRepository : BaseSqliteRepository , IUserDataRepository
2013-02-21 02:33:05 +01:00
{
2013-09-29 17:32:50 +02:00
private IDbConnection _connection ;
2015-05-03 00:59:01 +02:00
private readonly IApplicationPaths _appPaths ;
public SqliteUserDataRepository ( ILogManager logManager , IApplicationPaths appPaths ) : base ( logManager )
{
_appPaths = appPaths ;
}
2013-09-27 17:23:27 +02:00
2013-02-21 02:33:05 +01:00
/// <summary>
/// Gets the name of the repository
/// </summary>
/// <value>The name.</value>
public string Name
{
get
{
2013-06-18 21:16:27 +02:00
return "SQLite" ;
2013-02-21 02:33:05 +01:00
}
}
/// <summary>
/// Opens the connection to the database
/// </summary>
/// <returns>Task.</returns>
2013-06-18 11:43:07 +02:00
public async Task Initialize ( )
2013-02-21 02:33:05 +01:00
{
2013-09-27 17:23:27 +02:00
var dbFile = Path . Combine ( _appPaths . DataPath , "userdata_v2.db" ) ;
2013-06-18 11:43:07 +02:00
2015-05-03 00:59:01 +02:00
_connection = await SqliteExtensions . ConnectToDb ( dbFile , Logger ) . ConfigureAwait ( false ) ;
2013-06-18 11:43:07 +02:00
string [ ] queries = {
2013-09-27 17:23:27 +02:00
"create table if not exists userdata (key nvarchar, userId GUID, rating float null, played bit, playCount int, isFavorite bit, playbackPositionTicks bigint, lastPlayedDate datetime null)" ,
2013-06-18 11:43:07 +02:00
"create unique index if not exists userdataindex on userdata (key, userId)" ,
2013-09-27 17:23:27 +02:00
2013-06-18 11:43:07 +02:00
//pragmas
2013-12-08 02:42:15 +01:00
"pragma temp_store = memory" ,
"pragma shrink_memory"
2013-06-18 11:43:07 +02:00
} ;
2015-05-03 00:59:01 +02:00
_connection . RunQueries ( queries , Logger ) ;
2016-02-21 00:06:57 +01:00
_connection . AddColumn ( Logger , "userdata" , "AudioStreamIndex" , "int" ) ;
_connection . AddColumn ( Logger , "userdata" , "SubtitleStreamIndex" , "int" ) ;
2013-02-21 02:33:05 +01:00
}
/// <summary>
2013-04-02 21:25:16 +02:00
/// Saves the user data.
2013-02-21 02:33:05 +01:00
/// </summary>
2013-04-02 21:25:16 +02:00
/// <param name="userId">The user id.</param>
2013-04-13 20:02:30 +02:00
/// <param name="key">The key.</param>
2013-04-02 21:25:16 +02:00
/// <param name="userData">The user data.</param>
2013-02-21 02:33:05 +01:00
/// <param name="cancellationToken">The cancellation token.</param>
/// <returns>Task.</returns>
2013-04-13 20:02:30 +02:00
/// <exception cref="System.ArgumentNullException">userData
2013-04-02 21:25:16 +02:00
/// or
/// cancellationToken
/// or
/// userId
/// or
2013-04-13 20:02:30 +02:00
/// userDataId</exception>
2013-10-02 18:58:30 +02:00
public Task SaveUserData ( Guid userId , string key , UserItemData userData , CancellationToken cancellationToken )
2013-02-21 02:33:05 +01:00
{
2013-04-02 21:25:16 +02:00
if ( userData = = null )
2013-02-21 02:33:05 +01:00
{
2013-04-02 21:25:16 +02:00
throw new ArgumentNullException ( "userData" ) ;
2013-02-21 02:33:05 +01:00
}
2013-04-02 21:25:16 +02:00
if ( userId = = Guid . Empty )
{
throw new ArgumentNullException ( "userId" ) ;
}
2013-04-13 20:02:30 +02:00
if ( string . IsNullOrEmpty ( key ) )
2013-04-02 21:25:16 +02:00
{
2013-04-13 20:02:30 +02:00
throw new ArgumentNullException ( "key" ) ;
2013-04-02 21:25:16 +02:00
}
2013-10-02 18:58:30 +02:00
return PersistUserData ( userId , key , userData , cancellationToken ) ;
2013-04-20 01:43:00 +02:00
}
2014-09-20 23:57:16 +02:00
public Task SaveAllUserData ( Guid userId , IEnumerable < UserItemData > userData , CancellationToken cancellationToken )
{
if ( userData = = null )
{
throw new ArgumentNullException ( "userData" ) ;
}
if ( userId = = Guid . Empty )
{
throw new ArgumentNullException ( "userId" ) ;
}
return PersistAllUserData ( userId , userData , cancellationToken ) ;
}
2013-04-13 20:02:30 +02:00
/// <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 ( ) ;
2015-05-03 00:59:01 +02:00
await WriteLock . WaitAsync ( cancellationToken ) . ConfigureAwait ( false ) ;
2013-05-26 01:52:41 +02:00
2013-09-29 17:32:50 +02:00
IDbTransaction transaction = null ;
2013-05-28 19:32:50 +02:00
2013-06-17 22:35:43 +02:00
try
{
2013-06-18 21:16:27 +02:00
transaction = _connection . BeginTransaction ( ) ;
2013-06-18 11:43:07 +02:00
2013-06-18 21:16:27 +02:00
using ( var cmd = _connection . CreateCommand ( ) )
2013-06-18 11:43:07 +02:00
{
2016-02-21 00:06:57 +01:00
cmd . CommandText = "replace into userdata (key, userId, rating,played,playCount,isFavorite,playbackPositionTicks,lastPlayedDate,AudioStreamIndex,SubtitleStreamIndex) values (@key, @userId, @rating,@played,@playCount,@isFavorite,@playbackPositionTicks,@lastPlayedDate,@AudioStreamIndex,@SubtitleStreamIndex)" ;
2013-09-27 17:23:27 +02:00
2013-09-29 17:32:50 +02:00
cmd . Parameters . Add ( cmd , "@key" , DbType . String ) . Value = key ;
cmd . Parameters . Add ( cmd , "@userId" , DbType . Guid ) . Value = userId ;
cmd . Parameters . Add ( cmd , "@rating" , DbType . Double ) . Value = userData . Rating ;
cmd . Parameters . Add ( cmd , "@played" , DbType . Boolean ) . Value = userData . Played ;
cmd . Parameters . Add ( cmd , "@playCount" , DbType . Int32 ) . Value = userData . PlayCount ;
cmd . Parameters . Add ( cmd , "@isFavorite" , DbType . Boolean ) . Value = userData . IsFavorite ;
cmd . Parameters . Add ( cmd , "@playbackPositionTicks" , DbType . Int64 ) . Value = userData . PlaybackPositionTicks ;
cmd . Parameters . Add ( cmd , "@lastPlayedDate" , DbType . DateTime ) . Value = userData . LastPlayedDate ;
2016-02-21 00:06:57 +01:00
cmd . Parameters . Add ( cmd , "@AudioStreamIndex" , DbType . Int32 ) . Value = userData . AudioStreamIndex ;
cmd . Parameters . Add ( cmd , "@SubtitleStreamIndex" , DbType . Int32 ) . Value = userData . SubtitleStreamIndex ;
2013-09-27 17:23:27 +02:00
2013-06-18 11:43:07 +02:00
cmd . Transaction = transaction ;
2013-09-29 17:32:50 +02:00
cmd . ExecuteNonQuery ( ) ;
2013-06-18 11:43:07 +02:00
}
transaction . Commit ( ) ;
}
catch ( OperationCanceledException )
{
if ( transaction ! = null )
{
transaction . Rollback ( ) ;
}
throw ;
}
catch ( Exception e )
{
2015-05-03 00:59:01 +02:00
Logger . ErrorException ( "Failed to save user data:" , e ) ;
2013-06-18 11:43:07 +02:00
if ( transaction ! = null )
{
transaction . Rollback ( ) ;
}
throw ;
2013-05-26 01:52:41 +02:00
}
finally
{
2013-06-18 11:43:07 +02:00
if ( transaction ! = null )
{
transaction . Dispose ( ) ;
}
2015-05-03 00:59:01 +02:00
WriteLock . Release ( ) ;
2013-05-26 01:52:41 +02:00
}
2013-02-21 02:33:05 +01:00
}
2014-09-20 23:57:16 +02:00
/// <summary>
/// Persist all user data for the specified user
/// </summary>
/// <param name="userId"></param>
/// <param name="userData"></param>
/// <param name="cancellationToken"></param>
/// <returns></returns>
private async Task PersistAllUserData ( Guid userId , IEnumerable < UserItemData > userData , CancellationToken cancellationToken )
{
cancellationToken . ThrowIfCancellationRequested ( ) ;
2015-05-03 00:59:01 +02:00
await WriteLock . WaitAsync ( cancellationToken ) . ConfigureAwait ( false ) ;
2014-09-20 23:57:16 +02:00
IDbTransaction transaction = null ;
try
{
transaction = _connection . BeginTransaction ( ) ;
foreach ( var userItemData in userData )
{
using ( var cmd = _connection . CreateCommand ( ) )
{
2016-02-21 00:06:57 +01:00
cmd . CommandText = "replace into userdata (key, userId, rating,played,playCount,isFavorite,playbackPositionTicks,lastPlayedDate,AudioStreamIndex,SubtitleStreamIndex) values (@key, @userId, @rating,@played,@playCount,@isFavorite,@playbackPositionTicks,@lastPlayedDate,@AudioStreamIndex,@SubtitleStreamIndex)" ;
2014-09-20 23:57:16 +02:00
cmd . Parameters . Add ( cmd , "@key" , DbType . String ) . Value = userItemData . Key ;
cmd . Parameters . Add ( cmd , "@userId" , DbType . Guid ) . Value = userId ;
cmd . Parameters . Add ( cmd , "@rating" , DbType . Double ) . Value = userItemData . Rating ;
cmd . Parameters . Add ( cmd , "@played" , DbType . Boolean ) . Value = userItemData . Played ;
cmd . Parameters . Add ( cmd , "@playCount" , DbType . Int32 ) . Value = userItemData . PlayCount ;
cmd . Parameters . Add ( cmd , "@isFavorite" , DbType . Boolean ) . Value = userItemData . IsFavorite ;
cmd . Parameters . Add ( cmd , "@playbackPositionTicks" , DbType . Int64 ) . Value = userItemData . PlaybackPositionTicks ;
cmd . Parameters . Add ( cmd , "@lastPlayedDate" , DbType . DateTime ) . Value = userItemData . LastPlayedDate ;
2016-02-21 00:06:57 +01:00
cmd . Parameters . Add ( cmd , "@AudioStreamIndex" , DbType . Int32 ) . Value = userItemData . AudioStreamIndex ;
cmd . Parameters . Add ( cmd , "@SubtitleStreamIndex" , DbType . Int32 ) . Value = userItemData . SubtitleStreamIndex ;
2014-09-20 23:57:16 +02:00
cmd . Transaction = transaction ;
cmd . ExecuteNonQuery ( ) ;
}
2014-09-30 06:47:30 +02:00
2014-09-21 00:16:17 +02:00
cancellationToken . ThrowIfCancellationRequested ( ) ;
2014-09-20 23:57:16 +02:00
}
transaction . Commit ( ) ;
}
catch ( OperationCanceledException )
{
if ( transaction ! = null )
{
transaction . Rollback ( ) ;
}
throw ;
}
catch ( Exception e )
{
2015-05-03 00:59:01 +02:00
Logger . ErrorException ( "Failed to save user data:" , e ) ;
2014-09-20 23:57:16 +02:00
if ( transaction ! = null )
{
transaction . Rollback ( ) ;
}
throw ;
}
finally
{
if ( transaction ! = null )
{
transaction . Dispose ( ) ;
}
2015-05-03 00:59:01 +02:00
WriteLock . Release ( ) ;
2014-09-20 23:57:16 +02:00
}
}
2013-02-21 02:33:05 +01:00
/// <summary>
2013-04-02 21:25:16 +02:00
/// Gets the user data.
2013-02-21 02:33:05 +01:00
/// </summary>
2013-04-02 21:25:16 +02:00
/// <param name="userId">The user id.</param>
2013-04-13 20:02:30 +02:00
/// <param name="key">The key.</param>
2013-04-02 21:25:16 +02:00
/// <returns>Task{UserItemData}.</returns>
/// <exception cref="System.ArgumentNullException">
/// userId
/// or
2013-04-13 20:02:30 +02:00
/// key
2013-04-02 21:25:16 +02:00
/// </exception>
2013-06-17 22:35:43 +02:00
public UserItemData GetUserData ( Guid userId , string key )
2013-02-21 02:33:05 +01:00
{
2013-04-02 21:25:16 +02:00
if ( userId = = Guid . Empty )
2013-02-21 02:33:05 +01:00
{
2013-04-02 21:25:16 +02:00
throw new ArgumentNullException ( "userId" ) ;
}
2013-04-13 20:02:30 +02:00
if ( string . IsNullOrEmpty ( key ) )
2013-04-02 21:25:16 +02:00
{
2013-04-13 20:02:30 +02:00
throw new ArgumentNullException ( "key" ) ;
2013-02-21 02:33:05 +01:00
}
2013-05-09 22:29:50 +02:00
2013-06-18 21:16:27 +02:00
using ( var cmd = _connection . CreateCommand ( ) )
2013-05-08 22:58:52 +02:00
{
2016-02-21 00:06:57 +01:00
cmd . CommandText = "select key,userid,rating,played,playCount,isFavorite,playbackPositionTicks,lastPlayedDate,AudioStreamIndex,SubtitleStreamIndex from userdata where key = @key and userId=@userId" ;
2013-06-18 11:43:07 +02:00
2013-09-29 17:32:50 +02:00
cmd . Parameters . Add ( cmd , "@key" , DbType . String ) . Value = key ;
cmd . Parameters . Add ( cmd , "@userId" , DbType . Guid ) . Value = userId ;
2013-06-18 11:43:07 +02:00
using ( var reader = cmd . ExecuteReader ( CommandBehavior . SequentialAccess | CommandBehavior . SingleResult | CommandBehavior . SingleRow ) )
{
if ( reader . Read ( ) )
{
2014-09-30 06:47:30 +02:00
return ReadRow ( reader ) ;
2014-09-20 23:57:16 +02:00
}
}
2013-09-27 17:23:27 +02:00
2014-09-30 06:47:30 +02:00
return new UserItemData
{
UserId = userId ,
Key = key
} ;
2014-09-20 23:57:16 +02:00
}
}
2013-09-27 17:23:27 +02:00
2014-09-20 23:57:16 +02:00
/// <summary>
/// Return all user-data associated with the given user
/// </summary>
/// <param name="userId"></param>
/// <returns></returns>
public IEnumerable < UserItemData > GetAllUserData ( Guid userId )
{
if ( userId = = Guid . Empty )
{
throw new ArgumentNullException ( "userId" ) ;
}
using ( var cmd = _connection . CreateCommand ( ) )
{
2016-02-21 00:06:57 +01:00
cmd . CommandText = "select key,userid,rating,played,playCount,isFavorite,playbackPositionTicks,lastPlayedDate,AudioStreamIndex,SubtitleStreamIndex from userdata where userId=@userId" ;
2014-09-20 23:57:16 +02:00
cmd . Parameters . Add ( cmd , "@userId" , DbType . Guid ) . Value = userId ;
using ( var reader = cmd . ExecuteReader ( CommandBehavior . SequentialAccess | CommandBehavior . SingleResult ) )
{
while ( reader . Read ( ) )
{
2014-09-30 06:47:30 +02:00
yield return ReadRow ( reader ) ;
2013-06-18 11:43:07 +02:00
}
}
2013-05-08 22:58:52 +02:00
}
2014-09-20 23:57:16 +02:00
}
/// <summary>
/// Read a row from the specified reader into the provided userData object
/// </summary>
/// <param name="reader"></param>
2014-09-30 06:47:30 +02:00
private UserItemData ReadRow ( IDataReader reader )
2014-09-20 23:57:16 +02:00
{
2014-09-30 06:47:30 +02:00
var userData = new UserItemData ( ) ;
userData . Key = reader . GetString ( 0 ) ;
userData . UserId = reader . GetGuid ( 1 ) ;
if ( ! reader . IsDBNull ( 2 ) )
2014-09-20 23:57:16 +02:00
{
2014-09-30 06:47:30 +02:00
userData . Rating = reader . GetDouble ( 2 ) ;
2014-09-20 23:57:16 +02:00
}
2014-09-30 06:47:30 +02:00
userData . Played = reader . GetBoolean ( 3 ) ;
userData . PlayCount = reader . GetInt32 ( 4 ) ;
userData . IsFavorite = reader . GetBoolean ( 5 ) ;
userData . PlaybackPositionTicks = reader . GetInt64 ( 6 ) ;
2014-09-20 23:57:16 +02:00
2014-09-30 06:47:30 +02:00
if ( ! reader . IsDBNull ( 7 ) )
2014-09-20 23:57:16 +02:00
{
2014-09-30 06:47:30 +02:00
userData . LastPlayedDate = reader . GetDateTime ( 7 ) . ToUniversalTime ( ) ;
2014-09-20 23:57:16 +02:00
}
2014-09-30 06:47:30 +02:00
2016-02-21 00:06:57 +01:00
if ( ! reader . IsDBNull ( 8 ) )
{
userData . AudioStreamIndex = reader . GetInt32 ( 8 ) ;
}
if ( ! reader . IsDBNull ( 9 ) )
{
userData . SubtitleStreamIndex = reader . GetInt32 ( 9 ) ;
}
2014-09-30 06:47:30 +02:00
return userData ;
2013-02-21 02:33:05 +01:00
}
2013-06-18 21:16:27 +02:00
2015-05-03 00:59:01 +02:00
protected override void CloseConnection ( )
2013-06-18 21:16:27 +02:00
{
2015-05-03 00:59:01 +02:00
if ( _connection ! = null )
2013-06-18 21:16:27 +02:00
{
2015-05-03 00:59:01 +02:00
if ( _connection . IsOpen ( ) )
2013-06-18 21:16:27 +02:00
{
2015-05-03 00:59:01 +02:00
_connection . Close ( ) ;
2013-06-18 21:16:27 +02:00
}
2015-05-03 00:59:01 +02:00
_connection . Dispose ( ) ;
_connection = null ;
2013-06-18 21:16:27 +02:00
}
}
2013-02-21 02:33:05 +01:00
}
2013-05-09 22:29:50 +02:00
}