Module: JdbcSpec::MsSQL

Includes:
TSqlMethods
Defined in:
lib/jdbc_adapter/jdbc_mssql.rb

Defined Under Namespace

Modules: Column

Class Method Summary collapse

Instance Method Summary collapse

Methods included from TSqlMethods

#add_limit_offset!, #modify_types, #type_to_sql

Class Method Details

.adapter_matcher(name) ⇒ Object 



26
27
28
 
# File 'lib/jdbc_adapter/jdbc_mssql.rb', line 26

def self.adapter_matcher(name, *)
  name =~ /sqlserver|tds/i ? self : false
end

.column_selector ⇒ Object 



30
31
32
 
# File 'lib/jdbc_adapter/jdbc_mssql.rb', line 30

def self.column_selector
  [/sqlserver|tds/i, lambda {|cfg,col| col.extend(::JdbcSpec::MsSQL::Column)}]
end

.extended(mod) ⇒ Object 



7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
 
# File 'lib/jdbc_adapter/jdbc_mssql.rb', line 7

def self.extended(mod)
  unless @lob_callback_added
    ActiveRecord::Base.class_eval do
      def after_save_with_mssql_lob
      self.class.columns.select { |c| c.sql_type =~ /image/i }.each do |c|
        value = self[c.name]
        value = value.to_yaml if unserializable_attribute?(c.name, c)
        next if value.nil?  || (value == '')

        connection.write_large_object(c.type == :binary, c.name, self.class.table_name, self.class.primary_key, quote_value(id), value)
      end
      end
    end

    ActiveRecord::Base.after_save :after_save_with_mssql_lob
    @lob_callback_added = true
  end
end

.jdbc_connection_class ⇒ Object 



34
35
36
 
# File 'lib/jdbc_adapter/jdbc_mssql.rb', line 34

def self.jdbc_connection_class
  ::ActiveRecord::ConnectionAdapters::MssqlJdbcConnection
end

Instance Method Details

#_execute(sql, name = nil) ⇒ Object 



250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
 
# File 'lib/jdbc_adapter/jdbc_mssql.rb', line 250

def _execute(sql, name = nil)
  if sql.lstrip =~ /^insert/i
    if query_requires_identity_insert?(sql)
      table_name = get_table_name(sql)
      with_identity_insert_enabled(table_name) do
        id = @connection.execute_insert(sql)
      end
    else
      @connection.execute_insert(sql)
    end
  elsif sql.lstrip =~ /^\(?\s*(select|show)/i
    repair_special_columns(sql)
    @connection.execute_query(sql)
  else
    @connection.execute_update(sql)
  end
end

#add_column(table_name, column_name, type, options = {}) ⇒ Object

Adds a new column to the named table. See TableDefinition#column for details of the options you can use.



188
189
190
191
192
193
194
 
# File 'lib/jdbc_adapter/jdbc_mssql.rb', line 188

def add_column(table_name, column_name, type, options = {})
  add_column_sql = "ALTER TABLE #{table_name} ADD #{quote_column_name(column_name)} #{type_to_sql(type, options[:limit], options[:precision], options[:scale])}"
  add_column_options!(add_column_sql, options)
  # TODO: Add support to mimic date columns, using constraints to mark them as such in the database
  # add_column_sql << " CONSTRAINT ck__#{table_name}__#{column_name}__date_only CHECK ( CONVERT(CHAR(12), #{quote_column_name(column_name)}, 14)='00:00:00:000' )" if type == :date
  execute(add_column_sql)
end

#add_lock!(sql, options) ⇒ Object

SELECT .. FOR UPDATE is not supported on Microsoft SQL Server



269
270
271
 
# File 'lib/jdbc_adapter/jdbc_mssql.rb', line 269

def add_lock!(sql, options)
  sql
end

#change_column(table_name, column_name, type, options = {}) ⇒ Object

:nodoc:



200
201
202
203
204
205
206
207
208
209
 
# File 'lib/jdbc_adapter/jdbc_mssql.rb', line 200

def change_column(table_name, column_name, type, options = {}) #:nodoc:
  sql_commands = ["ALTER TABLE #{table_name} ALTER COLUMN #{column_name} #{type_to_sql(type, options[:limit], options[:precision], options[:scale])}"]
  if options_include_default?(options)
    remove_default_constraint(table_name, column_name)
    sql_commands << "ALTER TABLE #{table_name} ADD CONSTRAINT DF_#{table_name}_#{column_name} DEFAULT #{quote(options[:default], options[:column])} FOR #{column_name}"
  end
  sql_commands.each {|c|
    execute(c)
  }
end

#change_column_default(table_name, column_name, default) ⇒ Object

:nodoc:



210
211
212
213
 
# File 'lib/jdbc_adapter/jdbc_mssql.rb', line 210

def change_column_default(table_name, column_name, default) #:nodoc:
  remove_default_constraint(table_name, column_name)
  execute "ALTER TABLE #{table_name} ADD CONSTRAINT DF_#{table_name}_#{column_name} DEFAULT #{quote(default, column_name)} FOR #{column_name}"
end

#change_order_direction(order) ⇒ Object 



157
158
159
160
161
162
163
164
165
 
# File 'lib/jdbc_adapter/jdbc_mssql.rb', line 157

def change_order_direction(order)
  order.split(",").collect {|fragment|
    case fragment
    when  /\bDESC\b/i     then fragment.gsub(/\bDESC\b/i, "ASC")
    when  /\bASC\b/i      then fragment.gsub(/\bASC\b/i, "DESC")
    else                  String.new(fragment).split(',').join(' DESC,') + ' DESC'
    end
  }.join(",")
end

#columns(table_name, name = nil) ⇒ Object 



240
241
242
243
244
245
246
247
248
 
# File 'lib/jdbc_adapter/jdbc_mssql.rb', line 240

def columns(table_name, name = nil)
  return [] if table_name =~ /^information_schema\./i
  cc = super
  cc.each do |col|
    col.identity = true if col.sql_type =~ /identity/i
    col.is_special = true if col.sql_type =~ /text|ntext|image/i
  end
  cc
end

#create_database(name) ⇒ Object 



177
178
179
180
 
# File 'lib/jdbc_adapter/jdbc_mssql.rb', line 177

def create_database(name)
  execute "CREATE DATABASE #{name}"
  execute "USE #{name}"
end

#drop_database(name) ⇒ Object 



172
173
174
175
 
# File 'lib/jdbc_adapter/jdbc_mssql.rb', line 172

def drop_database(name)
  execute "USE master"
  execute "DROP DATABASE #{name}"
end

#quote(value, column = nil) ⇒ Object 



115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
 
# File 'lib/jdbc_adapter/jdbc_mssql.rb', line 115

def quote(value, column = nil)
  return value.quoted_id if value.respond_to?(:quoted_id)

  case value
  when String, ActiveSupport::Multibyte::Chars
    value = value.to_s
    if column && column.type == :binary
      "'#{quote_string(JdbcSpec::MsSQL::Column.string_to_binary(value))}'" # ' (for ruby-mode)
    elsif column && [:integer, :float].include?(column.type)
      value = column.type == :integer ? value.to_i : value.to_f
      value.to_s
    elsif column && column.respond_to?(:is_utf8?) && column.is_utf8?
      "N'#{quote_string(value)}'" # ' (for ruby-mode)
    else
      super
    end
  when TrueClass             then '1'
  when FalseClass            then '0'
  else                       super
  end
end

#quote_column_name(name) ⇒ Object 



145
146
147
 
# File 'lib/jdbc_adapter/jdbc_mssql.rb', line 145

def quote_column_name(name)
  "[#{name}]"
end

#quote_string(string) ⇒ Object 



137
138
139
 
# File 'lib/jdbc_adapter/jdbc_mssql.rb', line 137

def quote_string(string)
  string.gsub(/\'/, "''")
end

#quote_table_name(name) ⇒ Object 



141
142
143
 
# File 'lib/jdbc_adapter/jdbc_mssql.rb', line 141

def quote_table_name(name)
  name
end

#quoted_false ⇒ Object 



153
154
155
 
# File 'lib/jdbc_adapter/jdbc_mssql.rb', line 153

def quoted_false
  quote false
end

#quoted_true ⇒ Object 



149
150
151
 
# File 'lib/jdbc_adapter/jdbc_mssql.rb', line 149

def quoted_true
  quote true
end

#recreate_database(name) ⇒ Object 



167
168
169
170
 
# File 'lib/jdbc_adapter/jdbc_mssql.rb', line 167

def recreate_database(name)
  drop_database(name)
  create_database(name)
end

#remove_check_constraints(table_name, column_name) ⇒ Object 



227
228
229
230
231
232
233
 
# File 'lib/jdbc_adapter/jdbc_mssql.rb', line 227

def remove_check_constraints(table_name, column_name)
  # TODO remove all constraints in single method
  constraints = select "SELECT CONSTRAINT_NAME FROM INFORMATION_SCHEMA.CONSTRAINT_COLUMN_USAGE where TABLE_NAME = '#{table_name}' and COLUMN_NAME = '#{column_name}'"
  constraints.each do |constraint|
    execute "ALTER TABLE #{table_name} DROP CONSTRAINT #{constraint["CONSTRAINT_NAME"]}"
  end
end

#remove_column(table_name, column_name) ⇒ Object 



214
215
216
217
218
 
# File 'lib/jdbc_adapter/jdbc_mssql.rb', line 214

def remove_column(table_name, column_name)
  remove_check_constraints(table_name, column_name)
  remove_default_constraint(table_name, column_name)
  execute "ALTER TABLE #{table_name} DROP COLUMN [#{column_name}]"
end

#remove_default_constraint(table_name, column_name) ⇒ Object 



220
221
222
223
224
225
 
# File 'lib/jdbc_adapter/jdbc_mssql.rb', line 220

def remove_default_constraint(table_name, column_name)
  defaults = select "select def.name from sysobjects def, syscolumns col, sysobjects tab where col.cdefault = def.id and col.name = '#{column_name}' and tab.name = '#{table_name}' and col.id = tab.id"
  defaults.each {|constraint|
    execute "ALTER TABLE #{table_name} DROP CONSTRAINT #{constraint["name"]}"
  }
end

#remove_index(table_name, options = {}) ⇒ Object 



235
236
237
 
# File 'lib/jdbc_adapter/jdbc_mssql.rb', line 235

def remove_index(table_name, options = {})
  execute "DROP INDEX #{table_name}.#{index_name(table_name, options)}"
end

#rename_column(table, column, new_column_name) ⇒ Object 



196
197
198
 
# File 'lib/jdbc_adapter/jdbc_mssql.rb', line 196

def rename_column(table, column, new_column_name)
  execute "EXEC sp_rename '#{table}.#{column}', '#{new_column_name}'"
end

#rename_table(name, new_name) ⇒ Object 



182
183
184
 
# File 'lib/jdbc_adapter/jdbc_mssql.rb', line 182

def rename_table(name, new_name)
  execute "EXEC sp_rename '#{name}', '#{new_name}'"
end