jellyfin/Emby.Server.Implementations/Data/SqliteExtensions.cs

406 lines
12 KiB
C#
Raw Normal View History

using System;
2016-11-19 21:07:01 +00:00
using System.Collections.Generic;
2016-11-18 08:39:20 +00:00
using System.Globalization;
using System.IO;
2016-11-18 09:28:39 +00:00
using MediaBrowser.Model.Serialization;
2016-11-18 08:39:20 +00:00
using SQLitePCL.pretty;
namespace Emby.Server.Implementations.Data
{
public static class SqliteExtensions
{
public static void RunQueries(this SQLiteDatabaseConnection connection, string[] queries)
{
if (queries == null)
{
throw new ArgumentNullException(nameof(queries));
2016-11-18 08:39:20 +00:00
}
connection.RunInTransaction(conn =>
{
conn.ExecuteAll(string.Join(";", queries));
});
}
2017-05-07 20:02:32 +00:00
public static byte[] ToGuidBlob(this string str)
2016-11-18 08:39:20 +00:00
{
2017-05-07 20:02:32 +00:00
return ToGuidBlob(new Guid(str));
2016-11-18 09:28:39 +00:00
}
2017-05-07 20:02:32 +00:00
public static byte[] ToGuidBlob(this Guid guid)
2016-11-18 09:28:39 +00:00
{
return guid.ToByteArray();
2016-11-18 08:39:20 +00:00
}
2017-05-07 20:02:32 +00:00
public static Guid ReadGuidFromBlob(this IResultSetValue result)
2016-11-18 08:39:20 +00:00
{
2019-08-19 15:03:21 +00:00
// TODO: Remove ToArray when upgrading to netstandard2.1
return new Guid(result.ToBlob().ToArray());
2016-11-18 08:39:20 +00:00
}
public static string ToDateTimeParamValue(this DateTime dateValue)
{
var kind = DateTimeKind.Utc;
return (dateValue.Kind == DateTimeKind.Unspecified)
? DateTime.SpecifyKind(dateValue, kind).ToString(
GetDateTimeKindFormat(kind),
CultureInfo.InvariantCulture)
: dateValue.ToString(
GetDateTimeKindFormat(dateValue.Kind),
CultureInfo.InvariantCulture);
}
private static string GetDateTimeKindFormat(
DateTimeKind kind)
{
return (kind == DateTimeKind.Utc) ? _datetimeFormatUtc : _datetimeFormatLocal;
}
/// <summary>
/// An array of ISO-8601 DateTime formats that we support parsing.
/// </summary>
private static string[] _datetimeFormats = new string[] {
"THHmmssK",
"THHmmK",
"HH:mm:ss.FFFFFFFK",
"HH:mm:ssK",
"HH:mmK",
"yyyy-MM-dd HH:mm:ss.FFFFFFFK", /* NOTE: UTC default (5). */
"yyyy-MM-dd HH:mm:ssK",
"yyyy-MM-dd HH:mmK",
"yyyy-MM-ddTHH:mm:ss.FFFFFFFK",
"yyyy-MM-ddTHH:mmK",
"yyyy-MM-ddTHH:mm:ssK",
"yyyyMMddHHmmssK",
"yyyyMMddHHmmK",
"yyyyMMddTHHmmssFFFFFFFK",
"THHmmss",
"THHmm",
"HH:mm:ss.FFFFFFF",
"HH:mm:ss",
"HH:mm",
"yyyy-MM-dd HH:mm:ss.FFFFFFF", /* NOTE: Non-UTC default (19). */
"yyyy-MM-dd HH:mm:ss",
"yyyy-MM-dd HH:mm",
"yyyy-MM-ddTHH:mm:ss.FFFFFFF",
"yyyy-MM-ddTHH:mm",
"yyyy-MM-ddTHH:mm:ss",
"yyyyMMddHHmmss",
"yyyyMMddHHmm",
"yyyyMMddTHHmmssFFFFFFF",
"yyyy-MM-dd",
"yyyyMMdd",
"yy-MM-dd"
};
private static string _datetimeFormatUtc = _datetimeFormats[5];
private static string _datetimeFormatLocal = _datetimeFormats[19];
public static DateTime ReadDateTime(this IResultSetValue result)
{
var dateText = result.ToString();
return DateTime.ParseExact(
dateText, _datetimeFormats,
DateTimeFormatInfo.InvariantInfo,
DateTimeStyles.None).ToUniversalTime();
}
2016-11-18 09:28:39 +00:00
2018-09-12 17:26:21 +00:00
public static DateTime? TryReadDateTime(this IResultSetValue result)
{
var dateText = result.ToString();
if (DateTime.TryParseExact(dateText, _datetimeFormats, DateTimeFormatInfo.InvariantInfo, DateTimeStyles.None, out var dateTimeResult))
2018-09-12 17:26:21 +00:00
{
return dateTimeResult.ToUniversalTime();
}
return null;
}
2016-11-18 09:28:39 +00:00
/// <summary>
/// Serializes to bytes.
/// </summary>
/// <returns>System.Byte[][].</returns>
2019-01-13 20:37:13 +00:00
/// <exception cref="ArgumentNullException">obj</exception>
2018-09-12 17:26:21 +00:00
public static byte[] SerializeToBytes(this IJsonSerializer json, object obj)
2016-11-18 09:28:39 +00:00
{
if (obj == null)
{
throw new ArgumentNullException(nameof(obj));
2016-11-18 09:28:39 +00:00
}
2018-09-12 17:26:21 +00:00
using (var stream = new MemoryStream())
2016-11-18 09:28:39 +00:00
{
json.SerializeToStream(obj, stream);
return stream.ToArray();
}
}
2016-11-19 21:07:01 +00:00
2019-02-20 13:26:49 +00:00
public static void Attach(SQLiteDatabaseConnection db, string path, string alias)
2016-11-19 21:07:01 +00:00
{
2016-11-21 03:52:58 +00:00
var commandText = string.Format("attach @path as {0};", alias);
2016-11-19 21:07:01 +00:00
2016-11-21 03:52:58 +00:00
using (var statement = db.PrepareStatement(commandText))
{
statement.TryBind("@path", path);
statement.MoveNext();
}
2016-11-19 21:07:01 +00:00
}
2016-11-20 02:43:21 +00:00
public static bool IsDBNull(this IReadOnlyList<IResultSetValue> result, int index)
{
return result[index].SQLiteType == SQLiteType.Null;
}
public static string GetString(this IReadOnlyList<IResultSetValue> result, int index)
{
return result[index].ToString();
}
public static bool GetBoolean(this IReadOnlyList<IResultSetValue> result, int index)
{
return result[index].ToBool();
}
public static int GetInt32(this IReadOnlyList<IResultSetValue> result, int index)
{
return result[index].ToInt();
}
public static long GetInt64(this IReadOnlyList<IResultSetValue> result, int index)
{
return result[index].ToInt64();
}
public static float GetFloat(this IReadOnlyList<IResultSetValue> result, int index)
{
return result[index].ToFloat();
}
2016-11-20 05:59:36 +00:00
public static Guid GetGuid(this IReadOnlyList<IResultSetValue> result, int index)
2016-11-20 02:43:21 +00:00
{
2017-05-07 20:02:32 +00:00
return result[index].ReadGuidFromBlob();
2016-11-20 02:43:21 +00:00
}
2016-11-20 09:46:07 +00:00
private static void CheckName(string name)
{
#if DEBUG
//if (!name.IndexOf("@", StringComparison.OrdinalIgnoreCase) != 0)
{
throw new Exception("Invalid param name: " + name);
}
#endif
}
2016-11-20 07:10:07 +00:00
public static void TryBind(this IStatement statement, string name, double value)
2016-11-20 06:13:35 +00:00
{
if (statement.BindParameters.TryGetValue(name, out IBindParameter bindParam))
2016-11-20 06:13:35 +00:00
{
bindParam.Bind(value);
}
2016-11-20 09:46:07 +00:00
else
{
CheckName(name);
}
2016-11-20 06:13:35 +00:00
}
2016-11-20 07:10:07 +00:00
public static void TryBind(this IStatement statement, string name, string value)
2016-11-20 02:43:21 +00:00
{
if (statement.BindParameters.TryGetValue(name, out IBindParameter bindParam))
2016-11-20 05:59:36 +00:00
{
2016-11-20 09:46:07 +00:00
if (value == null)
{
bindParam.BindNull();
}
else
{
bindParam.Bind(value);
}
}
else
{
CheckName(name);
2016-11-20 05:59:36 +00:00
}
}
2016-11-20 07:10:07 +00:00
public static void TryBind(this IStatement statement, string name, bool value)
2016-11-20 05:59:36 +00:00
{
if (statement.BindParameters.TryGetValue(name, out IBindParameter bindParam))
2016-11-20 06:13:35 +00:00
{
bindParam.Bind(value);
}
2016-11-20 09:46:07 +00:00
else
{
CheckName(name);
}
2016-11-20 06:13:35 +00:00
}
2016-11-20 07:10:07 +00:00
public static void TryBind(this IStatement statement, string name, float value)
2016-11-20 06:13:35 +00:00
{
if (statement.BindParameters.TryGetValue(name, out IBindParameter bindParam))
2016-11-20 06:13:35 +00:00
{
bindParam.Bind(value);
}
2016-11-20 09:46:07 +00:00
else
{
CheckName(name);
}
2016-11-20 06:13:35 +00:00
}
2016-11-20 07:10:07 +00:00
public static void TryBind(this IStatement statement, string name, int value)
2016-11-20 06:13:35 +00:00
{
if (statement.BindParameters.TryGetValue(name, out IBindParameter bindParam))
2016-11-20 05:59:36 +00:00
{
bindParam.Bind(value);
}
2016-11-20 09:46:07 +00:00
else
{
CheckName(name);
}
2016-11-20 05:59:36 +00:00
}
2016-11-20 07:10:07 +00:00
public static void TryBind(this IStatement statement, string name, Guid value)
2016-11-20 05:59:36 +00:00
{
if (statement.BindParameters.TryGetValue(name, out IBindParameter bindParam))
2016-11-20 07:10:07 +00:00
{
2017-05-07 20:02:32 +00:00
bindParam.Bind(value.ToGuidBlob());
2016-11-20 07:10:07 +00:00
}
2016-11-20 09:46:07 +00:00
else
{
CheckName(name);
}
2016-11-20 07:10:07 +00:00
}
public static void TryBind(this IStatement statement, string name, DateTime value)
{
if (statement.BindParameters.TryGetValue(name, out IBindParameter bindParam))
2016-11-20 07:10:07 +00:00
{
bindParam.Bind(value.ToDateTimeParamValue());
}
2016-11-20 09:46:07 +00:00
else
{
CheckName(name);
}
2016-11-20 07:10:07 +00:00
}
public static void TryBind(this IStatement statement, string name, long value)
{
if (statement.BindParameters.TryGetValue(name, out IBindParameter bindParam))
2016-11-20 05:59:36 +00:00
{
bindParam.Bind(value);
}
2016-11-20 09:46:07 +00:00
else
{
CheckName(name);
}
2016-11-20 05:59:36 +00:00
}
2016-11-20 07:10:07 +00:00
public static void TryBind(this IStatement statement, string name, byte[] value)
2016-11-20 05:59:36 +00:00
{
if (statement.BindParameters.TryGetValue(name, out IBindParameter bindParam))
2016-11-20 07:10:07 +00:00
{
bindParam.Bind(value);
}
2016-11-20 09:46:07 +00:00
else
{
CheckName(name);
}
2016-11-20 07:10:07 +00:00
}
public static void TryBindNull(this IStatement statement, string name)
{
if (statement.BindParameters.TryGetValue(name, out IBindParameter bindParam))
2016-11-20 05:59:36 +00:00
{
bindParam.BindNull();
}
2016-11-20 09:46:07 +00:00
else
{
CheckName(name);
}
2016-11-20 05:59:36 +00:00
}
2016-11-20 07:10:07 +00:00
public static void TryBind(this IStatement statement, string name, DateTime? value)
{
if (value.HasValue)
{
TryBind(statement, name, value.Value);
}
else
{
TryBindNull(statement, name);
}
}
public static void TryBind(this IStatement statement, string name, Guid? value)
{
if (value.HasValue)
{
TryBind(statement, name, value.Value);
}
else
{
TryBindNull(statement, name);
}
}
2016-12-23 17:09:50 +00:00
public static void TryBind(this IStatement statement, string name, double? value)
{
if (value.HasValue)
{
TryBind(statement, name, value.Value);
}
else
{
TryBindNull(statement, name);
}
}
2016-11-20 07:10:07 +00:00
public static void TryBind(this IStatement statement, string name, int? value)
{
if (value.HasValue)
{
TryBind(statement, name, value.Value);
}
else
{
TryBindNull(statement, name);
}
}
public static void TryBind(this IStatement statement, string name, float? value)
{
if (value.HasValue)
{
TryBind(statement, name, value.Value);
}
else
{
TryBindNull(statement, name);
}
}
public static void TryBind(this IStatement statement, string name, bool? value)
{
if (value.HasValue)
{
TryBind(statement, name, value.Value);
}
else
{
TryBindNull(statement, name);
}
}
2016-11-20 05:59:36 +00:00
public static IEnumerable<IReadOnlyList<IResultSetValue>> ExecuteQuery(
this IStatement This)
{
while (This.MoveNext())
{
yield return This.Current;
}
2016-11-20 02:43:21 +00:00
}
2016-11-18 08:39:20 +00:00
}
}