Skip to content

Commit

Permalink
Removed Fetch - redundant. Added overload for passing in a connection…
Browse files Browse the repository at this point in the history 
… to Query. Refactored reader assignment to use single method. LOC is same
  • Loading branch information
@subsonic
subsonic committed Apr 5, 2011
1 parent eef91b4 commit 1b4d7c7
Showing 1 changed file with 45 additions and 44 deletions.
89 changes: 45 additions & 44 deletions Massive.cs
View file
Expand Up @@ -33,7 +33,7 @@ public static class ObjectExtensions {
p.Value = item.ToString();
p.DbType = DbType.String;
p.Size = 4000;
}else if(item.GetType()==typeof(ExpandoObject)){
} else if (item.GetType() == typeof(ExpandoObject)) {
var d = (IDictionary<string, object>)item;
p.Value = d.Values.FirstOrDefault();
} else {
Expand All @@ -51,22 +51,25 @@ public static class ObjectExtensions {
public static List<dynamic> ToExpandoList(this IDataReader rdr) {
var result = new List<dynamic>();
while (rdr.Read()) {
dynamic e = new ExpandoObject();
var d = e as IDictionary<string, object>;
for (int i = 0; i < rdr.FieldCount; i++)
d.Add(rdr.GetName(i), rdr[i]);
result.Add(e);
result.Add(rdr.RecordToExpando());
}
return result;
}
public static dynamic RecordToExpando(this IDataReader rdr) {
dynamic e = new ExpandoObject();
var d = e as IDictionary<string, object>;
for (int i = 0; i < rdr.FieldCount; i++)
d.Add(rdr.GetName(i), rdr[i]);
return e;
}
/// <summary>
/// Turns the object into an ExpandoObject
/// </summary>
public static dynamic ToExpando(this object o) {
var result = new ExpandoObject();
var d = result as IDictionary<string, object>; //work with the Expando as a Dictionary
if (o.GetType() == typeof(ExpandoObject)) return o; //shouldn't have to... but just in case
if (o.GetType() == typeof(NameValueCollection) || o.GetType().IsSubclassOf(typeof(NameValueCollection))){
if (o.GetType() == typeof(NameValueCollection) || o.GetType().IsSubclassOf(typeof(NameValueCollection))) {
var nv = (NameValueCollection)o;
nv.Cast<string>().Select(key => new KeyValuePair<string, object>(key, nv[key])).ToList().ForEach(i => d.Add(i));
} else {
Expand All @@ -87,12 +90,12 @@ public static class ObjectExtensions {
/// <summary>
/// A class that wraps your database table in Dynamic Funtime
/// </summary>
public class DynamicModel {
public class DynamicModel {
DbProviderFactory _factory;
string _connectionString;

public DynamicModel(string connectionStringName= "", string tableName = "", string primaryKeyField ="") {
TableName = tableName == "" ? this.GetType().Name : tableName;
public DynamicModel(string connectionStringName = "", string tableName = "", string primaryKeyField = "") {
TableName = tableName == "" ? this.GetType().Name : tableName;
PrimaryKeyField = string.IsNullOrEmpty(primaryKeyField) ? "ID" : primaryKeyField;
if (connectionStringName == "")
connectionStringName = ConfigurationManager.ConnectionStrings[0].Name;
Expand All @@ -111,21 +114,19 @@ public class DynamicModel {
/// </summary>
public virtual IEnumerable<dynamic> Query(string sql, params object[] args) {
using (var conn = OpenConnection()) {
var rdr = CreateCommand(sql, conn, args).ExecuteReader(CommandBehavior.CloseConnection);
var rdr = CreateCommand(sql, conn, args).ExecuteReader();
while (rdr.Read()) {
var e = new ExpandoObject();
var d = e as IDictionary<string, object>;
for (var i = 0; i < rdr.FieldCount; i++)
d.Add(rdr.GetName(i), rdr[i]);
yield return e;
yield return rdr.RecordToExpando(); ;
}
}
}
/// <summary>
/// Runs a query against the database
/// </summary>
public virtual IList<dynamic> Fetch(string sql, params object[] args) {
return Query(sql, args).ToList<dynamic>();
public virtual IEnumerable<dynamic> Query(string sql, DbConnection connection, params object[] args) {
using (var rdr = CreateCommand(sql, connection, args).ExecuteReader()) {
while (rdr.Read()) {
yield return rdr.RecordToExpando(); ;
}
}

}
/// <summary>
/// Returns a single result
Expand All @@ -136,13 +137,12 @@ public class DynamicModel {
result = CreateCommand(sql, conn, args).ExecuteScalar();
}
return result;
}
}
/// <summary>
/// Creates a DBCommand that you can use for loving your database.
/// </summary>
DbCommand CreateCommand(string sql, DbConnection conn, params object[] args) {
DbCommand result = null;
result = _factory.CreateCommand();
var result = _factory.CreateCommand();
result.Connection = conn;
result.CommandText = sql;
if (args.Length > 0)
Expand All @@ -153,10 +153,10 @@ public class DynamicModel {
/// Returns and OpenConnection
/// </summary>
public virtual DbConnection OpenConnection() {
var conn = _factory.CreateConnection();
conn.ConnectionString = _connectionString;
conn.Open();
return conn;
var result = _factory.CreateConnection();
result.ConnectionString = _connectionString;
result.Open();
return result;
}
/// <summary>
/// Builds a set of Insert and Update commands based on the passed-on objects.
Expand All @@ -167,8 +167,8 @@ public class DynamicModel {
var commands = new List<DbCommand>();
foreach (var item in things) {
if (HasPrimaryKey(item)) {
commands.Add(CreateUpdateCommand(item,GetPrimaryKey(item)));
}else{
commands.Add(CreateUpdateCommand(item, GetPrimaryKey(item)));
} else {
commands.Add(CreateInsertCommand(item));
}
}
Expand Down Expand Up @@ -197,7 +197,7 @@ public class DynamicModel {
foreach (var cmd in commands) {
cmd.Connection = conn;
cmd.Transaction = tx;
result+=cmd.ExecuteNonQuery();
result += cmd.ExecuteNonQuery();
}
tx.Commit();
}
Expand Down Expand Up @@ -232,7 +232,7 @@ public class DynamicModel {
var sbKeys = new StringBuilder();
var sbVals = new StringBuilder();
var stub = "INSERT INTO {0} ({1}) \r\n VALUES ({2})";
result = CreateCommand(stub,null);
result = CreateCommand(stub, null);
int counter = 0;
foreach (var item in settings) {
sbKeys.AppendFormat("{0},", item.Key);
Expand All @@ -257,7 +257,7 @@ public class DynamicModel {
var sbKeys = new StringBuilder();
var stub = "UPDATE {0} SET {1} WHERE {2} = @{3}";
var args = new List<object>();
var result = CreateCommand(stub,null);
var result = CreateCommand(stub, null);
int counter = 0;
foreach (var item in settings) {
var val = item.Value;
Expand All @@ -283,10 +283,10 @@ public class DynamicModel {
var sql = string.Format("DELETE FROM {0} ", TableName);
if (key != null) {
sql += string.Format("WHERE {0}=@0", PrimaryKeyField);
args = new object[]{key};
args = new object[] { key };
} else if (!string.IsNullOrEmpty(where)) {
sql += where.Trim().StartsWith("where", StringComparison.CurrentCultureIgnoreCase) ? where : "WHERE " + where;
}
}
return CreateCommand(sql, null, args);
}
/// <summary>
Expand Down Expand Up @@ -315,7 +315,7 @@ public class DynamicModel {
/// Removes one or more records from the DB according to the passed-in WHERE
/// </summary>
public int Delete(object key = null, string where = "", params object[] args) {
return Execute(CreateDeleteCommand(where: where, key:key, args: args));
return Execute(CreateDeleteCommand(where: where, key: key, args: args));
}
/// <summary>
/// Returns all records complying with the passed-in WHERE clause and arguments,
Expand All @@ -327,13 +327,13 @@ public class DynamicModel {
sql += where.Trim().StartsWith("where", StringComparison.CurrentCultureIgnoreCase) ? where : "WHERE " + where;
if (!String.IsNullOrEmpty(orderBy))
sql += orderBy.Trim().StartsWith("order by", StringComparison.CurrentCultureIgnoreCase) ? orderBy : " ORDER BY " + orderBy;
return Query(string.Format(sql, columns,TableName), args);
return Query(string.Format(sql, columns, TableName), args);
}

/// <summary>
/// Returns a dynamic PagedResult. Result properties are Items, TotalPages, and TotalRecords.
/// </summary>
public virtual dynamic Paged(string where = "", string orderBy = "", string columns = "*", int pageSize = 20, int currentPage =1, params object[] args) {
public virtual dynamic Paged(string where = "", string orderBy = "", string columns = "*", int pageSize = 20, int currentPage = 1, params object[] args) {
dynamic result = new ExpandoObject();
var countSQL = string.Format("SELECT COUNT({0}) FROM {1}", PrimaryKeyField, TableName);
if (String.IsNullOrEmpty(orderBy))
Expand All @@ -344,11 +344,11 @@ public class DynamicModel {
where = "WHERE " + where;
}
}
var sql = string.Format("SELECT {0} FROM (SELECT ROW_NUMBER() OVER (ORDER BY {2}) AS Row, {0} FROM {3} {4}) AS Paged ",columns,pageSize,orderBy,TableName, where);
var pageStart = (currentPage -1) * pageSize;
sql+= string.Format(" WHERE Row >={0} AND Row <={1}",pageStart, (pageStart + pageSize));
var sql = string.Format("SELECT {0} FROM (SELECT ROW_NUMBER() OVER (ORDER BY {2}) AS Row, {0} FROM {3} {4}) AS Paged ", columns, pageSize, orderBy, TableName, where);
var pageStart = (currentPage - 1) * pageSize;
sql += string.Format(" WHERE Row >={0} AND Row <={1}", pageStart, (pageStart + pageSize));
countSQL += where;
result.TotalRecords = Scalar(countSQL,args);
result.TotalRecords = Scalar(countSQL, args);
result.TotalPages = result.TotalRecords / pageSize;
if (result.TotalRecords % pageSize > 0)
result.TotalPages += 1;
Expand All @@ -359,8 +359,9 @@ public class DynamicModel {
/// Returns a single row from the database
/// </summary>
public virtual dynamic Single(object key, string columns = "*") {
var sql = string.Format("SELECT {0} FROM {1} WHERE {2} = @0", columns,TableName, PrimaryKeyField);
return Fetch(sql, key).FirstOrDefault();
var sql = string.Format("SELECT {0} FROM {1} WHERE {2} = @0", columns, TableName, PrimaryKeyField);
var items = Query(sql, key).ToList();
return items.FirstOrDefault();
}
}
}

0 comments on commit 1b4d7c7

Please sign in to comment.