Last active
March 14, 2019 21:01
-
-
Save jpann/7138806 to your computer and use it in GitHub Desktop.
SqlUtil class to make querying a SQL Server database easier and more contained.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
using System; | |
using System.Collections.Generic; | |
using System.Linq; | |
using System.Text; | |
using System.Data.SqlClient; | |
namespace Utilities | |
{ | |
public static class SqlUtil | |
{ | |
private static Func<SqlConnection> sqlConnectionFunc; | |
public static void Initialize(Func<SqlConnection> func) | |
{ | |
if (func == null) | |
throw new ArgumentNullException("func"); | |
sqlConnectionFunc = func; | |
} | |
public static T ExecuteScalar<T>(string query) | |
{ | |
return Execute<T>(c => c.ExecuteScalar<T>(query)); | |
} | |
public static int ExecuteNonQuery(string query) | |
{ | |
return Execute<int>(c => c.ExecuteNonQuery(query)); | |
} | |
private static T Execute<T>(Func<SqlConnection, T> func) | |
{ | |
if (sqlConnectionFunc == null) | |
throw new NullReferenceException("SqlUtil must be initialized first."); | |
using (SqlConnection conn = sqlConnectionFunc()) | |
{ | |
conn.Open(); | |
return func(conn); | |
} | |
} | |
public static IEnumerable<T> ExecuteQueryReader<T>(string query, Func<SqlDataReader, T> recordReader) | |
{ | |
if (sqlConnectionFunc == null) | |
throw new NullReferenceException("SqlUtil must be initialized first."); | |
using (SqlConnection conn = sqlConnectionFunc()) | |
{ | |
conn.Open(); | |
using (var command = new SqlCommand(query, conn)) | |
{ | |
using (var reader = command.ExecuteReader()) | |
{ | |
while (reader.Read()) | |
{ | |
yield return recordReader(reader); | |
} | |
} | |
} | |
} | |
} | |
#region SqlConnection Extensions | |
public static T ExecuteScalar<T>(this SqlConnection conn, string query) | |
{ | |
using (var command = new SqlCommand(query, conn)) | |
{ | |
return (T)command.ExecuteScalar(); | |
} | |
} | |
public static int ExecuteNonQuery(this SqlConnection conn, string query) | |
{ | |
using (var command = new SqlCommand(query, conn)) | |
{ | |
return command.ExecuteNonQuery(); | |
} | |
} | |
#endregion | |
} | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment