Module: JdbcSpec::Derby

Includes:
MissingFunctionalityHelper
Defined in:
lib/jdbc_adapter/jdbc_derby.rb

Defined Under Namespace

Modules: Column

Constant Summary collapse

COLUMN_INFO_STMT = 
"SELECT C.COLUMNNAME, C.REFERENCEID, C.COLUMNNUMBER FROM SYS.SYSCOLUMNS C, SYS.SYSTABLES T WHERE T.TABLEID = '%s' AND T.TABLEID = C.REFERENCEID ORDER BY C.COLUMNNUMBER"
COLUMN_TYPE_STMT = 
"SELECT COLUMNDATATYPE, COLUMNDEFAULT FROM SYS.SYSCOLUMNS WHERE REFERENCEID = '%s' AND COLUMNNAME = '%s'"
AUTO_INC_STMT = 
"SELECT AUTOINCREMENTSTART, AUTOINCREMENTINC, COLUMNNAME, REFERENCEID, COLUMNDEFAULT FROM SYS.SYSCOLUMNS WHERE REFERENCEID = '%s' AND COLUMNNAME = '%s'"
AUTO_INC_STMT2 = 
"SELECT AUTOINCREMENTSTART, AUTOINCREMENTINC, COLUMNNAME, REFERENCEID, COLUMNDEFAULT FROM SYS.SYSCOLUMNS WHERE REFERENCEID = (SELECT T.TABLEID FROM SYS.SYSTABLES T WHERE T.TABLENAME = '%s') AND COLUMNNAME = '%s'"
SIZEABLE = 
%w(VARCHAR CLOB BLOB)

Class Method Summary collapse

Instance Method Summary collapse

Methods included from MissingFunctionalityHelper

#alter_table, #copy_table, #copy_table_contents, #copy_table_indexes, #move_table

Class Method Details

.adapter_selector ⇒ Object 



17
18
19
 
# File 'lib/jdbc_adapter/jdbc_derby.rb', line 17

def self.adapter_selector
  [/derby/i, lambda {|cfg,adapt| adapt.extend(::JdbcSpec::Derby)}]
end

.column_selector ⇒ Object 



13
14
15
 
# File 'lib/jdbc_adapter/jdbc_derby.rb', line 13

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

.extended(*args) ⇒ Object 



42
43
44
 
# File 'lib/jdbc_adapter/jdbc_derby.rb', line 42

def self.extended(*args)
  monkey_rails
end

.included(*args) ⇒ Object 



46
47
48
 
# File 'lib/jdbc_adapter/jdbc_derby.rb', line 46

def self.included(*args)
  monkey_rails
end

.monkey_rails ⇒ Object 



21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
 
# File 'lib/jdbc_adapter/jdbc_derby.rb', line 21

def self.monkey_rails
  unless @already_monkeyd
    # Needed because Rails is broken wrt to quoting of 
    # some values. Most databases are nice about it,
    # but not Derby. The real issue is that you can't
    # compare a CHAR value to a NUMBER column.
    ::ActiveRecord::Associations::ClassMethods.module_eval do
      private

      def select_limited_ids_list(options, join_dependency)
        connection.select_all(
                              construct_finder_sql_for_association_limiting(options, join_dependency),
                              "#{name} Load IDs For Limited Eager Loading"
                              ).collect { |row| connection.quote(row[primary_key], columns_hash[primary_key]) }.join(", ")
      end           
    end

    @already_monkeyd = true
  end
end

Instance Method Details

#add_quotes(name) ⇒ Object 



128
129
130
131
 
# File 'lib/jdbc_adapter/jdbc_derby.rb', line 128

def add_quotes(name)
  return name unless name
  %Q{"#{name}"}
end

#auto_increment_stmt(tname, cname) ⇒ Object 



180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
 
# File 'lib/jdbc_adapter/jdbc_derby.rb', line 180

def auto_increment_stmt(tname, cname)
  stmt = AUTO_INC_STMT2 % [tname, strip_quotes(cname)]
  data = execute(stmt).first
  if data
    start = data['autoincrementstart']
    if start
      coldef = ""
      coldef << " GENERATED " << (data['columndefault'].nil? ? "ALWAYS" : "BY DEFAULT ")
      coldef << "AS IDENTITY (START WITH "
      coldef << start
      coldef << ", INCREMENT BY "
      coldef << data['autoincrementinc']
      coldef << ")"
      return coldef
    end
  end
  ""
end

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

Notes about changing in Derby:

http://db.apache.org/derby/docs/10.2/ref/rrefsqlj81859.html#rrefsqlj81859__rrefsqlj37860)

We support changing columns using the strategy outlined in:

https://issues.apache.org/jira/browse/DERBY-1515

This feature has not made it into a formal release and is not in Java 6. We will need to conditionally support this somehow (supposed to arrive for 10.3.0.0)



283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
 
# File 'lib/jdbc_adapter/jdbc_derby.rb', line 283

def change_column(table_name, column_name, type, options = {})
  # null/not nulling is easy, handle that separately
  if options.include?(:null)
    # This seems to only work with 10.2 of Derby
    if options.delete(:null) == false
      execute "ALTER TABLE #{table_name} ALTER COLUMN #{column_name} NOT NULL"
    else
      execute "ALTER TABLE #{table_name} ALTER COLUMN #{column_name} NULL"
    end
  end

  # anything left to do?
  unless options.empty?
    begin
      execute "ALTER TABLE #{table_name} ALTER COLUMN #{column_name} SET DATA TYPE #{type_to_sql(type, options[:limit])}"
    rescue
      transaction do
        temp_new_column_name = "#{column_name}_newtype"
        # 1) ALTER TABLE t ADD COLUMN c1_newtype NEWTYPE;
        add_column table_name, temp_new_column_name, type, options
        # 2) UPDATE t SET c1_newtype = c1;
        execute "UPDATE #{table_name} SET #{temp_new_column_name} = CAST(#{column_name} AS #{type_to_sql(type, options[:limit])})"
        # 3) ALTER TABLE t DROP COLUMN c1;
        remove_column table_name, column_name
        # 4) ALTER TABLE t RENAME COLUMN c1_newtype to c1;
        rename_column table_name, temp_new_column_name, column_name
      end
    end
  end
end

#classes_for_table_name(table) ⇒ Object 



90
91
92
 
# File 'lib/jdbc_adapter/jdbc_derby.rb', line 90

def classes_for_table_name(table)
  ActiveRecord::Base.send(:subclasses).select {|klass| klass.table_name == table}
end

#create_column(name, refid, colno) ⇒ Object 



199
200
201
202
203
204
205
206
207
208
209
210
211
212
 
# File 'lib/jdbc_adapter/jdbc_derby.rb', line 199

def create_column(name, refid, colno)
  stmt = COLUMN_TYPE_STMT % [refid, strip_quotes(name)]
  coldef = ""
  data = execute(stmt).first
  if data
    coldef << add_quotes(expand_double_quotes(strip_quotes(name)))
    coldef << " "
    coldef << data['columndatatype']
    if !reinstate_auto_increment(name, refid, coldef) && data['columndefault']
      coldef << " DEFAULT " << data['columndefault']
    end
  end
  coldef
end

#expand_double_quotes(name) ⇒ Object 



139
140
141
142
 
# File 'lib/jdbc_adapter/jdbc_derby.rb', line 139

def expand_double_quotes(name)
  return name unless name && name['"']
  name.gsub(/"/,'""')
end

#modify_types(tp) ⇒ Object 



82
83
84
85
86
87
88
 
# File 'lib/jdbc_adapter/jdbc_derby.rb', line 82

def modify_types(tp)
  tp[:primary_key] = "int generated by default as identity NOT NULL PRIMARY KEY"
  tp[:integer][:limit] = nil
  tp[:string][:limit] = 256
  tp[:boolean] = {:name => "smallint"}
  tp
end

#primary_key(table_name) ⇒ Object

:nodoc:



109
110
111
 
# File 'lib/jdbc_adapter/jdbc_derby.rb', line 109

def primary_key(table_name) #:nodoc:
  primary_keys(table_name).first
end

#primary_keys(table_name) ⇒ Object 



328
329
330
 
# File 'lib/jdbc_adapter/jdbc_derby.rb', line 328

def primary_keys(table_name)
  @connection.primary_keys table_name.to_s.upcase
end

#quote_column_name(name) ⇒ Object

For DDL it appears you can quote “” column names, but in queries (like insert it errors out?)



339
340
341
342
343
344
345
346
347
348
349
350
351
 
# File 'lib/jdbc_adapter/jdbc_derby.rb', line 339

def quote_column_name(name) #:nodoc:
  if /^references$/i =~ name
    %Q{"#{name.upcase}"}
  elsif /[A-Z]/ =~ name && /[a-z]/ =~ name
    %Q{"#{name}"}
  elsif name =~ /\s/
    %Q{"#{name.upcase}"}
  elsif name =~ /^[_\d]/
    %Q{"#{name.upcase}"}
  else
    name
  end
end

#quoted_false ⇒ Object 



357
358
359
 
# File 'lib/jdbc_adapter/jdbc_derby.rb', line 357

def quoted_false
  '0'
end

#quoted_true ⇒ Object 



353
354
355
 
# File 'lib/jdbc_adapter/jdbc_derby.rb', line 353

def quoted_true
  '1'
end

#recreate_database(db_name) ⇒ Object 



332
333
334
335
336
 
# File 'lib/jdbc_adapter/jdbc_derby.rb', line 332

def recreate_database(db_name)
  tables.each do |t|
    drop_table t
  end
end

#reinstate_auto_increment(name, refid, coldef) ⇒ Object 



144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
 
# File 'lib/jdbc_adapter/jdbc_derby.rb', line 144

def reinstate_auto_increment(name, refid, coldef)
  stmt = AUTO_INC_STMT % [refid, strip_quotes(name)]
  data = execute(stmt).first
  if data
    start = data['autoincrementstart']
    if start
      coldef << " GENERATED " << (data['columndefault'].nil? ? "ALWAYS" : "BY DEFAULT ")
      coldef << "AS IDENTITY (START WITH "
      coldef << start
      coldef << ", INCREMENT BY "
      coldef << data['autoincrementinc']
      coldef << ")"
      return true
    end
  end
  false
end

#remove_column(table_name, column_name) ⇒ Object

Support for removing columns added via derby bug issue: issues.apache.org/jira/browse/DERBY-1489

This feature has not made it into a formal release and is not in Java 6. If the normal strategy fails we fall back on a strategy by creating a new table without the new column and there after moving the data to the new



265
266
267
268
269
270
271
272
273
 
# File 'lib/jdbc_adapter/jdbc_derby.rb', line 265

def remove_column(table_name, column_name)
  begin
    execute "ALTER TABLE #{table_name} DROP COLUMN #{column_name} RESTRICT"
  rescue
    alter_table(table_name) do |definition|
      definition.columns.delete(definition[column_name])
    end
  end
end

#remove_index(table_name, options) ⇒ Object

:nodoc:



113
114
115
 
# File 'lib/jdbc_adapter/jdbc_derby.rb', line 113

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

#rename_column(table_name, column_name, new_column_name) ⇒ Object

Support for renaming columns: issues.apache.org/jira/browse/DERBY-1490

This feature is expect to arrive in version 10.3.0.0: wiki.apache.org/db-derby/DerbyTenThreeRelease)



320
321
322
323
324
325
326
 
# File 'lib/jdbc_adapter/jdbc_derby.rb', line 320

def rename_column(table_name, column_name, new_column_name) #:nodoc:
  begin
    execute "ALTER TABLE #{table_name} ALTER RENAME COLUMN #{column_name} TO #{new_column_name}"
  rescue
    alter_table(table_name, :rename => {column_name => new_column_name})
  end
end

#rename_table(name, new_name) ⇒ Object 



117
118
119
 
# File 'lib/jdbc_adapter/jdbc_derby.rb', line 117

def rename_table(name, new_name)
  execute "RENAME TABLE #{name} TO #{new_name}"
end

#reset_pk_sequence!(table, pk = nil, sequence = nil) ⇒ Object 



100
101
102
103
104
105
106
107
 
# File 'lib/jdbc_adapter/jdbc_derby.rb', line 100

def reset_pk_sequence!(table, pk = nil, sequence = nil)
  klasses = classes_for_table_name(table)
  klass   = klasses.nil? ? nil : klasses.first
  pk      = klass.primary_key unless klass.nil?
  if pk && klass.columns_hash[pk].type == :integer
    reset_sequence!(klass.table_name, pk)
  end
end

#reset_sequence!(table, column, sequence = nil) ⇒ Object

Set the sequence to the max value of the table’s column.



95
96
97
98
 
# File 'lib/jdbc_adapter/jdbc_derby.rb', line 95

def reset_sequence!(table, column, sequence = nil)
  mpk = select_value("SELECT MAX(#{quote_column_name column}) FROM #{table}")
  execute("ALTER TABLE #{table} ALTER COLUMN #{quote_column_name column} RESTART WITH #{mpk.to_i + 1}")      
end

#strip_quotes(str) ⇒ Object 



133
134
135
136
137
 
# File 'lib/jdbc_adapter/jdbc_derby.rb', line 133

def strip_quotes(str)
  return str unless str
  return str unless /^(["']).*\1$/ =~ str
  str[1..-2]
end

#structure_dump ⇒ Object

:nodoc:



216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
 
# File 'lib/jdbc_adapter/jdbc_derby.rb', line 216

def structure_dump #:nodoc:
  definition=""
  rs = @connection.connection.meta_data.getTables(nil,nil,nil,["TABLE"].to_java(:string))
  while rs.next
    tname = rs.getString(3)
    definition << "CREATE TABLE #{tname} (\n"
    rs2 = @connection.connection.meta_data.getColumns(nil,nil,tname,nil)
    first_col = true
    while rs2.next
      col_name = add_quotes(rs2.getString(4));
      default = ""
      d1 = rs2.getString(13)
      if d1 =~ /^GENERATED_/
        default = auto_increment_stmt(tname, col_name)
      elsif d1
        default = " DEFAULT #{d1}"
      end
      
      type = rs2.getString(6)
      col_size = rs2.getString(7)
      nulling = (rs2.getString(18) == 'NO' ? " NOT NULL" : "")
      create_col_string = add_quotes(expand_double_quotes(strip_quotes(col_name))) + 
        " " + 
        type +
        (SIZEABLE.include?(type) ? "(#{col_size})" : "") + 
        nulling +
        default
      if !first_col
        create_col_string = ",\n #{create_col_string}"
      else
        create_col_string = " #{create_col_string}"
      end
      
      definition << create_col_string
      
      first_col = false
    end
    definition << ");\n\n"
  end
  definition
end