activerecord-sqlserver-adapter 2.3.7 → 3.2.18

data/lib/active_record/connection_adapters/sqlserver/core_ext/database_statements.rb

@@ -0,0 +1,97 @@
+module ActiveRecord
+  module ConnectionAdapters
+    module Sqlserver
+      module CoreExt
+        module DatabaseStatements
+          
+          # This is a copy of the current (3.1.3) ActiveRecord's transaction method. We should propose
+          # a patch to the default transaction method to make it more callback for adapters that want to 
+          # do deadlock retry logic. Because this is a copy, we really need to keep an eye out on this when
+          # upgradding the adapter. 
+          def transaction_with_retry_deadlock_victim(options = {})
+            options.assert_valid_keys :requires_new, :joinable
+
+            last_transaction_joinable = defined?(@transaction_joinable) ? @transaction_joinable : nil
+            if options.has_key?(:joinable)
+              @transaction_joinable = options[:joinable]
+            else
+              @transaction_joinable = true
+            end
+            requires_new = options[:requires_new] || !last_transaction_joinable
+
+            transaction_open = false
+            @_current_transaction_records ||= []
+
+            begin
+              if block_given?
+                if requires_new || open_transactions == 0
+                  if open_transactions == 0
+                    begin_db_transaction
+                  elsif requires_new
+                    create_savepoint
+                  end
+                  increment_open_transactions
+                  transaction_open = true
+                  @_current_transaction_records.push([])
+                end
+                yield
+              end
+            rescue Exception => database_transaction_rollback
+              if transaction_open && !outside_transaction?
+                transaction_open = false
+                decrement_open_transactions
+                # handle deadlock victim retries at the outermost transaction
+                if open_transactions == 0
+                  if database_transaction_rollback.is_a?(::ActiveRecord::DeadlockVictim)
+                    # SQL Server has already rolled back, so rollback activerecord's history
+                    rollback_transaction_records(true)
+                    retry
+                  else
+                    rollback_db_transaction
+                    rollback_transaction_records(true)
+                  end
+                else
+                  rollback_to_savepoint
+                  rollback_transaction_records(false)
+                end
+              end
+              raise unless database_transaction_rollback.is_a?(::ActiveRecord::Rollback)
+            end
+          ensure
+            @transaction_joinable = last_transaction_joinable
+
+            if outside_transaction?
+              @open_transactions = 0
+            elsif transaction_open
+              decrement_open_transactions
+              begin
+                if open_transactions == 0
+                  commit_db_transaction
+                  commit_transaction_records
+                else
+                  release_savepoint
+                  save_point_records = @_current_transaction_records.pop
+                  unless save_point_records.blank?
+                    @_current_transaction_records.push([]) if @_current_transaction_records.empty?
+                    @_current_transaction_records.last.concat(save_point_records)
+                  end
+                end
+              rescue Exception => database_transaction_rollback
+                if open_transactions == 0
+                  rollback_db_transaction
+                  rollback_transaction_records(true)
+                else
+                  rollback_to_savepoint
+                  rollback_transaction_records(false)
+                end
+                raise
+              end
+            end
+          end
+          
+        end
+      end
+    end
+  end
+end
+

data/lib/active_record/connection_adapters/sqlserver/core_ext/explain.rb

@@ -0,0 +1,41 @@
+module ActiveRecord
+  module ConnectionAdapters
+    module Sqlserver
+      module CoreExt
+        module Explain
+          
+          SQLSERVER_STATEMENT_PREFIX = "EXEC sp_executesql "
+          SQLSERVER_PARAM_MATCHER = /@\d+ =/
+          
+          def exec_explain(queries)
+            unprepared_queries = queries.map { |sql, bind| [unprepare_sqlserver_statement(sql), bind] }
+            super(unprepared_queries)
+          end
+          
+          private
+          
+          # This is somewhat hacky, but it should reliably reformat our prepared sql statment 
+          # which uses sp_executesql to just the first argument, then unquote it. Likewise our 
+          # do_exec_query method should substitude the @n args withe the quoted values.
+          def unprepare_sqlserver_statement(sql)
+            if sql.starts_with?(SQLSERVER_STATEMENT_PREFIX)
+              executesql = sql.from(SQLSERVER_STATEMENT_PREFIX.length)
+              executesql_args = executesql.split(', ')
+              found_args = executesql_args.reject! { |arg| arg =~ SQLSERVER_PARAM_MATCHER }
+              executesql_args.pop if found_args && executesql_args.many?
+              executesql = executesql_args.join(', ').strip.match(/N'(.*)'/m)[1]
+              Utils.unquote_string(executesql)
+            else
+              sql
+            end
+          end
+
+          
+        end
+      end
+    end
+  end
+end
+
+ActiveRecord::Base.extend ActiveRecord::ConnectionAdapters::Sqlserver::CoreExt::Explain
+ActiveRecord::Relation.send :include, ActiveRecord::ConnectionAdapters::Sqlserver::CoreExt::Explain

