New tests.
[mono.git] / mcs / class / System.Data.Linq / src / DbLinq / Data / Linq / DataContext.cs
index 568dc358bde775e9415daa62f5c77fdd1042a624..9710a8aad0bec7110912565d75d906ac60c9b601 100644 (file)
-#region MIT license
-// 
-// MIT license
-//
-// Copyright (c) 2007-2008 Jiri Moudry, Pascal Craponne
-// 
-// Permission is hereby granted, free of charge, to any person obtaining a copy
-// of this software and associated documentation files (the "Software"), to deal
-// in the Software without restriction, including without limitation the rights
-// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-// copies of the Software, and to permit persons to whom the Software is
-// furnished to do so, subject to the following conditions:
-// 
-// The above copyright notice and this permission notice shall be included in
-// all copies or substantial portions of the Software.
-// 
-// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
-// THE SOFTWARE.
-// 
-#endregion
-
-using System;
-using System.Collections;
-using System.Data;
-using System.Data.Common;
-using System.Data.Linq.Mapping;
-using System.Collections.Generic;
-using System.IO;
-using System.Linq;
-using System.Reflection;
-
-#if MONO_STRICT
-using System.Data.Linq.Implementation;
-using System.Data.Linq.Sugar;
-using System.Data.Linq.Identity;
-using DbLinq.Util;
-using AttributeMappingSource = System.Data.Linq.Mapping.AttributeMappingSource;
-using MappingContext = System.Data.Linq.Mapping.MappingContext;
-using DbLinq;
-#else
-using DbLinq.Data.Linq.Implementation;
-using DbLinq.Data.Linq.Sugar;
-using DbLinq.Data.Linq.Identity;
-using DbLinq.Util;
-using AttributeMappingSource = DbLinq.Data.Linq.Mapping.AttributeMappingSource;
-using MappingContext = DbLinq.Data.Linq.Mapping.MappingContext;
-using System.Data.Linq;
-#endif
-
-using DbLinq.Factory;
-using DbLinq.Vendor;
-using DbLinq.Data.Linq.Database;
-using DbLinq.Data.Linq.Database.Implementation;
-using System.Linq.Expressions;
-using System.Reflection.Emit;
-
-#if MONO_STRICT
-namespace System.Data.Linq
-#else
-namespace DbLinq.Data.Linq
-#endif
-{
-    public partial class DataContext : IDisposable
-    {
-        //private readonly Dictionary<string, ITable> _tableMap = new Dictionary<string, ITable>();
-               private readonly Dictionary<Type, ITable> _tableMap = new Dictionary<Type, ITable>();
-
-        public MetaModel Mapping { get; private set; }
-        // PC question: at ctor, we get a IDbConnection and the Connection property exposes a DbConnection
-        //              WTF?
-        public DbConnection Connection { get { return DatabaseContext.Connection as DbConnection; } }
-
-        // all properties below are set public to optionally be injected
-        internal IVendor Vendor { get; set; }
-        internal IQueryBuilder QueryBuilder { get; set; }
-        internal IQueryRunner QueryRunner { get; set; }
-        internal IMemberModificationHandler MemberModificationHandler { get; set; }
-        internal IDatabaseContext DatabaseContext { get; private set; }
-        // /all properties...
-
-        private readonly EntityTracker entityTracker = new EntityTracker();
-
-        private IIdentityReaderFactory identityReaderFactory;
-        private readonly IDictionary<Type, IIdentityReader> identityReaders = new Dictionary<Type, IIdentityReader>();
-
-        /// <summary>
-        /// The default behavior creates one MappingContext.
-        /// </summary>
-        [DBLinqExtended]
-        internal virtual MappingContext _MappingContext { get; set; }
-
-        [DBLinqExtended]
-        internal IVendorProvider _VendorProvider { get; set; }
-
-        public DataContext(IDbConnection connection, MappingSource mapping)
-        {
-            Init(new DatabaseContext(connection), mapping, null);
-        }
-
-        public DataContext(IDbConnection connection)
-        {
-            Init(new DatabaseContext(connection), null, null);
-        }
-
-        [DbLinqToDo]
-        public DataContext(string fileOrServerOrConnection, MappingSource mapping)
-        {
-            throw new NotImplementedException();
-        }
-
-        /// <summary>
-        /// Construct DataContext, given a connectionString.
-        /// To determine which DB type to go against, we look for 'DbLinqProvider=xxx' substring.
-        /// If not found, we assume that we are dealing with MS Sql Server.
-        /// 
-        /// Valid values are names of provider DLLs (or any other DLL containing an IVendor implementation)
-        /// DbLinqProvider=Mysql
-        /// DbLinqProvider=Oracle etc.
-        /// </summary>
-        /// <param name="connectionString">specifies file or server connection</param>
-        [DbLinqToDo]
-        public DataContext(string connectionString)
-        {
-            #region DataContext connectionString ctor
-            if (connectionString == null)
-                throw new ArgumentNullException("connectionString");
-
-            System.Text.RegularExpressions.Regex reProvider
-                = new System.Text.RegularExpressions.Regex(@"DbLinqProvider=([\w\.]+)");
-
-            int startPos = connectionString.IndexOf("DbLinqProvider=");
-            string assemblyToLoad;
-            string vendorClassToLoad;
-            if (!reProvider.IsMatch(connectionString))
-            {
-                assemblyToLoad = "DbLinq.SqlServer.dll";
-                vendorClassToLoad = "SqlServerVendor";
-            }
-            else
-            {
-                System.Text.RegularExpressions.Match match = reProvider.Match(connectionString);
-#if MONO_STRICT
-                //Pascal says on the forum: 
-                //[in MONO] "all vendors are (will be) embedded in the System.Data.Linq assembly"
-                assemblyToLoad = "System.Data.Linq.dll";
-                vendorClassToLoad = match.Groups[1].Value; //eg. "MySql"
-#else
-                //plain DbLinq - non MONO: 
-                //IVendor classes are in DLLs such as "DbLinq.MySql.dll"
-                assemblyToLoad = match.Groups[1].Value; //eg. assemblyToLoad="DbLinq.MySql.dll"
-                if (assemblyToLoad.Contains("."))
-                {
-                    //already fully qualified DLL name?
-                    throw new ArgumentException("Please provide a short name, such as 'MySql', not '" + assemblyToLoad + "'");
-                }
-                else
-                {
-                    //we were given short name, such as MySql
-                    vendorClassToLoad = assemblyToLoad + "Vendor"; //eg. MySqlVendor
-                    assemblyToLoad = "DbLinq." + assemblyToLoad + ".dll"; //eg. DbLinq.MySql.dll
-                }
-#endif
-                //shorten: "DbLinqProvider=X;Server=Y" -> ";Server=Y"
-                string shortenedConnStr = reProvider.Replace(connectionString, "");
-                connectionString = shortenedConnStr;
-            }
-
-            Assembly assy;
-            try
-            {
-#if MONO_STRICT
-               assy = typeof (DataContext).Assembly; // System.Data.Linq.dll
-#else
-                //TODO: check if DLL is already loaded?
-                assy = Assembly.LoadFrom(assemblyToLoad);
-#endif
-            }
-            catch (Exception ex)
-            {
-                //TODO: add proper logging here
-                Console.WriteLine("DataContext ctor: Assembly load failed for " + assemblyToLoad + ": " + ex);
-                throw ex;
-            }
-
-            //find IDbProvider class in this assembly:
-            var ctors = (from mod in assy.GetModules()
-                         from cls in mod.GetTypes()
-                         where cls.GetInterfaces().Contains(typeof(IVendor))
-                            && cls.Name.ToLower() == vendorClassToLoad.ToLower()
-                         let ctorInfo = cls.GetConstructor(Type.EmptyTypes)
-                         where ctorInfo != null
-                         select ctorInfo).ToList();
-            if (ctors.Count == 0)
-            {
-                string msg = "Found no IVendor class in assembly " + assemblyToLoad + " having a string ctor";
-                throw new ArgumentException(msg);
-            }
-            else if (ctors.Count > 1)
-            {
-                string msg = "Found more than one IVendor class in assembly " + assemblyToLoad + " having a string ctor";
-                throw new ArgumentException(msg);
-            }
-            ConstructorInfo ctorInfo2 = ctors[0];
-
-            object ivendorObject;
-            try
-            {
-                ivendorObject = ctorInfo2.Invoke(new object[]{});
-            }
-            catch (Exception ex)
-            {
-                //TODO: add proper logging here
-                Console.WriteLine("DataContext ctor: Failed to invoke IVendor ctor " + ctorInfo2.Name + ": " + ex);
-                throw ex;
-            }
-            IVendor ivendor = (IVendor)ivendorObject;
-            IDbConnection dbConnection = ivendor.CreateDbConnection(connectionString);
-            Init(new DatabaseContext(dbConnection), null, ivendor);
-            #endregion
-        }
-
-        private void Init(IDatabaseContext databaseContext, MappingSource mappingSource, IVendor vendor)
-        {
-            if (databaseContext == null)
-                throw new ArgumentNullException("databaseContext");
-
-            _VendorProvider = ObjectFactory.Get<IVendorProvider>();
-            if (vendor == null)
-                Vendor = _VendorProvider.FindVendorByProviderType(typeof(SqlClient.Sql2005Provider));
-            else
-                Vendor = vendor;
-
-            DatabaseContext = databaseContext;
-
-            MemberModificationHandler = ObjectFactory.Create<IMemberModificationHandler>(); // not a singleton: object is stateful
-            QueryBuilder = ObjectFactory.Get<IQueryBuilder>();
-            QueryRunner = ObjectFactory.Get<IQueryRunner>();
-
-            //EntityMap = ObjectFactory.Create<IEntityMap>();
-            identityReaderFactory = ObjectFactory.Get<IIdentityReaderFactory>();
-
-            _MappingContext = new MappingContext();
-
-            // initialize the mapping information
-            if (mappingSource == null)
-                mappingSource = new AttributeMappingSource();
-            Mapping = mappingSource.GetModel(GetType());
-        }
-
-               /// <summary>
-               /// Checks if the table is allready mapped or maps it if not.
-               /// </summary>
-               /// <param name="tableType">Type of the table.</param>
-               /// <exception cref="InvalidOperationException">Thrown if the table is not mappable.</exception>
-               private void CheckTableMapping(Type tableType)
-               {
-                       //This will throw an exception if the table is not found
-                       if(Mapping.GetTable(tableType) == null)
-                       {
-                               throw new InvalidOperationException("The type '" + tableType.Name + "' is not mapped as a Table.");
-                       }
-               }
-
-               /// <summary>
-               /// Returns a Table for the type TEntity.
-               /// </summary>
-               /// <exception cref="InvalidOperationException">If the type TEntity is not mappable as a Table.</exception>
-               /// <typeparam name="TEntity">The table type.</typeparam>
-       public Table<TEntity> GetTable<TEntity>() where TEntity : class
-        {
-            return (Table<TEntity>)GetTable(typeof(TEntity));
-        }
-
-               /// <summary>
-               /// Returns a Table for the given type.
-               /// </summary>
-               /// <param name="type">The table type.</param>
-               /// <exception cref="InvalidOperationException">If the type is not mappable as a Table.</exception>
-        public ITable GetTable(Type type)
-        {
-            lock (_tableMap)
-            {
-                ITable tableExisting;
-                               if (_tableMap.TryGetValue(type, out tableExisting))
-                    return tableExisting;
-
-                               //Check for table mapping
-                               CheckTableMapping(type);
-
-                var tableNew = Activator.CreateInstance(
-                                  typeof(Table<>).MakeGenericType(type)
-                                  , BindingFlags.NonPublic | BindingFlags.Instance
-                                  , null
-                                  , new object[] { this }
-                                  , System.Globalization.CultureInfo.CurrentCulture) as ITable;
-
-                _tableMap[type] = tableNew;
-                return tableNew;
-            }
-        }
-
-        public void SubmitChanges()
-        {
-            SubmitChanges(ConflictMode.FailOnFirstConflict);
-        }
-
-        /// <summary>
-        /// Pings database
-        /// </summary>
-        /// <returns></returns>
-        public bool DatabaseExists()
-        {
-            try
-            {
-                return Vendor.Ping(this);
-            }
-            catch (Exception)
-            {
-                return false;
-            }
-        }
-
-        /// <summary>
-        /// Commits all pending changes to database 
-        /// </summary>
-        /// <param name="failureMode"></param>
-        public virtual void SubmitChanges(ConflictMode failureMode)
-        {
-            using (DatabaseContext.OpenConnection()) //ConnMgr will close connection for us
-            using (IDatabaseTransaction transaction = DatabaseContext.Transaction())
-            {
-                var queryContext = new QueryContext(this);
-                var entityTracks = entityTracker.EnumerateAll().ToList();
-                foreach (var entityTrack in entityTracks)
-                {
-                    switch (entityTrack.EntityState)
-                    {
-                        case EntityState.ToInsert:
-                            var insertQuery = QueryBuilder.GetInsertQuery(entityTrack.Entity, queryContext);
-                            QueryRunner.Insert(entityTrack.Entity, insertQuery);
-                            Register(entityTrack.Entity);
-                            break;
-                        case EntityState.ToWatch:
-                            if (MemberModificationHandler.IsModified(entityTrack.Entity, Mapping))
-                            {
-                                var modifiedMembers = MemberModificationHandler.GetModifiedProperties(entityTrack.Entity, Mapping);
-                                var updateQuery = QueryBuilder.GetUpdateQuery(entityTrack.Entity, modifiedMembers, queryContext);
-                                QueryRunner.Update(entityTrack.Entity, updateQuery, modifiedMembers);
-
-                                RegisterUpdateAgain(entityTrack.Entity);
-                            }
-                            break;
-                        case EntityState.ToDelete:
-                            var deleteQuery = QueryBuilder.GetDeleteQuery(entityTrack.Entity, queryContext);
-                            QueryRunner.Delete(entityTrack.Entity, deleteQuery);
-
-                            UnregisterDelete(entityTrack.Entity);
-                            break;
-                        default:
-                            throw new ArgumentOutOfRangeException();
-                    }
-                }
-                // TODO: handle conflicts (which can only occur when concurrency mode is implemented)
-                transaction.Commit();
-            }
-        }
-
-        /// <summary>
-        /// TODO - allow generated methods to call into stored procedures
-        /// </summary>
-        [DBLinqExtended]
-        internal IExecuteResult _ExecuteMethodCall(DataContext context, System.Reflection.MethodInfo method, params object[] sqlParams)
-        {
-            using (DatabaseContext.OpenConnection())
-            {
-                System.Data.Linq.IExecuteResult result = Vendor.ExecuteMethodCall(context, method, sqlParams);
-                return result;
-            }
-        }
-
-        [DbLinqToDo]
-        protected IExecuteResult ExecuteMethodCall(object instance, System.Reflection.MethodInfo methodInfo, params object[] parameters)
-        {
-            throw new NotImplementedException();
-        }
-
-        #region Identity management
-
-        [DBLinqExtended]
-        internal IIdentityReader _GetIdentityReader(Type t)
-        {
-            IIdentityReader identityReader;
-            lock (identityReaders)
-            {
-                if (!identityReaders.TryGetValue(t, out identityReader))
-                {
-                    identityReader = identityReaderFactory.GetReader(t, this);
-                    identityReaders[t] = identityReader;
-                }
-            }
-            return identityReader;
-        }
-
-        [DBLinqExtended]
-        internal object _GetRegisteredEntity(object entity)
-        {
-            // TODO: check what is faster: by identity or by ref
-            var identityReader = _GetIdentityReader(entity.GetType());
-            var identityKey = identityReader.GetIdentityKey(entity);
-            if (identityKey == null) // if we don't have an entitykey here, it means that the entity has no PK
-                return entity;
-            // even 
-            var registeredEntityTrack = entityTracker.FindByIdentity(identityKey);
-            if (registeredEntityTrack != null)
-                return registeredEntityTrack.Entity;
-            return null;
-        }
-
-        //internal object GetRegisteredEntityByKey(IdentityKey identityKey)
-        //{
-        //    return EntityMap[identityKey];
-        //}
-
-        /// <summary>
-        /// Registers an entity in a watch state
-        /// </summary>
-        /// <param name="entity"></param>
-        /// <returns></returns>
-        [DBLinqExtended]
-        internal object _GetOrRegisterEntity(object entity)
-        {
-            var identityReader = _GetIdentityReader(entity.GetType());
-            var identityKey = identityReader.GetIdentityKey(entity);
-            SetEntitySetsQueries(entity);
-            SetEntityRefQueries(entity);
-
-            // if we have no identity, we can't track it
-            if (identityKey == null)
-                return entity;
-
-            // try to find an already registered entity and return it
-            var registeredEntityTrack = entityTracker.FindByIdentity(identityKey);
-            if (registeredEntityTrack != null)
-                return registeredEntityTrack.Entity;
-
-            // otherwise, register and return
-            entityTracker.RegisterToWatch(entity, identityKey);
-            return entity;
-        }
-
-        readonly IDataMapper DataMapper = ObjectFactory.Get<IDataMapper>();
-        private void SetEntityRefQueries(object entity)
-        {
-            Type thisType = entity.GetType();
-            IList<MemberInfo> properties = DataMapper.GetEntityRefAssociations(thisType);
-
-
-            foreach (PropertyInfo prop in properties)
-            {
-                //example of entityRef:Order.Employee
-                AssociationAttribute associationInfo = prop.GetAttribute<AssociationAttribute>();
-                Type otherTableType = prop.PropertyType;
-                IList<MemberInfo> otherPKs = DataMapper.GetPrimaryKeys(Mapping.GetTable(otherTableType));
-
-                if (otherPKs.Count > 1)
-                    throw new NotSupportedException("Multiple keys object not supported yet.");
-
-                var otherTable = GetTable(otherTableType);
-
-                //ie:EmployeeTerritories.EmployeeID
-
-                var thisForeignKeyProperty = thisType.GetProperty(associationInfo.ThisKey);
-                object thisForeignKeyValue = thisForeignKeyProperty.GetValue(entity, null);
-
-                IEnumerable query = null;
-                if (thisForeignKeyValue != null)
-                {
-                    ParameterExpression p = Expression.Parameter(otherTableType, "other");
-                    Expression predicate;
-                    if (!(thisForeignKeyProperty.PropertyType.IsNullable()))
-                    {
-                        predicate = Expression.Equal(Expression.MakeMemberAccess(p, otherPKs.First()),
-                                                                    Expression.Constant(thisForeignKeyValue));
-                    }
-                    else
-                    {
-                        var ValueProperty = thisForeignKeyProperty.PropertyType.GetProperty("Value");
-                        predicate = Expression.Equal(Expression.MakeMemberAccess(p, otherPKs.First()),
-                                                                 Expression.Constant(ValueProperty.GetValue(thisForeignKeyValue, null)));
-                    }
-
-                    query = GetOtherTableQuery(predicate, p, otherTableType, otherTable) as IEnumerable;
-                    //it would be interesting surround the above query with a .Take(1) expression for performance.
-                }
-
-
-                FieldInfo entityRefField = entity.GetType().GetField(associationInfo.Storage, BindingFlags.NonPublic | BindingFlags.Instance);
-                object entityRefValue = null;
-                if (query != null)
-                    entityRefValue = Activator.CreateInstance(entityRefField.FieldType, query);
-                else
-                    entityRefValue = Activator.CreateInstance(entityRefField.FieldType);
-                entityRefField.SetValue(entity, entityRefValue);
-            }
-        }
-
-        /// <summary>
-        /// This method is executed when the entity is being registered. Each EntitySet property has a internal query that can be set using the EntitySet.SetSource method.
-        /// Here we set the query source of each EntitySetProperty
-        /// </summary>
-        /// <param name="entity"></param>
-        private void SetEntitySetsQueries(object entity)
-        {
-            IList<MemberInfo> properties = DataMapper.GetEntitySetAssociations(entity.GetType());
-            
-            if (properties.Any()) {
-                IList<MemberInfo> thisPKs = DataMapper.GetPrimaryKeys(Mapping.GetTable(entity.GetType()));
-
-                if (thisPKs.Count > 1)
-                    throw new NotSupportedException("Multiple keys object not supported yet.");
-
-                object primaryKeyValue = (thisPKs.First() as PropertyInfo).GetValue(entity, null);
-
-
-                foreach (PropertyInfo prop in properties)
-                {
-                    //example of entitySet: Employee.EmployeeTerritories
-                    var associationInfo = prop.GetAttribute<AssociationAttribute>();
-                    Type otherTableType = prop.PropertyType.GetGenericArguments().First();
-
-                    //other table:EmployeeTerritories
-                    var otherTable = GetTable(otherTableType);
-                    //other table member:EmployeeTerritories.EmployeeID
-                    var otherTableMember = otherTableType.GetProperty(associationInfo.OtherKey);
-
-
-                    ParameterExpression p = Expression.Parameter(otherTableType, "other");
-                    Expression predicate;
-                    if (!(otherTableMember.PropertyType.IsNullable()))
-                    {
-                        predicate = Expression.Equal(Expression.MakeMemberAccess(p, otherTableMember),
-                                                                    Expression.Constant(primaryKeyValue));
-                    }
-                    else
-                    {
-                        var ValueProperty = otherTableMember.PropertyType.GetProperty("Value");
-                        predicate = Expression.Equal(Expression.MakeMemberAccess(
-                                                                    Expression.MakeMemberAccess(p, otherTableMember),
-                                                                    ValueProperty),
-                                                                 Expression.Constant(primaryKeyValue));
-                    }
-
-                    var query = GetOtherTableQuery(predicate, p, otherTableType, otherTable);
-
-                    var entitySetValue = prop.GetValue(entity, null);
-
-                    if (entitySetValue == null)
-                    {
-                        entitySetValue = Activator.CreateInstance(prop.PropertyType);
-                        prop.SetValue(entity, entitySetValue, null);
-                    }
-
-                    var setSourceMethod = entitySetValue.GetType().GetMethod("SetSource");
-                    setSourceMethod.Invoke(entitySetValue, new[] { query });
-                    //employee.EmployeeTerritories.SetSource(Table[EmployeesTerritories].Where(other=>other.employeeID="WARTH"))
-                }
-            }
-        }
-
-        private object GetOtherTableQuery(Expression predicate, ParameterExpression parameter, Type otherTableType, IQueryable otherTable)
-        {
-            //predicate: other.EmployeeID== "WARTH"
-            Expression lambdaPredicate = Expression.Lambda(predicate, parameter);
-            //lambdaPredicate: other=>other.EmployeeID== "WARTH"
-
-            var whereMethod = typeof(Queryable)
-                              .GetMethods().First(m => m.Name == "Where")
-                              .MakeGenericMethod(otherTableType);
-
-
-            Expression call = Expression.Call(whereMethod, otherTable.Expression, lambdaPredicate);
-            //Table[EmployeesTerritories].Where(other=>other.employeeID="WARTH")
-
-            return otherTable.Provider.CreateQuery(call);
-        }
-
-        #endregion
-
-        #region Insert/Update/Delete management
-
-        /// <summary>
-        /// Registers an entity for insert
-        /// </summary>
-        /// <param name="entity"></param>
-        internal void RegisterInsert(object entity)
-        {
-            entityTracker.RegisterToInsert(entity);
-        }
-
-        /// <summary>
-        /// Registers an entity for update
-        /// The entity will be updated only if some of its members have changed after the registration
-        /// </summary>
-        /// <param name="entity"></param>
-        internal void RegisterUpdate(object entity)
-        {
-            var identityReader = _GetIdentityReader(entity.GetType());
-            var identityKey = identityReader.GetIdentityKey(entity);
-            // if we have no key, we can not watch
-            if (identityKey == null)
-                return;
-            // register entity
-            entityTracker.RegisterToWatch(entity, identityKey);
-        }
-
-        /// <summary>
-        /// Registers or re-registers an entity and clears its state
-        /// </summary>
-        /// <param name="entity"></param>
-        /// <returns></returns>
-        internal object Register(object entity)
-        {
-            var registeredEntity = _GetOrRegisterEntity(entity);
-            // the fact of registering again clears the modified state, so we're... clear with that
-            MemberModificationHandler.Register(registeredEntity, Mapping);
-            return registeredEntity;
-        }
-
-        /// <summary>
-        /// Registers an entity for update
-        /// The entity will be updated only if some of its members have changed after the registration
-        /// </summary>
-        /// <param name="entity"></param>
-        /// <param name="entityOriginalState"></param>
-        internal void RegisterUpdate(object entity, object entityOriginalState)
-        {
-            RegisterUpdate(entity);
-            MemberModificationHandler.Register(entity, entityOriginalState, Mapping);
-        }
-
-        /// <summary>
-        /// Clears the current state, and marks the object as clean
-        /// </summary>
-        /// <param name="entity"></param>
-        internal void RegisterUpdateAgain(object entity)
-        {
-            MemberModificationHandler.ClearModified(entity, Mapping);
-        }
-
-        /// <summary>
-        /// Registers an entity for delete
-        /// </summary>
-        /// <param name="entity"></param>
-        internal void RegisterDelete(object entity)
-        {
-            entityTracker.RegisterToDelete(entity);
-        }
-
-        /// <summary>
-        /// Unregisters entity after deletion
-        /// </summary>
-        /// <param name="entity"></param>
-        internal void UnregisterDelete(object entity)
-        {
-            entityTracker.RegisterDeleted(entity);
-        }
-
-        #endregion
-
-        /// <summary>
-        /// Changed object determine 
-        /// </summary>
-        /// <returns>Lists of inserted, updated, deleted objects</returns>
-        public ChangeSet GetChangeSet()
-        {
-            var inserts = new List<object>();
-            var updates = new List<object>();
-            var deletes = new List<object>();
-            foreach (var entityTrack in entityTracker.EnumerateAll())
-            {
-                switch (entityTrack.EntityState)
-                {
-                    case EntityState.ToInsert:
-                        inserts.Add(entityTrack.Entity);
-                        break;
-                    case EntityState.ToWatch:
-                        if (MemberModificationHandler.IsModified(entityTrack.Entity, Mapping))
-                            updates.Add(entityTrack.Entity);
-                        break;
-                    case EntityState.ToDelete:
-                        deletes.Add(entityTrack.Entity);
-                        break;
-                    default:
-                        throw new ArgumentOutOfRangeException();
-                }
-            }
-            return new ChangeSet(inserts, updates, deletes);
-        }
-
-        /// <summary>
-        /// use ExecuteCommand to call raw SQL
-        /// </summary>
-        public int ExecuteCommand(string command, params object[] parameters)
-        {
-            var directQuery = QueryBuilder.GetDirectQuery(command, new QueryContext(this));
-            return QueryRunner.Execute(directQuery, parameters);
-        }
-
-        /// <summary>
-        /// Execute raw SQL query and return object
-        /// </summary>
-        public IEnumerable<TResult> ExecuteQuery<TResult>(string query, params object[] parameters) where TResult : class, new()
-        {
-            //GetTable<TResult>();
-            foreach (TResult result in ExecuteQuery(typeof(TResult), query, parameters))
-                yield return result;
-        }
-
-        public IEnumerable ExecuteQuery(Type elementType, string query, params object[] parameters)
-        {
-            var queryContext = new QueryContext(this);
-            var directQuery = QueryBuilder.GetDirectQuery(query, queryContext);
-            return QueryRunner.ExecuteSelect(elementType, directQuery, parameters);
-        }
-
-        /// <summary>
-        /// Gets or sets the load options
-        /// </summary>
-        [DbLinqToDo]
-        public DataLoadOptions LoadOptions { get; set; }
-
-        public DbTransaction Transaction { get; set; }
-
-        /// <summary>
-        /// Runs the given reader and returns columns.
-        /// </summary>
-        /// <typeparam name="TResult">The type of the result.</typeparam>
-        /// <param name="reader">The reader.</param>
-        /// <returns></returns>
-        public IEnumerable<TResult> Translate<TResult>(DbDataReader reader)
-        {
-            foreach (TResult result in Translate(typeof(TResult), reader))
-                yield return result;
-        }
-
-        public IMultipleResults Translate(DbDataReader reader)
-        {
-            throw new NotImplementedException();
-        }
-
-        public IEnumerable Translate(Type elementType, DbDataReader reader)
-        {
-            return QueryRunner.EnumerateResult(elementType, reader, this);
-        }
-
-        public void Dispose()
-        {
-            //connection closing should not be done here.
-            //read: http://msdn2.microsoft.com/en-us/library/bb292288.aspx
-        }
-
-        [DbLinqToDo]
-        protected virtual void Dispose(bool disposing)
-        {
-            throw new NotImplementedException();
-        }
-
-        /// <summary>
-        /// Creates a IDbDataAdapter. Used internally by Vendors
-        /// </summary>
-        /// <returns></returns>
-        internal IDbDataAdapter CreateDataAdapter()
-        {
-            return DatabaseContext.CreateDataAdapter();
-        }
-
-        /// <summary>
-        /// Sets a TextWriter where generated SQL commands are written
-        /// </summary>
-        public TextWriter Log { get; set; }
-
-        /// <summary>
-        /// Writes text on Log (if not null)
-        /// Internal helper
-        /// </summary>
-        /// <param name="text"></param>
-        internal void WriteLog(string text)
-        {
-            if (Log != null)
-                Log.WriteLine(text);
-        }
-
-        /// <summary>
-        /// Write an IDbCommand to Log (if non null)
-        /// </summary>
-        /// <param name="command"></param>
-        internal void WriteLog(IDbCommand command)
-        {
-            if (Log != null)
-            {
-                Log.WriteLine(command.CommandText);
-                foreach (IDbDataParameter parameter in command.Parameters)
-                    WriteLog(parameter);
-                Log.Write("--");
-                Log.Write(" Context: {0}", Vendor.VendorName);
-                Log.Write(" Model: {0}", Mapping.GetType().Name);
-                Log.Write(" Build: {0}", Assembly.GetExecutingAssembly().GetName().Version);
-                Log.WriteLine();
-            }
-        }
-
-        /// <summary>
-        /// Writes and IDbDataParameter to Log (if non null)
-        /// </summary>
-        /// <param name="parameter"></param>
-        internal void WriteLog(IDbDataParameter parameter)
-        {
-            if (Log != null)
-            {
-                // -- @p0: Input Int (Size = 0; Prec = 0; Scale = 0) [2]
-                // -- <name>: <direction> <type> (...) [<value>]
-                Log.WriteLine("-- {0}: {1} {2} (Size = {3}; Prec = {4}; Scale = {5}) [{6}]",
-                    parameter.ParameterName, parameter.Direction, parameter.DbType,
-                    parameter.Size, parameter.Precision, parameter.Scale, parameter.Value);
-            }
-        }
-
-
-        [DbLinqToDo]
-        public bool ObjectTrackingEnabled
-        {
-            get { throw new NotImplementedException(); }
-            set { throw new NotImplementedException(); }
-        }
-
-        [DbLinqToDo]
-        public int CommandTimeout
-        {
-            get { throw new NotImplementedException(); }
-            set { throw new NotImplementedException(); }
-        }
-
-        [DbLinqToDo]
-        public bool DeferredLoadingEnabled
-        {
-            get { throw new NotImplementedException(); }
-            set { throw new NotImplementedException(); }
-        }
-
-        [DbLinqToDo]
-        public ChangeConflictCollection ChangeConflicts
-        {
-            get { throw new NotImplementedException(); }
-        }
-
-        [DbLinqToDo]
-        public DbCommand GetCommand(IQueryable query)
-        {
-            var qp = query.Provider as QueryProvider;
-            if (qp == null)
-                throw new InvalidOperationException();
-
-            IDbCommand dbCommand = qp.GetQuery(null).GetCommand().Command;
-            if (!(dbCommand is DbCommand))
-                throw new InvalidOperationException();
-
-            return (DbCommand)dbCommand;
-        }
-
-        [DbLinqToDo]
-        public void Refresh(RefreshMode mode, IEnumerable entities)
-        {
-            throw new NotImplementedException();
-        }
-
-        [DbLinqToDo]
-        public void Refresh(RefreshMode mode, params object[] entities)
-        {
-            throw new NotImplementedException();
-        }
-
-        [DbLinqToDo]
-        public void Refresh(RefreshMode mode, object entity)
-        {
-            throw new NotImplementedException();
-        }
-
-        [DbLinqToDo]
-        public void DeleteDatabase()
-        {
-            throw new NotImplementedException();
-        }
-
-        [DbLinqToDo]
-        public void CreateDatabase()
-        {
-            throw new NotImplementedException();
-        }
-
-        [DbLinqToDo]
-        protected internal IQueryable<TResult> CreateMethodCallQuery<TResult>(object instance, MethodInfo methodInfo, params object[] parameters)
-        {
-            throw new NotImplementedException();
-        }
-
-        [DbLinqToDo]
-        protected internal void ExecuteDynamicDelete(object entity)
-        {
-            throw new NotImplementedException();
-        }
-
-        [DbLinqToDo]
-        protected internal void ExecuteDynamicInsert(object entity)
-        {
-            throw new NotImplementedException();
-        }
-
-        [DbLinqToDo]
-        protected internal void ExecuteDynamicUpdate(object entity)
-        {
-            throw new NotImplementedException();
-        }
-    }
-}
+#region MIT license\r
+// \r
+// MIT license\r
+//\r
+// Copyright (c) 2007-2008 Jiri Moudry, Pascal Craponne\r
+// \r
+// Permission is hereby granted, free of charge, to any person obtaining a copy\r
+// of this software and associated documentation files (the "Software"), to deal\r
+// in the Software without restriction, including without limitation the rights\r
+// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\r
+// copies of the Software, and to permit persons to whom the Software is\r
+// furnished to do so, subject to the following conditions:\r
+// \r
+// The above copyright notice and this permission notice shall be included in\r
+// all copies or substantial portions of the Software.\r
+// \r
+// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\r
+// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\r
+// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\r
+// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\r
+// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\r
+// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\r
+// THE SOFTWARE.\r
+// \r
+#endregion\r
+\r
+using System;\r
+using System.Collections;\r
+using System.Data;\r
+using System.Data.Common;\r
+using System.Data.Linq;\r
+using System.Data.Linq.Mapping;\r
+using System.Linq.Expressions;\r
+using System.Collections.Generic;\r
+using System.IO;\r
+using System.Linq;\r
+using System.Reflection;\r
+using System.Reflection.Emit;\r
+\r
+#if MONO_STRICT\r
+using AttributeMappingSource  = System.Data.Linq.Mapping.AttributeMappingSource;\r
+#else\r
+using AttributeMappingSource  = DbLinq.Data.Linq.Mapping.AttributeMappingSource;\r
+#endif\r
+\r
+using DbLinq;\r
+using DbLinq.Data.Linq;\r
+using DbLinq.Data.Linq.Database;\r
+using DbLinq.Data.Linq.Database.Implementation;\r
+using DbLinq.Data.Linq.Identity;\r
+using DbLinq.Data.Linq.Implementation;\r
+using DbLinq.Data.Linq.Mapping;\r
+using DbLinq.Data.Linq.Sugar;\r
+using DbLinq.Factory;\r
+using DbLinq.Util;\r
+using DbLinq.Vendor;\r
+\r
+#if MONO_STRICT\r
+namespace System.Data.Linq\r
+#else\r
+namespace DbLinq.Data.Linq\r
+#endif\r
+{\r
+    public partial class DataContext : IDisposable\r
+    {\r
+        //private readonly Dictionary<string, ITable> _tableMap = new Dictionary<string, ITable>();\r
+        private readonly Dictionary<Type, ITable> _tableMap = new Dictionary<Type, ITable>();\r
+\r
+        public MetaModel Mapping { get; private set; }\r
+        // PC question: at ctor, we get a IDbConnection and the Connection property exposes a DbConnection\r
+        //              WTF?\r
+        public DbConnection Connection { get { return DatabaseContext.Connection as DbConnection; } }\r
+\r
+        // all properties below are set public to optionally be injected\r
+        internal IVendor Vendor { get; set; }\r
+        internal IQueryBuilder QueryBuilder { get; set; }\r
+        internal IQueryRunner QueryRunner { get; set; }\r
+        internal IMemberModificationHandler MemberModificationHandler { get; set; }\r
+        internal IDatabaseContext DatabaseContext { get; private set; }\r
+        // /all properties...\r
+\r
+        private bool objectTrackingEnabled = true;\r
+        private bool deferredLoadingEnabled = true;\r
+\r
+        private bool queryCacheEnabled = false;\r
+\r
+        /// <summary>\r
+        /// Disable the QueryCache: this is surely good for rarely used Select, since preparing\r
+        /// the SelectQuery to be cached could require more time than build the sql from scratch.\r
+        /// </summary>\r
+        [DBLinqExtended]\r
+        public bool QueryCacheEnabled \r
+        {\r
+            get { return queryCacheEnabled; }\r
+            set { queryCacheEnabled = value; }\r
+        }\r
+\r
+        private IEntityTracker currentTransactionEntities;\r
+        private IEntityTracker CurrentTransactionEntities\r
+        {\r
+            get\r
+            {\r
+                if (this.currentTransactionEntities == null)\r
+                {\r
+                    if (this.ObjectTrackingEnabled)\r
+                        this.currentTransactionEntities = new EntityTracker();\r
+                    else\r
+                        this.currentTransactionEntities = new DisabledEntityTracker();\r
+                }\r
+                return this.currentTransactionEntities;\r
+            }\r
+        }\r
+\r
+        private IEntityTracker allTrackedEntities;\r
+        private IEntityTracker AllTrackedEntities\r
+        {\r
+            get\r
+            {\r
+                if (this.allTrackedEntities == null)\r
+                {\r
+                    allTrackedEntities = ObjectTrackingEnabled\r
+                        ? (IEntityTracker) new EntityTracker()\r
+                        : (IEntityTracker) new DisabledEntityTracker();\r
+                }\r
+                return this.allTrackedEntities;\r
+            }\r
+        }\r
+\r
+        private IIdentityReaderFactory identityReaderFactory;\r
+        private readonly IDictionary<Type, IIdentityReader> identityReaders = new Dictionary<Type, IIdentityReader>();\r
+\r
+        /// <summary>\r
+        /// The default behavior creates one MappingContext.\r
+        /// </summary>\r
+        [DBLinqExtended]\r
+        internal virtual MappingContext _MappingContext { get; set; }\r
+\r
+        [DBLinqExtended]\r
+        internal IVendorProvider _VendorProvider { get; set; }\r
+\r
+        public DataContext(IDbConnection connection, MappingSource mapping)\r
+        {\r
+            Profiler.At("START DataContext(IDbConnection, MappingSource)");\r
+            Init(new DatabaseContext(connection), mapping, null);\r
+            Profiler.At("END DataContext(IDbConnection, MappingSource)");\r
+        }\r
+\r
+        public DataContext(IDbConnection connection)\r
+        {\r
+            Profiler.At("START DataContext(IDbConnection)");\r
+            if (connection == null)\r
+                throw new ArgumentNullException("connection");\r
+\r
+            Init(new DatabaseContext(connection), null, null);\r
+            Profiler.At("END DataContext(IDbConnection)");\r
+        }\r
+\r
+        [DbLinqToDo]\r
+        public DataContext(string fileOrServerOrConnection, MappingSource mapping)\r
+        {\r
+            Profiler.At("START DataContext(string, MappingSource)");\r
+            if (fileOrServerOrConnection == null)\r
+                throw new ArgumentNullException("fileOrServerOrConnection");\r
+            if (mapping == null)\r
+                throw new ArgumentNullException("mapping");\r
+\r
+            if (File.Exists(fileOrServerOrConnection))\r
+                throw new NotImplementedException("File names not supported.");\r
+\r
+            // Is this a decent server name check?\r
+            // It assumes that the connection string will have at least 2\r
+            // parameters (separated by ';')\r
+            if (!fileOrServerOrConnection.Contains(";"))\r
+                throw new NotImplementedException("Server name not supported.");\r
+\r
+            // Assume it's a connection string...\r
+            IVendor ivendor = GetVendor(ref fileOrServerOrConnection);\r
+\r
+            IDbConnection dbConnection = ivendor.CreateDbConnection(fileOrServerOrConnection);\r
+            Init(new DatabaseContext(dbConnection), mapping, ivendor);\r
+            Profiler.At("END DataContext(string, MappingSource)");\r
+        }\r
+\r
+        /// <summary>\r
+        /// Construct DataContext, given a connectionString.\r
+        /// To determine which DB type to go against, we look for 'DbLinqProvider=xxx' substring.\r
+        /// If not found, we assume that we are dealing with MS Sql Server.\r
+        /// \r
+        /// Valid values are names of provider DLLs (or any other DLL containing an IVendor implementation)\r
+        /// DbLinqProvider=Mysql\r
+        /// DbLinqProvider=Oracle etc.\r
+        /// </summary>\r
+        /// <param name="connectionString">specifies file or server connection</param>\r
+        [DbLinqToDo]\r
+        public DataContext(string connectionString)\r
+        {\r
+            Profiler.At("START DataContext(string)");\r
+            IVendor ivendor = GetVendor(ref connectionString);\r
+\r
+            IDbConnection dbConnection = ivendor.CreateDbConnection(connectionString);\r
+            Init(new DatabaseContext(dbConnection), null, ivendor);\r
+\r
+            Profiler.At("END DataContext(string)");\r
+        }\r
+\r
+        private IVendor GetVendor(ref string connectionString)\r
+        {\r
+            if (connectionString == null)\r
+                throw new ArgumentNullException("connectionString");\r
+\r
+            Assembly assy;\r
+            string vendorClassToLoad;\r
+            GetVendorInfo(ref connectionString, out assy, out vendorClassToLoad);\r
+\r
+            var types =\r
+                from type in assy.GetTypes()\r
+                where type.Name.ToLowerInvariant() == vendorClassToLoad.ToLowerInvariant() &&\r
+                    type.GetInterfaces().Contains(typeof(IVendor)) &&\r
+                    type.GetConstructor(Type.EmptyTypes) != null\r
+                select type;\r
+            if (!types.Any())\r
+            {\r
+                throw new ArgumentException(string.Format("Found no IVendor class in assembly `{0}' named `{1}' having a default constructor.",\r
+                    assy.GetName().Name, vendorClassToLoad));\r
+            }\r
+            else if (types.Count() > 1)\r
+            {\r
+                throw new ArgumentException(string.Format("Found too many IVendor classes in assembly `{0}' named `{1}' having a default constructor.",\r
+                    assy.GetName().Name, vendorClassToLoad));\r
+            }\r
+            return (IVendor) Activator.CreateInstance(types.First());\r
+        }\r
+\r
+        private void GetVendorInfo(ref string connectionString, out Assembly assembly, out string typeName)\r
+        {\r
+            System.Text.RegularExpressions.Regex reProvider\r
+                = new System.Text.RegularExpressions.Regex(@"DbLinqProvider=([\w\.]+);?");\r
+\r
+            string assemblyName = null;\r
+            string vendor;\r
+            if (!reProvider.IsMatch(connectionString))\r
+            {\r
+                vendor       = "SqlServer";\r
+                assemblyName = "DbLinq.SqlServer";\r
+            }\r
+            else\r
+            {\r
+                var match    = reProvider.Match(connectionString);\r
+                vendor       = match.Groups[1].Value;\r
+                assemblyName = "DbLinq." + vendor;\r
+\r
+                //plain DbLinq - non MONO: \r
+                //IVendor classes are in DLLs such as "DbLinq.MySql.dll"\r
+                if (vendor.Contains("."))\r
+                {\r
+                    //already fully qualified DLL name?\r
+                    throw new ArgumentException("Please provide a short name, such as 'MySql', not '" + vendor + "'");\r
+                }\r
+\r
+                //shorten: "DbLinqProvider=X;Server=Y" -> ";Server=Y"\r
+                connectionString = reProvider.Replace(connectionString, "");\r
+            }\r
+\r
+            typeName = vendor + "Vendor";\r
+\r
+            try\r
+            {\r
+#if MONO_STRICT\r
+                assembly = typeof (DataContext).Assembly; // System.Data.Linq.dll\r
+#else\r
+                assembly = Assembly.Load(assemblyName);\r
+#endif\r
+            }\r
+            catch (Exception e)\r
+            {\r
+                throw new ArgumentException(\r
+                        string.Format(\r
+                            "Unable to load the `{0}' DbLinq vendor within assembly '{1}.dll'.",\r
+                            assemblyName, vendor),\r
+                        "connectionString", e);\r
+            }\r
+        }\r
+\r
+        private void Init(IDatabaseContext databaseContext, MappingSource mappingSource, IVendor vendor)\r
+        {\r
+            if (databaseContext == null)\r
+                throw new ArgumentNullException("databaseContext");\r
+\r
+            // Yes, .NET throws an NRE for this.  Why it's not ArgumentNullException, I couldn't tell you.\r
+            if (databaseContext.Connection.ConnectionString == null)\r
+                throw new NullReferenceException();\r
+\r
+            string connectionString = databaseContext.Connection.ConnectionString;\r
+            _VendorProvider = ObjectFactory.Get<IVendorProvider>();\r
+            Vendor = vendor ?? \r
+                (connectionString != null ? GetVendor(ref connectionString) : null) ??\r
+                _VendorProvider.FindVendorByProviderType(typeof(SqlClient.Sql2005Provider));\r
+            \r
+            DatabaseContext = databaseContext;\r
+\r
+            MemberModificationHandler = ObjectFactory.Create<IMemberModificationHandler>(); // not a singleton: object is stateful\r
+            QueryBuilder = ObjectFactory.Get<IQueryBuilder>();\r
+            QueryRunner = ObjectFactory.Get<IQueryRunner>();\r
+\r
+            //EntityMap = ObjectFactory.Create<IEntityMap>();\r
+            identityReaderFactory = ObjectFactory.Get<IIdentityReaderFactory>();\r
+\r
+            _MappingContext = new MappingContext();\r
+\r
+            // initialize the mapping information\r
+            if (mappingSource == null)\r
+                mappingSource = new AttributeMappingSource();\r
+            Mapping = mappingSource.GetModel(GetType());\r
+        }\r
+\r
+        /// <summary>\r
+        /// Checks if the table is allready mapped or maps it if not.\r
+        /// </summary>\r
+        /// <param name="tableType">Type of the table.</param>\r
+        /// <exception cref="InvalidOperationException">Thrown if the table is not mappable.</exception>\r
+        private void CheckTableMapping(Type tableType)\r
+        {\r
+            //This will throw an exception if the table is not found\r
+            if(Mapping.GetTable(tableType) == null)\r
+            {\r
+                throw new InvalidOperationException("The type '" + tableType.Name + "' is not mapped as a Table.");\r
+            }\r
+        }\r
+\r
+        /// <summary>\r
+        /// Returns a Table for the type TEntity.\r
+        /// </summary>\r
+        /// <exception cref="InvalidOperationException">If the type TEntity is not mappable as a Table.</exception>\r
+        /// <typeparam name="TEntity">The table type.</typeparam>\r
+        public Table<TEntity> GetTable<TEntity>() where TEntity : class\r
+        {\r
+            return (Table<TEntity>)GetTable(typeof(TEntity));\r
+        }\r
+\r
+        /// <summary>\r
+        /// Returns a Table for the given type.\r
+        /// </summary>\r
+        /// <param name="type">The table type.</param>\r
+        /// <exception cref="InvalidOperationException">If the type is not mappable as a Table.</exception>\r
+        public ITable GetTable(Type type)\r
+        {\r
+            Profiler.At("DataContext.GetTable(typeof({0}))", type != null ? type.Name : null);\r
+            ITable tableExisting;\r
+            if (_tableMap.TryGetValue(type, out tableExisting))\r
+                return tableExisting;\r
+\r
+            //Check for table mapping\r
+            CheckTableMapping(type);\r
+\r
+            var tableNew = Activator.CreateInstance(\r
+                              typeof(Table<>).MakeGenericType(type)\r
+                              , BindingFlags.NonPublic | BindingFlags.Instance\r
+                              , null\r
+                              , new object[] { this }\r
+                              , System.Globalization.CultureInfo.CurrentCulture) as ITable;\r
+\r
+            _tableMap[type] = tableNew;\r
+            return tableNew;\r
+        }\r
+\r
+        public void SubmitChanges()\r
+        {\r
+            SubmitChanges(ConflictMode.FailOnFirstConflict);\r
+        }\r
+\r
+        /// <summary>\r
+        /// Pings database\r
+        /// </summary>\r
+        /// <returns></returns>\r
+        public bool DatabaseExists()\r
+        {\r
+            try\r
+            {\r
+                return Vendor.Ping(this);\r
+            }\r
+            catch (Exception)\r
+            {\r
+                return false;\r
+            }\r
+        }\r
+\r
+        /// <summary>\r
+        /// Commits all pending changes to database \r
+        /// </summary>\r
+        /// <param name="failureMode"></param>\r
+        public virtual void SubmitChanges(ConflictMode failureMode)\r
+        {\r
+            if (this.objectTrackingEnabled == false)\r
+                throw new InvalidOperationException("Object tracking is not enabled for the current data context instance.");\r
+            using (DatabaseContext.OpenConnection()) //ConnMgr will close connection for us\r
+            {\r
+                if (Transaction != null)\r
+                    SubmitChangesImpl(failureMode);\r
+                else\r
+                {\r
+                    using (IDbTransaction transaction = DatabaseContext.CreateTransaction())\r
+                    {\r
+                        try\r
+                        {\r
+                            Transaction = (DbTransaction) transaction;\r
+                            SubmitChangesImpl(failureMode);\r
+                            // TODO: handle conflicts (which can only occur when concurrency mode is implemented)\r
+                            transaction.Commit();\r
+                        }\r
+                        finally\r
+                        {\r
+                            Transaction = null;\r
+                        }\r
+                    }\r
+                }\r
+            }\r
+        }\r
+\r
+        void SubmitChangesImpl(ConflictMode failureMode)\r
+        {\r
+            var queryContext = new QueryContext(this);\r
+\r
+            // There's no sense in updating an entity when it's going to \r
+            // be deleted in the current transaction, so do deletes first.\r
+            foreach (var entityTrack in CurrentTransactionEntities.EnumerateAll().ToList())\r
+            {\r
+                switch (entityTrack.EntityState)\r
+                {\r
+                    case EntityState.ToDelete:\r
+                        var deleteQuery = QueryBuilder.GetDeleteQuery(entityTrack.Entity, queryContext);\r
+                        QueryRunner.Delete(entityTrack.Entity, deleteQuery);\r
+\r
+                        UnregisterDelete(entityTrack.Entity);\r
+                        AllTrackedEntities.RegisterToDelete(entityTrack.Entity);\r
+                        AllTrackedEntities.RegisterDeleted(entityTrack.Entity);\r
+                        break;\r
+                    default:\r
+                        // ignore.\r
+                        break;\r
+                }\r
+            }\r
+            foreach (var entityTrack in CurrentTransactionEntities.EnumerateAll()\r
+                    .Concat(AllTrackedEntities.EnumerateAll())\r
+                    .ToList())\r
+            {\r
+                switch (entityTrack.EntityState)\r
+                {\r
+                    case EntityState.ToInsert:\r
+                        foreach (var toInsert in GetReferencedObjects(entityTrack.Entity))\r
+                        {\r
+                            InsertEntity(toInsert, queryContext);\r
+                        }\r
+                        break;\r
+                    case EntityState.ToWatch:\r
+                        foreach (var toUpdate in GetReferencedObjects(entityTrack.Entity))\r
+                        {\r
+                            UpdateEntity(toUpdate, queryContext);\r
+                        }\r
+                        break;\r
+                    default:\r
+                        throw new ArgumentOutOfRangeException();\r
+                }\r
+            }\r
+        }\r
+\r
+        private IEnumerable<object> GetReferencedObjects(object value)\r
+        {\r
+            var values = new EntitySet<object>();\r
+            FillReferencedObjects(value, values);\r
+            return values;\r
+        }\r
+\r
+        // Breadth-first traversal of an object graph\r
+        private void FillReferencedObjects(object parent, EntitySet<object> values)\r
+        {\r
+            if (parent == null)\r
+                return;\r
+            var children = new Queue<object>();\r
+                       children.Enqueue(parent);\r
+                       while (children.Count > 0)\r
+                       {\r
+                object value = children.Dequeue();\r
+                values.Add(value);\r
+                IEnumerable<MetaAssociation> associationList = Mapping.GetMetaType(value.GetType()).Associations.Where(a => !a.IsForeignKey);\r
+                if (associationList.Any())\r
+                           {\r
+                                   foreach (MetaAssociation association in associationList)\r
+                    {\r
+                        var memberData = association.ThisMember;\r
+                        var entitySetValue = memberData.Member.GetMemberValue(value);\r
+\r
+                        if (entitySetValue != null)\r
+                        {\r
+                                                   var hasLoadedOrAssignedValues = entitySetValue.GetType().GetProperty("HasLoadedOrAssignedValues");\r
+                                                   if (!((bool)hasLoadedOrAssignedValues.GetValue(entitySetValue, null)))\r
+                                                           continue;   // execution deferred; ignore.\r
+                                                   foreach (var o in ((IEnumerable)entitySetValue))\r
+                                                           children.Enqueue(o);\r
+                                           }\r
+                    }\r
+                }\r
+                       }\r
+        }\r
+\r
+        private void InsertEntity(object entity, QueryContext queryContext)\r
+        {\r
+            var insertQuery = QueryBuilder.GetInsertQuery(entity, queryContext);\r
+            QueryRunner.Insert(entity, insertQuery);\r
+            Register(entity);\r
+            UpdateReferencedObjects(entity);\r
+            MoveToAllTrackedEntities(entity, true);\r
+        }\r
+\r
+        private void UpdateEntity(object entity, QueryContext queryContext)\r
+        {\r
+            if (!AllTrackedEntities.ContainsReference(entity))\r
+                InsertEntity(entity, queryContext);\r
+            else if (MemberModificationHandler.IsModified(entity, Mapping))\r
+            {\r
+                var modifiedMembers = MemberModificationHandler.GetModifiedProperties(entity, Mapping);\r
+                var updateQuery = QueryBuilder.GetUpdateQuery(entity, modifiedMembers, queryContext);\r
+                QueryRunner.Update(entity, updateQuery, modifiedMembers);\r
+\r
+                RegisterUpdateAgain(entity);\r
+                UpdateReferencedObjects(entity);\r
+                MoveToAllTrackedEntities(entity, false);\r
+            }\r
+        }\r
+\r
+        private void UpdateReferencedObjects(object root)\r
+        {\r
+            var metaType = Mapping.GetMetaType(root.GetType());\r
+            foreach (var assoc in metaType.Associations)\r
+            {\r
+                var memberData = assoc.ThisMember;\r
+                               //This is not correct - AutoSyncing applies to auto-updating columns, such as a TimeStamp, not to foreign key associations, which is always automatically synched\r
+                               //Confirmed against default .NET l2sql - association columns are always set, even if AutoSync==AutoSync.Never\r
+                               //if (memberData.Association.ThisKey.Any(m => (m.AutoSync != AutoSync.Always) && (m.AutoSync != sync)))\r
+                //    continue;\r
+                var oks = memberData.Association.OtherKey.Select(m => m.StorageMember).ToList();\r
+                if (oks.Count == 0)\r
+                    continue;\r
+                var pks = memberData.Association.ThisKey\r
+                    .Select(m => m.StorageMember.GetMemberValue(root))\r
+                    .ToList();\r
+                if (pks.Count != oks.Count)\r
+                    throw new InvalidOperationException(\r
+                        string.Format("Count of primary keys ({0}) doesn't match count of other keys ({1}).",\r
+                            pks.Count, oks.Count));\r
+                var members = memberData.Member.GetMemberValue(root) as IEnumerable;\r
+                if (members == null)\r
+                    continue;\r
+                foreach (var member in members)\r
+                {\r
+                    for (int i = 0; i < pks.Count; ++i)\r
+                    {\r
+                        oks[i].SetMemberValue(member, pks[i]);\r
+                    }\r
+                }\r
+            }\r
+        }\r
+\r
+        private void MoveToAllTrackedEntities(object entity, bool insert)\r
+        {\r
+            if (!ObjectTrackingEnabled)\r
+                return;\r
+            if (CurrentTransactionEntities.ContainsReference(entity))\r
+            {\r
+                CurrentTransactionEntities.RegisterToDelete(entity);\r
+                if (!insert)\r
+                    CurrentTransactionEntities.RegisterDeleted(entity);\r
+            }\r
+            if (!AllTrackedEntities.ContainsReference(entity))\r
+            {\r
+                var identityReader = _GetIdentityReader(entity.GetType());\r
+                AllTrackedEntities.RegisterToWatch(entity, identityReader.GetIdentityKey(entity));\r
+            }\r
+        }\r
+\r
+        /// <summary>\r
+        /// TODO - allow generated methods to call into stored procedures\r
+        /// </summary>\r
+        [DBLinqExtended]\r
+        internal IExecuteResult _ExecuteMethodCall(DataContext context, System.Reflection.MethodInfo method, params object[] sqlParams)\r
+        {\r
+            using (DatabaseContext.OpenConnection())\r
+            {\r
+                System.Data.Linq.IExecuteResult result = Vendor.ExecuteMethodCall(context, method, sqlParams);\r
+                return result;\r
+            }\r
+        }\r
+\r
+        [DbLinqToDo]\r
+        protected IExecuteResult ExecuteMethodCall(object instance, System.Reflection.MethodInfo methodInfo, params object[] parameters)\r
+        {\r
+            throw new NotImplementedException();\r
+        }\r
+\r
+        #region Identity management\r
+\r
+        [DBLinqExtended]\r
+        internal IIdentityReader _GetIdentityReader(Type t)\r
+        {\r
+            IIdentityReader identityReader;\r
+            if (!identityReaders.TryGetValue(t, out identityReader))\r
+            {\r
+                identityReader = identityReaderFactory.GetReader(t, this);\r
+                identityReaders[t] = identityReader;\r
+            }\r
+            return identityReader;\r
+        }\r
+\r
+        [DBLinqExtended]\r
+        internal object _GetRegisteredEntity(object entity)\r
+        {\r
+            // TODO: check what is faster: by identity or by ref\r
+            var identityReader = _GetIdentityReader(entity.GetType());\r
+            var identityKey = identityReader.GetIdentityKey(entity);\r
+            if (identityKey == null) // if we don't have an entitykey here, it means that the entity has no PK\r
+                return entity;\r
+            // even \r
+            var registeredEntityTrack = \r
+                CurrentTransactionEntities.FindByIdentity(identityKey) ??\r
+                AllTrackedEntities.FindByIdentity(identityKey);\r
+            if (registeredEntityTrack != null)\r
+                return registeredEntityTrack.Entity;\r
+            return null;\r
+        }\r
+\r
+        //internal object GetRegisteredEntityByKey(IdentityKey identityKey)\r
+        //{\r
+        //    return EntityMap[identityKey];\r
+        //}\r
+\r
+        /// <summary>\r
+        /// Registers an entity in a watch state\r
+        /// </summary>\r
+        /// <param name="entity"></param>\r
+        /// <returns></returns>\r
+        [DBLinqExtended]\r
+        internal object _GetOrRegisterEntity(object entity)\r
+        {\r
+            var identityReader = _GetIdentityReader(entity.GetType());\r
+            var identityKey = identityReader.GetIdentityKey(entity);\r
+            SetEntitySetsQueries(entity);\r
+            SetEntityRefQueries(entity);\r
+\r
+            // if we have no identity, we can't track it\r
+            if (identityKey == null)\r
+                return entity;\r
+\r
+            // try to find an already registered entity and return it\r
+            var registeredEntityTrack = \r
+                CurrentTransactionEntities.FindByIdentity(identityKey) ??\r
+                AllTrackedEntities.FindByIdentity(identityKey);\r
+            if (registeredEntityTrack != null)\r
+                return registeredEntityTrack.Entity;\r
+\r
+            // otherwise, register and return\r
+            AllTrackedEntities.RegisterToWatch(entity, identityKey);\r
+            return entity;\r
+        }\r
+\r
+        readonly IDataMapper DataMapper = ObjectFactory.Get<IDataMapper>();\r
+               private void SetEntityRefQueries(object entity)\r
+               {\r
+            if (!this.deferredLoadingEnabled)\r
+                return;\r
+\r
+            // BUG: This is ignoring External Mappings from XmlMappingSource.\r
+\r
+                       Type thisType = entity.GetType();\r
+                       IEnumerable<MetaAssociation> associationList = Mapping.GetMetaType(entity.GetType()).Associations.Where(a => a.IsForeignKey);\r
+                       foreach (MetaAssociation association in associationList)\r
+                       {\r
+                               //example of entityRef:Order.Employee\r
+                               var memberData = association.ThisMember;\r
+                               Type otherTableType = association.OtherType.Type;\r
+                               ParameterExpression p = Expression.Parameter(otherTableType, "other");\r
+\r
+                               var otherTable = GetTable(otherTableType);\r
+\r
+                               //ie:EmployeeTerritories.EmployeeID\r
+                               var foreignKeys = memberData.Association.ThisKey;\r
+                               BinaryExpression predicate = null;\r
+                               var otherPKs = memberData.Association.OtherKey;\r
+                               IEnumerator<MetaDataMember> otherPKEnumerator = otherPKs.GetEnumerator();\r
+\r
+                               if (otherPKs.Count != foreignKeys.Count)\r
+                                       throw new InvalidOperationException("Foreign keys don't match ThisKey");\r
+                               foreach (MetaDataMember key in foreignKeys)\r
+                               {\r
+                                       otherPKEnumerator.MoveNext();\r
+\r
+                                       var thisForeignKeyProperty = (PropertyInfo)key.Member;\r
+                                       object thisForeignKeyValue = thisForeignKeyProperty.GetValue(entity, null);\r
+\r
+                                       if (thisForeignKeyValue != null)\r
+                                       {\r
+                                               BinaryExpression keyPredicate;\r
+                                               if (!(thisForeignKeyProperty.PropertyType.IsNullable()))\r
+                                               {\r
+                                                       keyPredicate = Expression.Equal(Expression.MakeMemberAccess(p, otherPKEnumerator.Current.Member),\r
+                                                                                                                                               Expression.Constant(thisForeignKeyValue));\r
+                                               }\r
+                                               else\r
+                                               {\r
+                                                       var ValueProperty = thisForeignKeyProperty.PropertyType.GetProperty("Value");\r
+                                                       keyPredicate = Expression.Equal(Expression.MakeMemberAccess(p, otherPKEnumerator.Current.Member),\r
+                                                                                                                                        Expression.Constant(ValueProperty.GetValue(thisForeignKeyValue, null)));\r
+                                               }\r
+\r
+                                               if (predicate == null)\r
+                                                       predicate = keyPredicate;\r
+                                               else\r
+                                                       predicate = Expression.And(predicate, keyPredicate);\r
+                                       }\r
+                               }\r
+                               IEnumerable query = null;\r
+                               if (predicate != null)\r
+                               {\r
+                                       query = GetOtherTableQuery(predicate, p, otherTableType, otherTable) as IEnumerable;\r
+                                       //it would be interesting surround the above query with a .Take(1) expression for performance.\r
+                               }\r
+\r
+                               // If no separate Storage is specified, use the member directly\r
+                               MemberInfo storage = memberData.StorageMember;\r
+                               if (storage == null)\r
+                                       storage = memberData.Member;\r
+\r
+                                // Check that the storage is a field or a writable property\r
+                               if (!(storage is FieldInfo) && !(storage is PropertyInfo && ((PropertyInfo)storage).CanWrite)) {\r
+                                       throw new InvalidOperationException(String.Format(\r
+                                               "Member {0}.{1} is not a field nor a writable property",\r
+                                               storage.DeclaringType, storage.Name));\r
+                               }\r
+\r
+                               Type storageType = storage.GetMemberType();\r
+\r
+                               object entityRefValue = null;\r
+                               if (query != null)\r
+                                       entityRefValue = Activator.CreateInstance(storageType, query);\r
+                               else\r
+                                       entityRefValue = Activator.CreateInstance(storageType);\r
+\r
+                               storage.SetMemberValue(entity, entityRefValue);\r
+                       }\r
+               }\r
+\r
+        /// <summary>\r
+        /// This method is executed when the entity is being registered. Each EntitySet property has a internal query that can be set using the EntitySet.SetSource method.\r
+        /// Here we set the query source of each EntitySetProperty\r
+        /// </summary>\r
+        /// <param name="entity"></param>\r
+        private void SetEntitySetsQueries(object entity)\r
+        {\r
+            if (!this.deferredLoadingEnabled)\r
+                return;\r
+\r
+            // BUG: This is ignoring External Mappings from XmlMappingSource.\r
+\r
+                       IEnumerable<MetaAssociation> associationList = Mapping.GetMetaType(entity.GetType()).Associations.Where(a => !a.IsForeignKey);\r
+\r
+                       if (associationList.Any())\r
+                       {\r
+                               foreach (MetaAssociation association in associationList)\r
+                {\r
+                                       //example of entitySet: Employee.EmployeeTerritories\r
+                                       var memberData = association.ThisMember;\r
+                                       Type otherTableType = association.OtherType.Type;\r
+                    ParameterExpression p = Expression.Parameter(otherTableType, "other");\r
+\r
+                    //other table:EmployeeTerritories\r
+                    var otherTable = GetTable(otherTableType);\r
+\r
+                                       var otherKeys = memberData.Association.OtherKey;\r
+                                       var thisKeys = memberData.Association.ThisKey;\r
+                    if (otherKeys.Count != thisKeys.Count)\r
+                        throw new InvalidOperationException("This keys don't match OtherKey");\r
+                    BinaryExpression predicate = null;\r
+                    IEnumerator<MetaDataMember> thisKeyEnumerator = thisKeys.GetEnumerator();\r
+                                       foreach (MetaDataMember otherKey in otherKeys)\r
+                    {\r
+                        thisKeyEnumerator.MoveNext();\r
+                        //other table member:EmployeeTerritories.EmployeeID\r
+                                               var otherTableMember = (PropertyInfo)otherKey.Member;\r
+\r
+                        BinaryExpression keyPredicate;\r
+                        if (!(otherTableMember.PropertyType.IsNullable()))\r
+                        {\r
+                            keyPredicate = Expression.Equal(Expression.MakeMemberAccess(p, otherTableMember),\r
+                                                                        Expression.Constant(thisKeyEnumerator.Current.Member.GetMemberValue(entity)));\r
+                        }\r
+                        else\r
+                        {\r
+                            var ValueProperty = otherTableMember.PropertyType.GetProperty("Value");\r
+                            keyPredicate = Expression.Equal(Expression.MakeMemberAccess(\r
+                                                                        Expression.MakeMemberAccess(p, otherTableMember),\r
+                                                                        ValueProperty),\r
+                                                                     Expression.Constant(thisKeyEnumerator.Current.Member.GetMemberValue(entity)));\r
+                        }\r
+                        if (predicate == null)\r
+                            predicate = keyPredicate;\r
+                        else\r
+                            predicate = Expression.And(predicate, keyPredicate);\r
+                    }\r
+\r
+                    var query = GetOtherTableQuery(predicate, p, otherTableType, otherTable);\r
+\r
+                                       var entitySetValue = memberData.Member.GetMemberValue(entity);\r
+\r
+                    if (entitySetValue == null)\r
+                    {\r
+                                               entitySetValue = Activator.CreateInstance(memberData.Member.GetMemberType());\r
+                                               memberData.Member.SetMemberValue(entity, entitySetValue);\r
+                    }\r
+\r
+                    var hasLoadedOrAssignedValues = entitySetValue.GetType().GetProperty("HasLoadedOrAssignedValues");\r
+                    if ((bool)hasLoadedOrAssignedValues.GetValue(entitySetValue, null))\r
+                        continue;\r
+\r
+                    var setSourceMethod = entitySetValue.GetType().GetMethod("SetSource");\r
+                    setSourceMethod.Invoke(entitySetValue, new[] { query });\r
+                    //employee.EmployeeTerritories.SetSource(Table[EmployeesTerritories].Where(other=>other.employeeID="WARTH"))\r
+                }\r
+            }\r
+        }\r
+\r
+               private static MethodInfo _WhereMethod = typeof(Queryable).GetMethods().First(m => m.Name == "Where");\r
+        internal object GetOtherTableQuery(Expression predicate, ParameterExpression parameter, Type otherTableType, IQueryable otherTable)\r
+        {\r
+            //predicate: other.EmployeeID== "WARTH"\r
+            Expression lambdaPredicate = Expression.Lambda(predicate, parameter);\r
+            //lambdaPredicate: other=>other.EmployeeID== "WARTH"\r
+\r
+                       Expression call = Expression.Call(_WhereMethod.MakeGenericMethod(otherTableType), otherTable.Expression, lambdaPredicate);\r
+            //Table[EmployeesTerritories].Where(other=>other.employeeID="WARTH")\r
+\r
+            return otherTable.Provider.CreateQuery(call);\r
+        }\r
+\r
+        #endregion\r
+\r
+        #region Insert/Update/Delete management\r
+\r
+        /// <summary>\r
+        /// Registers an entity for insert\r
+        /// </summary>\r
+        /// <param name="entity"></param>\r
+        internal void RegisterInsert(object entity)\r
+        {\r
+            CurrentTransactionEntities.RegisterToInsert(entity);\r
+        }\r
+\r
+        private void DoRegisterUpdate(object entity)\r
+        {\r
+            if (entity == null)\r
+                throw new ArgumentNullException("entity");\r
+\r
+            if (!this.objectTrackingEnabled)\r
+                return;\r
+\r
+            var identityReader = _GetIdentityReader(entity.GetType());\r
+            var identityKey = identityReader.GetIdentityKey(entity);\r
+            // if we have no key, we can not watch\r
+            if (identityKey == null || identityKey.Keys.Count == 0)\r
+                return;\r
+            // register entity\r
+            AllTrackedEntities.RegisterToWatch(entity, identityKey);\r
+        }\r
+\r
+        /// <summary>\r
+        /// Registers an entity for update\r
+        /// The entity will be updated only if some of its members have changed after the registration\r
+        /// </summary>\r
+        /// <param name="entity"></param>\r
+        internal void RegisterUpdate(object entity)\r
+        {\r
+            DoRegisterUpdate(entity);\r
+                       MemberModificationHandler.Register(entity, Mapping);\r
+        }\r
+\r
+        /// <summary>\r
+        /// Registers or re-registers an entity and clears its state\r
+        /// </summary>\r
+        /// <param name="entity"></param>\r
+        /// <returns></returns>\r
+        internal object Register(object entity)\r
+        {\r
+            if (! this.objectTrackingEnabled)\r
+                return entity;\r
+            var registeredEntity = _GetOrRegisterEntity(entity);\r
+            // the fact of registering again clears the modified state, so we're... clear with that\r
+            MemberModificationHandler.Register(registeredEntity, Mapping);\r
+            return registeredEntity;\r
+        }\r
+\r
+        /// <summary>\r
+        /// Registers an entity for update\r
+        /// The entity will be updated only if some of its members have changed after the registration\r
+        /// </summary>\r
+        /// <param name="entity"></param>\r
+        /// <param name="entityOriginalState"></param>\r
+        internal void RegisterUpdate(object entity, object entityOriginalState)\r
+        {\r
+            if (!this.objectTrackingEnabled)\r
+                return;\r
+            DoRegisterUpdate(entity);\r
+            MemberModificationHandler.Register(entity, entityOriginalState, Mapping);\r
+        }\r
+\r
+        /// <summary>\r
+        /// Clears the current state, and marks the object as clean\r
+        /// </summary>\r
+        /// <param name="entity"></param>\r
+        internal void RegisterUpdateAgain(object entity)\r
+        {\r
+            if (!this.objectTrackingEnabled)\r
+                return;\r
+            MemberModificationHandler.ClearModified(entity, Mapping);\r
+        }\r
+\r
+        /// <summary>\r
+        /// Registers an entity for delete\r
+        /// </summary>\r
+        /// <param name="entity"></param>\r
+        internal void RegisterDelete(object entity)\r
+        {\r
+            if (!this.objectTrackingEnabled)\r
+                return;\r
+            CurrentTransactionEntities.RegisterToDelete(entity);\r
+        }\r
+\r
+        /// <summary>\r
+        /// Unregisters entity after deletion\r
+        /// </summary>\r
+        /// <param name="entity"></param>\r
+        internal void UnregisterDelete(object entity)\r
+        {\r
+            if (!this.objectTrackingEnabled)\r
+                return;\r
+            CurrentTransactionEntities.RegisterDeleted(entity);\r
+        }\r
+\r
+        #endregion\r
+\r
+        /// <summary>\r
+        /// Changed object determine \r
+        /// </summary>\r
+        /// <returns>Lists of inserted, updated, deleted objects</returns>\r
+        public ChangeSet GetChangeSet()\r
+        {\r
+            var inserts = new List<object>();\r
+            var updates = new List<object>();\r
+            var deletes = new List<object>();\r
+            foreach (var entityTrack in CurrentTransactionEntities.EnumerateAll()\r
+                    .Concat(AllTrackedEntities.EnumerateAll()))\r
+            {\r
+                switch (entityTrack.EntityState)\r
+                {\r
+                    case EntityState.ToInsert:\r
+                        inserts.Add(entityTrack.Entity);\r
+                        break;\r
+                    case EntityState.ToWatch:\r
+                        if (MemberModificationHandler.IsModified(entityTrack.Entity, Mapping))\r
+                            updates.Add(entityTrack.Entity);\r
+                        break;\r
+                    case EntityState.ToDelete:\r
+                        deletes.Add(entityTrack.Entity);\r
+                        break;\r
+                    default:\r
+                        throw new ArgumentOutOfRangeException();\r
+                }\r
+            }\r
+            return new ChangeSet(inserts, updates, deletes);\r
+        }\r
+\r
+        /// <summary>\r
+        /// use ExecuteCommand to call raw SQL\r
+        /// </summary>\r
+        public int ExecuteCommand(string command, params object[] parameters)\r
+        {\r
+            var directQuery = QueryBuilder.GetDirectQuery(command, new QueryContext(this));\r
+            return QueryRunner.Execute(directQuery, parameters);\r
+        }\r
+\r
+        /// <summary>\r
+        /// Execute raw SQL query and return object\r
+        /// </summary>\r
+        public IEnumerable<TResult> ExecuteQuery<TResult>(string query, params object[] parameters) where TResult : class, new()\r
+        {\r
+            if (query == null)\r
+                throw new ArgumentNullException("query");\r
+\r
+            return CreateExecuteQueryEnumerable<TResult>(query, parameters);\r
+        }\r
+\r
+        private IEnumerable<TResult> CreateExecuteQueryEnumerable<TResult>(string query, object[] parameters)\r
+            where TResult : class, new()\r
+        {\r
+            foreach (TResult result in ExecuteQuery(typeof(TResult), query, parameters))\r
+                yield return result;\r
+        }\r
+\r
+        public IEnumerable ExecuteQuery(Type elementType, string query, params object[] parameters)\r
+        {\r
+            if (elementType == null)\r
+                throw new ArgumentNullException("elementType");\r
+            if (query == null)\r
+                throw new ArgumentNullException("query");\r
+\r
+            var queryContext = new QueryContext(this);\r
+            var directQuery = QueryBuilder.GetDirectQuery(query, queryContext);\r
+            return QueryRunner.ExecuteSelect(elementType, directQuery, parameters);\r
+        }\r
+\r
+        /// <summary>\r
+        /// Gets or sets the load options\r
+        /// </summary>\r
+        [DbLinqToDo]\r
+               public DataLoadOptions LoadOptions\r
+               {\r
+                       get { throw new NotImplementedException(); }\r
+                       set { throw new NotImplementedException(); }\r
+               }\r
+\r
+        public DbTransaction Transaction {\r
+            get { return (DbTransaction) DatabaseContext.CurrentTransaction; }\r
+            set { DatabaseContext.CurrentTransaction = value; }\r
+        }\r
+\r
+        /// <summary>\r
+        /// Runs the given reader and returns columns.\r
+        /// </summary>\r
+        /// <typeparam name="TResult">The type of the result.</typeparam>\r
+        /// <param name="reader">The reader.</param>\r
+        /// <returns></returns>\r
+        public IEnumerable<TResult> Translate<TResult>(DbDataReader reader)\r
+        {\r
+            if (reader == null)\r
+                throw new ArgumentNullException("reader");\r
+            return CreateTranslateIterator<TResult>(reader);\r
+        }\r
+\r
+        IEnumerable<TResult> CreateTranslateIterator<TResult>(DbDataReader reader)\r
+        {\r
+            foreach (TResult result in Translate(typeof(TResult), reader))\r
+                yield return result;\r
+        }\r
+\r
+        public IMultipleResults Translate(DbDataReader reader)\r
+        {\r
+            throw new NotImplementedException();\r
+        }\r
+\r
+        public IEnumerable Translate(Type elementType, DbDataReader reader)\r
+        {\r
+            if (elementType == null)\r
+                throw new ArgumentNullException("elementType");\r
+            if (reader == null)\r
+                throw new ArgumentNullException("reader");\r
+\r
+            return QueryRunner.EnumerateResult(elementType, reader, this);\r
+        }\r
+\r
+        public void Dispose()\r
+        {\r
+            //connection closing should not be done here.\r
+            //read: http://msdn2.microsoft.com/en-us/library/bb292288.aspx\r
+\r
+                       //We own the instance of MemberModificationHandler - we must unregister listeners of entities we attached to\r
+                       MemberModificationHandler.UnregisterAll();\r
+        }\r
+\r
+        [DbLinqToDo]\r
+        protected virtual void Dispose(bool disposing)\r
+        {\r
+            throw new NotImplementedException();\r
+        }\r
+\r
+        /// <summary>\r
+        /// Creates a IDbDataAdapter. Used internally by Vendors\r
+        /// </summary>\r
+        /// <returns></returns>\r
+        internal IDbDataAdapter CreateDataAdapter()\r
+        {\r
+            return DatabaseContext.CreateDataAdapter();\r
+        }\r
+\r
+        /// <summary>\r
+        /// Sets a TextWriter where generated SQL commands are written\r
+        /// </summary>\r
+        public TextWriter Log { get; set; }\r
+\r
+        /// <summary>\r
+        /// Writes text on Log (if not null)\r
+        /// Internal helper\r
+        /// </summary>\r
+        /// <param name="text"></param>\r
+        internal void WriteLog(string text)\r
+        {\r
+            if (Log != null)\r
+                Log.WriteLine(text);\r
+        }\r
+\r
+        /// <summary>\r
+        /// Write an IDbCommand to Log (if non null)\r
+        /// </summary>\r
+        /// <param name="command"></param>\r
+        internal void WriteLog(IDbCommand command)\r
+        {\r
+            if (Log != null)\r
+            {\r
+                Log.WriteLine(command.CommandText);\r
+                foreach (IDbDataParameter parameter in command.Parameters)\r
+                    WriteLog(parameter);\r
+                Log.Write("--");\r
+                Log.Write(" Context: {0}", Vendor.VendorName);\r
+                Log.Write(" Model: {0}", Mapping.GetType().Name);\r
+                Log.Write(" Build: {0}", Assembly.GetExecutingAssembly().GetName().Version);\r
+                Log.WriteLine();\r
+            }\r
+        }\r
+\r
+        /// <summary>\r
+        /// Writes and IDbDataParameter to Log (if non null)\r
+        /// </summary>\r
+        /// <param name="parameter"></param>\r
+        internal void WriteLog(IDbDataParameter parameter)\r
+        {\r
+            if (Log != null)\r
+            {\r
+                // -- @p0: Input Int (Size = 0; Prec = 0; Scale = 0) [2]\r
+                // -- <name>: <direction> <type> (...) [<value>]\r
+                Log.WriteLine("-- {0}: {1} {2} (Size = {3}; Prec = {4}; Scale = {5}) [{6}]",\r
+                    parameter.ParameterName, parameter.Direction, parameter.DbType,\r
+                    parameter.Size, parameter.Precision, parameter.Scale, parameter.Value);\r
+            }\r
+        }\r
+\r
+        public bool ObjectTrackingEnabled\r
+        {\r
+            get { return this.objectTrackingEnabled; }\r
+            set \r
+            {\r
+                if (this.currentTransactionEntities != null && value != this.objectTrackingEnabled)\r
+                    throw new InvalidOperationException("Data context options cannot be modified after results have been returned from a query.");\r
+                this.objectTrackingEnabled = value;\r
+            }\r
+        }\r
+\r
+        [DbLinqToDo]\r
+        public int CommandTimeout\r
+        {\r
+            get { throw new NotImplementedException(); }\r
+            set { throw new NotImplementedException(); }\r
+        }\r
+\r
+        public bool DeferredLoadingEnabled\r
+        {\r
+            get { return this.deferredLoadingEnabled; }\r
+            set\r
+            {\r
+                if (this.currentTransactionEntities != null && value != this.deferredLoadingEnabled)\r
+                    throw new InvalidOperationException("Data context options cannot be modified after results have been returned from a query.");\r
+                this.deferredLoadingEnabled = value;\r
+            }\r
+        }\r
+\r
+        [DbLinqToDo]\r
+        public ChangeConflictCollection ChangeConflicts\r
+        {\r
+            get { throw new NotImplementedException(); }\r
+        }\r
+\r
+        [DbLinqToDo]\r
+        public DbCommand GetCommand(IQueryable query)\r
+        {\r
+            DbCommand dbCommand = GetIDbCommand(query) as DbCommand;\r
+            if (dbCommand == null)\r
+                throw new InvalidOperationException();\r
+\r
+            return dbCommand;\r
+        }\r
+\r
+        [DBLinqExtended]\r
+        public IDbCommand GetIDbCommand(IQueryable query)\r
+        {\r
+            if (query == null)\r
+                throw new ArgumentNullException("query");\r
+\r
+            var qp = query.Provider as QueryProvider;\r
+            if (qp == null)\r
+                throw new InvalidOperationException();\r
+\r
+            if (qp.ExpressionChain.Expressions.Count == 0)\r
+                qp.ExpressionChain.Expressions.Add(CreateDefaultQuery(query));\r
+\r
+            return qp.GetQuery(null).GetCommand().Command;\r
+        }\r
+\r
+        private Expression CreateDefaultQuery(IQueryable query)\r
+        {\r
+            // Manually create the expression tree for: IQueryable<TableType>.Select(e => e)\r
+            var identityParameter = Expression.Parameter(query.ElementType, "e");\r
+            var identityBody = Expression.Lambda(\r
+                typeof(Func<,>).MakeGenericType(query.ElementType, query.ElementType),\r
+                identityParameter,\r
+                new[] { identityParameter }\r
+            );\r
+\r
+            return Expression.Call(\r
+                typeof(Queryable),\r
+                "Select",\r
+                new[] { query.ElementType, query.ElementType },\r
+                query.Expression,\r
+                Expression.Quote(identityBody)\r
+            );\r
+        }\r
+\r
+        [DbLinqToDo]\r
+        public void Refresh(RefreshMode mode, IEnumerable entities)\r
+        {\r
+            throw new NotImplementedException();\r
+        }\r
+\r
+        [DbLinqToDo]\r
+        public void Refresh(RefreshMode mode, params object[] entities)\r
+        {\r
+            throw new NotImplementedException();\r
+        }\r
+\r
+        [DbLinqToDo]\r
+        public void Refresh(RefreshMode mode, object entity)\r
+        {\r
+            throw new NotImplementedException();\r
+        }\r
+\r
+        [DbLinqToDo]\r
+        public void DeleteDatabase()\r
+        {\r
+            throw new NotImplementedException();\r
+        }\r
+\r
+        [DbLinqToDo]\r
+        public void CreateDatabase()\r
+        {\r
+            throw new NotImplementedException();\r
+        }\r
+\r
+        [DbLinqToDo]\r
+        protected internal IQueryable<TResult> CreateMethodCallQuery<TResult>(object instance, MethodInfo methodInfo, params object[] parameters)\r
+        {\r
+            throw new NotImplementedException();\r
+        }\r
+\r
+        [DbLinqToDo]\r
+        protected internal void ExecuteDynamicDelete(object entity)\r
+        {\r
+            throw new NotImplementedException();\r
+        }\r
+\r
+        [DbLinqToDo]\r
+        protected internal void ExecuteDynamicInsert(object entity)\r
+        {\r
+            throw new NotImplementedException();\r
+        }\r
+\r
+        [DbLinqToDo]\r
+        protected internal void ExecuteDynamicUpdate(object entity)\r
+        {\r
+            throw new NotImplementedException();\r
+        }\r
+    }\r
+}\r