sequel 3.21.0 → 3.41.0

data/lib/sequel/adapters/jdbc/db2.rb

@@ -0,0 +1,78 @@
+Sequel.require 'adapters/shared/db2'
+Sequel.require 'adapters/jdbc/transactions'
+
+module Sequel
+  module JDBC
+    class Database
+      # Alias the generic JDBC versions so they can be called directly later
+      alias jdbc_schema_parse_table schema_parse_table
+      alias jdbc_tables tables
+      alias jdbc_views views
+      alias jdbc_indexes indexes
+    end
+    
+    # Database and Dataset instance methods for DB2 specific
+    # support via JDBC.
+    module DB2
+      # Database instance methods for DB2 databases accessed via JDBC.
+      module DatabaseMethods
+        PRIMARY_KEY_INDEX_RE = /\Asql\d+\z/i.freeze
+
+        include Sequel::DB2::DatabaseMethods
+        include Sequel::JDBC::Transactions
+        IDENTITY_VAL_LOCAL = "SELECT IDENTITY_VAL_LOCAL() FROM SYSIBM.SYSDUMMY1".freeze
+        
+        %w'schema_parse_table tables views indexes'.each do |s|
+          class_eval("def #{s}(*a) jdbc_#{s}(*a) end", __FILE__, __LINE__)
+        end
+
+        private
+
+        def set_ps_arg(cps, arg, i)
+          case arg
+          when Sequel::SQL::Blob
+            cps.setString(i, arg)
+          else
+            super
+          end
+        end
+        
+        def last_insert_id(conn, opts={})
+          statement(conn) do |stmt|
+            sql = IDENTITY_VAL_LOCAL
+            rs = log_yield(sql){stmt.executeQuery(sql)}
+            rs.next
+            rs.getInt(1)
+          end
+        end
+        
+        # Primary key indexes appear to be named sqlNNNN on DB2
+        def primary_key_index_re
+          PRIMARY_KEY_INDEX_RE
+        end
+      end
+
+      class Dataset < JDBC::Dataset
+        include Sequel::DB2::DatasetMethods
+
+        class ::Sequel::JDBC::Dataset::TYPE_TRANSLATOR
+          def db2_clob(v) Sequel::SQL::Blob.new(v.getSubString(1, v.length)) end
+        end
+
+        DB2_CLOB_METHOD = TYPE_TRANSLATOR_INSTANCE.method(:db2_clob)
+      
+        private
+
+        # Return clob as blob if use_clob_as_blob is true
+        def convert_type_proc(v)
+          case v
+          when JAVA_SQL_CLOB
+            ::Sequel::DB2::use_clob_as_blob ? DB2_CLOB_METHOD : super
+          else
+            super
+          end
+        end
+      end
+    end
+  end
+end

data/lib/sequel/adapters/jdbc/derby.rb

@@ -0,0 +1,325 @@
+Sequel.require 'adapters/jdbc/transactions'
+
+module Sequel
+  module JDBC
+    # Database and Dataset support for Derby databases accessed via JDBC.
+    module Derby
+      # Instance methods for Derby Database objects accessed via JDBC.
+      module DatabaseMethods
+        PRIMARY_KEY_INDEX_RE = /\Asql\d+\z/i.freeze
+
+        include ::Sequel::JDBC::Transactions
+
+        # Derby doesn't support casting integer to varchar, only integer to char,
+        # and char(254) appears to have the widest support (with char(255) failing).
+        # This does add a bunch of extra spaces at the end, but those will be trimmed
+        # elsewhere.
+        def cast_type_literal(type)
+          (type == String) ? 'CHAR(254)' : super
+        end
+
+        # Derby uses the :derby database type.
+        def database_type
+          :derby
+        end
+
+        # Derby uses an IDENTITY sequence for autoincrementing columns.
+        def serial_primary_key_options
+          {:primary_key => true, :type => :integer, :identity=>true, :start_with=>1}
+        end
+
+        # The SVN version of the database.
+        def svn_version
+          @svn_version ||= begin
+            v = synchronize{|c| c.get_meta_data.get_database_product_version}
+            v =~ /\((\d+)\)\z/
+            $1.to_i
+          end
+        end
+        
+        # Derby supports transaction DDL statements.
+        def supports_transactional_ddl?
+          true
+        end
+
+        private
+        
+        # Derby optimizes away Sequel's default check of SELECT NULL FROM table,
+        # so use a SELECT * FROM table there.
+        def _table_exists?(ds)
+          ds.first
+        end
+    
+        # Derby-specific syntax for renaming columns and changing a columns type/nullity.
+        def alter_table_sql(table, op)
+          case op[:op]
+          when :rename_column
+            "RENAME COLUMN #{quote_schema_table(table)}.#{quote_identifier(op[:name])} TO #{quote_identifier(op[:new_name])}"
+          when :set_column_type
+            # Derby is very limited in changing a columns type, so adding a new column and then dropping the existing column is
+            # the best approach, as mentioned in the Derby documentation.
+            temp_name = :x_sequel_temp_column_x
+            [alter_table_sql(table, op.merge(:op=>:add_column, :name=>temp_name)),
+             from(table).update_sql(temp_name=>::Sequel::SQL::Cast.new(op[:name], op[:type])),
+             alter_table_sql(table, op.merge(:op=>:drop_column)),
+             alter_table_sql(table, op.merge(:op=>:rename_column, :name=>temp_name, :new_name=>op[:name]))]
+          when :set_column_null
+            "ALTER TABLE #{quote_schema_table(table)} ALTER COLUMN #{quote_identifier(op[:name])} #{op[:null] ? 'NULL' : 'NOT NULL'}"
+          else
+            super
+          end
+        end
+
+        # Derby doesn't allow specifying NULL for columns, only NOT NULL.
+        def column_definition_null_sql(sql, column)
+          sql << " NOT NULL" if column.fetch(:null, column[:allow_null]) == false
+        end
+    
+        # Add NOT LOGGED for temporary tables to improve performance.
+        def create_table_sql(name, generator, options)
+          s = super
+          s << ' NOT LOGGED' if options[:temp]
+          s
+        end
+
+        # Insert data from the current table into the new table after
+        # creating the table, since it is not possible to do it in one step.
+        def create_table_as(name, sql, options)
+          super
+          from(name).insert(sql.is_a?(Dataset) ? sql : dataset.with_sql(sql))
+        end
+
+        # Derby currently only requires WITH NO DATA, with a separate insert
+        # to import data.
+        def create_table_as_sql(name, sql, options)
+          "#{create_table_prefix_sql(name, options)} AS #{sql} WITH NO DATA"
+        end
+
+        # Temporary table creation on Derby uses DECLARE instead of CREATE.
+        def create_table_prefix_sql(name, options)
+          if options[:temp]
+            "DECLARE GLOBAL TEMPORARY TABLE #{quote_identifier(name)}"
+          else
+            super
+          end
+        end
+
+        # Use IDENTITY_VAL_LOCAL() to get the last inserted id.
+        def last_insert_id(conn, opts={})
+          statement(conn) do |stmt|
+            sql = 'SELECT IDENTITY_VAL_LOCAL() FROM sysibm.sysdummy1'
+            rs = log_yield(sql){stmt.executeQuery(sql)}
+            rs.next
+            rs.getInt(1)
+          end
+        end
+
+        # Handle nil values by using setNull with the correct parameter type.
+        def set_ps_arg_nil(cps, i)
+          cps.setNull(i, cps.getParameterMetaData.getParameterType(i))
+        end
+      
+        # Derby uses RENAME TABLE syntax to rename tables.
+        def rename_table_sql(name, new_name)
+          "RENAME TABLE #{quote_schema_table(name)} TO #{quote_schema_table(new_name)}"
+        end
+
+        # Primary key indexes appear to be named sqlNNNN on Derby
+        def primary_key_index_re
+          PRIMARY_KEY_INDEX_RE
+        end
+
+        # If an :identity option is present in the column, add the necessary IDENTITY SQL.
+        def type_literal(column)
+          if column[:identity]
+            sql = "#{super} GENERATED BY DEFAULT AS IDENTITY"
+            if sw = column[:start_with]
+              sql << " (START WITH #{sw.to_i}"
+              sql << " INCREMENT BY #{column[:increment_by].to_i}" if column[:increment_by]
+              sql << ")"
+            end
+            sql
+          else
+            super
+          end
+        end
+
+        # Derby uses clob for text types.
+        def uses_clob_for_text?
+          true
+        end
+
+        # The SQL query to issue to check if a connection is valid.
+        def valid_connection_sql
+          @valid_connection_sql ||= select(1).sql
+        end
+      end
+      
+      # Dataset class for Derby datasets accessed via JDBC.
+      class Dataset < JDBC::Dataset
+        PAREN_CLOSE = Dataset::PAREN_CLOSE
+        PAREN_OPEN = Dataset::PAREN_OPEN
+        OFFSET = Dataset::OFFSET
+        CAST_STRING_OPEN = "RTRIM(".freeze
+        BITCOMP_OPEN = "((0 - ".freeze
+        BITCOMP_CLOSE = ") - 1)".freeze
+        BLOB_OPEN = "CAST(X'".freeze
+        BLOB_CLOSE = "' AS BLOB)".freeze
+        HSTAR = "H*".freeze
+        TIME_FORMAT = "'%H:%M:%S'".freeze
+        DEFAULT_FROM = " FROM sysibm.sysdummy1".freeze
+        ROWS = " ROWS".freeze
+        FETCH_FIRST = " FETCH FIRST ".freeze
+        ROWS_ONLY = " ROWS ONLY".freeze
+        BOOL_TRUE_OLD = '(1 = 1)'.freeze
+        BOOL_FALSE_OLD = '(1 = 0)'.freeze
+        BOOL_TRUE = 'TRUE'.freeze
+        BOOL_FALSE = 'FALSE'.freeze
+        SELECT_CLAUSE_METHODS = clause_methods(:select, %w'select distinct columns from join where group having compounds order limit lock')
+        EMULATED_FUNCTION_MAP = {:char_length=>'length'.freeze}
+
+        # Derby doesn't support an expression between CASE and WHEN,
+        # so remove 
+        # conditions.
+        def case_expression_sql_append(sql, ce)
+          super(sql, ce.with_merged_expression)
+        end
+
+        # If the type is String, trim the extra spaces since CHAR is used instead
+        # of varchar.  This can cause problems if you are casting a char/varchar to
+        # a string and the ending whitespace is important.
+        def cast_sql_append(sql, expr, type)
+          if type == String
+            sql << CAST_STRING_OPEN
+            super
+            sql << PAREN_CLOSE
+          else
+            super
+          end
+        end
+
+        # Handle Derby specific LIKE, extract, and some bitwise compliment support.
+        def complex_expression_sql_append(sql, op, args)
+          case op
+          when :ILIKE
+            super(sql, :LIKE, [SQL::Function.new(:upper, args.at(0)), SQL::Function.new(:upper, args.at(1))])
+          when :"NOT ILIKE"
+            super(sql, :"NOT LIKE", [SQL::Function.new(:upper, args.at(0)), SQL::Function.new(:upper, args.at(1))])
+          when :%
+            sql << complex_expression_arg_pairs(args){|a, b| "MOD(#{literal(a)}, #{literal(b)})"}
+          when :&, :|, :^, :<<, :>>
+            raise Error, "Derby doesn't support the #{op} operator"
+          when :'B~'
+            sql << BITCOMP_OPEN
+            literal_append(sql, args.at(0))
+            sql << BITCOMP_CLOSE
+          when :extract
+            sql << args.at(0).to_s << PAREN_OPEN
+            literal_append(sql, args.at(1))
+            sql << PAREN_CLOSE
+          else
+            super
+          end
+        end
+
+        # Derby supports GROUP BY ROLLUP (but not CUBE)
+        def supports_group_rollup?
+          true
+        end
+
+        # Derby does not support IS TRUE.
+        def supports_is_true?
+          false
+        end
+
+        # Derby does not support IN/NOT IN with multiple columns
+        def supports_multiple_column_in?
+          false
+        end
+
+        private
+
+        JAVA_SQL_CLOB         = Java::JavaSQL::Clob
+
+        class ::Sequel::JDBC::Dataset::TYPE_TRANSLATOR
+          def derby_clob(v) v.getSubString(1, v.length) end
+        end
+
+        DERBY_CLOB_METHOD = TYPE_TRANSLATOR_INSTANCE.method(:derby_clob)
+      
+        # Handle clobs on Derby as strings.
+        def convert_type_proc(v)
+          if v.is_a?(JAVA_SQL_CLOB)
+            DERBY_CLOB_METHOD
+          else
+            super
+          end
+        end
+        
+        # Derby needs a hex string casted to BLOB for blobs.
+        def literal_blob_append(sql, v)
+          sql << BLOB_OPEN << v.unpack(HSTAR).first << BLOB_CLOSE
+        end
+
+        # Derby needs the standard workaround to insert all default values into
+        # a table with more than one column.
+        def insert_supports_empty_values?
+          false
+        end
+
+        # Derby uses an expression yielding false for false values.
+        # Newer versions can use the FALSE literal, but the latest gem version cannot.
+        def literal_false
+          if db.svn_version >= 1040133
+            BOOL_FALSE
+          else
+            BOOL_FALSE_OLD
+          end
+        end
+
+        # Derby handles fractional seconds in timestamps, but not in times
+        def literal_sqltime(v)
+          v.strftime(TIME_FORMAT)
+        end
+
+        # Derby uses an expression yielding true for true values.
+        # Newer versions can use the TRUE literal, but the latest gem version cannot.
+        def literal_true
+          if db.svn_version >= 1040133
+            BOOL_TRUE
+          else
+            BOOL_TRUE_OLD
+          end
+        end
+
+        # Derby doesn't support common table expressions.
+        def select_clause_methods
+          SELECT_CLAUSE_METHODS
+        end
+
+        # Use a default FROM table if the dataset does not contain a FROM table.
+        def select_from_sql(sql)
+          if @opts[:from]
+            super
+          else
+            sql << DEFAULT_FROM
+          end
+        end
+
+        # Offset comes before limit in Derby
+        def select_limit_sql(sql)
+          if o = @opts[:offset]
+            sql << OFFSET
+            literal_append(sql, o)
+            sql << ROWS
+          end
+          if l = @opts[:limit]
+            sql << FETCH_FIRST
+            literal_append(sql, l)
+            sql << ROWS_ONLY
+          end
+        end
+      end
+    end
+  end
+end

data/lib/sequel/adapters/jdbc/firebird.rb

@@ -0,0 +1,24 @@
+Sequel.require 'adapters/shared/firebird'
+Sequel.require 'adapters/jdbc/transactions'
+
+module Sequel
+  module JDBC
+    # Database and Dataset instance methods for Firebird specific
+    # support via JDBC.
+    module Firebird
+      # Database instance methods for Firebird databases accessed via JDBC.
+      module DatabaseMethods
+        include Sequel::Firebird::DatabaseMethods
+        include Sequel::JDBC::Transactions
+        
+        # Add the primary_keys and primary_key_sequences instance variables,
+        # so we can get the correct return values for inserted rows.
+        def self.extended(db)
+          db.instance_eval do
+            @primary_keys = {}
+          end
+        end
+      end
+    end
+  end
+end

data/lib/sequel/adapters/jdbc/h2.rb

@@ -17,11 +17,6 @@ module Sequel
           :h2
         end
 
-        # Return Sequel::JDBC::H2::Dataset object with the given opts.
-        def dataset(opts=nil)
-          Sequel::JDBC::H2::Dataset.new(self, opts)
-        end
-        
         # Rollback an existing prepared transaction with the given transaction
         # identifier string.
         def rollback_prepared_transaction(transaction_id)