data/lib/active_record/connection_adapters/sqlserver/core_ext/explain_subscriber.rb

@@ -0,0 +1,26 @@
+module ActiveRecord
+  module ConnectionAdapters
+    module Sqlserver
+      module CoreExt
+        class ExplainSubscriber
+          def call(*args)
+            if queries = Thread.current[:available_queries_for_explain]
+              payload = args.last
+              queries << payload.values_at(:sql, :binds) unless ignore_sqlserver_payload?(payload)
+            end
+          end
+
+          IGNORED_PAYLOADS = %w(SCHEMA EXPLAIN CACHE)
+          SQLSERVER_EXPLAINED_SQLS = /(select|update|delete|insert)/i
+
+          # Need to modify the regex for the TSQL generated by this adapter so we can explain the proper sql statements
+          def ignore_sqlserver_payload?(payload)
+            payload[:exception] || IGNORED_PAYLOADS.include?(payload[:name]) || payload[:sql] !~ SQLSERVER_EXPLAINED_SQLS
+          end
+
+          ActiveSupport::Notifications.subscribe("sql.active_record", new)
+        end
+      end
+    end
+  end
+end

data/lib/active_record/connection_adapters/sqlserver/core_ext/odbc.rb

@@ -0,0 +1,38 @@
+module ActiveRecord
+  module ConnectionAdapters
+    module Sqlserver
+      module CoreExt
+        module ODBC
+
+          module Statement
+            
+            def finished?
+              begin
+                connected?
+                false
+              rescue ::ODBC::Error
+                true
+              end
+            end
+            
+          end
+
+          module Database
+            
+            def run_block(*args)
+              yield sth = run(*args)
+              sth.drop
+            end
+            
+          end
+
+        end
+      end
+    end
+  end
+end
+
+
+ODBC::Statement.send :include, ActiveRecord::ConnectionAdapters::Sqlserver::CoreExt::ODBC::Statement
+ODBC::Database.send :include, ActiveRecord::ConnectionAdapters::Sqlserver::CoreExt::ODBC::Database
+

data/lib/active_record/connection_adapters/sqlserver/core_ext/relation.rb

@@ -0,0 +1,19 @@
+module ActiveRecord
+  module ConnectionAdapters
+    module Sqlserver
+      module CoreExt
+        module Relation
+          
+          private
+          
+          def tables_in_string(string)
+            super - ['__rnt']
+          end
+          
+        end
+      end
+    end
+  end
+end
+
+ActiveRecord::Relation.send :include, ActiveRecord::ConnectionAdapters::Sqlserver::CoreExt::Relation

data/lib/active_record/connection_adapters/sqlserver/database_limits.rb

@@ -0,0 +1,49 @@
+module ActiveRecord
+  module ConnectionAdapters
+    module Sqlserver
+      module DatabaseLimits
+        
+        def table_alias_length
+          128
+        end
+
+        def column_name_length
+          128
+        end
+
+        def table_name_length
+          128
+        end
+
+        def index_name_length
+          128
+        end
+
+        def columns_per_table
+          1024
+        end
+
+        def indexes_per_table
+          999
+        end
+
+        def columns_per_multicolumn_index
+          16
+        end
+
+        def in_clause_length
+          65536
+        end
+
+        def sql_query_length
+          65536 * 4096
+        end
+
+        def joins_per_query
+          256
+        end
+        
+      end
+    end
+  end
+end