upsert 2.9.10-java

data/lib/upsert/column_definition/postgresql.rb

@@ -0,0 +1,66 @@
+class Upsert
+  class ColumnDefinition
+    # @private
+    class Postgresql < ColumnDefinition
+      class << self
+        # activerecord-3.2.5/lib/active_record/connection_adapters/postgresql_adapter.rb#column_definitions
+        def all(connection, quoted_table_name)
+          res = connection.execute <<-EOS
+  SELECT a.attname AS name, format_type(a.atttypid, a.atttypmod) AS sql_type, d.adsrc AS default
+  FROM pg_attribute a LEFT JOIN pg_attrdef d
+  ON a.attrelid = d.adrelid AND a.attnum = d.adnum
+  WHERE a.attrelid = '#{quoted_table_name}'::regclass
+  AND a.attnum > 0 AND NOT a.attisdropped
+  EOS
+          res.map do |row|
+            new connection, row['name'], row['sql_type'], row['default']
+          end.sort_by do |cd|
+            cd.name
+          end
+        end
+      end
+
+      # NOTE not using this because it can't be indexed
+      # def equality(left, right)
+      #   "#{left} IS NOT DISTINCT FROM #{right}"
+      # end
+
+      HSTORE_DETECTOR = /hstore/i
+
+      def initialize(*)
+        super
+        @hstore_query = !!(sql_type =~ HSTORE_DETECTOR)
+      end
+
+      def hstore?
+        @hstore_query
+      end
+
+      def arg_type
+        if hstore?
+          'text'
+        else
+          # JDBC uses prepared statements and properly sends date objects (which are otherwise escaped)
+          RUBY_PLATFORM == "java" ? sql_type : super
+        end
+      end
+
+      def to_setter_value
+        if hstore?
+          "#{quoted_setter_name}::hstore"
+        else
+          super
+        end
+      end
+
+      def to_setter
+        if hstore?
+          # http://stackoverflow.com/questions/9317971/adding-a-key-to-an-empty-hstore-column
+          "#{quoted_name} = COALESCE(#{quoted_name}, hstore(array[]::varchar[])) || #{to_setter_value}"
+        else
+          super
+        end
+      end
+    end
+  end
+end

data/lib/upsert/column_definition/sqlite3.rb

@@ -0,0 +1,34 @@
+class Upsert
+  class ColumnDefinition
+    # @private
+    class Sqlite3 < ColumnDefinition
+      class << self
+        def all(connection, quoted_table_name)
+          # activerecord-3.2.13/lib/active_record/connection_adapters/sqlite_adapter.rb
+          connection.execute("PRAGMA table_info(#{quoted_table_name})").map do |row|#, 'SCHEMA').to_hash
+            if connection.metal.respond_to?(:results_as_hash) and not connection.metal.results_as_hash
+              row = {'name' => row[1], 'type' => row[2], 'dflt_value' => row[4]}
+            end
+            default = case row["dflt_value"]
+            when /^null$/i
+              nil
+            when /^'(.*)'$/
+              $1.gsub(/''/, "'")
+            when /^"(.*)"$/
+              $1.gsub(/""/, '"')
+            else
+              row["dflt_value"]
+            end
+            new connection, row['name'], row['type'], default
+          end.sort_by do |cd|
+            cd.name
+          end
+        end
+      end
+
+      def equality(left, right)
+        "(#{left} IS #{right} OR (#{left} IS NULL AND #{right} IS NULL))"
+      end
+    end
+  end
+end

data/lib/upsert/connection.rb

@@ -0,0 +1,37 @@
+class Upsert
+  # @private
+  class Connection
+    attr_reader :controller
+    attr_reader :metal
+
+    def initialize(controller, metal)
+      @controller = controller
+      @metal = metal
+    end
+
+    def convert_binary(bind_values)
+      bind_values.map do |v|
+        case v
+        when Upsert::Binary
+          binary v
+        else
+          v
+        end
+      end
+    end
+
+    def bind_value(v)
+      case v
+      when Time, DateTime
+        Upsert.utc_iso8601 v
+      when Date
+        v.strftime ISO8601_DATE
+      when Symbol
+        v.to_s
+      else
+        v
+      end
+    end
+
+  end
+end

data/lib/upsert/connection/Java_ComMysqlJdbc_JDBC4Connection.rb

@@ -0,0 +1,31 @@
+require 'upsert/connection/jdbc'
+
+class Upsert
+  class Connection
+    # @private
+    class Java_ComMysqlJdbc_JDBC4Connection < Connection
+      include Jdbc
+
+      def quote_ident(k)
+        if metal.useAnsiQuotedIdentifiers
+          DOUBLE_QUOTE + k.to_s.gsub(DOUBLE_QUOTE, '""') + DOUBLE_QUOTE
+        else
+          # Escape backticks by doubling them.  Ref http://dev.mysql.com/doc/refman/5.7/en/identifiers.html
+          BACKTICK + k.to_s.gsub(BACKTICK, BACKTICK + BACKTICK) + BACKTICK
+        end
+      end
+
+      def bind_value(v)
+        case v
+        when DateTime, Time
+          date = v.utc
+          java.time.LocalDateTime.of(date.year, date.month, date.day, date.hour, date.min, date.sec, date.nsec)
+        when Date
+          java.time.LocalDate.of(v.year, v.month, v.day)
+        else
+          super
+        end
+      end
+    end
+  end
+end

