do_mysql 0.2.0

data/ext/mysql_c.i

@@ -0,0 +1,67 @@
+%module mysql_c
+%{
+ #include <mysql.h>
+ #include <errmsg.h>
+ #include <mysqld_error.h>
+
+ VALUE mysql_c_fetch_field_names(MYSQL_RES *reader, int count) {
+   VALUE arr = rb_ary_new();
+   int i;
+   for(i = 0; i < count; i++) {
+     rb_ary_push(arr, rb_str_new2(mysql_fetch_field_direct(reader, i)->name));
+   }
+   return arr;
+ }
+
+ VALUE mysql_c_fetch_field_types(MYSQL_RES *reader, int count) {
+   VALUE arr = rb_ary_new();
+   int i;
+   for(i = 0; i < count; i++) {
+     rb_ary_push(arr, INT2NUM(mysql_fetch_field_direct(reader, i)->type));
+   }
+   return arr;
+ }
+
+ VALUE mysql_c_fetch_row(MYSQL_RES *reader) {
+   VALUE arr = rb_ary_new();
+   MYSQL_ROW result = (MYSQL_ROW)mysql_fetch_row(reader);
+   if(!result) return Qnil;
+   int i;
+   
+   for(i = 0; i < reader->field_count; i++) {
+     if(result[i] == NULL) rb_ary_push(arr, Qnil);
+     else rb_ary_push(arr, rb_str_new2(result[i]));
+   }
+   return arr;
+ }
+
+%}
+
+%ignore st_mysql_options;
+%include "/usr/local/mysql-5.0.45-osx10.4-i686/include/mysql.h"
+
+VALUE mysql_c_fetch_field_names(MYSQL_RES *reader, int count);
+VALUE mysql_c_fetch_field_types(MYSQL_RES *reader, int count);
+VALUE mysql_c_fetch_row(MYSQL_RES *reader);
+
+enum enum_field_types { MYSQL_TYPE_DECIMAL, MYSQL_TYPE_TINY,
+			MYSQL_TYPE_SHORT,  MYSQL_TYPE_LONG,
+			MYSQL_TYPE_FLOAT,  MYSQL_TYPE_DOUBLE,
+			MYSQL_TYPE_NULL,   MYSQL_TYPE_TIMESTAMP,
+			MYSQL_TYPE_LONGLONG,MYSQL_TYPE_INT24,
+			MYSQL_TYPE_DATE,   MYSQL_TYPE_TIME,
+			MYSQL_TYPE_DATETIME, MYSQL_TYPE_YEAR,
+			MYSQL_TYPE_NEWDATE, MYSQL_TYPE_VARCHAR,
+			MYSQL_TYPE_BIT,
+                        MYSQL_TYPE_NEWDECIMAL=246,
+			MYSQL_TYPE_ENUM=247,
+			MYSQL_TYPE_SET=248,
+			MYSQL_TYPE_TINY_BLOB=249,
+			MYSQL_TYPE_MEDIUM_BLOB=250,
+			MYSQL_TYPE_LONG_BLOB=251,
+			MYSQL_TYPE_BLOB=252,
+			MYSQL_TYPE_VAR_STRING=253,
+			MYSQL_TYPE_STRING=254,
+			MYSQL_TYPE_GEOMETRY=255
+
+};

data/lib/do_mysql.rb

