module Sequel::DB2::DatabaseMethods

   # File lib/sequel/adapters/shared/db2.rb
14 def database_type
15   :db2
16 end

   # File lib/sequel/adapters/shared/db2.rb
20 def db2_version
21   return @db2_version if defined?(@db2_version)
22   @db2_version = metadata_dataset.with_sql("select service_level from sysibmadm.env_inst_info").first[:service_level]
23 end

   # File lib/sequel/adapters/shared/db2.rb
26 def freeze
27   db2_version
28   offset_strategy
29   super
30 end

   # File lib/sequel/adapters/shared/db2.rb
71 def indexes(table, opts = OPTS)
72   m = output_identifier_meth
73   table = table.value if table.is_a?(Sequel::SQL::Identifier)
74   indexes = {}
75   metadata_dataset.
76    from(Sequel[:syscat][:indexes]).
77    select(:indname, :uniquerule, :colnames).
78    where(:tabname=>input_identifier_meth.call(table), :system_required=>0).
79    each do |r|
80     indexes[m.call(r[:indname])] = {:unique=>(r[:uniquerule]=='U'), :columns=>r[:colnames][1..-1].split('+').map{|v| m.call(v)}}
81   end
82   indexes
83 end

   # File lib/sequel/adapters/shared/db2.rb
85 def offset_strategy
86   return @offset_strategy if defined?(@offset_strategy)
87 
88   @offset_strategy = case strategy = opts[:offset_strategy].to_s
89   when "limit_offset", "offset_fetch"
90     opts[:offset_strategy] = strategy.to_sym
91   else
92     opts[:offset_strategy] = :emulate
93   end
94 end

   # File lib/sequel/adapters/shared/db2.rb
33 def schema_parse_table(table, opts = OPTS)
34   m = output_identifier_meth(opts[:dataset])
35   im = input_identifier_meth(opts[:dataset])
36   metadata_dataset.with_sql("SELECT * FROM SYSIBM.SYSCOLUMNS WHERE TBNAME = #{literal(im.call(table))} ORDER BY COLNO").
37     collect do |column| 
38       column[:db_type] = column.delete(:typename)
39       if column[:db_type] =~ /\A(VAR)?CHAR\z/
40         column[:db_type] << "(#{column[:length]})"
41       end
42       if column[:db_type] == "DECIMAL"
43         column[:db_type] << "(#{column[:longlength]},#{column[:scale]})"
44       end
45       column[:allow_null] = column.delete(:nulls) == 'Y'
46       identity = column.delete(:identity) == 'Y'
47       if column[:primary_key] = identity || !column[:keyseq].nil?
48         column[:auto_increment] = identity
49       end
50       column[:type]        = schema_column_type(column[:db_type])
51       column[:max_length]  = column[:longlength] if column[:type] == :string
52       [ m.call(column.delete(:name)), column]
53     end
54 end

   # File lib/sequel/adapters/shared/db2.rb
97 def supports_transaction_isolation_levels?
98   true
99 end

    # File lib/sequel/adapters/shared/db2.rb
104 def table_exists?(name)
105   v ||= false # only retry once
106   sch, table_name = schema_and_table(name)
107   name = SQL::QualifiedIdentifier.new(sch, table_name) if sch
108   from(name).first
109   true
110 rescue DatabaseError => e
111   if e.to_s.include?('Operation not allowed for reason code "7" on table') && v == false
112     # table probably needs reorg
113     reorg(name)
114     v = true
115     retry 
116   end
117   false
118 end

   # File lib/sequel/adapters/shared/db2.rb
57 def tables
58   metadata_dataset.
59     with_sql("SELECT TABNAME FROM SYSCAT.TABLES WHERE TYPE='T' AND OWNER = #{literal(input_identifier_meth.call(opts[:user]))}").
60     all.map{|h| output_identifier_meth.call(h[:tabname]) }
61 end

   # File lib/sequel/adapters/shared/db2.rb
64 def views
65   metadata_dataset.
66     with_sql("SELECT TABNAME FROM SYSCAT.TABLES WHERE TYPE='V' AND OWNER = #{literal(input_identifier_meth.call(opts[:user]))}").
67     all.map{|h| output_identifier_meth.call(h[:tabname]) }
68 end

    # File lib/sequel/adapters/shared/db2.rb
122 def alter_table_sql(table, op)
123   case op[:op]
124   when :add_column
125     if op[:primary_key] && op[:auto_increment] && op[:type] == Integer
126       [
127       "ALTER TABLE #{quote_schema_table(table)} ADD #{column_definition_sql(op.merge(:auto_increment=>false, :primary_key=>false, :default=>0, :null=>false))}",
128       "ALTER TABLE #{quote_schema_table(table)} ALTER COLUMN #{literal(op[:name])} DROP DEFAULT",
129       "ALTER TABLE #{quote_schema_table(table)} ALTER COLUMN #{literal(op[:name])} SET #{auto_increment_sql}"
130       ]
131     else
132       "ALTER TABLE #{quote_schema_table(table)} ADD #{column_definition_sql(op)}"
133     end
134   when :drop_column
135     "ALTER TABLE #{quote_schema_table(table)} DROP #{column_definition_sql(op)}"
136   when :rename_column       # renaming is only possible after db2 v9.7
137     "ALTER TABLE #{quote_schema_table(table)} RENAME COLUMN #{quote_identifier(op[:name])} TO #{quote_identifier(op[:new_name])}"
138   when :set_column_type
139     "ALTER TABLE #{quote_schema_table(table)} ALTER COLUMN #{quote_identifier(op[:name])} SET DATA TYPE #{type_literal(op)}"
140   when :set_column_default
141     "ALTER TABLE #{quote_schema_table(table)} ALTER COLUMN #{quote_identifier(op[:name])} SET DEFAULT #{literal(op[:default])}"
142   when :add_constraint
143     if op[:type] == :unique
144       sqls = op[:columns].map{|c| ["ALTER TABLE #{quote_schema_table(table)} ALTER COLUMN #{quote_identifier(c)} SET NOT NULL", reorg_sql(table)]}
145       sqls << super
146       sqls.flatten
147     else
148       super
149     end
150   else
151     super
152   end
153 end

    # File lib/sequel/adapters/shared/db2.rb
158 def apply_alter_table(name, ops)
159   alter_table_sql_list(name, ops).each do |sql|
160     execute_ddl(sql)
161     reorg(name)
162   end
163 end

    # File lib/sequel/adapters/shared/db2.rb
166 def auto_increment_sql
167   'GENERATED ALWAYS AS IDENTITY'
168 end

    # File lib/sequel/adapters/shared/db2.rb
171 def can_add_primary_key_constraint_on_nullable_columns?
172   false
173 end

    # File lib/sequel/adapters/shared/db2.rb
177 def column_list_sql(g)
178   ks = {}
179   g.constraints.each do |c|
180     case c[:type]
181     when :primary_key, :unique
182       c[:columns].each{|c| ks[c] = true}
183     end
184   end
185   g.columns.each{|c| c[:null] = false if ks.include?(c[:name]) }
186   super
187 end

    # File lib/sequel/adapters/shared/db2.rb
191 def create_table_as(name, sql, options)
192   super
193   if options[:temp]
194     name = Sequel.qualify(:session, name)
195   end
196   from(name).insert(sql.is_a?(Dataset) ? sql : dataset.with_sql(sql))
197 end

    # File lib/sequel/adapters/shared/db2.rb
200 def create_table_as_sql(name, sql, options)
201   "#{create_table_prefix_sql(name, options)} AS (#{sql}) DEFINITION ONLY"
202 end

    # File lib/sequel/adapters/shared/db2.rb
207 def create_table_prefix_sql(name, options)
208   if options[:temp]
209     "DECLARE GLOBAL TEMPORARY TABLE #{create_table_table_name_sql(name, options)}"
210   else
211     super
212   end
213 end

    # File lib/sequel/adapters/shared/db2.rb
222 def database_error_regexps
223   DATABASE_ERROR_REGEXPS
224 end

    # File lib/sequel/adapters/shared/db2.rb
227 def disconnect_error?(exception, opts)
228   sqlstate = database_exception_sqlstate(exception, opts)
229 
230   case sqlstate
231   when *DISCONNECT_SQL_STATES
232     true
233   else
234     super
235   end
236 end

    # File lib/sequel/adapters/shared/db2.rb
240 def quote_identifiers_default
241   false
242 end

    # File lib/sequel/adapters/shared/db2.rb
245 def rename_table_sql(name, new_name)
246   "RENAME TABLE #{quote_schema_table(name)} TO #{quote_schema_table(new_name)}"
247 end

    # File lib/sequel/adapters/shared/db2.rb
251 def reorg(table)
252   execute_ddl(reorg_sql(table))
253 end

    # File lib/sequel/adapters/shared/db2.rb
256 def reorg_sql(table)
257   "CALL SYSPROC.ADMIN_CMD(#{literal("REORG TABLE #{quote_schema_table(table)}")})"
258 end

    # File lib/sequel/adapters/shared/db2.rb
261 def schema_column_type(db_type)
262   (use_clob_as_blob && db_type.downcase == 'clob') ? :blob : super
263 end

    # File lib/sequel/adapters/shared/db2.rb
266 def set_transaction_isolation_sql(level)
267   "SET CURRENT ISOLATION #{Database::TRANSACTION_ISOLATION_LEVELS[level]}"
268 end

    # File lib/sequel/adapters/shared/db2.rb
274 def type_literal_generic_file(column)
275   use_clob_as_blob ? :clob : :blob
276 end

    # File lib/sequel/adapters/shared/db2.rb
279 def type_literal_generic_trueclass(column)
280   :smallint
281 end

    # File lib/sequel/adapters/shared/db2.rb
285 def uses_clob_for_text?
286   true
287 end

    # File lib/sequel/adapters/shared/db2.rb
290 def view_with_check_option_support
291   :local
292 end