data/lib/upsert/connection/Java_OrgPostgresqlJdbc_PgConnection.rb

@@ -0,0 +1,33 @@
+require_relative "jdbc"
+require_relative "postgresql"
+
+class Upsert
+  class Connection
+    # @private
+    class Java_OrgPostgresqlJdbc_PgConnection < Connection
+      include Jdbc
+      include Postgresql
+
+      def quote_ident(k)
+        DOUBLE_QUOTE + k.to_s.gsub(DOUBLE_QUOTE, '""') + DOUBLE_QUOTE
+      end
+
+      def in_transaction?
+        # https://github.com/kares/activerecord-jdbc-adapter/commit/4d6e0e0c52d12b0166810dffc9f898141a23bee6
+        ![0, 4].include?(metal.get_transaction_state)
+      end
+
+      def bind_value(v)
+        case v
+        when DateTime, Time
+          date = v.utc
+          java.time.LocalDateTime.of(date.year, date.month, date.day, date.hour, date.min, date.sec, date.nsec)
+        when Date
+          java.time.LocalDate.of(v.year, v.month, v.day)
+        else
+          super
+        end
+      end
+    end
+  end
+end

data/lib/upsert/connection/Java_OrgSqlite_Conn.rb

@@ -0,0 +1,17 @@
+require 'upsert/connection/jdbc'
+require 'upsert/connection/sqlite3'
+
+class Upsert
+  class Connection
+    # @private
+    class Java_OrgSqlite_Conn < Connection
+      include Jdbc
+      include Sqlite3
+
+      # ?
+      def quote_ident(k)
+        DOUBLE_QUOTE + k.to_s.gsub(DOUBLE_QUOTE, '""') + DOUBLE_QUOTE
+      end
+    end
+  end
+end

data/lib/upsert/connection/Mysql2_Client.rb

