activerecord-jdbc-alt-adapter 50.3.0-java

data/src/java/arjdbc/db2/DB2Module.java

@@ -0,0 +1,76 @@
+/*
+ * The MIT License
+ *
+ * Copyright 2013 Karol Bucek.
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to deal
+ * in the Software without restriction, including without limitation the rights
+ * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ * copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in
+ * all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ * THE SOFTWARE.
+ */
+package arjdbc.db2;
+
+import static arjdbc.util.QuotingUtils.BYTES_0;
+import static arjdbc.util.QuotingUtils.BYTES_1;
+import static arjdbc.util.QuotingUtils.quoteSingleQuotesWithFallback;
+
+import org.jruby.Ruby;
+import org.jruby.RubyModule;
+import org.jruby.RubyString;
+import org.jruby.anno.JRubyMethod;
+import org.jruby.runtime.ThreadContext;
+import org.jruby.runtime.builtin.IRubyObject;
+
+/**
+ * ArJdbc::DB2
+ *
+ * @author kares
+ */
+public class DB2Module {
+
+    public static RubyModule load(final RubyModule arJdbc) {
+        RubyModule db2 = arJdbc.defineModuleUnder("DB2");
+        db2.defineAnnotatedMethods( DB2Module.class );
+        return db2;
+    }
+
+    public static RubyModule load(final Ruby runtime) {
+        return load( arjdbc.ArJdbcModule.get(runtime) );
+    }
+
+    @JRubyMethod(name = "quote_string", required = 1, frame = false)
+    public static IRubyObject quote_string(
+            final ThreadContext context,
+            final IRubyObject self,
+            final IRubyObject string) {
+        return quoteSingleQuotesWithFallback(context, string);
+    }
+
+    @JRubyMethod(name = "quoted_true", required = 0, frame = false)
+    public static IRubyObject quoted_true(
+            final ThreadContext context,
+            final IRubyObject self) {
+        return RubyString.newString(context.runtime, BYTES_1);
+    }
+
+    @JRubyMethod(name = "quoted_false", required = 0, frame = false)
+    public static IRubyObject quoted_false(
+            final ThreadContext context,
+            final IRubyObject self) {
+        return RubyString.newString(context.runtime, BYTES_0);
+    }
+
+}

data/src/java/arjdbc/db2/DB2RubyJdbcConnection.java

@@ -0,0 +1,126 @@
+/***** BEGIN LICENSE BLOCK *****
+ * Copyright (c) 2012-2013 Karol Bucek <self@kares.org>
+ * Copyright (c) 2006-2010 Nick Sieger <nick@nicksieger.com>
+ * Copyright (c) 2006-2007 Ola Bini <ola.bini@gmail.com>
+ * Copyright (c) 2008-2009 Thomas E Enebo <enebo@acm.org>
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining
+ * a copy of this software and associated documentation files (the
+ * "Software"), to deal in the Software without restriction, including
+ * without limitation the rights to use, copy, modify, merge, publish,
+ * distribute, sublicense, and/or sell copies of the Software, and to
+ * permit persons to whom the Software is furnished to do so, subject to
+ * the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be
+ * included in all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+ * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+ * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+ * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+ * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+ * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+ * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+ ***** END LICENSE BLOCK *****/
+package arjdbc.db2;
+
+import arjdbc.jdbc.Callable;
+import arjdbc.jdbc.RubyJdbcConnection;
+import arjdbc.util.StringHelper;
+
+import java.sql.Connection;
+import java.sql.PreparedStatement;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+
+import org.jruby.Ruby;
+import org.jruby.RubyBoolean;
+import org.jruby.RubyClass;
+import org.jruby.RubyString;
+import org.jruby.anno.JRubyMethod;
+import org.jruby.runtime.ObjectAllocator;
+import org.jruby.runtime.ThreadContext;
+import org.jruby.runtime.builtin.IRubyObject;
+import org.jruby.util.ByteList;
+
+/**
+ *
+ * @author mikestone
+ */
+public class DB2RubyJdbcConnection extends RubyJdbcConnection {
+
+    public DB2RubyJdbcConnection(Ruby runtime, RubyClass metaClass) {
+        super(runtime, metaClass);
+    }
+
+    public static RubyClass createDB2JdbcConnectionClass(Ruby runtime, RubyClass jdbcConnection) {
+        RubyClass clazz = getConnectionAdapters(runtime).
+            defineClassUnder("DB2JdbcConnection", jdbcConnection, ALLOCATOR);
+        clazz.defineAnnotatedMethods(DB2RubyJdbcConnection.class);
+
+        return clazz;
+    }
+
+    public static RubyClass load(final Ruby runtime) {
+        RubyClass jdbcConnection = getJdbcConnection(runtime);
+        return createDB2JdbcConnectionClass(runtime, jdbcConnection);
+    }
+
+    protected static ObjectAllocator ALLOCATOR = new ObjectAllocator() {
+        public IRubyObject allocate(Ruby runtime, RubyClass klass) {
+            return new DB2RubyJdbcConnection(runtime, klass);
+        }
+    };
+
+    @JRubyMethod(name = "select?", required = 1, meta = true, frame = false)
+    public static RubyBoolean select_p(final ThreadContext context,
+        final IRubyObject self, final IRubyObject sql) {
+        final RubyString sqlStr = sql.asString();
+        if ( isValues(sqlStr) ) return context.runtime.getTrue();
+        return arjdbc.jdbc.RubyJdbcConnection.select_p(context, self, sqlStr);
+    }
+
+    // DB2 supports 'stand-alone' VALUES expressions
+    private static final byte[] VALUES = new byte[]{ 'v','a','l','u', 'e', 's' };
+
+    private static boolean isValues(final RubyString sql) {
+        final ByteList sqlBytes = sql.getByteList();
+        return StringHelper.startsWithIgnoreCase(sqlBytes, VALUES);
+    }
+
+    private static final String[] TABLE_TYPES = new String[] {
+        "TABLE", "VIEW", "SYNONYM", "MATERIALIZED QUERY TABLE", "ALIAS"
+    };
+
+    @Override
+    protected String[] getTableTypes() {
+        return TABLE_TYPES;
+    }
+
+    @Override
+    protected boolean databaseSupportsSchemas() {
+        return true;
+    }
+
+    @JRubyMethod(name = {"identity_val_local", "last_insert_id"})
+    public IRubyObject identity_val_local(final ThreadContext context)
+        throws SQLException {
+        return withConnection(context, new Callable<IRubyObject>() {
+            public IRubyObject call(final Connection connection) throws SQLException {
+                PreparedStatement statement = null; ResultSet genKeys = null;
+                try {
+                    statement = connection.prepareStatement("VALUES IDENTITY_VAL_LOCAL()");
+                    genKeys = statement.executeQuery();
+                    return doMapGeneratedKeys(context.runtime, genKeys, true);
+                }
+                catch (final SQLException e) {
+                    debugMessage(context.runtime, "failed to get generated keys: ", e);
+                    throw e;
+                }
+                finally { close(genKeys); close(statement); }
+            }
+        });
+    }
+
+}

data/src/java/arjdbc/derby/DerbyModule.java

@@ -0,0 +1,178 @@
+/***** BEGIN LICENSE BLOCK *****
+ * Copyright (c) 2012-2013 Karol Bucek <self@kares.org>
+ * Copyright (c) 2006-2011 Nick Sieger <nick@nicksieger.com>
+ * Copyright (c) 2006-2007 Ola Bini <ola.bini@gmail.com>
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining
+ * a copy of this software and associated documentation files (the
+ * "Software"), to deal in the Software without restriction, including
+ * without limitation the rights to use, copy, modify, merge, publish,
+ * distribute, sublicense, and/or sell copies of the Software, and to
+ * permit persons to whom the Software is furnished to do so, subject to
+ * the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be
+ * included in all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+ * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+ * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+ * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+ * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+ * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+ * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+ ***** END LICENSE BLOCK *****/
+
+package arjdbc.derby;
+
+import static arjdbc.util.QuotingUtils.BYTES_0;
+import static arjdbc.util.QuotingUtils.BYTES_1;
+import static arjdbc.util.QuotingUtils.BYTES_SINGLE_Q_x2;
+
+import org.jruby.Ruby;
+import org.jruby.RubyModule;
+import org.jruby.RubyString;
+import org.jruby.anno.JRubyMethod;
+import org.jruby.runtime.ThreadContext;
+import org.jruby.runtime.builtin.IRubyObject;
+import org.jruby.util.ByteList;
+
+public class DerbyModule {
+
+    public static RubyModule load(final RubyModule arJdbc) {
+        RubyModule derby = arJdbc.defineModuleUnder("Derby");
+        derby.defineAnnotatedMethods( DerbyModule.class );
+        RubyModule column = derby.defineModuleUnder("Column");
+        column.defineAnnotatedMethods(Column.class);
+        return derby;
+    }
+
+    public static RubyModule load(final Ruby runtime) {
+        return load( arjdbc.ArJdbcModule.get(runtime) );
+    }
+
+    public static class Column {
+
+        @JRubyMethod(name = "type_cast", required = 1)
+        public static IRubyObject type_cast(final ThreadContext context,
+            final IRubyObject self, final IRubyObject value) {
+
+            if ( value.isNil() ||
+            ( (value instanceof RubyString) && value.toString().trim().equalsIgnoreCase("null") ) ) {
+                return context.nil;
+            }
+
+            final String type = self.getInstanceVariables().getInstanceVariable("@type").toString();
+
+            switch (type.charAt(0)) {
+            case 's': //string
+                return value;
+            case 't': //text, timestamp, time
+                if ( type.equals("time") ) {
+                    return self.getMetaClass().callMethod(context, "string_to_dummy_time", value);
+                }
+                if ( type.equals("timestamp") ) {
+                    return self.getMetaClass().callMethod(context, "string_to_time", value);
+                }
+                return value; // text
+            case 'i': //integer
+            case 'p': //primary key
+                if ( value.respondsTo("to_i") ) {
+                    return value.callMethod(context, "to_i");
+                }
+                return context.getRuntime().newFixnum( value.isTrue() ? 1 : 0 );
+            case 'd': //decimal, datetime, date
+                if ( type.equals("datetime") ) {
+                    return self.getMetaClass().callMethod(context, "string_to_time", value);
+                }
+                if ( type.equals("date") ) {
+                    return self.getMetaClass().callMethod(context, "string_to_date", value);
+                }
+                return self.getMetaClass().callMethod(context, "value_to_decimal", value);
+            case 'f': //float
+                return value.callMethod(context, "to_f");
+            case 'b': //binary, boolean
+                return type.equals("binary") ?
+                    self.getMetaClass().callMethod(context, "binary_to_string", value) :
+                    self.getMetaClass().callMethod(context, "value_to_boolean", value) ;
+            }
+            return value;
+        }
+
+    }
+
+    @JRubyMethod(name = "quote_binary", required = 1)
+    public static IRubyObject quote_binary(final ThreadContext context,
+        final IRubyObject self, final IRubyObject string) {
+        final byte[] empty = ByteList.NULL_ARRAY; // ""
+        return quoteStringHex(context.runtime, empty, string, empty);
+    }
+
+    private final static byte[] HEX = {'0','1','2','3','4','5','6','7','8','9','A','B','C','D','E','F'};
+
+    private static RubyString quoteStringHex(final Ruby runtime,
+        final byte[] before, final IRubyObject string, final byte[] after) {
+
+        final ByteList input = ((RubyString) string).getByteList();
+        final int size = input.getRealSize();
+        final ByteList output = new ByteList( before.length + size * 2 + after.length );
+        output.append( before );
+
+        final byte[] inputBytes = input.unsafeBytes();
+
+        int written = 0;
+        for (int i = input.getBegin(); i < input.getBegin() + size; i++) {
+            final byte b = inputBytes[i];
+            byte h = HEX[ ( ((char) b) >> 4 ) % 16 ];
+            byte l = HEX[ ( (char) b ) % 16 ];
+            output.append(h).append(l);
+            written += 2;
+            if ( written >= 16334 ) { // max hex length = 16334
+                output.append("'||X'".getBytes());
+                written = 0;
+            }
+        }
+
+        output.append( after );
+        return RubyString.newString(runtime, output);
+    }
+
+    @JRubyMethod(name = "quote_string", required = 1)
+    public static IRubyObject quote_string(final IRubyObject self, IRubyObject string) {
+        if ( ! ( string instanceof RubyString ) ) {
+            string = string.asString(); // e.g. Multibyte::Chars
+        }
+
+        ByteList bytes = ((RubyString) string).getByteList();
+
+        boolean newBytes = false;
+        for ( int i = 0; i < bytes.length(); i++ ) {
+            switch ( bytes.get(i) ) {
+                case '\'': break;
+                default: continue;
+            }
+            // on first replacement allocate so we don't manip original
+            if ( ! newBytes ) {
+                bytes = new ByteList(bytes);
+                newBytes = true;
+            }
+            bytes.replace(i, 1, BYTES_SINGLE_Q_x2);
+            i += 1;
+        }
+
+        return newBytes ? RubyString.newString(self.getRuntime(), bytes) : string;
+    }
+
+    @JRubyMethod(name = "quoted_true", required = 0, frame = false)
+    public static RubyString quoted_true(
+        final ThreadContext context, final IRubyObject self) {
+        return RubyString.newStringShared(context.runtime, BYTES_1);
+    }
+
+    @JRubyMethod(name = "quoted_false", required = 0, frame = false)
+    public static RubyString quoted_false(
+        final ThreadContext context, final IRubyObject self) {
+        return RubyString.newStringShared(context.runtime, BYTES_0);
+    }
+
+}

