activerecord-sqlserver-adapter_new 4.2.15

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

@@ -0,0 +1,66 @@
+require 'active_record/connection_adapters/sqlserver/showplan/printer_table'
+require 'active_record/connection_adapters/sqlserver/showplan/printer_xml'
+
+module ActiveRecord
+  module ConnectionAdapters
+    module SQLServer
+      module Showplan
+
+        OPTION_ALL  = 'SHOWPLAN_ALL'
+        OPTION_TEXT = 'SHOWPLAN_TEXT'
+        OPTION_XML  = 'SHOWPLAN_XML'
+        OPTIONS = [OPTION_ALL, OPTION_TEXT, OPTION_XML]
+
+        def explain(arel, binds = [])
+          sql = to_sql(arel)
+          result = with_showplan_on { sp_executesql(sql, 'EXPLAIN', binds) }
+          printer = showplan_printer.new(result)
+          printer.pp
+        end
+
+        protected
+
+        def with_showplan_on
+          set_showplan_option(true)
+          yield
+        ensure
+          set_showplan_option(false)
+        end
+
+        def set_showplan_option(enable = true)
+          sql = "SET #{option} #{enable ? 'ON' : 'OFF'}"
+          raw_connection_do(sql)
+        rescue Exception
+          raise ActiveRecordError, "#{option} could not be turned #{enable ? 'ON' : 'OFF'}, perhaps you do not have SHOWPLAN permissions?"
+        end
+
+        def option
+          (SQLServerAdapter.showplan_option || OPTION_ALL).tap do |opt|
+            raise(ArgumentError, "Unknown SHOWPLAN option #{opt.inspect} found.") if OPTIONS.exclude?(opt)
+          end
+        end
+
+        def showplan_all?
+          option == OPTION_ALL
+        end
+
+        def showplan_text?
+          option == OPTION_TEXT
+        end
+
+        def showplan_xml?
+          option == OPTION_XML
+        end
+
+        def showplan_printer
+          case option
+          when OPTION_XML then PrinterXml
+          when OPTION_ALL, OPTION_TEXT then PrinterTable
+          else PrinterTable
+          end
+        end
+
+      end
+    end
+  end
+end

data/lib/active_record/connection_adapters/sqlserver/showplan/printer_table.rb

@@ -0,0 +1,66 @@
+module ActiveRecord
+  module ConnectionAdapters
+    module SQLServer
+      module Showplan
+        class PrinterTable
+          cattr_accessor :max_column_width, :cell_padding
+          self.max_column_width = 50
+          self.cell_padding = 1
+
+          attr_reader :result
+
+          def initialize(result)
+            @result = result
+          end
+
+          def pp
+            @widths = compute_column_widths
+            @separator = build_separator
+            pp = []
+            pp << @separator
+            pp << build_cells(result.columns)
+            pp << @separator
+            result.rows.each do |row|
+              pp << build_cells(row)
+            end
+            pp << @separator
+            pp.join("\n") + "\n"
+          end
+
+          private
+
+          def compute_column_widths
+            [].tap do |computed_widths|
+              result.columns.each_with_index do |column, i|
+                cells_in_column = [column] + result.rows.map { |r| cast_item(r[i]) }
+                computed_width = cells_in_column.map(&:length).max
+                final_width = computed_width > max_column_width ? max_column_width : computed_width
+                computed_widths << final_width
+              end
+            end
+          end
+
+          def build_separator
+            '+' + @widths.map { |w| '-' * (w + (cell_padding * 2)) }.join('+') + '+'
+          end
+
+          def build_cells(items)
+            cells = []
+            items.each_with_index do |item, i|
+              cells << cast_item(item).ljust(@widths[i])
+            end
+            "| #{cells.join(' | ')} |"
+          end
+
+          def cast_item(item)
+            case item
+            when NilClass then 'NULL'
+            when Float then item.to_s.to(9)
+            else item.to_s.truncate(max_column_width)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/active_record/connection_adapters/sqlserver/showplan/printer_xml.rb

@@ -0,0 +1,22 @@
+module ActiveRecord
+  module ConnectionAdapters
+    module SQLServer
+      module Showplan
+        class PrinterXml
+          def initialize(result)
+            @result = result
+          end
+
+          def pp
+            xml = @result.rows.first.first
+            if defined?(Nokogiri)
+              Nokogiri::XML(xml).to_xml indent: 2, encoding: 'UTF-8'
+            else
+              xml
+            end
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,76 @@
+module ActiveRecord
+  module ConnectionAdapters
+    module SQLServer
+      class TableDefinition < ActiveRecord::ConnectionAdapters::TableDefinition
+
+        def primary_key(name, type = :primary_key, options = {})
+          return super unless type == :uuid
+          options[:default] = options.fetch(:default, 'NEWID()')
+          options[:primary_key] = true
+          column name, type, options
+        end
+
+        def real(name, options = {})
+          column(name, :real, options)
+        end
+
+        def money(name, options = {})
+          column(name, :money, options)
+        end
+
+        def datetime2(name, options = {})
+          column(name, :datetime2, options)
+        end
+
+        def datetimeoffset(name, options = {})
+          column(name, :datetimeoffset, options)
+        end
+
+        def smallmoney(name, options = {})
+          column(name, :smallmoney, options)
+        end
+
+        def char(name, options = {})
+          column(name, :char, options)
+        end
+
+        def varchar(name, options = {})
+          column(name, :varchar, options)
+        end
+
+        def varchar_max(name, options = {})
+          column(name, :varchar_max, options)
+        end
+
+        def text_basic(name, options = {})
+          column(name, :text_basic, options)
+        end
+
+        def nchar(name, options = {})
+          column(name, :nchar, options)
+        end
+
+        def ntext(name, options = {})
+          column(name, :ntext, options)
+        end
+
+        def binary_basic(name, options = {})
+          column(name, :binary_basic, options)
+        end
+
+        def varbinary(name, options = {})
+          column(name, :varbinary, options)
+        end
+
+        def uuid(name, options = {})
+          column(name, :uniqueidentifier, options)
+        end
+
+        def ss_timestamp(name, options = {})
+          column(name, :ss_timestamp, options)
+        end
+
+      end
+    end
+  end
+end

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

@@ -0,0 +1,57 @@
+require 'active_record/connection_adapters/abstract/transaction'
+
+module ActiveRecord
+  module ConnectionAdapters
+
+    module SQLServerTransaction
+
+      private
+
+      def sqlserver?
+        connection.respond_to?(:sqlserver?) && connection.sqlserver?
+      end
+
+      def current_isolation_level
+        return unless sqlserver?
+        level = connection.user_options_isolation_level
+        level.blank? ? 'READ COMMITTED' : level.upcase
+      end
+
+    end
+
+    Transaction.send :include, SQLServerTransaction
+
+    module SQLServerRealTransaction
+
+      attr_reader :starting_isolation_level
+
+      def initialize(connection, options)
+        @connection = connection
+        @starting_isolation_level = current_isolation_level if options[:isolation]
+        super
+      end
+
+      def commit
+        super
+        reset_starting_isolation_level
+      end
+
+      def rollback
+        super
+        reset_starting_isolation_level
+      end
+
+      private
+
+      def reset_starting_isolation_level
+        if sqlserver? && starting_isolation_level
+          connection.set_transaction_isolation_level(starting_isolation_level)
+        end
+      end
+
+    end
+
+    RealTransaction.send :include, SQLServerRealTransaction
+
+  end
+end

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

