activerecord-jdbcsqlserver-adapter 50.0.0

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 #{showplan_option} #{enable ? 'ON' : 'OFF'}"
+          raw_connection_do(sql)
+        rescue Exception
+          raise ActiveRecordError, "#{showplan_option} could not be turned #{enable ? 'ON' : 'OFF'}, perhaps you do not have SHOWPLAN permissions?"
+        end
+
+        def showplan_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?
+          showplan_option == OPTION_ALL
+        end
+
+        def showplan_text?
+          showplan_option == OPTION_TEXT
+        end
+
+        def showplan_xml?
+          showplan_option == OPTION_XML
+        end
+
+        def showplan_printer
+          case showplan_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/sql_type_metadata.rb

@@ -0,0 +1,20 @@
+module ActiveRecord
+  module ConnectionAdapters
+    module SQLServer
+      class SqlTypeMetadata < ActiveRecord::ConnectionAdapters::SqlTypeMetadata
+
+        def initialize(**kwargs)
+          @sqlserver_options = kwargs.extract!(:sqlserver_options)
+          super(**kwargs)
+        end
+
+        protected
+
+        def attributes_for_hash
+          super + [@sqlserver_options]
+        end
+
+      end
+    end
+  end
+end

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

@@ -0,0 +1,112 @@
+module ActiveRecord
+  module ConnectionAdapters
+    module SQLServer
+
+      module ColumnMethods
+
+        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 primary_key_nonclustered(*args, **options)
+          args.each { |name| column(name, :primary_key_nonclustered, options) }
+        end
+
+        def real(*args, **options)
+          args.each { |name| column(name, :real, options) }
+        end
+
+        def money(*args, **options)
+          args.each { |name| column(name, :money, options) }
+        end
+
+        def smalldatetime(*args, **options)
+          args.each { |name| column(name, :smalldatetime, options) }
+        end
+
+        def datetime(*args, **options)
+          args.each do |name|
+            if options[:precision]
+              datetime2(name, options)
+            else
+              column(name, :datetime, options)
+            end
+          end
+        end
+
+        def datetime2(*args, **options)
+          args.each { |name| column(name, :datetime2, options) }
+        end
+
+        def datetimeoffset(*args, **options)
+          args.each { |name| column(name, :datetimeoffset, options) }
+        end
+
+        def smallmoney(*args, **options)
+          args.each { |name| column(name, :smallmoney, options) }
+        end
+
+        def char(*args, **options)
+          args.each { |name| column(name, :char, options) }
+        end
+
+        def varchar(*args, **options)
+          args.each { |name| column(name, :varchar, options) }
+        end
+
+        def varchar_max(*args, **options)
+          args.each { |name| column(name, :varchar_max, options) }
+        end
+
+        def text_basic(*args, **options)
+          args.each { |name| column(name, :text_basic, options) }
+        end
+
+        def nchar(*args, **options)
+          args.each { |name| column(name, :nchar, options) }
+        end
+
+        def ntext(*args, **options)
+          args.each { |name| column(name, :ntext, options) }
+        end
+
+        def binary_basic(*args, **options)
+          args.each { |name| column(name, :binary_basic, options) }
+        end
+
+        def varbinary(*args, **options)
+          args.each { |name| column(name, :varbinary, options) }
+        end
+
+        def uuid(*args, **options)
+          args.each { |name| column(name, :uniqueidentifier, options) }
+        end
+
+        def ss_timestamp(*args, **options)
+          args.each { |name| column(name, :ss_timestamp, options) }
+        end
+
+        def json(*args, **options)
+          args.each { |name| column(name, :text, options) }
+        end
+
+      end
+
+      class TableDefinition < ActiveRecord::ConnectionAdapters::TableDefinition
+        include ColumnMethods
+
+        def new_column_definition(name, type, options)
+          type = :datetime2 if type == :datetime && options[:precision]
+          super name, type, options
+        end
+      end
+
+      class Table < ActiveRecord::ConnectionAdapters::Table
+        include ColumnMethods
+      end
+    end
+  end
+end

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