data/src/java/arjdbc/derby/DerbyRubyJdbcConnection.java

@@ -0,0 +1,152 @@
+/*
+ * The MIT License
+ *
+ * Copyright 2013 Karol Bucek.
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to deal
+ * in the Software without restriction, including without limitation the rights
+ * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ * copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in
+ * all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ * THE SOFTWARE.
+ */
+package arjdbc.derby;
+
+import arjdbc.jdbc.Callable;
+import arjdbc.jdbc.DriverWrapper;
+import arjdbc.jdbc.RubyJdbcConnection;
+
+import java.sql.Connection;
+import java.sql.PreparedStatement;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+
+import org.jruby.Ruby;
+import org.jruby.RubyBoolean;
+import org.jruby.RubyClass;
+import org.jruby.RubyString;
+import org.jruby.anno.JRubyMethod;
+import org.jruby.exceptions.RaiseException;
+import org.jruby.runtime.ObjectAllocator;
+import org.jruby.runtime.ThreadContext;
+import org.jruby.runtime.builtin.IRubyObject;
+import org.jruby.util.ByteList;
+
+/**
+ * ArJdbc::DerbyJdbcConnection
+ *
+ * @author kares
+ */
+public class DerbyRubyJdbcConnection extends RubyJdbcConnection {
+    private static final long serialVersionUID = 4809475910953623325L;
+
+    public DerbyRubyJdbcConnection(Ruby runtime, RubyClass metaClass) {
+        super(runtime, metaClass);
+    }
+
+    public static RubyClass createDerbyJdbcConnectionClass(Ruby runtime, RubyClass jdbcConnection) {
+        final RubyClass clazz = getConnectionAdapters(runtime). // ActiveRecord::ConnectionAdapters
+            defineClassUnder("DerbyJdbcConnection", jdbcConnection, ALLOCATOR);
+        clazz.defineAnnotatedMethods(DerbyRubyJdbcConnection.class);
+        return clazz;
+    }
+
+    public static RubyClass load(final Ruby runtime) {
+        RubyClass jdbcConnection = getJdbcConnection(runtime);
+        return createDerbyJdbcConnectionClass(runtime, jdbcConnection);
+    }
+
+    protected static ObjectAllocator ALLOCATOR = new ObjectAllocator() {
+        public IRubyObject allocate(Ruby runtime, RubyClass klass) {
+            return new DerbyRubyJdbcConnection(runtime, klass);
+        }
+    };
+
+    @Override
+    protected DriverWrapper newDriverWrapper(final ThreadContext context, final String driver) {
+        DriverWrapper driverWrapper = super.newDriverWrapper(context, driver);
+
+        final java.sql.Driver jdbcDriver = driverWrapper.getDriverInstance();
+        if ( jdbcDriver.getClass().getName().startsWith("org.apache.derby.") ) {
+            final int major = jdbcDriver.getMajorVersion();
+            final int minor = jdbcDriver.getMinorVersion();
+            if ( major < 10 || ( major == 10 && minor < 5 ) ) {
+                final RubyClass errorClass = getConnectionNotEstablished(context.runtime);
+                throw context.runtime.newRaiseException(errorClass,
+                    "adapter requires Derby >= 10.5 got: " + major + "." + minor + "");
+            }
+            if ( major == 10 && minor < 8 ) { // 10.8 ~ supports JDBC 4.1
+                // config[:connection_alive_sql] ||= 'SELECT 1 FROM SYS.SYSSCHEMAS FETCH FIRST 1 ROWS ONLY'
+                setConfigValueIfNotSet(context, "connection_alive_sql", // FROM clause mandatory
+                    context.runtime.newString("SELECT 1 FROM SYS.SYSSCHEMAS FETCH FIRST 1 ROWS ONLY"));
+            }
+        }
+
+        return driverWrapper;
+    }
+
+    @JRubyMethod(name = "select?", required = 1, meta = true, frame = false)
+    public static RubyBoolean select_p(final ThreadContext context,
+        final IRubyObject self, final IRubyObject sql) {
+        final RubyString sqlStr = sql.asString();
+        if ( isValues(sqlStr) ) return context.runtime.getTrue();
+        return arjdbc.jdbc.RubyJdbcConnection.select_p(context, self, sqlStr);
+    }
+
+    // Derby supports 'stand-alone' VALUES expressions
+    private static final byte[] VALUES = new byte[]{ 'v','a','l','u','e','s' };
+
+    private static boolean isValues(final RubyString sql) {
+        final ByteList sqlBytes = sql.getByteList();
+        return startsWithIgnoreCase(sqlBytes, VALUES);
+    }
+
+    @JRubyMethod(name = {"identity_val_local", "last_insert_id"})
+    public IRubyObject identity_val_local(final ThreadContext context)
+        throws SQLException {
+        return withConnection(context, new Callable<IRubyObject>() {
+            public IRubyObject call(final Connection connection) throws SQLException {
+                PreparedStatement statement = null; ResultSet genKeys = null;
+                try {
+                    statement = connection.prepareStatement("values IDENTITY_VAL_LOCAL()");
+                    genKeys = statement.executeQuery();
+                    return doMapGeneratedKeys(context.getRuntime(), genKeys, true);
+                }
+                catch (final SQLException e) {
+                    debugMessage(context.runtime, "failed to get generated keys: ", e);
+                    throw e;
+                }
+                finally { close(genKeys); close(statement); }
+            }
+        });
+    }
+
+    @Override
+    protected boolean supportsGeneratedKeys(final Connection connection) throws SQLException {
+        return true; // driver reports false from supportsGetGeneratedKeys()
+    }
+
+    @Override
+    protected IRubyObject matchTables(final ThreadContext context,
+            final Connection connection,
+            final String catalog, String schemaPattern,
+            final String tablePattern, final String[] types,
+            final boolean checkExistsOnly) throws SQLException {
+        if (schemaPattern != null && schemaPattern.equals("")) {
+            schemaPattern = null; // Derby doesn't like empty-string schema name
+        }
+        return super.matchTables(context, connection, catalog, schemaPattern, tablePattern, types, checkExistsOnly);
+    }
+
+}