@@ -0,0 +1,216 @@
+require 'mysql_c'
+require 'data_objects'
+
+module DataObject
+  module Mysql
+    TYPES = Hash[*Mysql_c.constants.select {|x| x.include?("MYSQL_TYPE")}.map {|x| [Mysql_c.const_get(x), x.gsub(/^MYSQL_TYPE_/, "")]}.flatten]    
+    
+    QUOTE_STRING = "\""
+    QUOTE_COLUMN = "`"
+    
+    class Connection < DataObject::Connection
+      
+      attr_reader :db
+      
+      def initialize(connection_string)        
+        @state = STATE_CLOSED
+        @connection_string = connection_string
+        opts = connection_string.split(" ")
+        opts.each do |opt|
+          k, v = opt.split("=")
+          raise ArgumentError, "you specified an invalid connection component: #{opt}" unless k && v
+          instance_variable_set("@#{k}", v)
+        end
+      end
+      
+      def change_database(database_name)
+        @dbname = database_name
+        @connection_string.gsub(/db_name=[^ ]*/, "db_name=#{database_name}")
+      end
+      
+      def open
+        @db = Mysql_c.mysql_init(nil)
+        raise ConnectionFailed, "could not allocate a MySQL connection" unless @db
+        conn = Mysql_c.mysql_real_connect(@db, @host, @user, @password, @dbname, @port || 0, @socket, @flags || 0)
+        raise ConnectionFailed, "The connection with connection string #{@connection_string} failed\n#{Mysql_c.mysql_error(@db)}" unless conn
+        @state = STATE_OPEN
+        true
+      end
+      
+      def close
+        if @state == STATE_OPEN
+          Mysql_c.mysql_close(@db)
+          @state = STATE_CLOSED        
+          true
+        else
+          false
+        end
+      end
+      
+      def create_command(text)
+        Command.new(self, text)
+      end
+      
+    end
+    
+    class Field
+      attr_reader :name, :type
+      
+      def initialize(ptr)
+        @name, @type = ptr.name.to_s, ptr.type.to_s
+      end
+    end
+    
+    class Reader < DataObject::Reader
+      
+      def initialize(db, reader)        
+        @reader = reader
+        unless @reader
+          if Mysql_c.mysql_field_count(db) == 0
+            @records_affected = Mysql_c.mysql_affected_rows(db)
+            close
+          else
+            raise UnknownError, "An unknown error has occured while trying to process a MySQL query.\n#{Mysql_c.mysql_error(db)}"
+          end
+        else
+          @field_count = @reader.field_count
+          @state = STATE_OPEN
+          
+          @native_fields, @fields = Mysql_c.mysql_c_fetch_field_types(@reader, @field_count), Mysql_c.mysql_c_fetch_field_names(@reader, @field_count)
+
+          raise UnknownError, "An unknown error has occured while trying to process a MySQL query. There were no fields in the resultset\n#{Mysql_c.mysql_error(db)}" if @native_fields.empty?
+          
+          @has_rows = !(@row = Mysql_c.mysql_c_fetch_row(@reader)).nil?
+        end
+      end
+      
+      def close
+        if @state == STATE_OPEN
+          Mysql_c.mysql_free_result(@reader)
+          @state = STATE_CLOSED
+          true
+        else
+          false
+        end
+      end
+      
+      def name(col)
+        super
+        @fields[col]
+      end
+      
+      def get_index(name)
+        super
+        @fields.index(name)
+      end
+      
+      def null?(idx)
+        super
+        @row[idx] == nil
+      end
+      
+      def current_row
+        @row
+      end
+      
+      def item(idx)
+        super
+        typecast(@row[idx], idx)
+      end
+      
+      def next
+        super
+        @row = Mysql_c.mysql_c_fetch_row(@reader)
+        close if @row.nil?
+        @row ? true : nil
+      end      
+      
+      def each
+        return unless has_rows?
+        
+        while(true) do
+          yield
+          break unless self.next
+        end
+      end
+      
+      protected
+      def native_type(col)
+        super
+        TYPES[@native_fields[col].type]
+      end
+      
+      def typecast(val, idx)
+        return nil if val.nil?
+        field = @native_fields[idx]
+        case TYPES[field]
+          when "TINY"
+            val != "0"
+          when "BIT"
+            val.to_i(2)
+          when "SHORT", "LONG", "INT24", "LONGLONG"
+            val.to_i
+          when "DECIMAL", "NEWDECIMAL", "FLOAT", "DOUBLE", "YEAR"
+            val.to_f
+          when "TIMESTAMP", "DATETIME"
+            DateTime.parse(val) rescue nil
+          when "TIME"
+            DateTime.parse(val).to_time rescue nil
+          when "DATE"
+            Date.parse(val) rescue nil
+          when "NULL"
+            nil
+          else
+            val
+        end
+      end      
+    end
+    
+    class Command < DataObject::Command
+      
+      def execute_reader(*args)
+        super
+        sql = escape_sql(args)
+        @connection.logger.debug { sql }
+        result = Mysql_c.mysql_query(@connection.db, sql)
+        # TODO: Real Error
+        raise QueryError, "Your query failed.\n#{Mysql_c.mysql_error(@connection.db)}\n#{@text}" unless result == 0
+        reader = Reader.new(@connection.db, Mysql_c.mysql_use_result(@connection.db))
+        if block_given?
+          result = yield(reader)
+          reader.close
+          result
+        else
+          reader
+        end
+      end
+      
+      def execute_non_query(*args)
+        super
+        sql = escape_sql(args)
+        @connection.logger.debug { sql }
+        result = Mysql_c.mysql_query(@connection.db, sql)
+        raise QueryError, "Your query failed.\n#{Mysql_c.mysql_error(@connection.db)}\n#{@text}" unless result == 0         
+        reader = Mysql_c.mysql_store_result(@connection.db)
+        raise QueryError, "You called execute_non_query on a query: #{@text}" if reader
+        rows_affected = Mysql_c.mysql_affected_rows(@connection.db)
+        Mysql_c.mysql_free_result(reader)
+        return ResultData.new(@connection, rows_affected, Mysql_c.mysql_insert_id(@connection.db))
+      end
+      
+      def quote_time(value)
+        # TIMESTAMP() used for both time and datetime columns
+        quote_datetime(value)
+      end
+      
+      def quote_datetime(value)
+        "TIMESTAMP('#{value.strftime("%Y-%m-%d %H:%M:%S")}')"
+      end
+      
+      def quote_date(value)
+        "DATE('#{value.strftime("%Y-%m-%d")}')"
+      end
+    end
+    
+  end
+end

metadata

@@ -0,0 +1,63 @@
+--- !ruby/object:Gem::Specification 
+rubygems_version: 0.9.4
+specification_version: 1
+name: do_mysql
+version: !ruby/object:Gem::Version 
+  version: 0.2.0
+date: 2007-11-11 00:00:00 -08:00
+summary: A DataObject.rb driver for mysql
+require_paths: 
+- lib
+email: wycats@gmail.com
+homepage: http://dataobjects.devjavu.com
+rubyforge_project: 
+description: A DataObject.rb driver for mysql
+autorequire: do_mysql
+default_executable: 
+bindir: bin
+has_rdoc: true
+required_ruby_version: !ruby/object:Gem::Version::Requirement 
+  requirements: 
+  - - ">"
+    - !ruby/object:Gem::Version 
+      version: 0.0.0
+  version: 
+platform: ruby
+signing_key: 
+cert_chain: 
+post_install_message: 
+authors: 
+- Yehuda Katz
+files: 
+- LICENSE
+- README
+- Rakefile
+- TODO
+- lib/do_mysql.rb
+- ext/extconf.rb
+- ext/mysql_c.c
+- ext/mysql_c.i
+test_files: []
+
+rdoc_options: []
+
+extra_rdoc_files: 
+- README
+- LICENSE
+- TODO
+executables: []
+
+extensions: []
+
+requirements: []
+
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: data_objects
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Version::Requirement 
+    requirements: 
+    - - ">"
+      - !ruby/object:Gem::Version 
+        version: 0.0.0
+    version: