/ * PetaPoco v4 . 0.2 - A Tiny ORMish thing for your POCO ' s .
* Copyright © 2011 Topten Software . All Rights Reserved .
*
* Apache License 2.0 - http : //www.toptensoftware.com/petapoco/license
*
* Special thanks to Rob Conery ( @robconery ) for original inspiration ( ie : Massive ) and for
* use of Subsonic ' s T4 templates , Rob Sullivan ( @DataChomp ) for hard core DBA advice
* and Adam Schroder ( @schotime ) for lots of suggestions , improvements and Oracle support
* /
#define PETAPOCO_NO_DYNAMIC //in your project settings on .NET 3.5
using System ;
using System.Collections ;
using System.Collections.Generic ;
using System.Configuration ;
using System.Data ;
using System.Data.Common ;
using System.Diagnostics ;
using System.Linq ;
using System.Linq.Expressions ;
using System.Reflection ;
using System.Reflection.Emit ;
using System.Text ;
using System.Text.RegularExpressions ;
using System.Threading ;
using MvcMiniProfiler ;
namespace PetaPoco
{
// Poco's marked [Explicit] require all column properties to be marked
[AttributeUsage(AttributeTargets.Class)]
public class ExplicitColumnsAttribute : Attribute
{
}
// For non-explicit pocos, causes a property to be ignored
[AttributeUsage(AttributeTargets.Property)]
public class IgnoreAttribute : Attribute
{
}
// For explicit pocos, marks property as a column
[AttributeUsage(AttributeTargets.Property)]
public class ColumnAttribute : Attribute
{
public ColumnAttribute ( ) { }
public ColumnAttribute ( string name ) { Name = name ; }
public string Name { get ; set ; }
}
// For explicit pocos, marks property as a column
[AttributeUsage(AttributeTargets.Property)]
public class ResultColumnAttribute : ColumnAttribute
{
public ResultColumnAttribute ( ) { }
public ResultColumnAttribute ( string name ) : base ( name ) { }
}
// Specify the table name of a poco
[AttributeUsage(AttributeTargets.Class)]
public class TableNameAttribute : Attribute
{
public TableNameAttribute ( string tableName )
{
Value = tableName ;
}
public string Value { get ; private set ; }
}
// Specific the primary key of a poco class (and optional sequence name for Oracle)
[AttributeUsage(AttributeTargets.Class)]
public class PrimaryKeyAttribute : Attribute
{
public PrimaryKeyAttribute ( string primaryKey )
{
Value = primaryKey ;
autoIncrement = true ;
}
public string Value { get ; private set ; }
public string sequenceName { get ; set ; }
public bool autoIncrement { get ; set ; }
}
[AttributeUsage(AttributeTargets.Property)]
public class AutoJoinAttribute : Attribute
{
public AutoJoinAttribute ( ) { }
}
[AttributeUsage(AttributeTargets.Property)]
public class VersionColumnAttribute : ColumnAttribute
{
public VersionColumnAttribute ( ) { }
public VersionColumnAttribute ( string name ) : base ( name ) { }
}
// Results from paged request
public class Page < T >
{
public long CurrentPage { get ; set ; }
public long TotalPages { get ; set ; }
public long TotalItems { get ; set ; }
public long ItemsPerPage { get ; set ; }
public List < T > Items { get ; set ; }
public object Context { get ; set ; }
}
// Pass as parameter value to force to DBType.AnsiString
public class AnsiString
{
public AnsiString ( string str )
{
Value = str ;
}
public string Value { get ; private set ; }
}
// Used by IMapper to override table bindings for an object
public class TableInfo
{
public string TableName { get ; set ; }
public string PrimaryKey { get ; set ; }
public bool AutoIncrement { get ; set ; }
public string SequenceName { get ; set ; }
}
// Optionally provide and implementation of this to Database.Mapper
public interface IMapper
{
void GetTableInfo ( Type t , TableInfo ti ) ;
bool MapPropertyToColumn ( PropertyInfo pi , ref string columnName , ref bool resultColumn ) ;
Func < object , object > GetFromDbConverter ( DestinationInfo destinationInfo , Type SourceType ) ;
Func < object , object > GetToDbConverter ( Type SourceType ) ;
}
public class DefaultMapper : IMapper
{
public virtual void GetTableInfo ( Type t , TableInfo ti ) { }
public virtual bool MapPropertyToColumn ( PropertyInfo pi , ref string columnName , ref bool resultColumn )
{
return true ;
}
public virtual Func < object , object > GetFromDbConverter ( DestinationInfo destinationInfo , Type SourceType )
{
return null ;
}
public virtual Func < object , object > GetToDbConverter ( Type SourceType )
{
return null ;
}
}
public class DestinationInfo
{
public DestinationInfo ( Type type )
{
Type = type ;
}
public DestinationInfo ( PropertyInfo propertyInfo )
{
PropertyInfo = propertyInfo ;
Type = propertyInfo . PropertyType ;
}
public PropertyInfo PropertyInfo { get ; private set ; }
public Type Type { get ; private set ; }
}
public interface IDatabaseQuery
{
void OpenSharedConnection ( ) ;
void CloseSharedConnection ( ) ;
int Execute ( string sql , params object [ ] args ) ;
int Execute ( Sql sql ) ;
T ExecuteScalar < T > ( string sql , params object [ ] args ) ;
T ExecuteScalar < T > ( Sql sql ) ;
List < T > Fetch < T > ( ) ;
List < T > Fetch < T > ( string sql , params object [ ] args ) ;
List < T > Fetch < T > ( Sql sql ) ;
List < T > Fetch < T > ( long page , long itemsPerPage , string sql , params object [ ] args ) ;
List < T > Fetch < T > ( long page , long itemsPerPage , Sql sql ) ;
Page < T > Page < T > ( long page , long itemsPerPage , string sql , params object [ ] args ) ;
Page < T > Page < T > ( long page , long itemsPerPage , Sql sql ) ;
List < T > SkipTake < T > ( long skip , long take , string sql , params object [ ] args ) ;
List < T > SkipTake < T > ( long skip , long take , Sql sql ) ;
List < TRet > Fetch < T1 , T2 , TRet > ( Func < T1 , T2 , TRet > cb , string sql , params object [ ] args ) ;
List < TRet > Fetch < T1 , T2 , T3 , TRet > ( Func < T1 , T2 , T3 , TRet > cb , string sql , params object [ ] args ) ;
List < TRet > Fetch < T1 , T2 , T3 , T4 , TRet > ( Func < T1 , T2 , T3 , T4 , TRet > cb , string sql , params object [ ] args ) ;
IEnumerable < TRet > Query < T1 , T2 , TRet > ( Func < T1 , T2 , TRet > cb , string sql , params object [ ] args ) ;
IEnumerable < TRet > Query < T1 , T2 , T3 , TRet > ( Func < T1 , T2 , T3 , TRet > cb , string sql , params object [ ] args ) ;
IEnumerable < TRet > Query < T1 , T2 , T3 , T4 , TRet > ( Func < T1 , T2 , T3 , T4 , TRet > cb , string sql , params object [ ] args ) ;
List < TRet > Fetch < T1 , T2 , TRet > ( Func < T1 , T2 , TRet > cb , Sql sql ) ;
List < TRet > Fetch < T1 , T2 , T3 , TRet > ( Func < T1 , T2 , T3 , TRet > cb , Sql sql ) ;
List < TRet > Fetch < T1 , T2 , T3 , T4 , TRet > ( Func < T1 , T2 , T3 , T4 , TRet > cb , Sql sql ) ;
IEnumerable < TRet > Query < T1 , T2 , TRet > ( Func < T1 , T2 , TRet > cb , Sql sql ) ;
IEnumerable < TRet > Query < T1 , T2 , T3 , TRet > ( Func < T1 , T2 , T3 , TRet > cb , Sql sql ) ;
IEnumerable < TRet > Query < T1 , T2 , T3 , T4 , TRet > ( Func < T1 , T2 , T3 , T4 , TRet > cb , Sql sql ) ;
List < T1 > Fetch < T1 , T2 > ( string sql , params object [ ] args ) ;
List < T1 > Fetch < T1 , T2 , T3 > ( string sql , params object [ ] args ) ;
List < T1 > Fetch < T1 , T2 , T3 , T4 > ( string sql , params object [ ] args ) ;
IEnumerable < T1 > Query < T1 , T2 > ( string sql , params object [ ] args ) ;
IEnumerable < T1 > Query < T1 , T2 , T3 > ( string sql , params object [ ] args ) ;
IEnumerable < T1 > Query < T1 , T2 , T3 , T4 > ( string sql , params object [ ] args ) ;
IEnumerable < TRet > Query < TRet > ( Type [ ] types , object cb , string sql , params object [ ] args ) ;
List < T1 > Fetch < T1 , T2 > ( Sql sql ) ;
List < T1 > Fetch < T1 , T2 , T3 > ( Sql sql ) ;
List < T1 > Fetch < T1 , T2 , T3 , T4 > ( Sql sql ) ;
IEnumerable < T1 > Query < T1 , T2 > ( Sql sql ) ;
IEnumerable < T1 > Query < T1 , T2 , T3 > ( Sql sql ) ;
IEnumerable < T1 > Query < T1 , T2 , T3 , T4 > ( Sql sql ) ;
IEnumerable < T > Query < T > ( string sql , params object [ ] args ) ;
IEnumerable < T > Query < T > ( Sql sql ) ;
T Single < T > ( object primaryKey ) ;
T SingleOrDefault < T > ( object primaryKey ) ;
T Single < T > ( string sql , params object [ ] args ) ;
T SingleOrDefault < T > ( string sql , params object [ ] args ) ;
T First < T > ( string sql , params object [ ] args ) ;
T FirstOrDefault < T > ( string sql , params object [ ] args ) ;
T Single < T > ( Sql sql ) ;
T SingleOrDefault < T > ( Sql sql ) ;
T First < T > ( Sql sql ) ;
T FirstOrDefault < T > ( Sql sql ) ;
bool Exists < T > ( object primaryKey ) ;
int OneTimeCommandTimeout { get ; set ; }
bool Exists < T > ( string sql , params object [ ] args ) ;
}
public interface IDatabase : IDatabaseQuery
{
void Dispose ( ) ;
IDbConnection Connection { get ; }
ITransaction GetTransaction ( ) ;
void BeginTransaction ( ) ;
void AbortTransaction ( ) ;
void CompleteTransaction ( ) ;
object Insert ( string tableName , string primaryKeyName , bool autoIncrement , object poco ) ;
object Insert ( string tableName , string primaryKeyName , object poco ) ;
object Insert ( object poco ) ;
int Update ( string tableName , string primaryKeyName , object poco , object primaryKeyValue ) ;
int Update ( string tableName , string primaryKeyName , object poco ) ;
int Update ( object poco ) ;
int Update ( object poco , object primaryKeyValue ) ;
int Update < T > ( string sql , params object [ ] args ) ;
int Update < T > ( Sql sql ) ;
void UpdateMany < T > ( IEnumerable < T > pocoList ) ;
int Delete ( string tableName , string primaryKeyName , object poco ) ;
int Delete ( string tableName , string primaryKeyName , object poco , object primaryKeyValue ) ;
int Delete ( object poco ) ;
int Delete < T > ( string sql , params object [ ] args ) ;
int Delete < T > ( Sql sql ) ;
int Delete < T > ( object pocoOrPrimaryKey ) ;
void Save ( string tableName , string primaryKeyName , object poco ) ;
void Save ( object poco ) ;
void InsertMany < T > ( IEnumerable < T > pocoList ) ;
void SaveMany < T > ( IEnumerable < T > pocoList ) ;
}
// Database class ... this is where most of the action happens
public class Database : IDisposable , IDatabase
{
public const string MsSqlClientProvider = "System.Data.SqlClient" ;
public Database ( IDbConnection connection )
{
_sharedConnection = connection ;
_connectionString = connection . ConnectionString ;
_sharedConnectionDepth = 2 ; // Prevent closing external connection
CommonConstruct ( ) ;
}
public Database ( string connectionString , string providerName )
{
_connectionString = connectionString ;
_providerName = providerName ;
CommonConstruct ( ) ;
}
public Database ( string connectionString , DbProviderFactory provider )
{
_connectionString = connectionString ;
_factory = provider ;
CommonConstruct ( ) ;
}
public Database ( string connectionStringName )
{
// Use first?
if ( connectionStringName = = "" )
connectionStringName = ConfigurationManager . ConnectionStrings [ 0 ] . Name ;
// Work out connection string and provider name
var providerName = "System.Data.SqlClient" ;
if ( ConfigurationManager . ConnectionStrings [ connectionStringName ] ! = null )
{
if ( ! string . IsNullOrEmpty ( ConfigurationManager . ConnectionStrings [ connectionStringName ] . ProviderName ) )
providerName = ConfigurationManager . ConnectionStrings [ connectionStringName ] . ProviderName ;
}
else
{
throw new InvalidOperationException ( "Can't find a connection string with the name '" + connectionStringName + "'" ) ;
}
// Store factory and connection string
_connectionString = ConfigurationManager . ConnectionStrings [ connectionStringName ] . ConnectionString ;
_providerName = providerName ;
CommonConstruct ( ) ;
}
enum DBType
{
SqlServer ,
SqlServerCE ,
MySql ,
PostgreSQL ,
Oracle ,
SQLite
}
DBType _dbType = DBType . SqlServerCE ;
// Common initialization
private void CommonConstruct ( )
{
_transactionDepth = 0 ;
ForceDateTimesToUtc = true ;
EnableAutoSelect = true ;
if ( _providerName ! = null )
_factory = DbProviderFactories . GetFactory ( _providerName ) ;
string dbtype = ( _factory = = null ? _sharedConnection . GetType ( ) : _factory . GetType ( ) ) . Name ;
if ( dbtype . StartsWith ( "MySql" ) ) _dbType = DBType . MySql ;
else if ( dbtype . StartsWith ( "SqlCe" ) ) _dbType = DBType . SqlServerCE ;
else if ( dbtype . StartsWith ( "Npgsql" ) ) _dbType = DBType . PostgreSQL ;
else if ( dbtype . StartsWith ( "Oracle" ) ) _dbType = DBType . Oracle ;
else if ( dbtype . StartsWith ( "SQLite" ) ) _dbType = DBType . SQLite ;
if ( _dbType = = DBType . MySql & & _connectionString ! = null & & _connectionString . IndexOf ( "Allow User Variables=true" ) > = 0 )
_paramPrefix = "?" ;
if ( _dbType = = DBType . Oracle )
_paramPrefix = ":" ;
}
// Automatically close one open shared connection
public void Dispose ( )
{
// Automatically close one open connection reference
// (Works with KeepConnectionAlive and manually opening a shared connection)
CloseSharedConnection ( ) ;
}
// Set to true to keep the first opened connection alive until this object is disposed
public bool KeepConnectionAlive { get ; set ; }
// Open a connection (can be nested)
public void OpenSharedConnection ( )
{
using ( MvcMiniProfiler . MiniProfiler . StepStatic ( "OpenSharedConnection" ) )
{
if ( _sharedConnectionDepth = = 0 )
{
_sharedConnection = _factory . CreateConnection ( ) ;
_sharedConnection . ConnectionString = _connectionString ;
_sharedConnection . Open ( ) ;
if ( KeepConnectionAlive )
_sharedConnectionDepth + + ; // Make sure you call Dispose
}
_sharedConnectionDepth + + ;
}
}
/// <summary>
/// Close a previously opened connection
/// </summary>
public void CloseSharedConnection ( )
{
if ( _sharedConnectionDepth > 0 )
{
_sharedConnectionDepth - - ;
if ( _sharedConnectionDepth = = 0 )
{
_sharedConnection . Dispose ( ) ;
_sharedConnection = null ;
}
}
}
// Access to our shared connection
public IDbConnection Connection
{
get { return _sharedConnection ; }
}
// Helper to create a transaction scope
public ITransaction GetTransaction ( )
{
return new Transaction ( this ) ;
}
// Use by derived repo generated by T4 templates
public virtual void OnBeginTransaction ( ) { }
public virtual void OnEndTransaction ( ) { }
// Start a new transaction, can be nested, every call must be
// matched by a call to AbortTransaction or CompleteTransaction
// Use `using (var scope=db.Transaction) { scope.Complete(); }` to ensure correct semantics
public void BeginTransaction ( )
{
_transactionDepth + + ;
if ( _transactionDepth = = 1 )
{
OpenSharedConnection ( ) ;
_transaction = _sharedConnection . BeginTransaction ( ) ;
_transactionCancelled = false ;
OnBeginTransaction ( ) ;
}
}
// Internal helper to cleanup transaction stuff
void CleanupTransaction ( )
{
OnEndTransaction ( ) ;
if ( _transactionCancelled )
_transaction . Rollback ( ) ;
else
_transaction . Commit ( ) ;
_transaction . Dispose ( ) ;
_transaction = null ;
CloseSharedConnection ( ) ;
}
// Abort the entire outer most transaction scope
public void AbortTransaction ( )
{
_transactionCancelled = true ;
if ( ( - - _transactionDepth ) = = 0 )
CleanupTransaction ( ) ;
}
// Complete the transaction
public void CompleteTransaction ( )
{
if ( ( - - _transactionDepth ) = = 0 )
CleanupTransaction ( ) ;
}
// Helper to handle named parameters from object properties
static Regex rxParams = new Regex ( @"(?<!@)@\w+" , RegexOptions . Compiled ) ;
public static string ProcessParams ( string _sql , object [ ] args_src , List < object > args_dest )
{
return rxParams . Replace ( _sql , m = >
{
string param = m . Value . Substring ( 1 ) ;
object arg_val ;
int paramIndex ;
if ( int . TryParse ( param , out paramIndex ) )
{
// Numbered parameter
if ( paramIndex < 0 | | paramIndex > = args_src . Length )
throw new ArgumentOutOfRangeException ( string . Format ( "Parameter '@{0}' specified but only {1} parameters supplied (in `{2}`)" , paramIndex , args_src . Length , _sql ) ) ;
arg_val = args_src [ paramIndex ] ;
}
else
{
// Look for a property on one of the arguments with this name
bool found = false ;
arg_val = null ;
foreach ( var o in args_src )
{
var pi = o . GetType ( ) . GetProperty ( param ) ;
if ( pi ! = null )
{
arg_val = pi . GetValue ( o , null ) ;
found = true ;
break ;
}
}
if ( ! found )
throw new ArgumentException ( string . Format ( "Parameter '@{0}' specified but none of the passed arguments have a property with this name (in '{1}')" , param , _sql ) ) ;
}
// Expand collections to parameter lists
if ( ( arg_val as IEnumerable ) ! = null & &
( arg_val as string ) = = null & &
( arg_val as byte [ ] ) = = null )
{
var sb = new StringBuilder ( ) ;
foreach ( var i in arg_val as IEnumerable )
{
sb . Append ( ( sb . Length = = 0 ? "@" : ",@" ) + args_dest . Count . ToString ( ) ) ;
args_dest . Add ( i ) ;
}
return sb . ToString ( ) ;
}
else
{
args_dest . Add ( arg_val ) ;
return "@" + ( args_dest . Count - 1 ) . ToString ( ) ;
}
}
) ;
}
// Add a parameter to a DB command
void AddParam ( IDbCommand cmd , object item , string ParameterPrefix )
{
// Convert value to from poco type to db type
if ( Mapper ! = null & & item ! = null )
{
var fn = Mapper . GetToDbConverter ( item . GetType ( ) ) ;
if ( fn ! = null )
item = fn ( item ) ;
}
// Support passed in parameters
var idbParam = item as IDbDataParameter ;
if ( idbParam ! = null )
{
idbParam . ParameterName = string . Format ( "{0}{1}" , ParameterPrefix , cmd . Parameters . Count ) ;
cmd . Parameters . Add ( idbParam ) ;
return ;
}
var p = cmd . CreateParameter ( ) ;
p . ParameterName = string . Format ( "{0}{1}" , ParameterPrefix , cmd . Parameters . Count ) ;
if ( item = = null )
{
p . Value = DBNull . Value ;
}
else
{
var t = item . GetType ( ) ;
if ( t . IsEnum ) // PostgreSQL .NET driver wont cast enum to int
{
p . Value = ( int ) item ;
}
else if ( t = = typeof ( Guid ) )
{
p . Value = item . ToString ( ) ;
p . DbType = DbType . String ;
p . Size = 40 ;
}
else if ( t = = typeof ( string ) )
{
p . Size = Math . Max ( ( item as string ) . Length + 1 , 4000 ) ; // Help query plan caching by using common size
p . Value = item ;
}
else if ( t = = typeof ( AnsiString ) )
{
// Thanks @DataChomp for pointing out the SQL Server indexing performance hit of using wrong string type on varchar
p . Size = Math . Max ( ( item as AnsiString ) . Value . Length + 1 , 4000 ) ;
p . Value = ( item as AnsiString ) . Value ;
p . DbType = DbType . AnsiString ;
}
else if ( t = = typeof ( bool ) & & _dbType ! = DBType . PostgreSQL )
{
p . Value = ( ( bool ) item ) ? 1 : 0 ;
}
else if ( item . GetType ( ) . Name = = "SqlGeography" ) //SqlGeography is a CLR Type
{
p . GetType ( ) . GetProperty ( "UdtTypeName" ) . SetValue ( p , "geography" , null ) ; //geography is the equivalent SQL Server Type
p . Value = item ;
}
else if ( item . GetType ( ) . Name = = "SqlGeometry" ) //SqlGeometry is a CLR Type
{
p . GetType ( ) . GetProperty ( "UdtTypeName" ) . SetValue ( p , "geometry" , null ) ; //geography is the equivalent SQL Server Type
p . Value = item ;
}
else
{
p . Value = item ;
}
}
cmd . Parameters . Add ( p ) ;
}
// Create a command
static Regex rxParamsPrefix = new Regex ( @"(?<!@)@\w+" , RegexOptions . Compiled ) ;
public IDbCommand CreateCommand ( IDbConnection connection , Sql sqlStatement )
{
var sql = sqlStatement . SQL ;
var args = sqlStatement . Arguments ;
// Perform parameter prefix replacements
if ( _paramPrefix ! = "@" )
sql = rxParamsPrefix . Replace ( sql , m = > _paramPrefix + m . Value . Substring ( 1 ) ) ;
sql = sql . Replace ( "@@" , "@" ) ; // <- double @@ escapes a single @
// Create the command and add parameters
IDbCommand cmd = _factory = = null ? connection . CreateCommand ( ) : _factory . CreateCommand ( ) ;
cmd . Connection = connection ;
cmd . CommandText = sql ;
cmd . Transaction = _transaction ;
foreach ( var item in args )
{
AddParam ( cmd , item , _paramPrefix ) ;
}
if ( _dbType = = DBType . Oracle )
{
cmd . GetType ( ) . GetProperty ( "BindByName" ) . SetValue ( cmd , true , null ) ;
}
if ( ! String . IsNullOrEmpty ( sql ) )
DoPreExecute ( cmd ) ;
return cmd ;
}
// Create a command
IDbCommand CreateCommand ( IDbConnection connection , string sql , params object [ ] args )
{
var sqlStatement = new Sql ( sql , args ) ;
return CreateCommand ( connection , sqlStatement ) ;
}
// Override this to log/capture exceptions
public virtual void OnException ( Exception x )
{
Debug . WriteLine ( x . ToString ( ) ) ;
Debug . WriteLine ( LastCommand ) ;
}
// Override this to log commands, or modify command before execution
public virtual void OnExecutingCommand ( IDbCommand cmd ) { }
public virtual void OnExecutedCommand ( IDbCommand cmd ) { }
// Execute a non-query command
public int Execute ( string sql , params object [ ] args )
{
return Execute ( new Sql ( sql , args ) ) ;
}
public int Execute ( Sql sql )
{
using ( MiniProfiler . StepStatic ( "Peta Execute SQL" ) )
{
try
{
OpenSharedConnection ( ) ;
try
{
using ( var cmd = CreateCommand ( _sharedConnection , sql ) )
{
var result = cmd . ExecuteNonQuery ( ) ;
OnExecutedCommand ( cmd ) ;
return result ;
}
}
finally
{
CloseSharedConnection ( ) ;
}
}
catch ( Exception x )
{
OnException ( x ) ;
throw ;
}
}
}
// Execute and cast a scalar property
public T ExecuteScalar < T > ( string sql , params object [ ] args )
{
return ExecuteScalar < T > ( new Sql ( sql , args ) ) ;
}
public T ExecuteScalar < T > ( Sql sql )
{
using ( MiniProfiler . StepStatic ( "Peta ExecuteScalar<T>" ) )
{
try
{
OpenSharedConnection ( ) ;
try
{
using ( var cmd = CreateCommand ( _sharedConnection , sql ) )
{
object val = cmd . ExecuteScalar ( ) ;
OnExecutedCommand ( cmd ) ;
return ( T ) Convert . ChangeType ( val , typeof ( T ) ) ;
}
}
finally
{
CloseSharedConnection ( ) ;
}
}
catch ( Exception x )
{
OnException ( x ) ;
throw ;
}
}
}
Regex rxSelect = new Regex ( @"\A\s*(SELECT|EXECUTE|CALL)\s" , RegexOptions . Compiled | RegexOptions . Singleline | RegexOptions . IgnoreCase | RegexOptions . Multiline ) ;
Regex rxFrom = new Regex ( @"\A\s*FROM\s" , RegexOptions . Compiled | RegexOptions . Singleline | RegexOptions . IgnoreCase | RegexOptions . Multiline ) ;
string AddSelectClause < T > ( string sql )
{
if ( sql . StartsWith ( ";" ) )
return sql . Substring ( 1 ) ;
if ( ! rxSelect . IsMatch ( sql ) )
{
var pd = PocoData . ForType ( typeof ( T ) ) ;
var tableName = EscapeTableName ( pd . TableInfo . TableName ) ;
string cols = string . Join ( ", " , ( from c in pd . QueryColumns select EscapeSqlIdentifier ( c ) ) . ToArray ( ) ) ;
if ( ! rxFrom . IsMatch ( sql ) )
sql = string . Format ( "SELECT {0} FROM {1} {2}" , cols , tableName , sql ) ;
else
sql = string . Format ( "SELECT {0} {1}" , cols , sql ) ;
}
return sql ;
}
public bool ForceDateTimesToUtc { get ; set ; }
public bool EnableAutoSelect { get ; set ; }
// Return a typed list of pocos
public List < T > Fetch < T > ( string sql , params object [ ] args )
{
if ( EnableAutoSelect )
sql = AddSelectClause < T > ( sql ) ;
return Fetch < T > ( new Sql ( sql , args ) ) ;
}
public List < T > Fetch < T > ( Sql sql )
{
return Query < T > ( sql ) . ToList ( ) ;
}
public List < T > Fetch < T > ( long page , long itemsPerPage , string sql , params object [ ] args )
{
string sqlCount , sqlPage ;
BuildPageQueries < T > ( page , itemsPerPage , sql , ref args , out sqlCount , out sqlPage ) ;
return Fetch < T > ( sqlPage , args ) ;
}
public List < T > Fetch < T > ( long page , long itemsPerPage , Sql sql )
{
return Fetch < T > ( page , itemsPerPage , sql . SQL , sql . Arguments ) ;
}
public List < T > Fetch < T > ( )
{
return Fetch < T > ( AddSelectClause < T > ( "" ) ) ;
}
static Regex rxColumns = new Regex ( @"\A\s*SELECT\s+((?:\((?>\((?<depth>)|\)(?<-depth>)|.?)*(?(depth)(?!))\)|.)*?)(?<!,\s+)\bFROM\b" , RegexOptions . IgnoreCase | RegexOptions . Multiline | RegexOptions . Singleline | RegexOptions . Compiled ) ;
static Regex rxOrderBy = new Regex ( @"\bORDER\s+BY\s+(?:\((?>\((?<depth>)|\)(?<-depth>)|.?)*(?(depth)(?!))\)|[\w\(\)\.])+(?:\s+(?:ASC|DESC))?(?:\s*,\s*(?:\((?>\((?<depth>)|\)(?<-depth>)|.?)*(?(depth)(?!))\)|[\w\(\)\.])+(?:\s+(?:ASC|DESC))?)*" , RegexOptions . IgnoreCase | RegexOptions . Multiline | RegexOptions . Singleline | RegexOptions . Compiled ) ;
public static bool SplitSqlForPaging < T > ( string sql , out string sqlCount , out string sqlSelectRemoved , out string sqlOrderBy )
{
sqlSelectRemoved = null ;
sqlCount = null ;
sqlOrderBy = null ;
// Extract the columns from "SELECT <whatever> FROM"
var m = rxColumns . Match ( sql ) ;
if ( ! m . Success )
return false ;
// Save column list and replace with COUNT(*)
Group g = m . Groups [ 1 ] ;
sqlCount = sql . Substring ( 0 , g . Index ) + "COUNT(*) " + sql . Substring ( g . Index + g . Length ) ;
sqlSelectRemoved = sql . Substring ( g . Index ) ;
// Look for an "ORDER BY <whatever>" clause or primarykey from pocodata
var data = PocoData . ForType ( typeof ( T ) ) ;
m = rxOrderBy . Match ( sqlCount ) ;
if ( ! m . Success
& & ( string . IsNullOrEmpty ( data . TableInfo . PrimaryKey ) | |
( ! data . TableInfo . PrimaryKey . Split ( ',' ) . All ( x = > data . Columns . Values . Any ( y = > y . ColumnName . Equals ( x , StringComparison . OrdinalIgnoreCase ) ) ) ) ) )
{
return false ;
}
g = m . Groups [ 0 ] ;
sqlOrderBy = m . Success ? g . ToString ( ) : "ORDER BY " + data . TableInfo . PrimaryKey ;
sqlCount = sqlCount . Substring ( 0 , g . Index ) + sqlCount . Substring ( g . Index + g . Length ) ;
return true ;
}
private void BuildPageQueries < T > ( long skip , long take , string sql , ref object [ ] args , out string sqlCount , out string sqlPage )
{
// Add auto select clause
sql = AddSelectClause < T > ( sql ) ;
// Split the SQL into the bits we need
string sqlSelectRemoved , sqlOrderBy ;
if ( ! SplitSqlForPaging < T > ( sql , out sqlCount , out sqlSelectRemoved , out sqlOrderBy ) )
throw new Exception ( "Unable to parse SQL statement for paged query" ) ;
if ( _dbType = = DBType . Oracle & & sqlSelectRemoved . StartsWith ( "*" ) )
throw new Exception ( "Query must alias '*' when performing a paged query.\neg. select t.* from table t order by t.id" ) ;
// Build the SQL for the actual final result
if ( _dbType = = DBType . SqlServer | | _dbType = = DBType . Oracle )
{
var fromIndex = sqlSelectRemoved . IndexOf ( "from" , StringComparison . OrdinalIgnoreCase ) ;
sqlSelectRemoved = rxOrderBy . Replace ( sqlSelectRemoved , "" ) ;
sqlPage = string . Format ( "SELECT * FROM (SELECT {2}, ROW_NUMBER() OVER ({0}) peta_rn {1}) peta_paged WHERE peta_rn>@{3} AND peta_rn<=@{4}" ,
sqlOrderBy , sqlSelectRemoved . Substring ( fromIndex ) , sqlSelectRemoved . Substring ( 0 , fromIndex - 1 ) , args . Length , args . Length + 1 ) ;
args = args . Concat ( new object [ ] { skip , skip + take } ) . ToArray ( ) ;
}
else if ( _dbType = = DBType . SqlServerCE )
{
sqlPage = string . Format ( "{0}\nOFFSET @{1} ROWS FETCH NEXT @{2} ROWS ONLY" , sql , args . Length , args . Length + 1 ) ;
args = args . Concat ( new object [ ] { skip , take } ) . ToArray ( ) ;
}
else
{
sqlPage = string . Format ( "{0}\nLIMIT @{1} OFFSET @{2}" , sql , args . Length , args . Length + 1 ) ;
args = args . Concat ( new object [ ] { take , skip } ) . ToArray ( ) ;
}
}
// Fetch a page
public Page < T > Page < T > ( long page , long itemsPerPage , string sql , params object [ ] args )
{
string sqlCount , sqlPage ;
long skip = ( page - 1 ) * itemsPerPage ;
long take = itemsPerPage ;
BuildPageQueries < T > ( skip , take , sql , ref args , out sqlCount , out sqlPage ) ;
// Save the one-time command time out and use it for both queries
int saveTimeout = OneTimeCommandTimeout ;
// Setup the paged result
var result = new Page < T > ( ) ;
result . CurrentPage = page ;
result . ItemsPerPage = itemsPerPage ;
result . TotalItems = ExecuteScalar < long > ( sqlCount , args ) ;
result . TotalPages = result . TotalItems / itemsPerPage ;
if ( ( result . TotalItems % itemsPerPage ) ! = 0 )
result . TotalPages + + ;
OneTimeCommandTimeout = saveTimeout ;
// Get the records
result . Items = Fetch < T > ( sqlPage , args ) ;
// Done
return result ;
}
public Page < T > Page < T > ( long page , long itemsPerPage , Sql sql )
{
return Page < T > ( page , itemsPerPage , sql . SQL , sql . Arguments ) ;
}
public List < T > SkipTake < T > ( long skip , long take , string sql , params object [ ] args )
{
string sqlCount , sqlPage ;
BuildPageQueries < T > ( skip , take , sql , ref args , out sqlCount , out sqlPage ) ;
var result = Fetch < T > ( sqlPage , args ) ;
return result ;
}
public List < T > SkipTake < T > ( long skip , long take , Sql sql )
{
return SkipTake < T > ( skip , take , sql . SQL , sql . Arguments ) ;
}
// Return an enumerable collection of pocos
public IEnumerable < T > Query < T > ( string sql , params object [ ] args )
{
if ( EnableAutoSelect )
sql = AddSelectClause < T > ( sql ) ;
return Query < T > ( new Sql ( sql , args ) ) ;
}
public IEnumerable < T > Query < T > ( Sql sql )
{
using ( MiniProfiler . StepStatic ( "Peta Query SQL" ) )
{
OpenSharedConnection ( ) ;
try
{
using ( var cmd = CreateCommand ( _sharedConnection , sql ) )
{
IDataReader r ;
var pd = PocoData . ForType ( typeof ( T ) ) ;
try
{
r = cmd . ExecuteReader ( ) ;
OnExecutedCommand ( cmd ) ;
}
catch ( Exception x )
{
OnException ( x ) ;
throw ;
}
using ( r )
{
var factory =
pd . GetFactory ( cmd . CommandText , _sharedConnection . ConnectionString , ForceDateTimesToUtc , 0 , r . FieldCount , r )
as Func < IDataReader , T > ;
while ( true )
{
T poco ;
try
{
if ( ! r . Read ( ) )
yield break ;
poco = factory ( r ) ;
}
catch ( Exception x )
{
OnException ( x ) ;
throw ;
}
yield return poco ;
}
}
}
}
finally
{
CloseSharedConnection ( ) ;
}
}
}
// Multi Fetch
public List < TRet > Fetch < T1 , T2 , TRet > ( Func < T1 , T2 , TRet > cb , string sql , params object [ ] args ) { return Query < T1 , T2 , TRet > ( cb , sql , args ) . ToList ( ) ; }
public List < TRet > Fetch < T1 , T2 , T3 , TRet > ( Func < T1 , T2 , T3 , TRet > cb , string sql , params object [ ] args ) { return Query < T1 , T2 , T3 , TRet > ( cb , sql , args ) . ToList ( ) ; }
public List < TRet > Fetch < T1 , T2 , T3 , T4 , TRet > ( Func < T1 , T2 , T3 , T4 , TRet > cb , string sql , params object [ ] args ) { return Query < T1 , T2 , T3 , T4 , TRet > ( cb , sql , args ) . ToList ( ) ; }
// Multi Query
public IEnumerable < TRet > Query < T1 , T2 , TRet > ( Func < T1 , T2 , TRet > cb , string sql , params object [ ] args ) { return Query < TRet > ( new Type [ ] { typeof ( T1 ) , typeof ( T2 ) } , cb , sql , args ) ; }
public IEnumerable < TRet > Query < T1 , T2 , T3 , TRet > ( Func < T1 , T2 , T3 , TRet > cb , string sql , params object [ ] args ) { return Query < TRet > ( new Type [ ] { typeof ( T1 ) , typeof ( T2 ) , typeof ( T3 ) } , cb , sql , args ) ; }
public IEnumerable < TRet > Query < T1 , T2 , T3 , T4 , TRet > ( Func < T1 , T2 , T3 , T4 , TRet > cb , string sql , params object [ ] args ) { return Query < TRet > ( new Type [ ] { typeof ( T1 ) , typeof ( T2 ) , typeof ( T3 ) , typeof ( T4 ) } , cb , sql , args ) ; }
// Multi Fetch (SQL builder)
public List < TRet > Fetch < T1 , T2 , TRet > ( Func < T1 , T2 , TRet > cb , Sql sql ) { return Query < T1 , T2 , TRet > ( cb , sql . SQL , sql . Arguments ) . ToList ( ) ; }
public List < TRet > Fetch < T1 , T2 , T3 , TRet > ( Func < T1 , T2 , T3 , TRet > cb , Sql sql ) { return Query < T1 , T2 , T3 , TRet > ( cb , sql . SQL , sql . Arguments ) . ToList ( ) ; }
public List < TRet > Fetch < T1 , T2 , T3 , T4 , TRet > ( Func < T1 , T2 , T3 , T4 , TRet > cb , Sql sql ) { return Query < T1 , T2 , T3 , T4 , TRet > ( cb , sql . SQL , sql . Arguments ) . ToList ( ) ; }
// Multi Query (SQL builder)
public IEnumerable < TRet > Query < T1 , T2 , TRet > ( Func < T1 , T2 , TRet > cb , Sql sql ) { return Query < TRet > ( new Type [ ] { typeof ( T1 ) , typeof ( T2 ) } , cb , sql . SQL , sql . Arguments ) ; }
public IEnumerable < TRet > Query < T1 , T2 , T3 , TRet > ( Func < T1 , T2 , T3 , TRet > cb , Sql sql ) { return Query < TRet > ( new Type [ ] { typeof ( T1 ) , typeof ( T2 ) , typeof ( T3 ) } , cb , sql . SQL , sql . Arguments ) ; }
public IEnumerable < TRet > Query < T1 , T2 , T3 , T4 , TRet > ( Func < T1 , T2 , T3 , T4 , TRet > cb , Sql sql ) { return Query < TRet > ( new Type [ ] { typeof ( T1 ) , typeof ( T2 ) , typeof ( T3 ) , typeof ( T4 ) } , cb , sql . SQL , sql . Arguments ) ; }
// Multi Fetch (Simple)
public List < T1 > Fetch < T1 , T2 > ( string sql , params object [ ] args ) { return Query < T1 , T2 > ( sql , args ) . ToList ( ) ; }
public List < T1 > Fetch < T1 , T2 , T3 > ( string sql , params object [ ] args ) { return Query < T1 , T2 , T3 > ( sql , args ) . ToList ( ) ; }
public List < T1 > Fetch < T1 , T2 , T3 , T4 > ( string sql , params object [ ] args ) { return Query < T1 , T2 , T3 , T4 > ( sql , args ) . ToList ( ) ; }
// Multi Query (Simple)
public IEnumerable < T1 > Query < T1 , T2 > ( string sql , params object [ ] args ) { return Query < T1 > ( new Type [ ] { typeof ( T1 ) , typeof ( T2 ) } , null , sql , args ) ; }
public IEnumerable < T1 > Query < T1 , T2 , T3 > ( string sql , params object [ ] args ) { return Query < T1 > ( new Type [ ] { typeof ( T1 ) , typeof ( T2 ) , typeof ( T3 ) } , null , sql , args ) ; }
public IEnumerable < T1 > Query < T1 , T2 , T3 , T4 > ( string sql , params object [ ] args ) { return Query < T1 > ( new Type [ ] { typeof ( T1 ) , typeof ( T2 ) , typeof ( T3 ) , typeof ( T4 ) } , null , sql , args ) ; }
// Multi Fetch (Simple) (SQL builder)
public List < T1 > Fetch < T1 , T2 > ( Sql sql ) { return Query < T1 , T2 > ( sql . SQL , sql . Arguments ) . ToList ( ) ; }
public List < T1 > Fetch < T1 , T2 , T3 > ( Sql sql ) { return Query < T1 , T2 , T3 > ( sql . SQL , sql . Arguments ) . ToList ( ) ; }
public List < T1 > Fetch < T1 , T2 , T3 , T4 > ( Sql sql ) { return Query < T1 , T2 , T3 , T4 > ( sql . SQL , sql . Arguments ) . ToList ( ) ; }
// Multi Query (Simple) (SQL builder)
public IEnumerable < T1 > Query < T1 , T2 > ( Sql sql ) { return Query < T1 > ( new Type [ ] { typeof ( T1 ) , typeof ( T2 ) } , null , sql . SQL , sql . Arguments ) ; }
public IEnumerable < T1 > Query < T1 , T2 , T3 > ( Sql sql ) { return Query < T1 > ( new Type [ ] { typeof ( T1 ) , typeof ( T2 ) , typeof ( T3 ) } , null , sql . SQL , sql . Arguments ) ; }
public IEnumerable < T1 > Query < T1 , T2 , T3 , T4 > ( Sql sql ) { return Query < T1 > ( new Type [ ] { typeof ( T1 ) , typeof ( T2 ) , typeof ( T3 ) , typeof ( T4 ) } , null , sql . SQL , sql . Arguments ) ; }
// Automagically guess the property relationships between various POCOs and create a delegate that will set them up
object GetAutoMapper ( Type [ ] types )
{
// Build a key
var kb = new StringBuilder ( ) ;
foreach ( var t in types )
{
kb . Append ( t . ToString ( ) ) ;
kb . Append ( ":" ) ;
}
var key = kb . ToString ( ) ;
// Check cache
RWLock . EnterReadLock ( ) ;
try
{
object mapper ;
if ( AutoMappers . TryGetValue ( key , out mapper ) )
return mapper ;
}
finally
{
RWLock . ExitReadLock ( ) ;
}
// Create it
RWLock . EnterWriteLock ( ) ;
try
{
// Try again
object mapper ;
if ( AutoMappers . TryGetValue ( key , out mapper ) )
return mapper ;
// Create a method
var m = new DynamicMethod ( "petapoco_automapper" , types [ 0 ] , types , true ) ;
var il = m . GetILGenerator ( ) ;
for ( int i = 1 ; i < types . Length ; i + + )
{
bool handled = false ;
for ( int j = i - 1 ; j > = 0 ; j - - )
{
// Find the property
var candidates = from p in types [ j ] . GetProperties ( ) where p . PropertyType = = types [ i ] select p ;
if ( candidates . Count ( ) = = 0 )
continue ;
if ( candidates . Count ( ) > 1 )
throw new InvalidOperationException ( string . Format ( "Can't auto join {0} as {1} has more than one property of type {0}" , types [ i ] , types [ j ] ) ) ;
// Generate code
il . Emit ( OpCodes . Ldarg_S , j ) ;
il . Emit ( OpCodes . Ldarg_S , i ) ;
il . Emit ( OpCodes . Callvirt , candidates . First ( ) . GetSetMethod ( true ) ) ;
handled = true ;
}
if ( ! handled )
throw new InvalidOperationException ( string . Format ( "Can't auto join {0}" , types [ i ] ) ) ;
}
il . Emit ( OpCodes . Ldarg_0 ) ;
il . Emit ( OpCodes . Ret ) ;
// Cache it
var del = m . CreateDelegate ( Expression . GetFuncType ( types . Concat ( types . Take ( 1 ) ) . ToArray ( ) ) ) ;
AutoMappers . Add ( key , del ) ;
return del ;
}
finally
{
RWLock . ExitWriteLock ( ) ;
}
}
// Find the split point in a result set for two different pocos and return the poco factory for the first
Delegate FindSplitPoint ( Type typeThis , Type typeNext , string sql , IDataReader r , ref int pos )
{
// Last?
if ( typeNext = = null )
return PocoData . ForType ( typeThis ) . GetFactory ( sql , _sharedConnection . ConnectionString , ForceDateTimesToUtc , pos , r . FieldCount - pos , r ) ;
// Get PocoData for the two types
PocoData pdThis = PocoData . ForType ( typeThis ) ;
PocoData pdNext = PocoData . ForType ( typeNext ) ;
// Find split point
int firstColumn = pos ;
var usedColumns = new Dictionary < string , bool > ( ) ;
for ( ; pos < r . FieldCount ; pos + + )
{
// Split if field name has already been used, or if the field doesn't exist in current poco but does in the next
string fieldName = r . GetName ( pos ) ;
if ( usedColumns . ContainsKey ( fieldName ) | | ( ! pdThis . Columns . ContainsKey ( fieldName ) & & pdNext . Columns . ContainsKey ( fieldName ) ) )
{
return pdThis . GetFactory ( sql , _sharedConnection . ConnectionString , ForceDateTimesToUtc , firstColumn , pos - firstColumn , r ) ;
}
usedColumns . Add ( fieldName , true ) ;
}
throw new InvalidOperationException ( string . Format ( "Couldn't find split point between {0} and {1}" , typeThis , typeNext ) ) ;
}
// Instance data used by the Multipoco factory delegate - essentially a list of the nested poco factories to call
class MultiPocoFactory
{
public List < Delegate > m_Delegates ;
public Delegate GetItem ( int index ) { return m_Delegates [ index ] ; }
}
// Create a multi-poco factory
Func < IDataReader , object , TRet > CreateMultiPocoFactory < TRet > ( Type [ ] types , string sql , IDataReader r )
{
var m = new DynamicMethod ( "petapoco_multipoco_factory" , typeof ( TRet ) , new Type [ ] { typeof ( MultiPocoFactory ) , typeof ( IDataReader ) , typeof ( object ) } , typeof ( MultiPocoFactory ) ) ;
var il = m . GetILGenerator ( ) ;
// Load the callback
il . Emit ( OpCodes . Ldarg_2 ) ;
// Call each delegate
var dels = new List < Delegate > ( ) ;
int pos = 0 ;
for ( int i = 0 ; i < types . Length ; i + + )
{
// Add to list of delegates to call
var del = FindSplitPoint ( types [ i ] , i + 1 < types . Length ? types [ i + 1 ] : null , sql , r , ref pos ) ;
dels . Add ( del ) ;
// Get the delegate
il . Emit ( OpCodes . Ldarg_0 ) ; // callback,this
il . Emit ( OpCodes . Ldc_I4 , i ) ; // callback,this,Index
il . Emit ( OpCodes . Callvirt , typeof ( MultiPocoFactory ) . GetMethod ( "GetItem" ) ) ; // callback,Delegate
il . Emit ( OpCodes . Ldarg_1 ) ; // callback,delegate, datareader
// Call Invoke
var tDelInvoke = del . GetType ( ) . GetMethod ( "Invoke" ) ;
il . Emit ( OpCodes . Callvirt , tDelInvoke ) ; // Poco left on stack
}
// By now we should have the callback and the N pocos all on the stack. Call the callback and we're done
il . Emit ( OpCodes . Callvirt , Expression . GetFuncType ( types . Concat ( new Type [ ] { typeof ( TRet ) } ) . ToArray ( ) ) . GetMethod ( "Invoke" ) ) ;
il . Emit ( OpCodes . Ret ) ;
// Finish up
return ( Func < IDataReader , object , TRet > ) m . CreateDelegate ( typeof ( Func < IDataReader , object , TRet > ) , new MultiPocoFactory ( ) { m_Delegates = dels } ) ;
}
// Various cached stuff
static Dictionary < string , object > MultiPocoFactories = new Dictionary < string , object > ( ) ;
static Dictionary < string , object > AutoMappers = new Dictionary < string , object > ( ) ;
static ReaderWriterLockSlim RWLock = new ReaderWriterLockSlim ( ) ;
// Get (or create) the multi-poco factory for a query
Func < IDataReader , object , TRet > GetMultiPocoFactory < TRet > ( Type [ ] types , string sql , IDataReader r )
{
// Build a key string (this is crap, should address this at some point)
var kb = new StringBuilder ( ) ;
kb . Append ( typeof ( TRet ) . ToString ( ) ) ;
kb . Append ( ":" ) ;
foreach ( var t in types )
{
kb . Append ( ":" ) ;
kb . Append ( t . ToString ( ) ) ;
}
kb . Append ( ":" ) ; kb . Append ( _sharedConnection . ConnectionString ) ;
kb . Append ( ":" ) ; kb . Append ( ForceDateTimesToUtc ) ;
kb . Append ( ":" ) ; kb . Append ( sql ) ;
string key = kb . ToString ( ) ;
// Check cache
RWLock . EnterReadLock ( ) ;
try
{
object oFactory ;
if ( MultiPocoFactories . TryGetValue ( key , out oFactory ) )
return ( Func < IDataReader , object , TRet > ) oFactory ;
}
finally
{
RWLock . ExitReadLock ( ) ;
}
// Cache it
RWLock . EnterWriteLock ( ) ;
try
{
// Check again
object oFactory ;
if ( MultiPocoFactories . TryGetValue ( key , out oFactory ) )
return ( Func < IDataReader , object , TRet > ) oFactory ;
// Create the factory
var Factory = CreateMultiPocoFactory < TRet > ( types , sql , r ) ;
MultiPocoFactories . Add ( key , Factory ) ;
return Factory ;
}
finally
{
RWLock . ExitWriteLock ( ) ;
}
}
// Actual implementation of the multi-poco query
public IEnumerable < TRet > Query < TRet > ( Type [ ] types , object cb , string sql , params object [ ] args )
{
using ( MiniProfiler . StepStatic ( "Peta Query Type[]" ) )
{
OpenSharedConnection ( ) ;
try
{
using ( var cmd = CreateCommand ( _sharedConnection , sql , args ) )
{
IDataReader r ;
try
{
r = cmd . ExecuteReader ( ) ;
OnExecutedCommand ( cmd ) ;
}
catch ( Exception x )
{
OnException ( x ) ;
throw ;
}
var factory = GetMultiPocoFactory < TRet > ( types , sql , r ) ;
if ( cb = = null )
cb = GetAutoMapper ( types . ToArray ( ) ) ;
bool bNeedTerminator = false ;
using ( r )
{
while ( true )
{
TRet poco ;
try
{
if ( ! r . Read ( ) )
break ;
poco = factory ( r , cb ) ;
}
catch ( Exception x )
{
OnException ( x ) ;
throw ;
}
if ( poco ! = null )
yield return poco ;
else
bNeedTerminator = true ;
}
if ( bNeedTerminator )
{
var poco = ( TRet ) ( cb as Delegate ) . DynamicInvoke ( new object [ types . Length ] ) ;
if ( poco ! = null )
yield return poco ;
else
yield break ;
}
}
}
}
finally
{
CloseSharedConnection ( ) ;
}
}
}
public bool Exists < T > ( object primaryKey )
{
var index = 0 ;
var primaryKeyValuePairs = GetPrimaryKeyValues ( PocoData . ForType ( typeof ( T ) ) . TableInfo . PrimaryKey , primaryKey ) ;
return FirstOrDefault < T > ( string . Format ( "WHERE {0}" , BuildPrimaryKeySql ( primaryKeyValuePairs , ref index ) ) , primaryKeyValuePairs . Select ( x = > x . Value ) . ToArray ( ) ) ! = null ;
}
public bool Exists < T > ( string sql , params object [ ] args )
{
var poco = PocoData . ForType ( typeof ( T ) ) . TableInfo ;
string existsTemplate ;
switch ( _dbType )
{
case DBType . SQLite :
case DBType . MySql :
{
existsTemplate = "SELECT EXISTS (SELECT 1 FROM {0} {1})" ;
break ;
}
case DBType . SqlServer :
{
existsTemplate = "IF EXISTS (SELECT 1 FROM {0} {1}) SELECT 1 ELSE SELECT 0" ;
break ;
}
default :
{
existsTemplate = "SELECT COUNT(*) FROM {0} {1}" ;
break ;
}
}
return ExecuteScalar < int > ( string . Format ( existsTemplate , poco . TableName , sql ) , args ) ! = 0 ;
}
public T Single < T > ( object primaryKey )
{
var index = 0 ;
var primaryKeyValuePairs = GetPrimaryKeyValues ( PocoData . ForType ( typeof ( T ) ) . TableInfo . PrimaryKey , primaryKey ) ;
return Single < T > ( string . Format ( "WHERE {0}" , BuildPrimaryKeySql ( primaryKeyValuePairs , ref index ) ) , primaryKeyValuePairs . Select ( x = > x . Value ) . ToArray ( ) ) ;
}
public T SingleOrDefault < T > ( object primaryKey )
{
var index = 0 ;
var primaryKeyValuePairs = GetPrimaryKeyValues ( PocoData . ForType ( typeof ( T ) ) . TableInfo . PrimaryKey , primaryKey ) ;
return SingleOrDefault < T > ( string . Format ( "WHERE {0}" , BuildPrimaryKeySql ( primaryKeyValuePairs , ref index ) ) , primaryKeyValuePairs . Select ( x = > x . Value ) . ToArray ( ) ) ;
}
public T Single < T > ( string sql , params object [ ] args )
{
return Query < T > ( sql , args ) . Single ( ) ;
}
public T SingleOrDefault < T > ( string sql , params object [ ] args )
{
return Query < T > ( sql , args ) . SingleOrDefault ( ) ;
}
public T First < T > ( string sql , params object [ ] args )
{
return Query < T > ( sql , args ) . First ( ) ;
}
public T FirstOrDefault < T > ( string sql , params object [ ] args )
{
return Query < T > ( sql , args ) . FirstOrDefault ( ) ;
}
public T Single < T > ( Sql sql )
{
return Query < T > ( sql ) . Single ( ) ;
}
public T SingleOrDefault < T > ( Sql sql )
{
return Query < T > ( sql ) . SingleOrDefault ( ) ;
}
public T First < T > ( Sql sql )
{
return Query < T > ( sql ) . First ( ) ;
}
public T FirstOrDefault < T > ( Sql sql )
{
return Query < T > ( sql ) . FirstOrDefault ( ) ;
}
public string EscapeTableName ( string str )
{
// Assume table names with "dot", or opening sq is already escaped
return str . IndexOf ( '.' ) > = 0 ? str : EscapeSqlIdentifier ( str ) ;
}
public string EscapeSqlIdentifier ( string str )
{
switch ( _dbType )
{
case DBType . MySql :
return string . Format ( "`{0}`" , str ) ;
case DBType . PostgreSQL :
case DBType . Oracle :
return string . Format ( "\"{0}\"" , str ) ;
default :
return string . Format ( "[{0}]" , str ) ;
}
}
public object Insert ( string tableName , string primaryKeyName , object poco )
{
return Insert ( tableName , primaryKeyName , true , poco ) ;
}
// Insert a poco into a table. If the poco has a property with the same name
// as the primary key the id of the new record is assigned to it. Either way,
// the new id is returned.
public object Insert ( string tableName , string primaryKeyName , bool autoIncrement , object poco )
{
using ( MiniProfiler . StepStatic ( "Peta Insert " + tableName ) )
{
try
{
OpenSharedConnection ( ) ;
try
{
using ( var cmd = CreateCommand ( _sharedConnection , "" ) )
{
var pd = PocoData . ForObject ( poco , primaryKeyName ) ;
var names = new List < string > ( ) ;
var values = new List < string > ( ) ;
var index = 0 ;
var versionName = "" ;
foreach ( var i in pd . Columns )
{
// Don't insert result columns
if ( i . Value . ResultColumn )
continue ;
// Don't insert the primary key (except under oracle where we need bring in the next sequence value)
if ( autoIncrement & & primaryKeyName ! = null & & string . Compare ( i . Key , primaryKeyName , true ) = = 0 )
{
if ( _dbType = = DBType . Oracle & & ! string . IsNullOrEmpty ( pd . TableInfo . SequenceName ) )
{
names . Add ( i . Key ) ;
values . Add ( string . Format ( "{0}.nextval" , pd . TableInfo . SequenceName ) ) ;
}
continue ;
}
names . Add ( EscapeSqlIdentifier ( i . Key ) ) ;
values . Add ( string . Format ( "{0}{1}" , _paramPrefix , index + + ) ) ;
object val = i . Value . GetValue ( poco ) ;
if ( i . Value . VersionColumn )
{
val = 1 ;
versionName = i . Key ;
}
AddParam ( cmd , val , _paramPrefix ) ;
}
cmd . CommandText = string . Format ( "INSERT INTO {0} ({1}) VALUES ({2})" ,
EscapeTableName ( tableName ) ,
string . Join ( "," , names . ToArray ( ) ) ,
string . Join ( "," , values . ToArray ( ) )
) ;
if ( ! autoIncrement )
{
DoPreExecute ( cmd ) ;
cmd . ExecuteNonQuery ( ) ;
OnExecutedCommand ( cmd ) ;
return true ;
}
object id ;
switch ( _dbType )
{
case DBType . SqlServerCE :
DoPreExecute ( cmd ) ;
cmd . ExecuteNonQuery ( ) ;
OnExecutedCommand ( cmd ) ;
id = ExecuteScalar < object > ( "SELECT @@@IDENTITY AS NewID;" ) ;
break ;
case DBType . SqlServer :
cmd . CommandText + = ";\nSELECT SCOPE_IDENTITY() AS NewID;" ;
DoPreExecute ( cmd ) ;
id = cmd . ExecuteScalar ( ) ;
OnExecutedCommand ( cmd ) ;
break ;
case DBType . PostgreSQL :
if ( primaryKeyName ! = null )
{
cmd . CommandText + = string . Format ( "returning {0} as NewID" , EscapeSqlIdentifier ( primaryKeyName ) ) ;
DoPreExecute ( cmd ) ;
id = cmd . ExecuteScalar ( ) ;
}
else
{
id = - 1 ;
DoPreExecute ( cmd ) ;
cmd . ExecuteNonQuery ( ) ;
}
OnExecutedCommand ( cmd ) ;
break ;
case DBType . Oracle :
if ( primaryKeyName ! = null )
{
cmd . CommandText + = string . Format ( " returning {0} into :newid" , EscapeSqlIdentifier ( primaryKeyName ) ) ;
var param = cmd . CreateParameter ( ) ;
param . ParameterName = ":newid" ;
param . Value = DBNull . Value ;
param . Direction = ParameterDirection . ReturnValue ;
param . DbType = DbType . Int64 ;
cmd . Parameters . Add ( param ) ;
DoPreExecute ( cmd ) ;
cmd . ExecuteNonQuery ( ) ;
id = param . Value ;
}
else
{
id = - 1 ;
DoPreExecute ( cmd ) ;
cmd . ExecuteNonQuery ( ) ;
}
OnExecutedCommand ( cmd ) ;
break ;
case DBType . SQLite :
if ( primaryKeyName ! = null )
{
cmd . CommandText + = ";\nSELECT last_insert_rowid();" ;
DoPreExecute ( cmd ) ;
id = cmd . ExecuteScalar ( ) ;
}
else
{
id = - 1 ;
DoPreExecute ( cmd ) ;
cmd . ExecuteNonQuery ( ) ;
}
OnExecutedCommand ( cmd ) ;
break ;
default :
cmd . CommandText + = ";\nSELECT @@IDENTITY AS NewID;" ;
DoPreExecute ( cmd ) ;
id = cmd . ExecuteScalar ( ) ;
OnExecutedCommand ( cmd ) ;
break ;
}
// Assign the ID back to the primary key property
if ( primaryKeyName ! = null )
{
PocoColumn pc ;
if ( pd . Columns . TryGetValue ( primaryKeyName , out pc ) )
{
pc . SetValue ( poco , pc . ChangeType ( id ) ) ;
}
}
// Assign the Version column
if ( ! string . IsNullOrEmpty ( versionName ) )
{
PocoColumn pc ;
if ( pd . Columns . TryGetValue ( versionName , out pc ) )
{
pc . SetValue ( poco , pc . ChangeType ( 1 ) ) ;
}
}
return id ;
}
}
finally
{
CloseSharedConnection ( ) ;
}
}
catch ( Exception x )
{
OnException ( x ) ;
throw ;
}
}
}
// Insert an annotated poco object
public object Insert ( object poco )
{
var pd = PocoData . ForType ( poco . GetType ( ) ) ;
return Insert ( pd . TableInfo . TableName , pd . TableInfo . PrimaryKey , pd . TableInfo . AutoIncrement , poco ) ;
}
public void InsertMany < T > ( IEnumerable < T > pocoList )
{
using ( var tran = GetTransaction ( ) )
{
foreach ( var poco in pocoList )
{
Insert ( poco ) ;
}
tran . Complete ( ) ;
}
}
// Update a record with values from a poco. primary key value can be either supplied or read from the poco
public int Update ( string tableName , string primaryKeyName , object poco , object primaryKeyValue )
{
using ( MiniProfiler . StepStatic ( "Peta Update " + tableName ) )
{
try
{
OpenSharedConnection ( ) ;
try
{
using ( var cmd = CreateCommand ( _sharedConnection , "" ) )
{
var sb = new StringBuilder ( ) ;
var index = 0 ;
var pd = PocoData . ForObject ( poco , primaryKeyName ) ;
string versionName = null ;
object versionValue = null ;
var primaryKeyValuePairs = GetPrimaryKeyValues ( primaryKeyName , primaryKeyValue ) ;
foreach ( var i in pd . Columns )
{
// Don't update the primary key, but grab the value if we don't have it
if ( primaryKeyValue = = null & & primaryKeyValuePairs . ContainsKey ( i . Key ) )
{
primaryKeyValuePairs [ i . Key ] = i . Value . PropertyInfo . GetValue ( poco , null ) ;
continue ;
}
// Dont update result only columns
if ( i . Value . ResultColumn )
continue ;
object value = i . Value . PropertyInfo . GetValue ( poco , null ) ;
if ( i . Value . VersionColumn )
{
versionName = i . Key ;
versionValue = value ;
value = Convert . ToInt64 ( value ) + 1 ;
}
// Build the sql
if ( index > 0 )
sb . Append ( ", " ) ;
sb . AppendFormat ( "{0} = {1}{2}" , EscapeSqlIdentifier ( i . Key ) , _paramPrefix , index + + ) ;
// Store the parameter in the command
AddParam ( cmd , value , _paramPrefix ) ;
}
cmd . CommandText = string . Format ( "UPDATE {0} SET {1} WHERE {2}" ,
EscapeSqlIdentifier ( tableName ) , sb . ToString ( ) ,
BuildPrimaryKeySql ( primaryKeyValuePairs , ref index ) ) ;
foreach ( var keyValue in primaryKeyValuePairs )
{
AddParam ( cmd , keyValue . Value , _paramPrefix ) ;
}
if ( ! string . IsNullOrEmpty ( versionName ) )
{
cmd . CommandText + = string . Format ( " AND {0} = {1}{2}" , EscapeSqlIdentifier ( versionName ) , _paramPrefix ,
index + + ) ;
AddParam ( cmd , versionValue , _paramPrefix ) ;
}
DoPreExecute ( cmd ) ;
// Do it
var result = cmd . ExecuteNonQuery ( ) ;
OnExecutedCommand ( cmd ) ;
// Set Version
if ( ! string . IsNullOrEmpty ( versionName ) )
{
PocoColumn pc ;
if ( pd . Columns . TryGetValue ( versionName , out pc ) )
{
pc . PropertyInfo . SetValue ( poco ,
Convert . ChangeType ( Convert . ToInt64 ( versionValue ) + 1 ,
pc . PropertyInfo . PropertyType ) , null ) ;
}
}
return result ;
}
}
finally
{
CloseSharedConnection ( ) ;
}
}
catch ( Exception x )
{
OnException ( x ) ;
throw ;
}
}
}
private string BuildPrimaryKeySql ( Dictionary < string , object > primaryKeyValuePair , ref int index )
{
var tempIndex = index ;
index + = primaryKeyValuePair . Count ;
return string . Join ( " AND " , primaryKeyValuePair . Select ( ( x , i ) = > string . Format ( "{0} = {1}{2}" , EscapeSqlIdentifier ( x . Key ) , _paramPrefix , tempIndex + i ) ) . ToArray ( ) ) ;
}
private Dictionary < string , object > GetPrimaryKeyValues ( string primaryKeyName , object primaryKeyValue )
{
Dictionary < string , object > primaryKeyValues ;
var multiplePrimaryKeysNames = primaryKeyName . Split ( new [ ] { ',' } , StringSplitOptions . RemoveEmptyEntries ) . Select ( x = > x . Trim ( ) ) . ToArray ( ) ;
if ( primaryKeyValue ! = null )
{
if ( multiplePrimaryKeysNames . Length = = 1 )
primaryKeyValues = new Dictionary < string , object > ( StringComparer . OrdinalIgnoreCase ) { { primaryKeyName , primaryKeyValue } } ;
else
primaryKeyValues = multiplePrimaryKeysNames . ToDictionary ( x = > x ,
x = > primaryKeyValue . GetType ( ) . GetProperties ( )
. Where ( y = > string . Equals ( x , y . Name , StringComparison . OrdinalIgnoreCase ) )
. Single ( ) . GetValue ( primaryKeyValue , null ) , StringComparer . OrdinalIgnoreCase ) ;
}
else
{
primaryKeyValues = multiplePrimaryKeysNames . ToDictionary ( x = > x , x = > ( object ) null , StringComparer . OrdinalIgnoreCase ) ;
}
return primaryKeyValues ;
}
public int Update ( string tableName , string primaryKeyName , object poco )
{
return Update ( tableName , primaryKeyName , poco , null ) ;
}
public int Update ( object poco )
{
return Update ( poco , null ) ;
}
public int Update ( object poco , object primaryKeyValue )
{
var pd = PocoData . ForType ( poco . GetType ( ) ) ;
return Update ( pd . TableInfo . TableName , pd . TableInfo . PrimaryKey , poco , primaryKeyValue ) ;
}
public int Update < T > ( string sql , params object [ ] args )
{
var pd = PocoData . ForType ( typeof ( T ) ) ;
return Execute ( string . Format ( "UPDATE {0} {1}" , EscapeTableName ( pd . TableInfo . TableName ) , sql ) , args ) ;
}
public int Update < T > ( Sql sql )
{
var pd = PocoData . ForType ( typeof ( T ) ) ;
return Execute ( new Sql ( string . Format ( "UPDATE {0}" , EscapeTableName ( pd . TableInfo . TableName ) ) ) . Append ( sql ) ) ;
}
public void UpdateMany < T > ( IEnumerable < T > pocoList )
{
using ( var tran = GetTransaction ( ) )
{
foreach ( var poco in pocoList )
{
Update ( poco ) ;
}
tran . Complete ( ) ;
}
}
public int Delete ( string tableName , string primaryKeyName , object poco )
{
return Delete ( tableName , primaryKeyName , poco , null ) ;
}
public int Delete ( string tableName , string primaryKeyName , object poco , object primaryKeyValue )
{
var primaryKeyValuePairs = GetPrimaryKeyValues ( primaryKeyName , primaryKeyValue ) ;
// If primary key value not specified, pick it up from the object
if ( primaryKeyValue = = null )
{
var pd = PocoData . ForObject ( poco , primaryKeyName ) ;
foreach ( var i in pd . Columns )
{
if ( primaryKeyValuePairs . ContainsKey ( i . Key ) )
{
primaryKeyValuePairs [ i . Key ] = i . Value . PropertyInfo . GetValue ( poco , null ) ;
}
}
}
// Do it
var index = 0 ;
var sql = string . Format ( "DELETE FROM {0} WHERE {1}" , tableName , BuildPrimaryKeySql ( primaryKeyValuePairs , ref index ) ) ;
return Execute ( sql , primaryKeyValuePairs . Select ( x = > x . Value ) . ToArray ( ) ) ;
}
public int Delete ( object poco )
{
var pd = PocoData . ForType ( poco . GetType ( ) ) ;
return Delete ( pd . TableInfo . TableName , pd . TableInfo . PrimaryKey , poco ) ;
}
public int Delete < T > ( object pocoOrPrimaryKey )
{
if ( pocoOrPrimaryKey . GetType ( ) = = typeof ( T ) )
return Delete ( pocoOrPrimaryKey ) ;
var pd = PocoData . ForType ( typeof ( T ) ) ;
return Delete ( pd . TableInfo . TableName , pd . TableInfo . PrimaryKey , null , pocoOrPrimaryKey ) ;
}
public int Delete < T > ( string sql , params object [ ] args )
{
var pd = PocoData . ForType ( typeof ( T ) ) ;
return Execute ( string . Format ( "DELETE FROM {0} {1}" , EscapeTableName ( pd . TableInfo . TableName ) , sql ) , args ) ;
}
public int Delete < T > ( Sql sql )
{
var pd = PocoData . ForType ( typeof ( T ) ) ;
return Execute ( new Sql ( string . Format ( "DELETE FROM {0}" , EscapeTableName ( pd . TableInfo . TableName ) ) ) . Append ( sql ) ) ;
}
// Check if a poco represents a new record
public bool IsNew ( string primaryKeyName , object poco )
{
var pd = PocoData . ForObject ( poco , primaryKeyName ) ;
object pk ;
PocoColumn pc ;
if ( pd . Columns . TryGetValue ( primaryKeyName , out pc ) )
{
pk = pc . GetValue ( poco ) ;
}
#if !PETAPOCO_NO_DYNAMIC
else if ( poco . GetType ( ) = = typeof ( System . Dynamic . ExpandoObject ) )
{
return true ;
}
# endif
else
{
var pi = poco . GetType ( ) . GetProperty ( primaryKeyName ) ;
if ( pi = = null )
throw new ArgumentException ( string . Format ( "The object doesn't have a property matching the primary key column name '{0}'" , primaryKeyName ) ) ;
pk = pi . GetValue ( poco , null ) ;
}
if ( pk = = null )
return true ;
var type = pk . GetType ( ) ;
if ( type . IsValueType )
{
// Common primary key types
if ( type = = typeof ( long ) )
return ( long ) pk = = 0 ;
else if ( type = = typeof ( ulong ) )
return ( ulong ) pk = = 0 ;
else if ( type = = typeof ( int ) )
return ( int ) pk = = 0 ;
else if ( type = = typeof ( uint ) )
return ( uint ) pk = = 0 ;
// Create a default instance and compare
return pk = = Activator . CreateInstance ( pk . GetType ( ) ) ;
}
else
{
return pk = = null ;
}
}
public bool IsNew ( object poco )
{
var pd = PocoData . ForType ( poco . GetType ( ) ) ;
if ( ! pd . TableInfo . AutoIncrement )
throw new InvalidOperationException ( "IsNew() and Save() are only supported on tables with auto-increment/identity primary key columns" ) ;
return IsNew ( pd . TableInfo . PrimaryKey , poco ) ;
}
// Insert new record or Update existing record
public void Save ( string tableName , string primaryKeyName , object poco )
{
if ( IsNew ( primaryKeyName , poco ) )
{
Insert ( tableName , primaryKeyName , true , poco ) ;
}
else
{
Update ( tableName , primaryKeyName , poco ) ;
}
}
public void Save ( object poco )
{
var pd = PocoData . ForType ( poco . GetType ( ) ) ;
Save ( pd . TableInfo . TableName , pd . TableInfo . PrimaryKey , poco ) ;
}
public void SaveMany < T > ( IEnumerable < T > pocoList )
{
using ( var tran = GetTransaction ( ) )
{
foreach ( var poco in pocoList )
{
Save ( poco ) ;
}
tran . Complete ( ) ;
}
}
public int CommandTimeout { get ; set ; }
public int OneTimeCommandTimeout { get ; set ; }
void DoPreExecute ( IDbCommand cmd )
{
// Setup command timeout
if ( OneTimeCommandTimeout ! = 0 )
{
cmd . CommandTimeout = OneTimeCommandTimeout ;
OneTimeCommandTimeout = 0 ;
}
else if ( CommandTimeout ! = 0 )
{
cmd . CommandTimeout = CommandTimeout ;
}
// Call hook
OnExecutingCommand ( cmd ) ;
// Save it
_lastSql = cmd . CommandText ;
_lastArgs = ( from IDataParameter parameter in cmd . Parameters select parameter . Value ) . ToArray ( ) ;
}
public string LastSQL { get { return _lastSql ; } }
public object [ ] LastArgs { get { return _lastArgs ; } }
public string LastCommand
{
get { return FormatCommand ( _lastSql , _lastArgs ) ; }
}
public string FormatCommand ( IDbCommand cmd )
{
return FormatCommand ( cmd . CommandText , ( from IDataParameter parameter in cmd . Parameters select parameter . Value ) . ToArray ( ) ) ;
}
public string FormatCommand ( string sql , object [ ] args )
{
var sb = new StringBuilder ( ) ;
if ( sql = = null )
return "" ;
sb . Append ( sql ) ;
if ( args ! = null & & args . Length > 0 )
{
sb . Append ( "\n" ) ;
for ( int i = 0 ; i < args . Length ; i + + )
{
sb . AppendFormat ( "\t -> {0}{1} [{2}] = \"{3}\"\n" , _paramPrefix , i , args [ i ] . GetType ( ) . Name , args [ i ] ) ;
}
sb . Remove ( sb . Length - 1 , 1 ) ;
}
return sb . ToString ( ) ;
}
public static IMapper Mapper
{
get ;
set ;
}
public class PocoColumn
{
public string ColumnName ;
public PropertyInfo PropertyInfo ;
public bool ResultColumn ;
public bool VersionColumn ;
public virtual void SetValue ( object target , object val ) { PropertyInfo . SetValue ( target , val , null ) ; }
public virtual object GetValue ( object target ) { return PropertyInfo . GetValue ( target , null ) ; }
public virtual object ChangeType ( object val ) { return Convert . ChangeType ( val , PropertyInfo . PropertyType ) ; }
}
internal class ExpandoColumn : PocoColumn
{
public override void SetValue ( object target , object val ) { ( target as IDictionary < string , object > ) [ ColumnName ] = val ; }
public override object GetValue ( object target )
{
object val = null ;
( target as IDictionary < string , object > ) . TryGetValue ( ColumnName , out val ) ;
return val ;
}
public override object ChangeType ( object val ) { return val ; }
}
internal class PocoData
{
public static PocoData ForObject ( object o , string primaryKeyName )
{
var t = o . GetType ( ) ;
#if !PETAPOCO_NO_DYNAMIC
if ( t = = typeof ( System . Dynamic . ExpandoObject ) )
{
var pd = new PocoData ( ) ;
pd . TableInfo = new TableInfo ( ) ;
pd . Columns = new Dictionary < string , PocoColumn > ( StringComparer . OrdinalIgnoreCase ) ;
pd . Columns . Add ( primaryKeyName , new ExpandoColumn ( ) { ColumnName = primaryKeyName } ) ;
pd . TableInfo . PrimaryKey = primaryKeyName ;
pd . TableInfo . AutoIncrement = true ;
foreach ( var col in ( o as IDictionary < string , object > ) . Keys )
{
if ( col ! = primaryKeyName )
pd . Columns . Add ( col , new ExpandoColumn ( ) { ColumnName = col } ) ;
}
return pd ;
}
else
# endif
return ForType ( t ) ;
}
static ReaderWriterLockSlim RWLock = new ReaderWriterLockSlim ( ) ;
public static PocoData ForType ( Type t )
{
#if !PETAPOCO_NO_DYNAMIC
if ( t = = typeof ( System . Dynamic . ExpandoObject ) )
throw new InvalidOperationException ( "Can't use dynamic types with this method" ) ;
# endif
// Check cache
RWLock . EnterReadLock ( ) ;
PocoData pd ;
try
{
if ( m_PocoDatas . TryGetValue ( t , out pd ) )
return pd ;
}
finally
{
RWLock . ExitReadLock ( ) ;
}
// Cache it
RWLock . EnterWriteLock ( ) ;
try
{
// Check again
if ( m_PocoDatas . TryGetValue ( t , out pd ) )
return pd ;
// Create it
pd = new PocoData ( t ) ;
m_PocoDatas . Add ( t , pd ) ;
}
finally
{
RWLock . ExitWriteLock ( ) ;
}
return pd ;
}
public PocoData ( )
{
}
public PocoData ( Type t )
{
type = t ;
TableInfo = new TableInfo ( ) ;
// Get the table name
var a = t . GetCustomAttributes ( typeof ( TableNameAttribute ) , true ) ;
TableInfo . TableName = a . Length = = 0 ? t . Name : ( a [ 0 ] as TableNameAttribute ) . Value ;
// Get the primary key
a = t . GetCustomAttributes ( typeof ( PrimaryKeyAttribute ) , true ) ;
TableInfo . PrimaryKey = a . Length = = 0 ? "ID" : ( a [ 0 ] as PrimaryKeyAttribute ) . Value ;
TableInfo . SequenceName = a . Length = = 0 ? null : ( a [ 0 ] as PrimaryKeyAttribute ) . sequenceName ;
TableInfo . AutoIncrement = a . Length = = 0 ? false : ( a [ 0 ] as PrimaryKeyAttribute ) . autoIncrement ;
// Set autoincrement false if primary key has multiple columns
TableInfo . AutoIncrement = TableInfo . AutoIncrement ? ! TableInfo . PrimaryKey . Contains ( ',' ) : TableInfo . AutoIncrement ;
// Call column mapper
if ( Mapper ! = null )
Mapper . GetTableInfo ( t , TableInfo ) ;
// Work out bound properties
bool ExplicitColumns = t . GetCustomAttributes ( typeof ( ExplicitColumnsAttribute ) , true ) . Length > 0 ;
Columns = new Dictionary < string , PocoColumn > ( StringComparer . OrdinalIgnoreCase ) ;
foreach ( var pi in t . GetProperties ( ) )
{
// Work out if properties is to be included
var ColAttrs = pi . GetCustomAttributes ( typeof ( ColumnAttribute ) , true ) ;
if ( ExplicitColumns )
{
if ( ColAttrs . Length = = 0 )
continue ;
}
else
{
if ( pi . GetCustomAttributes ( typeof ( IgnoreAttribute ) , true ) . Length ! = 0 )
continue ;
}
var pc = new PocoColumn ( ) ;
pc . PropertyInfo = pi ;
// Work out the DB column name
if ( ColAttrs . Length > 0 )
{
var colattr = ( ColumnAttribute ) ColAttrs [ 0 ] ;
pc . ColumnName = colattr . Name ;
if ( ( colattr as ResultColumnAttribute ) ! = null )
pc . ResultColumn = true ;
if ( ( colattr as VersionColumnAttribute ) ! = null )
pc . VersionColumn = true ;
}
if ( pc . ColumnName = = null )
{
pc . ColumnName = pi . Name ;
if ( Mapper ! = null & & ! Mapper . MapPropertyToColumn ( pi , ref pc . ColumnName , ref pc . ResultColumn ) )
continue ;
}
// Store it
Columns . Add ( pc . ColumnName , pc ) ;
}
// Build column list for automatic select
QueryColumns = ( from c in Columns where ! c . Value . ResultColumn select c . Key ) . ToArray ( ) ;
}
static bool IsIntegralType ( Type t )
{
var tc = Type . GetTypeCode ( t ) ;
return tc > = TypeCode . SByte & & tc < = TypeCode . UInt64 ;
}
// Create factory function that can convert a IDataReader record into a POCO
public Delegate GetFactory ( string sql , string connString , bool ForceDateTimesToUtc , int firstColumn , int countColumns , IDataReader r )
{
// Check cache
var key = string . Format ( "{0}:{1}:{2}:{3}:{4}" , sql , connString , ForceDateTimesToUtc , firstColumn , countColumns ) ;
RWLock . EnterReadLock ( ) ;
try
{
// Have we already created it?
Delegate factory ;
if ( PocoFactories . TryGetValue ( key , out factory ) )
return factory ;
}
finally
{
RWLock . ExitReadLock ( ) ;
}
// Take the writer lock
RWLock . EnterWriteLock ( ) ;
try
{
// Check again, just in case
Delegate factory ;
if ( PocoFactories . TryGetValue ( key , out factory ) )
return factory ;
// Create the method
var m = new DynamicMethod ( "petapoco_factory_" + PocoFactories . Count . ToString ( ) , type , new Type [ ] { typeof ( IDataReader ) } , true ) ;
var il = m . GetILGenerator ( ) ;
#if !PETAPOCO_NO_DYNAMIC
if ( type = = typeof ( object ) )
{
// var poco=new T()
il . Emit ( OpCodes . Newobj , typeof ( System . Dynamic . ExpandoObject ) . GetConstructor ( Type . EmptyTypes ) ) ; // obj
MethodInfo fnAdd = typeof ( IDictionary < string , object > ) . GetMethod ( "Add" ) ;
// Enumerate all fields generating a set assignment for the column
for ( int i = firstColumn ; i < firstColumn + countColumns ; i + + )
{
var srcType = r . GetFieldType ( i ) ;
il . Emit ( OpCodes . Dup ) ; // obj, obj
il . Emit ( OpCodes . Ldstr , r . GetName ( i ) ) ; // obj, obj, fieldname
// Get the converter
Func < object , object > converter = null ;
if ( Database . Mapper ! = null )
converter = Database . Mapper . GetFromDbConverter ( null , srcType ) ;
if ( ForceDateTimesToUtc & & converter = = null & & srcType = = typeof ( DateTime ) )
converter = delegate ( object src ) { return new DateTime ( ( ( DateTime ) src ) . Ticks , DateTimeKind . Utc ) ; } ;
// Setup stack for call to converter
AddConverterToStack ( il , converter ) ;
// r[i]
il . Emit ( OpCodes . Ldarg_0 ) ; // obj, obj, fieldname, converter?, rdr
il . Emit ( OpCodes . Ldc_I4 , i ) ; // obj, obj, fieldname, converter?, rdr,i
il . Emit ( OpCodes . Callvirt , fnGetValue ) ; // obj, obj, fieldname, converter?, value
// Convert DBNull to null
il . Emit ( OpCodes . Dup ) ; // obj, obj, fieldname, converter?, value, value
il . Emit ( OpCodes . Isinst , typeof ( DBNull ) ) ; // obj, obj, fieldname, converter?, value, (value or null)
var lblNotNull = il . DefineLabel ( ) ;
il . Emit ( OpCodes . Brfalse_S , lblNotNull ) ; // obj, obj, fieldname, converter?, value
il . Emit ( OpCodes . Pop ) ; // obj, obj, fieldname, converter?
if ( converter ! = null )
il . Emit ( OpCodes . Pop ) ; // obj, obj, fieldname,
il . Emit ( OpCodes . Ldnull ) ; // obj, obj, fieldname, null
if ( converter ! = null )
{
var lblReady = il . DefineLabel ( ) ;
il . Emit ( OpCodes . Br_S , lblReady ) ;
il . MarkLabel ( lblNotNull ) ;
il . Emit ( OpCodes . Callvirt , fnInvoke ) ;
il . MarkLabel ( lblReady ) ;
}
else
{
il . MarkLabel ( lblNotNull ) ;
}
il . Emit ( OpCodes . Callvirt , fnAdd ) ;
}
}
else
# endif
if ( type . IsValueType | | type = = typeof ( string ) | | type = = typeof ( byte [ ] ) )
{
// Do we need to install a converter?
var srcType = r . GetFieldType ( 0 ) ;
var converter = GetConverter ( ForceDateTimesToUtc , null , srcType , type ) ;
// "if (!rdr.IsDBNull(i))"
il . Emit ( OpCodes . Ldarg_0 ) ; // rdr
il . Emit ( OpCodes . Ldc_I4_0 ) ; // rdr,0
il . Emit ( OpCodes . Callvirt , fnIsDBNull ) ; // bool
var lblCont = il . DefineLabel ( ) ;
il . Emit ( OpCodes . Brfalse_S , lblCont ) ;
il . Emit ( OpCodes . Ldnull ) ; // null
var lblFin = il . DefineLabel ( ) ;
il . Emit ( OpCodes . Br_S , lblFin ) ;
il . MarkLabel ( lblCont ) ;
// Setup stack for call to converter
AddConverterToStack ( il , converter ) ;
il . Emit ( OpCodes . Ldarg_0 ) ; // rdr
il . Emit ( OpCodes . Ldc_I4_0 ) ; // rdr,0
il . Emit ( OpCodes . Callvirt , fnGetValue ) ; // value
// Call the converter
if ( converter ! = null )
il . Emit ( OpCodes . Callvirt , fnInvoke ) ;
il . MarkLabel ( lblFin ) ;
il . Emit ( OpCodes . Unbox_Any , type ) ; // value converted
}
else
{
// var poco=new T()
il . Emit ( OpCodes . Newobj , type . GetConstructor ( BindingFlags . Instance | BindingFlags . Public | BindingFlags . NonPublic , null , new Type [ 0 ] , null ) ) ;
// Enumerate all fields generating a set assignment for the column
for ( int i = firstColumn ; i < firstColumn + countColumns ; i + + )
{
// Get the PocoColumn for this db column, ignore if not known
PocoColumn pc ;
if ( ! Columns . TryGetValue ( r . GetName ( i ) , out pc ) )
continue ;
// Get the source type for this column
var srcType = r . GetFieldType ( i ) ;
var dstType = pc . PropertyInfo . PropertyType ;
// "if (!rdr.IsDBNull(i))"
il . Emit ( OpCodes . Ldarg_0 ) ; // poco,rdr
il . Emit ( OpCodes . Ldc_I4 , i ) ; // poco,rdr,i
il . Emit ( OpCodes . Callvirt , fnIsDBNull ) ; // poco,bool
var lblNext = il . DefineLabel ( ) ;
il . Emit ( OpCodes . Brtrue_S , lblNext ) ; // poco
il . Emit ( OpCodes . Dup ) ; // poco,poco
// Do we need to install a converter?
var converter = GetConverter ( ForceDateTimesToUtc , pc , srcType , dstType ) ;
// Fast
bool Handled = false ;
if ( converter = = null )
{
var valuegetter = typeof ( IDataRecord ) . GetMethod ( "Get" + srcType . Name , new Type [ ] { typeof ( int ) } ) ;
if ( valuegetter ! = null
& & valuegetter . ReturnType = = srcType
& & ( valuegetter . ReturnType = = dstType | | valuegetter . ReturnType = = Nullable . GetUnderlyingType ( dstType ) ) )
{
il . Emit ( OpCodes . Ldarg_0 ) ; // *,rdr
il . Emit ( OpCodes . Ldc_I4 , i ) ; // *,rdr,i
il . Emit ( OpCodes . Callvirt , valuegetter ) ; // *,value
// Convert to Nullable
if ( Nullable . GetUnderlyingType ( dstType ) ! = null )
{
il . Emit ( OpCodes . Newobj , dstType . GetConstructor ( new Type [ ] { Nullable . GetUnderlyingType ( dstType ) } ) ) ;
}
il . Emit ( OpCodes . Callvirt , pc . PropertyInfo . GetSetMethod ( true ) ) ; // poco
Handled = true ;
}
}
// Not so fast
if ( ! Handled )
{
// Setup stack for call to converter
AddConverterToStack ( il , converter ) ;
// "value = rdr.GetValue(i)"
il . Emit ( OpCodes . Ldarg_0 ) ; // *,rdr
il . Emit ( OpCodes . Ldc_I4 , i ) ; // *,rdr,i
il . Emit ( OpCodes . Callvirt , fnGetValue ) ; // *,value
// Call the converter
if ( converter ! = null )
il . Emit ( OpCodes . Callvirt , fnInvoke ) ;
// Assign it
il . Emit ( OpCodes . Unbox_Any , pc . PropertyInfo . PropertyType ) ; // poco,poco,value
il . Emit ( OpCodes . Callvirt , pc . PropertyInfo . GetSetMethod ( true ) ) ; // poco
}
il . MarkLabel ( lblNext ) ;
}
}
il . Emit ( OpCodes . Ret ) ;
// Cache it, return it
var del = m . CreateDelegate ( Expression . GetFuncType ( typeof ( IDataReader ) , type ) ) ;
PocoFactories . Add ( key , del ) ;
return del ;
}
finally
{
RWLock . ExitWriteLock ( ) ;
}
}
private static void AddConverterToStack ( ILGenerator il , Func < object , object > converter )
{
if ( converter ! = null )
{
// Add the converter
int converterIndex = m_Converters . Count ;
m_Converters . Add ( converter ) ;
// Generate IL to push the converter onto the stack
il . Emit ( OpCodes . Ldsfld , fldConverters ) ;
il . Emit ( OpCodes . Ldc_I4 , converterIndex ) ;
il . Emit ( OpCodes . Callvirt , fnListGetItem ) ; // Converter
}
}
private static Func < object , object > GetConverter ( bool forceDateTimesToUtc , PocoColumn pc , Type srcType , Type dstType )
{
Func < object , object > converter = null ;
// Get converter from the mapper
if ( Mapper ! = null )
{
DestinationInfo destinationInfo = pc ! = null
? new DestinationInfo ( pc . PropertyInfo )
: new DestinationInfo ( dstType ) ;
converter = Mapper . GetFromDbConverter ( destinationInfo , srcType ) ;
}
// Standard DateTime->Utc mapper
if ( forceDateTimesToUtc & & converter = = null & & srcType = = typeof ( DateTime ) & & ( dstType = = typeof ( DateTime ) | | dstType = = typeof ( DateTime ? ) ) )
{
converter = delegate ( object src ) { return new DateTime ( ( ( DateTime ) src ) . Ticks , DateTimeKind . Utc ) ; } ;
}
// Forced type conversion including integral types -> enum
if ( converter = = null )
{
if ( dstType . IsEnum & & IsIntegralType ( srcType ) )
{
if ( srcType ! = typeof ( int ) )
{
converter = delegate ( object src ) { return Convert . ChangeType ( src , typeof ( int ) , null ) ; } ;
}
}
else if ( ! dstType . IsAssignableFrom ( srcType ) )
{
converter = delegate ( object src ) { return Convert . ChangeType ( src , dstType , null ) ; } ;
}
}
return converter ;
}
static Dictionary < Type , PocoData > m_PocoDatas = new Dictionary < Type , PocoData > ( ) ;
static List < Func < object , object > > m_Converters = new List < Func < object , object > > ( ) ;
static MethodInfo fnGetValue = typeof ( IDataRecord ) . GetMethod ( "GetValue" , new Type [ ] { typeof ( int ) } ) ;
static MethodInfo fnIsDBNull = typeof ( IDataRecord ) . GetMethod ( "IsDBNull" ) ;
static FieldInfo fldConverters = typeof ( PocoData ) . GetField ( "m_Converters" , BindingFlags . Static | BindingFlags . GetField | BindingFlags . NonPublic ) ;
static MethodInfo fnListGetItem = typeof ( List < Func < object , object > > ) . GetProperty ( "Item" ) . GetGetMethod ( ) ;
static MethodInfo fnInvoke = typeof ( Func < object , object > ) . GetMethod ( "Invoke" ) ;
public Type type ;
public string [ ] QueryColumns { get ; private set ; }
public TableInfo TableInfo { get ; private set ; }
public Dictionary < string , PocoColumn > Columns { get ; private set ; }
Dictionary < string , Delegate > PocoFactories = new Dictionary < string , Delegate > ( ) ;
}
// Member variables
string _connectionString ;
string _providerName ;
DbProviderFactory _factory ;
IDbConnection _sharedConnection ;
IDbTransaction _transaction ;
int _sharedConnectionDepth ;
int _transactionDepth ;
bool _transactionCancelled ;
string _lastSql ;
object [ ] _lastArgs ;
string _paramPrefix = "@" ;
}
// Transaction object helps maintain transaction depth counts
public interface ITransaction : IDisposable
{
void Complete ( ) ;
}
public class Transaction : ITransaction
{
public Transaction ( Database db )
{
_db = db ;
_db . BeginTransaction ( ) ;
}
public void Complete ( )
{
_db . CompleteTransaction ( ) ;
_db = null ;
}
public void Dispose ( )
{
if ( _db ! = null )
_db . AbortTransaction ( ) ;
}
Database _db ;
}
// Simple helper class for building SQL statments
public class Sql
{
public Sql ( )
{
}
public Sql ( string sql , params object [ ] args )
{
_sql = sql ;
_args = args ;
}
public static Sql Builder
{
get { return new Sql ( ) ; }
}
string _sql ;
object [ ] _args ;
Sql _rhs ;
string _sqlFinal ;
object [ ] _argsFinal ;
private void Build ( )
{
// already built?
if ( _sqlFinal ! = null )
return ;
// Build it
var sb = new StringBuilder ( ) ;
var args = new List < object > ( ) ;
Build ( sb , args , null ) ;
_sqlFinal = sb . ToString ( ) ;
_argsFinal = args . ToArray ( ) ;
}
public string SQL
{
get
{
Build ( ) ;
return _sqlFinal ;
}
}
public object [ ] Arguments
{
get
{
Build ( ) ;
return _argsFinal ;
}
}
public Sql Append ( Sql sql )
{
if ( _rhs ! = null )
_rhs . Append ( sql ) ;
else
_rhs = sql ;
return this ;
}
public Sql Append ( string sql , params object [ ] args )
{
return Append ( new Sql ( sql , args ) ) ;
}
static bool Is ( Sql sql , string sqltype )
{
return sql ! = null & & sql . _sql ! = null & & sql . _sql . StartsWith ( sqltype , StringComparison . InvariantCultureIgnoreCase ) ;
}
private void Build ( StringBuilder sb , List < object > args , Sql lhs )
{
if ( ! String . IsNullOrEmpty ( _sql ) )
{
// Add SQL to the string
if ( sb . Length > 0 )
{
sb . Append ( "\n" ) ;
}
var sql = Database . ProcessParams ( _sql , _args , args ) ;
if ( Is ( lhs , "WHERE " ) & & Is ( this , "WHERE " ) )
sql = "AND " + sql . Substring ( 6 ) ;
if ( Is ( lhs , "ORDER BY " ) & & Is ( this , "ORDER BY " ) )
sql = ", " + sql . Substring ( 9 ) ;
sb . Append ( sql ) ;
}
// Now do rhs
if ( _rhs ! = null )
_rhs . Build ( sb , args , this ) ;
}
public Sql Where ( string sql , params object [ ] args )
{
return Append ( new Sql ( "WHERE (" + sql + ")" , args ) ) ;
}
public Sql OrderBy ( params object [ ] columns )
{
return Append ( new Sql ( "ORDER BY " + String . Join ( ", " , ( from x in columns select x . ToString ( ) ) . ToArray ( ) ) ) ) ;
}
public Sql Select ( params object [ ] columns )
{
return Append ( new Sql ( "SELECT " + String . Join ( ", " , ( from x in columns select x . ToString ( ) ) . ToArray ( ) ) ) ) ;
}
public Sql From ( params object [ ] tables )
{
return Append ( new Sql ( "FROM " + String . Join ( ", " , ( from x in tables select x . ToString ( ) ) . ToArray ( ) ) ) ) ;
}
public Sql GroupBy ( params object [ ] columns )
{
return Append ( new Sql ( "GROUP BY " + String . Join ( ", " , ( from x in columns select x . ToString ( ) ) . ToArray ( ) ) ) ) ;
}
private SqlJoinClause Join ( string JoinType , string table )
{
return new SqlJoinClause ( Append ( new Sql ( JoinType + table ) ) ) ;
}
public SqlJoinClause InnerJoin ( string table ) { return Join ( "INNER JOIN " , table ) ; }
public SqlJoinClause LeftJoin ( string table ) { return Join ( "LEFT JOIN " , table ) ; }
public class SqlJoinClause
{
private readonly Sql _sql ;
public SqlJoinClause ( Sql sql )
{
_sql = sql ;
}
public Sql On ( string onClause , params object [ ] args )
{
return _sql . Append ( "ON " + onClause , args ) ;
}
}
}
}