@@ -0,0 +1,64 @@
+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
+        # When READ_COMMITTED_SNAPSHOT is set to ON,
+        # user_options_isolation_level will be equal to 'read committed
+        # snapshot' which is not a valid isolation level
+        if level.blank? || level == 'read committed snapshot'
+          'READ COMMITTED'
+        else
+          level.upcase
+        end
+      end
+
+    end
+
+    Transaction.send :include, SQLServerTransaction
+
+    module SQLServerRealTransaction
+
+      attr_reader :starting_isolation_level
+
+      def initialize(connection, options, run_commit_callbacks: false)
+        @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,49 @@
+require 'active_record/type'
+# Behaviors
+require 'active_record/connection_adapters/sqlserver/type/data'
+require 'active_record/connection_adapters/sqlserver/type/time_value_fractional'
+# Exact Numerics
+require 'active_record/connection_adapters/sqlserver/type/integer'
+require 'active_record/connection_adapters/sqlserver/type/big_integer'
+require 'active_record/connection_adapters/sqlserver/type/small_integer'
+require 'active_record/connection_adapters/sqlserver/type/tiny_integer'
+require 'active_record/connection_adapters/sqlserver/type/boolean'
+require 'active_record/connection_adapters/sqlserver/type/decimal'
+require 'active_record/connection_adapters/sqlserver/type/money'
+require 'active_record/connection_adapters/sqlserver/type/small_money'
+# Approximate Numerics
+require 'active_record/connection_adapters/sqlserver/type/float'
+require 'active_record/connection_adapters/sqlserver/type/real'
+# Date and Time
+require 'active_record/connection_adapters/sqlserver/type/date'
+require 'active_record/connection_adapters/sqlserver/type/datetime'
+require 'active_record/connection_adapters/sqlserver/type/datetime2'
+require 'active_record/connection_adapters/sqlserver/type/datetimeoffset'
+require 'active_record/connection_adapters/sqlserver/type/smalldatetime'
+require 'active_record/connection_adapters/sqlserver/type/time'
+# Character Strings
+require 'active_record/connection_adapters/sqlserver/type/string'
+require 'active_record/connection_adapters/sqlserver/type/char'
+require 'active_record/connection_adapters/sqlserver/type/varchar'
+require 'active_record/connection_adapters/sqlserver/type/varchar_max'
+require 'active_record/connection_adapters/sqlserver/type/text'
+# Unicode Character Strings
+require 'active_record/connection_adapters/sqlserver/type/unicode_string'
+require 'active_record/connection_adapters/sqlserver/type/unicode_char'
+require 'active_record/connection_adapters/sqlserver/type/unicode_varchar'
+require 'active_record/connection_adapters/sqlserver/type/unicode_varchar_max'
+require 'active_record/connection_adapters/sqlserver/type/unicode_text'
+# Binary Strings
+require 'active_record/connection_adapters/sqlserver/type/binary'
+require 'active_record/connection_adapters/sqlserver/type/varbinary'
+require 'active_record/connection_adapters/sqlserver/type/varbinary_max'
+# Other Data Types
+require 'active_record/connection_adapters/sqlserver/type/uuid'
+require 'active_record/connection_adapters/sqlserver/type/timestamp'
+require 'active_record/connection_adapters/sqlserver/type/json'
+
+module ActiveRecord
+  module Type
+    SQLServer = ConnectionAdapters::SQLServer::Type
+  end
+end

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

@@ -0,0 +1,19 @@
+module ActiveRecord
+  module ConnectionAdapters
+    module SQLServer
+      module Type
+        class BigInteger < Integer
+
+          def type
+            :bigint
+          end
+
+          def sqlserver_type
+            'bigint'.freeze
+          end
+
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,21 @@
+module ActiveRecord
+  module ConnectionAdapters
+    module SQLServer
+      module Type
+        class Binary < ActiveRecord::Type::Binary
+
+          def type
+            :binary_basic
+          end
+
+          def sqlserver_type
+            'binary'.tap do |type|
+              type << "(#{limit})" if limit
+            end
+          end
+
+        end
+      end
+    end
+  end
+end