@@ -0,0 +1,76 @@
+class Upsert
+  class Connection
+    # @private
+    class Mysql2_Client < Connection
+      def execute(sql)
+        Upsert.logger.debug { %{[upsert] #{sql}} }
+        if results = metal.query(sql)
+          rows = []
+          results.each { |row| rows << row }
+          if rows[0].is_a? Array
+            # you don't know if mysql2 is going to give you an array or a hash... and you shouldn't specify, because it's sticky
+            fields = results.fields
+            rows.map { |row| Hash[fields.zip(row)] }
+          else
+            rows
+          end
+        end
+      end
+
+      def quote_value(v)
+        case v
+        when NilClass
+          NULL_WORD
+        when Upsert::Binary
+          quote_binary v.value
+        when String
+          quote_string v
+        when TrueClass, FalseClass
+          quote_boolean v
+        when BigDecimal
+          quote_big_decimal v
+        when Numeric
+          v
+        when Symbol
+          quote_string v.to_s
+        when DateTime, Time
+          # mysql doesn't like it when you send timezone to a datetime
+          quote_string Upsert.utc_iso8601(v, false)
+        when Date
+          quote_date v
+        else
+          raise "not sure how to quote #{v.class}: #{v.inspect}"
+        end
+      end
+
+      def quote_boolean(v)
+        v ? 'TRUE' : 'FALSE'
+      end
+
+      def quote_string(v)
+        SINGLE_QUOTE + metal.escape(v) + SINGLE_QUOTE
+      end
+
+      # This doubles the size of the representation.
+      def quote_binary(v)
+        X_AND_SINGLE_QUOTE + v.unpack("H*")[0] + SINGLE_QUOTE
+      end
+
+      # put raw binary straight into sql
+      # might work if we could get the encoding issues fixed when joining together the values for the sql
+      # alias_method :quote_binary, :quote_string
+
+      def quote_date(v)
+        quote_string v.strftime(ISO8601_DATE)
+      end
+
+      def quote_ident(k)
+        BACKTICK + metal.escape(k.to_s) + BACKTICK
+      end
+
+      def quote_big_decimal(v)
+        v.to_s('F')
+      end
+    end
+  end
+end

data/lib/upsert/connection/PG_Connection.rb

@@ -0,0 +1,35 @@
+require_relative "postgresql"
+
+class Upsert
+  class Connection
+    # @private
+    class PG_Connection < Connection
+      include Postgresql
+
+      def execute(sql, params = nil)
+        if params
+          # Upsert.logger.debug { %{[upsert] #{sql} with #{params.inspect}} }
+          # The following will blow up if you pass a value that cannot be automatically type-casted,
+          #   such as passing a string to an integer field.  You'll get an error something along the
+          #   lines of: "invalid input syntax for <type>: <value>"
+          metal.exec sql, convert_binary(params)
+        else
+          Upsert.logger.debug { %{[upsert] #{sql}} }
+          metal.exec sql
+        end
+      end
+
+      def quote_ident(k)
+        metal.quote_ident k.to_s
+      end
+
+      def binary(v)
+        { :value => v.value, :format => 1 }
+      end
+
+      def in_transaction?
+        ![PG::PQTRANS_IDLE, PG::PQTRANS_UNKNOWN].include?(metal.transaction_status)
+      end
+    end
+  end
+end

data/lib/upsert/connection/SQLite3_Database.rb

@@ -0,0 +1,28 @@
+require 'upsert/connection/sqlite3'
+
+class Upsert
+  class Connection
+    # @private
+    class SQLite3_Database < Connection
+      include Sqlite3
+      
+      def execute(sql, params = nil)
+        if params
+          Upsert.logger.debug { %{[upsert] #{sql} with #{params.inspect}} }
+          metal.execute sql, convert_binary(params)
+        else
+          Upsert.logger.debug { %{[upsert] #{sql}} }
+          metal.execute sql
+        end
+      end
+      
+      def quote_ident(k)
+        DOUBLE_QUOTE + SQLite3::Database.quote(k.to_s) + DOUBLE_QUOTE
+      end
+
+      def binary(v)
+        SQLite3::Blob.new v.value
+      end
+    end
+  end
+end

data/lib/upsert/connection/jdbc.rb

@@ -0,0 +1,105 @@
+class Upsert
+  class Connection
+    # @private
+    module Jdbc
+      # /Users/seamusabshere/.rvm/gems/jruby-head/gems/activerecord-jdbc-adapter-1.2.2.1/src/java/arjdbc/jdbc/RubyJdbcConnection.java
+      GETTER = {
+        java.sql.Types::CHAR        => 'getString',
+        java.sql.Types::VARCHAR     => 'getString',
+        java.sql.Types::OTHER       => 'getString', # ?! i guess unicode text?
+        java.sql.Types::BINARY      => 'getBlob',
+        java.sql.Types::LONGVARCHAR => 'getString',
+        java.sql.Types::BIGINT      => 'getLong',
+        java.sql.Types::INTEGER     => 'getInt',
+        java.sql.Types::REAL        => "getLong",
+        java.sql.Types::ARRAY       => ->(r, i){ r.getArray(i).array.to_ary }
+      }
+      java.sql.Types.constants.each do |type_name|
+        i = java.sql.Types.const_get type_name
+        unless GETTER.has_key?(i)
+          GETTER[i] = 'get' + type_name[0].upcase + type_name[1..-1].downcase
+        end
+      end
+      SETTER = Hash.new do |hash, k|
+        hash[k] = 'set' + k
+      end.merge(
+        'TrueClass'  => 'setBoolean',
+        'FalseClass' => 'setBoolean',
+        'Fixnum'     => 'setInt',
+        'Integer'    => 'setInt'
+      )
+
+      def binary(v)
+        v.value.to_java_bytes.java_object
+      end
+
+      def execute(sql, params = nil)
+        has_result = if params
+          Upsert.logger.debug { %{[upsert] #{sql} with #{params.inspect}} }
+          setters = self.class.const_get(:SETTER)
+          statement = metal.prepareStatement sql
+          params.each_with_index do |v, i|
+            if v.is_a?(Fixnum) && v > 2_147_483_647
+              statement.setLong i+1, v
+              next
+            end
+
+            case v
+            when Upsert::Binary
+              statement.setBytes i+1, binary(v)
+            when Float, BigDecimal
+              statement.setBigDecimal i+1, java.math.BigDecimal.new(v.to_s)
+            when NilClass
+              # http://stackoverflow.com/questions/4243513/why-does-preparedstatement-setnull-requires-sqltype
+              statement.setObject i+1, nil
+            when java.time.LocalDateTime, java.time.Instant, java.time.LocalDate
+              statement.setObject i+1, v
+            else
+              setter = setters[v.class.name]
+              Upsert.logger.debug { "Setting [#{v.class}, #{v}] via #{setter}" }
+              statement.send setter, i+1, v
+            end
+          end
+          statement.execute
+        else
+          Upsert.logger.debug { %{[upsert] #{sql}} }
+          statement = metal.createStatement
+          statement.execute sql
+        end
+        if not has_result
+          statement.close
+          return
+        end
+        getters = self.class.const_get(:GETTER)
+        raw_result = statement.getResultSet
+        meta = raw_result.getMetaData
+        count = meta.getColumnCount
+        column_name_and_getter = (1..count).inject({}) do |memo, i|
+          memo[i] = [ meta.getColumnName(i), getters[meta.getColumnType(i)] ]
+          memo
+        end
+        result = []
+        while raw_result.next
+          row = {}
+          column_name_and_getter.each do |i, cg|
+            column_name, getter = cg
+            if getter == 'getNull'
+              row[column_name] = nil
+            elsif getter.respond_to?(:call)
+              row[column_name] = getter.call(raw_result, i)
+            else
+              row[column_name] = raw_result.send(getter, i)
+            end
+          end
+          result << row
+        end
+        statement.close
+        result
+      end
+
+      def in_transaction?
+        raise "Not implemented"
+      end
+    end
+  end
+end