예제 #1
0
        /// <summary>
        /// Gets an open READ UNCOMMITTED connection using the specified connection string, optionally timing out on the initial connect
        /// </summary>
        /// <param name="connectionString">The connection string to use for the connection</param>
        /// <param name="connectionTimeout">Milliseconds to wait to connect, optional</param>
        /// <returns>A READ UNCOMMITTED connection to the specified connection string</returns>
        public static async Task <DbConnection> GetOpenAsync(string connectionString, int?connectionTimeout = null)
        {
            var conn = new ProfiledDbConnection(new SqlConnection(connectionString), MiniProfiler.Current);

            if (connectionTimeout.GetValueOrDefault(0) == 0)
            {
                await conn.OpenAsync();

                await conn.SetReadUncommitted();
            }
            else
            {
                // In the case of remote monitoring, the timeout will be at the NIC level, not responding to traffic,
                // in that scenario, connection timeouts don't really do much, because they're never reached, the timeout happens
                // before their timer starts.  Because of that, we need to spin up our own overall timeout
                using (MiniProfiler.Current.Step($"Opening Connection, Timeout: {conn.ConnectionTimeout}"))
                    using (var tokenSource = new CancellationTokenSource())
                    {
                        tokenSource.CancelAfter(connectionTimeout.Value);
                        try
                        {
                            await conn.OpenAsync(tokenSource.Token); // Throwing Null Refs

                            await conn.SetReadUncommitted();
                        }
                        catch (TaskCanceledException e)
                        {
                            conn.Close();
                            var csb          = new SqlConnectionStringBuilder(connectionString);
                            var sqlException = $"Error opening connection to {csb.InitialCatalog} at {csb.DataSource}, timeout out at {connectionTimeout.ToComma()} ms";
                            throw new Exception(sqlException, e);
                        }
                        catch (SqlException e)
                        {
                            conn.Close();
                            var csb          = new SqlConnectionStringBuilder(connectionString);
                            var sqlException = $"Error opening connection to {csb.InitialCatalog} at {csb.DataSource}: {e.Message}";
                            throw new Exception(sqlException, e);
                        }
                        if (conn.State == ConnectionState.Connecting)
                        {
                            tokenSource.Cancel();
                            var b = new SqlConnectionStringBuilder {
                                ConnectionString = connectionString
                            };
                            throw new TimeoutException($"Timeout expired connecting to {b.InitialCatalog} on {b.DataSource} on in the alloted {connectionTimeout.Value.ToComma()} ms");
                        }
                    }
            }
            return(conn);
        }