Commit 047eef9f authored by Sam Saffron's avatar Sam Saffron

added execute command

parent 53e7f518
...@@ -109,33 +109,28 @@ public bool Equals(Identity other) ...@@ -109,33 +109,28 @@ public bool Equals(Identity other)
static Dictionary<Type, SqlDbType> typeMap; static Dictionary<Type, SqlDbType> typeMap;
/// <summary>
/// Execute parameterized SQL
/// </summary>
/// <returns>Number of rows affected</returns>
public static int ExecuteMapperCommand(this IDbConnection cnn, string sql, object param = null, SqlTransaction transaction = null)
{
return ExecuteCommand(cnn, transaction, sql, GetParamInfo(param));
}
public static List<dynamic> ExecuteMapperQuery (this IDbConnection cnn, string sql, object param = null, SqlTransaction transaction = null) public static List<dynamic> ExecuteMapperQuery (this IDbConnection cnn, string sql, object param = null, SqlTransaction transaction = null)
{ {
// TODO: get rid of casting hackery // TODO: get rid of casting hackery
return ExecuteMapperQuery<ExpandoObject>(cnn, sql, param, transaction).Select(s => s as dynamic).ToList(); return ExecuteMapperQuery<ExpandoObject>(cnn, sql, param, transaction).Select(s => s as dynamic).ToList();
} }
public static List<T> ExecuteMapperQuery<T>(this IDbConnection cnn, string sql, object param = null, SqlTransaction transaction = null) public static List<T> ExecuteMapperQuery<T>(this IDbConnection cnn, string sql, object param = null, SqlTransaction transaction = null)
{ {
var identity = new Identity(sql, typeof(T)); var identity = new Identity(sql, typeof(T));
var rval = new List<T>(); var rval = new List<T>();
using (var reader = GetReader(cnn, transaction, sql, GetParamInfo(param)))
Func<object, List<ParamInfo>> paramInfoGenerator;
List<ParamInfo> paramInfo = null;
if (param != null)
{
if (!cachedParamReaders.TryGetValue(param.GetType(), out paramInfoGenerator))
{
paramInfoGenerator = CreateParamInfoGenerator(param.GetType());
cachedParamReaders[param.GetType()] = paramInfoGenerator;
}
paramInfo = paramInfoGenerator(param);
}
using (var reader = GetReader<T>(cnn, transaction, sql, paramInfo))
{ {
object oDeserializer; object oDeserializer;
if (!cachedSerializers.TryGetValue(identity, out oDeserializer)) if (!cachedSerializers.TryGetValue(identity, out oDeserializer))
...@@ -167,6 +162,24 @@ public static List<T> ExecuteMapperQuery<T>(this IDbConnection cnn, string sql, ...@@ -167,6 +162,24 @@ public static List<T> ExecuteMapperQuery<T>(this IDbConnection cnn, string sql,
return rval; return rval;
} }
private static List<ParamInfo> GetParamInfo(object param)
{
Func<object, List<ParamInfo>> paramInfoGenerator;
List<ParamInfo> paramInfo = null;
if (param != null)
{
if (!cachedParamReaders.TryGetValue(param.GetType(), out paramInfoGenerator))
{
paramInfoGenerator = CreateParamInfoGenerator(param.GetType());
cachedParamReaders[param.GetType()] = paramInfoGenerator;
}
paramInfo = paramInfoGenerator(param);
}
return paramInfo;
}
private static object GetDynamicDeserializer(IDataReader reader) private static object GetDynamicDeserializer(IDataReader reader)
{ {
List<string> colNames = new List<string>(); List<string> colNames = new List<string>();
...@@ -228,11 +241,10 @@ private static object GetDynamicDeserializer(IDataReader reader) ...@@ -228,11 +241,10 @@ private static object GetDynamicDeserializer(IDataReader reader)
return (Func<object, List<ParamInfo>>)dm.CreateDelegate(typeof(Func<object, List<ParamInfo>>)); return (Func<object, List<ParamInfo>>)dm.CreateDelegate(typeof(Func<object, List<ParamInfo>>));
} }
private static IDbCommand SetupCommand(IDbConnection cnn, SqlTransaction tranaction, string sql, List<ParamInfo> paramInfo)
private static IDataReader GetReader<T>(IDbConnection cnn, SqlTransaction tranaction, string sql, List<ParamInfo> paramInfo)
{
using (var cmd = cnn.CreateCommand())
{ {
var cmd = cnn.CreateCommand();
cmd.Transaction = tranaction; cmd.Transaction = tranaction;
cmd.CommandText = sql; cmd.CommandText = sql;
if (paramInfo != null) if (paramInfo != null)
...@@ -282,6 +294,22 @@ private static IDataReader GetReader<T>(IDbConnection cnn, SqlTransaction tranac ...@@ -282,6 +294,22 @@ private static IDataReader GetReader<T>(IDbConnection cnn, SqlTransaction tranac
} }
} }
} }
return cmd;
}
private static int ExecuteCommand(IDbConnection cnn, SqlTransaction tranaction, string sql, List<ParamInfo> paramInfo)
{
using (var cmd = SetupCommand(cnn, tranaction, sql, paramInfo))
{
return cmd.ExecuteNonQuery();
}
}
private static IDataReader GetReader(IDbConnection cnn, SqlTransaction tranaction, string sql, List<ParamInfo> paramInfo)
{
using (var cmd = SetupCommand(cnn, tranaction, sql, paramInfo))
{
return cmd.ExecuteReader(); return cmd.ExecuteReader();
} }
} }
......
...@@ -123,5 +123,17 @@ public void TestStringList() ...@@ -123,5 +123,17 @@ public void TestStringList()
.IsSequenceEqual(new[] {"a","b","c"}); .IsSequenceEqual(new[] {"a","b","c"});
} }
public void TestExecuteCommand()
{
connection.ExecuteMapperCommand(@"
set nocount on
create table #t(i int)
set nocount off
insert #t
select @a a union all select @b
set nocount on
drop table #t", new {a=1, b=2 }).IsEquals(2);
}
} }
} }
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment