Merge pull request #4621 from alexanderkyte/strdup_env
[mono.git] / mcs / class / System.Data.Linq / src / DbLinq / Data / Linq / DataContext.cs
index c56791e9d7f3cd8ffaaf3394d6fb72a54a9879e5..db68b96777b3473655a9a97785e4452b77f7157f 100644 (file)
@@ -192,12 +192,12 @@ namespace DbLinq.Data.Linq
         /// </summary>\r
         /// <param name="connectionString">specifies file or server connection</param>\r
         [DbLinqToDo]\r
-        public DataContext(string connectionString)\r
+        public DataContext(string fileOrServerOrConnection)\r
         {\r
             Profiler.At("START DataContext(string)");\r
-            IVendor ivendor = GetVendor(ref connectionString);\r
+            IVendor ivendor = GetVendor(ref fileOrServerOrConnection);\r
 \r
-            IDbConnection dbConnection = ivendor.CreateDbConnection(connectionString);\r
+            IDbConnection dbConnection = ivendor.CreateDbConnection(fileOrServerOrConnection);\r
             Init(new DatabaseContext(dbConnection), null, ivendor);\r
 \r
             Profiler.At("END DataContext(string)");\r
@@ -236,18 +236,18 @@ namespace DbLinq.Data.Linq
             System.Text.RegularExpressions.Regex reProvider\r
                 = new System.Text.RegularExpressions.Regex(@"DbLinqProvider=([\w\.]+);?");\r
 \r
-            string assemblyFile = null;\r
+            string assemblyName = null;\r
             string vendor;\r
             if (!reProvider.IsMatch(connectionString))\r
             {\r
                 vendor       = "SqlServer";\r
-                assemblyFile = "DbLinq.SqlServer.dll";\r
+                assemblyName = "DbLinq.SqlServer";\r
             }\r
             else\r
             {\r
                 var match    = reProvider.Match(connectionString);\r
                 vendor       = match.Groups[1].Value;\r
-                assemblyFile = "DbLinq." + vendor + ".dll";\r
+                assemblyName = "DbLinq." + vendor;\r
 \r
                 //plain DbLinq - non MONO: \r
                 //IVendor classes are in DLLs such as "DbLinq.MySql.dll"\r
@@ -268,16 +268,15 @@ namespace DbLinq.Data.Linq
 #if MONO_STRICT\r
                 assembly = typeof (DataContext).Assembly; // System.Data.Linq.dll\r
 #else\r
-                //TODO: check if DLL is already loaded?\r
-                assembly = Assembly.LoadFrom(assemblyFile);\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}'.",\r
-                            assemblyFile, vendor),\r
+                            "Unable to load the `{0}' DbLinq vendor within assembly '{1}.dll'.",\r
+                            assemblyName, vendor),\r
                         "connectionString", e);\r
             }\r
         }\r
@@ -295,7 +294,11 @@ namespace DbLinq.Data.Linq
             _VendorProvider = ObjectFactory.Get<IVendorProvider>();\r
             Vendor = vendor ?? \r
                 (connectionString != null ? GetVendor(ref connectionString) : null) ??\r
+#if MOBILE\r
+                _VendorProvider.FindVendorByProviderType(typeof(DbLinq.Sqlite.SqliteSqlProvider));\r
+#else\r
                 _VendorProvider.FindVendorByProviderType(typeof(SqlClient.Sql2005Provider));\r
+#endif\r
             \r
             DatabaseContext = databaseContext;\r
 \r
@@ -394,53 +397,73 @@ namespace DbLinq.Data.Linq
             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
-            using (IDatabaseTransaction transaction = DatabaseContext.Transaction())\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
+                if (Transaction != null)\r
+                    SubmitChangesImpl(failureMode);\r
+                else\r
                 {\r
-                    switch (entityTrack.EntityState)\r
+                    using (IDbTransaction transaction = DatabaseContext.CreateTransaction())\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
+                        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
-                foreach (var entityTrack in CurrentTransactionEntities.EnumerateAll()\r
-                        .Concat(AllTrackedEntities.EnumerateAll())\r
-                        .ToList())\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
-                    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
+                    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
-                // TODO: handle conflicts (which can only occur when concurrency mode is implemented)\r
-                transaction.Commit();\r
             }\r
         }\r
 \r
@@ -704,14 +727,27 @@ namespace DbLinq.Data.Linq
                                        //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
-                               FieldInfo entityRefField = (FieldInfo)memberData.StorageMember;\r
                                object entityRefValue = null;\r
                                if (query != null)\r
-                                       entityRefValue = Activator.CreateInstance(entityRefField.FieldType, query);\r
+                                       entityRefValue = Activator.CreateInstance(storageType, query);\r
                                else\r
-                                       entityRefValue = Activator.CreateInstance(entityRefField.FieldType);\r
-                               entityRefField.SetValue(entity, entityRefValue);\r
+                                       entityRefValue = Activator.CreateInstance(storageType);\r
+\r
+                               storage.SetMemberValue(entity, entityRefValue);\r
                        }\r
                }\r
 \r
@@ -794,9 +830,12 @@ namespace DbLinq.Data.Linq
             }\r
         }\r
 \r
-               private static MethodInfo _WhereMethod = typeof(Queryable).GetMethods().First(m => m.Name == "Where");\r
-        private object GetOtherTableQuery(Expression predicate, ParameterExpression parameter, Type otherTableType, IQueryable otherTable)\r
+               private static MethodInfo _WhereMethod;\r
+        internal object GetOtherTableQuery(Expression predicate, ParameterExpression parameter, Type otherTableType, IQueryable otherTable)\r
         {\r
+            if (_WhereMethod == null)\r
+                System.Threading.Interlocked.CompareExchange (ref _WhereMethod, typeof(Queryable).GetMethods().First(m => m.Name == "Where"), null);\r
+\r
             //predicate: other.EmployeeID== "WARTH"\r
             Expression lambdaPredicate = Expression.Lambda(predicate, parameter);\r
             //lambdaPredicate: other=>other.EmployeeID== "WARTH"\r
@@ -955,7 +994,7 @@ namespace DbLinq.Data.Linq
         /// <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
+        public IEnumerable<TResult> ExecuteQuery<TResult>(string query, params object[] parameters) where TResult : new()\r
         {\r
             if (query == null)\r
                 throw new ArgumentNullException("query");\r
@@ -964,7 +1003,7 @@ namespace DbLinq.Data.Linq
         }\r
 \r
         private IEnumerable<TResult> CreateExecuteQueryEnumerable<TResult>(string query, object[] parameters)\r
-            where TResult : class, new()\r
+            where TResult : new()\r
         {\r
             foreach (TResult result in ExecuteQuery(typeof(TResult), query, parameters))\r
                 yield return result;\r
@@ -992,7 +1031,10 @@ namespace DbLinq.Data.Linq
                        set { throw new NotImplementedException(); }\r
                }\r
 \r
-        public DbTransaction Transaction { get; set; }\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