Swap to dapper Mk. 2

src/NzbDrone.Core/Datastore/Extensions/BuilderExtensions.cs

@@ -0,0 +1,135 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Linq.Expressions;
+using System.Reflection;
+using System.Text;
+using Dapper;
+using NLog;
+using NzbDrone.Common.Extensions;
+using NzbDrone.Common.Instrumentation;
+using NzbDrone.Common.Serializer;
+
+namespace NzbDrone.Core.Datastore
+{
+    public static class SqlBuilderExtensions
+    {
+        public static bool LogSql { get; set; }
+        private static readonly Logger Logger = NzbDroneLogger.GetLogger(typeof(SqlBuilderExtensions));
+
+        public static SqlBuilder SelectAll(this SqlBuilder builder)
+        {
+            return builder.Select("*");
+        }
+
+        public static SqlBuilder SelectCount(this SqlBuilder builder)
+        {
+            return builder.Select("COUNT(*)");
+        }
+
+        public static SqlBuilder Where<TModel>(this SqlBuilder builder, Expression<Func<TModel, bool>> filter)
+        {
+            var wb = new WhereBuilder(filter, true);
+
+            return builder.Where(wb.ToString(), wb.Parameters);
+        }
+
+        public static SqlBuilder OrWhere<TModel>(this SqlBuilder builder, Expression<Func<TModel, bool>> filter)
+        {
+            var wb = new WhereBuilder(filter, true);
+
+            return builder.OrWhere(wb.ToString(), wb.Parameters);
+        }
+
+        public static SqlBuilder Join<TLeft, TRight>(this SqlBuilder builder, Expression<Func<TLeft, TRight, bool>> filter)
+        {
+            var wb = new WhereBuilder(filter, false);
+
+            var rightTable = TableMapping.Mapper.TableNameMapping(typeof(TRight));
+
+            return builder.Join($"{rightTable} ON {wb.ToString()}");
+        }
+
+        public static SqlBuilder LeftJoin<TLeft, TRight>(this SqlBuilder builder, Expression<Func<TLeft, TRight, bool>> filter)
+        {
+            var wb = new WhereBuilder(filter, false);
+
+            var rightTable = TableMapping.Mapper.TableNameMapping(typeof(TRight));
+
+            return builder.LeftJoin($"{rightTable} ON {wb.ToString()}");
+        }
+
+        public static SqlBuilder.Template LogQuery(this SqlBuilder.Template template)
+        {
+            if (LogSql)
+            {
+                var sb = new StringBuilder();
+                sb.AppendLine();
+                sb.AppendLine("==== Begin Query Trace ====");
+                sb.AppendLine();
+                sb.AppendLine("QUERY TEXT:");
+                sb.AppendLine(template.RawSql);
+                sb.AppendLine();
+                sb.AppendLine("PARAMETERS:");
+                foreach (var p in ((DynamicParameters)template.Parameters).ToDictionary())
+                {
+                    object val = (p.Value is string) ? string.Format("\"{0}\"", p.Value) : p.Value;
+                    sb.AppendFormat("{0} = [{1}]", p.Key, val.ToJson() ?? "NULL").AppendLine();
+                }
+
+                sb.AppendLine();
+                sb.AppendLine("==== End Query Trace ====");
+                sb.AppendLine();
+
+                Logger.Trace(sb.ToString());
+            }
+
+            return template;
+        }
+
+        private static Dictionary<string, object> ToDictionary(this DynamicParameters dynamicParams)
+        {
+            var argsDictionary = new Dictionary<string, object>();
+            var iLookup = (SqlMapper.IParameterLookup)dynamicParams;
+
+            foreach (var paramName in dynamicParams.ParameterNames)
+            {
+                var value = iLookup[paramName];
+                argsDictionary.Add(paramName, value);
+            }
+
+            var templates = dynamicParams.GetType().GetField("templates", BindingFlags.NonPublic | BindingFlags.Instance);
+            if (templates != null)
+            {
+                var list = templates.GetValue(dynamicParams) as List<object>;
+                if (list != null)
+                {
+                    foreach (var objProps in list.Select(obj => obj.GetPropertyValuePairs().ToList()))
+                    {
+                        objProps.ForEach(p => argsDictionary.Add(p.Key, p.Value));
+                    }
+                }
+            }
+
+            return argsDictionary;
+        }
+
+        private static Dictionary<string, object> GetPropertyValuePairs(this object obj, string[] hidden = null)
+        {
+            var type = obj.GetType();
+            var pairs = hidden == null
+                ? type.GetProperties()
+                .DistinctBy(propertyInfo => propertyInfo.Name)
+                .ToDictionary(
+                    propertyInfo => propertyInfo.Name,
+                    propertyInfo => propertyInfo.GetValue(obj, null))
+                : type.GetProperties()
+                .Where(it => !hidden.Contains(it.Name))
+                .DistinctBy(propertyInfo => propertyInfo.Name)
+                .ToDictionary(
+                    propertyInfo => propertyInfo.Name,
+                    propertyInfo => propertyInfo.GetValue(obj, null));
+            return pairs;
+        }
+    }
+}