2016-11-20 04:46:07 -05:00
using System ;
using System.Collections.Generic ;
using System.IO ;
using System.Linq ;
using System.Threading ;
using System.Threading.Tasks ;
using MediaBrowser.Common.Configuration ;
using MediaBrowser.Controller.Entities ;
using MediaBrowser.Controller.Persistence ;
2016-11-21 15:22:43 -05:00
using MediaBrowser.Model.IO ;
2016-11-20 04:46:07 -05:00
using MediaBrowser.Model.Logging ;
using SQLitePCL.pretty ;
namespace Emby.Server.Implementations.Data
{
public class SqliteUserDataRepository : BaseSqliteRepository , IUserDataRepository
{
2016-11-21 15:22:43 -05:00
private readonly string _importFile ;
private readonly IFileSystem _fileSystem ;
public SqliteUserDataRepository ( ILogger logger , IApplicationPaths appPaths , IFileSystem fileSystem )
2016-11-20 04:46:07 -05:00
: base ( logger )
{
2016-11-21 15:22:43 -05:00
_fileSystem = fileSystem ;
DbFilePath = Path . Combine ( appPaths . DataPath , "library.db" ) ;
_importFile = Path . Combine ( appPaths . DataPath , "userdata_v2.db" ) ;
2016-11-20 04:46:07 -05:00
}
/// <summary>
/// Gets the name of the repository
/// </summary>
/// <value>The name.</value>
public string Name
{
get
{
return "SQLite" ;
}
}
/// <summary>
/// Opens the connection to the database
/// </summary>
/// <returns>Task.</returns>
2016-11-20 22:52:58 -05:00
public void Initialize ( ReaderWriterLockSlim writeLock )
2016-11-20 04:46:07 -05:00
{
WriteLock . Dispose ( ) ;
WriteLock = writeLock ;
2016-11-20 22:52:58 -05:00
using ( var connection = CreateConnection ( ) )
{
string [ ] queries = {
2016-11-21 15:22:43 -05:00
"pragma temp_store = memory" ,
2016-11-20 04:46:07 -05:00
2016-11-21 15:22:43 -05: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)" ,
2016-11-20 04:46:07 -05:00
2016-11-21 15:22:43 -05:00
"create table if not exists DataSettings (IsUserDataImported bit)" ,
2016-11-20 04:46:07 -05:00
2016-11-21 15:22:43 -05:00
"drop index if exists idx_userdata" ,
"drop index if exists idx_userdata1" ,
"drop index if exists idx_userdata2" ,
"drop index if exists userdataindex1" ,
2016-11-20 04:46:07 -05:00
2016-11-21 15:22:43 -05:00
"create unique index if not exists userdataindex on userdata (key, userId)" ,
"create index if not exists userdataindex2 on userdata (key, userId, played)" ,
"create index if not exists userdataindex3 on userdata (key, userId, playbackPositionTicks)" ,
"create index if not exists userdataindex4 on userdata (key, userId, isFavorite)" ,
2016-11-20 04:46:07 -05:00
"pragma shrink_memory"
} ;
2016-11-20 22:52:58 -05:00
connection . RunQueries ( queries ) ;
2016-11-20 04:46:07 -05:00
2016-11-20 22:52:58 -05:00
connection . RunInTransaction ( db = >
{
var existingColumnNames = GetColumnNames ( db , "userdata" ) ;
2016-11-20 04:46:07 -05:00
2016-11-20 22:52:58 -05:00
AddColumn ( db , "userdata" , "AudioStreamIndex" , "int" , existingColumnNames ) ;
AddColumn ( db , "userdata" , "SubtitleStreamIndex" , "int" , existingColumnNames ) ;
} ) ;
2016-11-21 15:22:43 -05:00
ImportUserDataIfNeeded ( connection ) ;
2016-11-20 22:52:58 -05:00
}
2016-11-20 04:46:07 -05:00
}
2016-11-21 15:22:43 -05:00
private void ImportUserDataIfNeeded ( IDatabaseConnection connection )
{
if ( ! _fileSystem . FileExists ( _importFile ) )
{
return ;
}
var fileToImport = _importFile ;
var isImported = connection . Query ( "select IsUserDataImported from DataSettings" ) . SelectScalarBool ( ) . FirstOrDefault ( ) ;
if ( isImported )
{
return ;
}
ImportUserData ( connection , fileToImport ) ;
connection . RunInTransaction ( db = >
{
using ( var statement = db . PrepareStatement ( "replace into DataSettings (IsUserDataImported) values (@IsUserDataImported)" ) )
{
statement . TryBind ( "@IsUserDataImported" , true ) ;
statement . MoveNext ( ) ;
}
} ) ;
}
private void ImportUserData ( IDatabaseConnection connection , string file )
{
SqliteExtensions . Attach ( connection , file , "UserDataBackup" ) ;
var columns = "key, userId, rating, played, playCount, isFavorite, playbackPositionTicks, lastPlayedDate, AudioStreamIndex, SubtitleStreamIndex" ;
connection . RunInTransaction ( db = >
{
db . Execute ( "REPLACE INTO userdata(" + columns + ") SELECT " + columns + " FROM UserDataBackup.userdata;" ) ;
} ) ;
}
2016-11-20 04:46:07 -05:00
/// <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 Task SaveUserData ( Guid userId , string key , UserItemData userData , CancellationToken cancellationToken )
{
if ( userData = = null )
{
throw new ArgumentNullException ( "userData" ) ;
}
if ( userId = = Guid . Empty )
{
throw new ArgumentNullException ( "userId" ) ;
}
if ( string . IsNullOrEmpty ( key ) )
{
throw new ArgumentNullException ( "key" ) ;
}
return PersistUserData ( userId , key , userData , cancellationToken ) ;
}
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 . ToList ( ) , cancellationToken ) ;
}
/// <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 ( ) ;
2016-11-20 22:52:58 -05:00
using ( var connection = CreateConnection ( ) )
2016-11-20 04:46:07 -05:00
{
2016-11-20 22:52:58 -05:00
using ( WriteLock . Write ( ) )
2016-11-20 04:46:07 -05:00
{
2016-11-20 22:52:58 -05:00
connection . RunInTransaction ( db = >
{
SaveUserData ( db , userId , key , userData ) ;
} ) ;
}
2016-11-20 04:46:07 -05:00
}
}
private void SaveUserData ( IDatabaseConnection db , Guid userId , string key , UserItemData userData )
{
2016-11-20 22:52:58 -05:00
using ( var statement = db . PrepareStatement ( "replace into userdata (key, userId, rating,played,playCount,isFavorite,playbackPositionTicks,lastPlayedDate,AudioStreamIndex,SubtitleStreamIndex) values (@key, @userId, @rating,@played,@playCount,@isFavorite,@playbackPositionTicks,@lastPlayedDate,@AudioStreamIndex,@SubtitleStreamIndex)" ) )
2016-11-20 04:46:07 -05:00
{
2016-11-21 13:49:07 -05:00
statement . TryBind ( "@userId" , userId . ToGuidParamValue ( ) ) ;
statement . TryBind ( "@key" , key ) ;
2016-11-20 04:46:07 -05:00
if ( userData . Rating . HasValue )
{
statement . TryBind ( "@rating" , userData . Rating . Value ) ;
}
else
{
statement . TryBindNull ( "@rating" ) ;
}
statement . TryBind ( "@played" , userData . Played ) ;
statement . TryBind ( "@playCount" , userData . PlayCount ) ;
statement . TryBind ( "@isFavorite" , userData . IsFavorite ) ;
statement . TryBind ( "@playbackPositionTicks" , userData . PlaybackPositionTicks ) ;
if ( userData . LastPlayedDate . HasValue )
{
statement . TryBind ( "@lastPlayedDate" , userData . LastPlayedDate . Value . ToDateTimeParamValue ( ) ) ;
}
else
{
statement . TryBindNull ( "@lastPlayedDate" ) ;
}
if ( userData . AudioStreamIndex . HasValue )
{
statement . TryBind ( "@AudioStreamIndex" , userData . AudioStreamIndex . Value ) ;
}
else
{
statement . TryBindNull ( "@AudioStreamIndex" ) ;
}
if ( userData . SubtitleStreamIndex . HasValue )
{
statement . TryBind ( "@SubtitleStreamIndex" , userData . SubtitleStreamIndex . Value ) ;
}
else
{
statement . TryBindNull ( "@SubtitleStreamIndex" ) ;
}
statement . MoveNext ( ) ;
}
}
/// <summary>
/// Persist all user data for the specified user
/// </summary>
private async Task PersistAllUserData ( Guid userId , List < UserItemData > userDataList , CancellationToken cancellationToken )
{
cancellationToken . ThrowIfCancellationRequested ( ) ;
2016-11-20 22:52:58 -05:00
using ( var connection = CreateConnection ( ) )
2016-11-20 04:46:07 -05:00
{
2016-11-20 22:52:58 -05:00
using ( WriteLock . Write ( ) )
2016-11-20 04:46:07 -05:00
{
2016-11-20 22:52:58 -05:00
connection . RunInTransaction ( db = >
2016-11-20 04:46:07 -05:00
{
2016-11-20 22:52:58 -05:00
foreach ( var userItemData in userDataList )
{
SaveUserData ( db , userId , userItemData . Key , userItemData ) ;
}
} ) ;
}
2016-11-20 04:46:07 -05:00
}
}
/// <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" ) ;
}
2016-11-20 22:52:58 -05:00
using ( var connection = CreateConnection ( true ) )
2016-11-20 04:46:07 -05:00
{
2016-11-20 22:52:58 -05:00
using ( WriteLock . Read ( ) )
2016-11-20 04:46:07 -05:00
{
2016-11-20 22:52:58 -05:00
using ( var statement = connection . PrepareStatement ( "select key,userid,rating,played,playCount,isFavorite,playbackPositionTicks,lastPlayedDate,AudioStreamIndex,SubtitleStreamIndex from userdata where key =@Key and userId=@UserId" ) )
2016-11-20 04:46:07 -05:00
{
2016-11-20 22:52:58 -05:00
statement . TryBind ( "@UserId" , userId . ToGuidParamValue ( ) ) ;
statement . TryBind ( "@Key" , key ) ;
foreach ( var row in statement . ExecuteQuery ( ) )
{
return ReadRow ( row ) ;
}
2016-11-20 04:46:07 -05:00
}
}
}
return null ;
}
public UserItemData GetUserData ( Guid userId , List < string > keys )
{
if ( userId = = Guid . Empty )
{
throw new ArgumentNullException ( "userId" ) ;
}
if ( keys = = null )
{
throw new ArgumentNullException ( "keys" ) ;
}
if ( keys . Count = = 0 )
{
return null ;
}
return GetUserData ( userId , keys [ 0 ] ) ;
}
/// <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" ) ;
}
var list = new List < UserItemData > ( ) ;
2016-11-20 22:52:58 -05:00
using ( var connection = CreateConnection ( ) )
2016-11-20 04:46:07 -05:00
{
2016-11-20 22:52:58 -05:00
using ( WriteLock . Read ( ) )
2016-11-20 04:46:07 -05:00
{
2016-11-20 22:52:58 -05:00
using ( var statement = connection . PrepareStatement ( "select key,userid,rating,played,playCount,isFavorite,playbackPositionTicks,lastPlayedDate,AudioStreamIndex,SubtitleStreamIndex from userdata where userId=@UserId" ) )
2016-11-20 04:46:07 -05:00
{
2016-11-20 22:52:58 -05:00
statement . TryBind ( "@UserId" , userId . ToGuidParamValue ( ) ) ;
foreach ( var row in statement . ExecuteQuery ( ) )
{
list . Add ( ReadRow ( row ) ) ;
}
2016-11-20 04:46:07 -05:00
}
}
}
return list ;
}
/// <summary>
/// Read a row from the specified reader into the provided userData object
/// </summary>
/// <param name="reader"></param>
private UserItemData ReadRow ( IReadOnlyList < IResultSetValue > reader )
{
var userData = new UserItemData ( ) ;
userData . Key = reader [ 0 ] . ToString ( ) ;
userData . UserId = reader [ 1 ] . ReadGuid ( ) ;
if ( reader [ 2 ] . SQLiteType ! = SQLiteType . Null )
{
userData . Rating = reader [ 2 ] . ToDouble ( ) ;
}
userData . Played = reader [ 3 ] . ToBool ( ) ;
userData . PlayCount = reader [ 4 ] . ToInt ( ) ;
userData . IsFavorite = reader [ 5 ] . ToBool ( ) ;
userData . PlaybackPositionTicks = reader [ 6 ] . ToInt64 ( ) ;
if ( reader [ 7 ] . SQLiteType ! = SQLiteType . Null )
{
userData . LastPlayedDate = reader [ 7 ] . ReadDateTime ( ) ;
}
if ( reader [ 8 ] . SQLiteType ! = SQLiteType . Null )
{
userData . AudioStreamIndex = reader [ 8 ] . ToInt ( ) ;
}
if ( reader [ 9 ] . SQLiteType ! = SQLiteType . Null )
{
userData . SubtitleStreamIndex = reader [ 9 ] . ToInt ( ) ;
}
return userData ;
}
protected override void Dispose ( bool dispose )
{
// handled by library database
}
protected override void CloseConnection ( )
{
// handled by library database
}
}
}