@@ -30,9 +25,14 @@ module Sequel
 
         # H2 uses an IDENTITY type
         def serial_primary_key_options
-          {:primary_key => true, :type => :identity}
+          {:primary_key => true, :type => :identity, :identity=>true}
         end
 
+        # H2 supports CREATE TABLE IF NOT EXISTS syntax.
+        def supports_create_table_if_not_exists?
+          true
+        end
+      
         # H2 supports prepared transactions
         def supports_prepared_transactions?
           true
@@ -48,8 +48,8 @@ module Sequel
         # If the :prepare option is given and we aren't in a savepoint,
         # prepare the transaction for a two-phase commit.
         def commit_transaction(conn, opts={})
-          if opts[:prepare] && Thread.current[:sequel_transaction_depth] <= 1
-            log_connection_execute(conn, "PREPARE COMMIT #{opts[:prepare]}")
+          if (s = opts[:prepare]) && _trans(conn)[:savepoint_level] <= 1
+            log_connection_execute(conn, "PREPARE COMMIT #{s}")
           else
             super
           end
@@ -75,7 +75,16 @@ module Sequel
           when :set_column_null
             "ALTER TABLE #{quote_schema_table(table)} ALTER COLUMN #{quote_identifier(op[:name])} SET#{' NOT' unless op[:null]} NULL"
           when :set_column_type
-            "ALTER TABLE #{quote_schema_table(table)} ALTER COLUMN #{quote_identifier(op[:name])} #{type_literal(op)}"
+            if sch = schema(table)
+              if cs = sch.each{|k, v| break v if k == op[:name]; nil}
+                cs = cs.dup
+                cs[:default] = cs[:ruby_default]
+                op = cs.merge!(op)
+              end
+            end
+            sql = "ALTER TABLE #{quote_schema_table(table)} ALTER COLUMN #{quote_identifier(op[:name])} #{type_literal(op)}"
+            column_definition_order.each{|m| send(:"column_definition_#{m}_sql", sql, op)}
+            sql
           else
             super(table, op)
           end
@@ -90,7 +99,8 @@ module Sequel
         # Use IDENTITY() to get the last inserted id.
         def last_insert_id(conn, opts={})
           statement(conn) do |stmt|
-            rs = stmt.executeQuery('SELECT IDENTITY();')
+            sql = 'SELECT IDENTITY();'
+            rs = log_yield(sql){stmt.executeQuery(sql)}
             rs.next
             rs.getInt(1)
           end
@@ -99,28 +109,42 @@ module Sequel
         def primary_key_index_re
           PRIMARY_KEY_INDEX_RE
         end
+
+        # Use BIGINT IDENTITY for identity columns that use bigint, fixes
+        # the case where primary_key :column, :type=>Bignum is used.
+        def type_literal_generic_bignum(column)
+          column[:identity] ? 'BIGINT IDENTITY' : super
+        end
       end
       
       # Dataset class for H2 datasets accessed via JDBC.
       class Dataset < JDBC::Dataset
-        SELECT_CLAUSE_METHODS = clause_methods(:select, %w'distinct columns from join where group having compounds order limit')
+        SELECT_CLAUSE_METHODS = clause_methods(:select, %w'select distinct columns from join where group having compounds order limit')
         BITWISE_METHOD_MAP = {:& =>:BITAND, :| => :BITOR, :^ => :BITXOR}
+        APOS = Dataset::APOS
+        HSTAR = "H*".freeze
+        BITCOMP_OPEN = "((0 - ".freeze
+        BITCOMP_CLOSE = ") - 1)".freeze
+        ILIKE_PLACEHOLDER = ["CAST(".freeze, " AS VARCHAR_IGNORECASE)".freeze].freeze
+        TIME_FORMAT = "'%H:%M:%S'".freeze
         
-        # Work around H2's lack of a case insensitive LIKE operator
-        def complex_expression_sql(op, args)
+        # Emulate the case insensitive LIKE operator and the bitwise operators.
+        def complex_expression_sql_append(sql, op, args)
           case op
-          when :ILIKE
-            super(:LIKE, [SQL::PlaceholderLiteralString.new("CAST(? AS VARCHAR_IGNORECASE)", [args.at(0)]), args.at(1)])
-          when :"NOT ILIKE"
-            super(:"NOT LIKE", [SQL::PlaceholderLiteralString.new("CAST(? AS VARCHAR_IGNORECASE)", [args.at(0)]), args.at(1)])
+          when :ILIKE, :"NOT ILIKE"
+            super(sql, (op == :ILIKE ? :LIKE : :"NOT LIKE"), [SQL::PlaceholderLiteralString.new(ILIKE_PLACEHOLDER, [args.at(0)]), args.at(1)])
           when :&, :|, :^
-            literal(SQL::Function.new(BITWISE_METHOD_MAP[op], *args))
+            sql << complex_expression_arg_pairs(args){|a, b| literal(SQL::Function.new(BITWISE_METHOD_MAP[op], a, b))}
           when :<<
-            "(#{literal(args[0])} * POWER(2, #{literal(args[1])}))"
+            sql << complex_expression_arg_pairs(args){|a, b| "(#{literal(a)} * POWER(2, #{literal(b)}))"}
           when :>>
-            "(#{literal(args[0])} / POWER(2, #{literal(args[1])}))"
+            sql << complex_expression_arg_pairs(args){|a, b| "(#{literal(a)} / POWER(2, #{literal(b)}))"}
+          when :'B~'
+            sql << BITCOMP_OPEN
+            literal_append(sql, args.at(0))
+            sql << BITCOMP_CLOSE
           else
-            super(op, args)
+            super
           end
         end
         
@@ -145,21 +169,34 @@ module Sequel
         end 
 
         private
-      
-        # H2 expects hexadecimal strings for blob values
-        def literal_blob(v)
-          literal_string v.unpack("H*").first
+
+        #JAVA_H2_CLOB = Java::OrgH2Jdbc::JdbcClob
+
+        class ::Sequel::JDBC::Dataset::TYPE_TRANSLATOR
+          def h2_clob(v) v.getSubString(1, v.length) end
         end
-        
-        def convert_type(v)
-          case v
-          when Java::OrgH2Jdbc::JdbcClob
-            convert_type(v.getSubString(1, v.length))
+
+        H2_CLOB_METHOD = TYPE_TRANSLATOR_INSTANCE.method(:h2_clob)
+      
+        # Handle H2 specific clobs as strings.
+        def convert_type_proc(v)
+          if v.is_a?(Java::OrgH2Jdbc::JdbcClob)
+            H2_CLOB_METHOD
           else
-            super(v)
+            super
           end
         end
         
+        # H2 expects hexadecimal strings for blob values
+        def literal_blob_append(sql, v)
+          sql << APOS << v.unpack(HSTAR).first << APOS
+        end
+        
+        # H2 handles fractional seconds in timestamps, but not in times
+        def literal_sqltime(v)
+          v.strftime(TIME_FORMAT)
+        end
+
         def select_clause_methods
           SELECT_CLAUSE_METHODS
         end