@@ -0,0 +1,46 @@
+require 'active_record/type'
+# Exact Numerics
+require 'active_record/connection_adapters/sqlserver/type/integer.rb'
+require 'active_record/connection_adapters/sqlserver/type/big_integer.rb'
+require 'active_record/connection_adapters/sqlserver/type/small_integer.rb'
+require 'active_record/connection_adapters/sqlserver/type/tiny_integer.rb'
+require 'active_record/connection_adapters/sqlserver/type/boolean.rb'
+require 'active_record/connection_adapters/sqlserver/type/decimal.rb'
+require 'active_record/connection_adapters/sqlserver/type/money.rb'
+require 'active_record/connection_adapters/sqlserver/type/small_money.rb'
+# Approximate Numerics
+require 'active_record/connection_adapters/sqlserver/type/float.rb'
+require 'active_record/connection_adapters/sqlserver/type/real.rb'
+# Date and Time
+require 'active_record/connection_adapters/sqlserver/type/time_value_fractional.rb'
+require 'active_record/connection_adapters/sqlserver/type/date.rb'
+require 'active_record/connection_adapters/sqlserver/type/datetime.rb'
+require 'active_record/connection_adapters/sqlserver/type/datetime2.rb'
+require 'active_record/connection_adapters/sqlserver/type/datetimeoffset.rb'
+require 'active_record/connection_adapters/sqlserver/type/smalldatetime.rb'
+require 'active_record/connection_adapters/sqlserver/type/time.rb'
+# Character Strings
+require 'active_record/connection_adapters/sqlserver/type/string.rb'
+require 'active_record/connection_adapters/sqlserver/type/char.rb'
+require 'active_record/connection_adapters/sqlserver/type/varchar.rb'
+require 'active_record/connection_adapters/sqlserver/type/varchar_max.rb'
+require 'active_record/connection_adapters/sqlserver/type/text.rb'
+# Unicode Character Strings
+require 'active_record/connection_adapters/sqlserver/type/unicode_string.rb'
+require 'active_record/connection_adapters/sqlserver/type/unicode_char.rb'
+require 'active_record/connection_adapters/sqlserver/type/unicode_varchar.rb'
+require 'active_record/connection_adapters/sqlserver/type/unicode_varchar_max.rb'
+require 'active_record/connection_adapters/sqlserver/type/unicode_text.rb'
+# Binary Strings
+require 'active_record/connection_adapters/sqlserver/type/binary.rb'
+require 'active_record/connection_adapters/sqlserver/type/varbinary.rb'
+require 'active_record/connection_adapters/sqlserver/type/varbinary_max.rb'
+# Other Data Types
+require 'active_record/connection_adapters/sqlserver/type/uuid.rb'
+require 'active_record/connection_adapters/sqlserver/type/timestamp.rb'
+
+module ActiveRecord
+  module Type
+    SQLServer = ConnectionAdapters::SQLServer::Type
+  end
+end

data/lib/active_record/connection_adapters/sqlserver/type/big_integer.rb

@@ -0,0 +1,15 @@
+module ActiveRecord
+  module ConnectionAdapters
+    module SQLServer
+      module Type
+        class BigInteger < Integer
+
+          def type
+            :bigint
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/active_record/connection_adapters/sqlserver/type/binary.rb

@@ -0,0 +1,15 @@
+module ActiveRecord
+  module ConnectionAdapters
+    module SQLServer
+      module Type
+        class Binary < ActiveRecord::Type::Binary
+
+          def type
+            :binary_basic
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/active_record/connection_adapters/sqlserver/type/boolean.rb

@@ -0,0 +1,12 @@
+module ActiveRecord
+  module ConnectionAdapters
+    module SQLServer
+      module Type
+        class Boolean < ActiveRecord::Type::Boolean
+
+
+        end
+      end
+    end
+  end
+end

data/lib/active_record/connection_adapters/sqlserver/type/char.rb

@@ -0,0 +1,38 @@
+module ActiveRecord
+  module ConnectionAdapters
+    module SQLServer
+      module Type
+        class Char < String
+
+          def type
+            :char
+          end
+
+          def type_cast_for_database(value)
+            return if value.nil?
+            return value if value.is_a?(Data)
+            Data.new(super)
+          end
+
+          class Data
+
+            def initialize(value)
+              @value = value.to_s
+            end
+
+            def quoted
+              "'#{Utils.quote_string(@value)}'"
+            end
+
+            def to_s
+              @value
+            end
+            alias_method :to_str, :to_s
+
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/active_record/connection_adapters/sqlserver/type/date.rb

@@ -0,0 +1,21 @@
+module ActiveRecord
+  module ConnectionAdapters
+    module SQLServer
+      module Type
+        class Date < ActiveRecord::Type::Date
+
+          def type_cast_for_database(value)
+            return unless value.present?
+            return value if value.acts_like?(:string)
+            value.to_s(:_sqlserver_dateformat)
+          end
+
+          def type_cast_for_schema(value)
+            type_cast_for_database(value).inspect
+          end
+
+        end
+      end
+    end
+  end
+end