imparcial 0.0.4 → 0.0.5

data/lib/imparcial/driver/mysql/sql/index.rb

@@ -0,0 +1,42 @@
+module Imparcial
+  module Driver
+    module MysqlSQL 
+      module Index
+        
+        private
+        
+        def sql_for_dropping_index ( options )
+          
+          %{DROP INDEX #{options[:index_name]} ON 
+            #{quote(options[:table_name])}
+           }
+          
+        end
+        
+        # Generate SQL statement for verifying if a given index exists.
+
+        def sql_for_index_exists? ( options )
+          
+          %{SELECT 1 FROM INFORMATION_SCHEMA.statistics
+            WHERE table_schema != "mysql" AND
+            table_name = #{quote_value(options[:table_name])} AND
+            INDEX_NAME = #{quote_value(options[:index_name])}
+           }
+          
+        end            
+        
+        def sql_for_getting_indexes ( options )
+          
+          %{SELECT table_name AS 'table', column_name AS 'column',\
+           index_name AS 'index' FROM INFORMATION_SCHEMA.statistics\
+           WHERE table_schema != 'mysql' 
+           AND table_name = #{quote_value(options[:table_name])}
+           }
+          
+        end
+        
+        
+      end
+    end
+  end
+end

data/lib/imparcial/driver/mysql/sql/sequence.rb

@@ -0,0 +1,39 @@
+module Imparcial
+  module Driver
+    module MysqlSQL
+      
+      # Mysql doesn't support sequences.
+      # We have to disable all them.
+    
+      module Sequence
+      
+        private
+        
+        def sql_for_creating_sequence ( options )
+      
+          raise FeatureNotFound
+        
+        end
+      
+        def sql_for_dropping_sequence ( options )
+
+          raise FeatureNotFound
+      
+        end
+            
+        def sql_for_sequence_exists? ( options )
+      
+          raise FeatureNotFound
+      
+        end
+            
+        def sql_for_retrieving_sequences
+      
+          raise FeatureNotFound
+      
+        end                        
+        
+      end
+    end
+  end
+end

data/lib/imparcial/driver/mysql/sql/table.rb

@@ -0,0 +1,67 @@
+module Imparcial
+  module Driver
+    module MysqlSQL
+      module Table
+        
+        private
+        
+        def sql_for_creating_table ( options = {} )
+          
+          syntax  = super
+          
+          syntax += 'ENGINE = '
+          
+          # If any engine has not been specified, we make INNODB as default.
+          
+          if options[:engine]
+            
+            syntax += options[:engine].to_s
+                        
+          else
+            
+            syntax += 'INNODB'
+            
+          end
+          
+          syntax
+          
+        end           
+        
+        private
+                 
+        # Generate SQL statement for retrieving tables.
+        
+        def sql_for_getting_tables
+      
+          'SHOW TABLES'
+                
+        end  
+                
+        # Generate SQL statement for retrieving columns.
+=begin        
+        def sql_for_retrieving_columns ( options = {} )
+        
+          %{
+            SELECT 
+            A.column_name, A.data_type, A.character_maximum_length as size,
+            (A.is_nullable = 'YES') as allow_null,
+            (A.column_name = B.column_name) as pk, 
+            A.column_default, 
+            A.extra  = 'auto_increment' as auto_inc,
+            A.column_name in (
+              SELECT s.COLUMN_NAME FROM INFORMATION_SCHEMA.statistics s 
+                WHERE s.COLUMN_NAME = A.column_name
+                             ) as indexed     
+                             
+            FROM information_schema.columns A 
+              LEFT JOIN information_schema.key_column_usage B ON  A.table_name = B.table_name 
+            WHERE A.table_name = #{quote_value(options[:table_name])}
+
+          }
+        
+        end          
+=end        
+      end
+    end
+  end
+end

data/lib/imparcial/driver/mysql/sql.rb

@@ -0,0 +1,15 @@
+FileList[File.dirname(__FILE__)  + '/sql/*.rb'].each do |f|
+  require f
+end
+
+module Imparcial
+  module Driver
+    module MysqlSQL
+      include Table
+      include Sequence
+      include Index
+      include Constraint
+      include Column
+    end
+  end
+end

data/lib/imparcial/driver/mysql/typemap.rb

@@ -0,0 +1,13 @@
+module Imparcial
+  module Driver    
+    module MysqlTypemap
+    
+      def sql_types
+      
+        super.dup.merge!('INT' => :integer) 
+        
+      end            
+    
+    end
+  end
+end

data/lib/imparcial/driver/mysql/util.rb

@@ -0,0 +1,13 @@
+module Imparcial
+  module Driver
+    module MysqlUtil
+    
+      def quote ( val )
+      
+        "`#{val}`"   
+      
+      end      
+        
+    end
+  end
+end

data/lib/imparcial/driver/mysql.rb

@@ -0,0 +1,49 @@
+require 'mysql'
+
+require 'imparcial/driver/mysql/util'
+require 'imparcial/driver/mysql/result'
+require 'imparcial/driver/mysql/typemap'
+require 'imparcial/driver/mysql/sql'
+require 'imparcial/driver/mysql/expression'
+
+module Imparcial
+  module Driver    
+    class MysqlAdapter < AbstractAdapter
+        
+      include MysqlUtil
+      include MysqlTypemap
+      include MysqlSQL
+      include MysqlExpression
+      
+      def adapter_specific_exception
+      
+        Mysql::Error
+        
+      end
+      
+      def connect
+        
+        @conn = Mysql.real_connect @host, @username, @password, @database, @port, @socket
+      
+      rescue adapter_specific_exception => ex
+           
+        raise AdapterConnectionError.new(ex.message)
+        
+      end
+          
+      def query ( sql )
+        
+        result = conn.query sql
+        @result = MysqlResult.new result
+        
+      end      
+          
+      def last_insert_id
+        
+        conn.last_insert_id
+        
+      end
+          
+    end    
+  end
+end

data/lib/imparcial/driver/postgre/expression.rb

@@ -0,0 +1,32 @@
+FileList[File.dirname(__FILE__)  + '/expression/*.rb'].each do |f|
+  require f
+end
+
+module Imparcial
+  module Driver
+    module PostgreExpression
+      
+      def insert ( options = {} )
+        
+        super options
+
+        begin
+          
+          seq_name = 'seq_' + options[:table_name].to_s + '_id'  
+          id = conn.query "SELECT currval('#{seq_name}')"
+        
+          @last_insert_id = id[0][0].to_i
+          
+        rescue adapter_specific_exception; end
+        
+      end
+      
+      def last_insert_id
+        
+        @last_insert_id
+        
+      end
+      
+    end
+  end
+end

data/lib/imparcial/driver/postgre/result.rb

@@ -0,0 +1,35 @@
+module Imparcial
+  module Driver
+    class PostgreResult < AbstractResult
+    
+      def rows
+      
+       @specific.result.length
+      
+      end
+      
+      def fetch
+
+        fields = @specific.fields
+        rows   = @specific.result
+        
+        rows.each do |row|
+        
+          v = []
+        
+          row.each_with_index do |value, index|
+            
+            v << Row.new(fields[index], value)
+            
+          end
+        
+          yield(*v)
+        
+        end        
+        
+
+      end
+              
+    end
+  end
+end

data/lib/imparcial/driver/postgre/sql/column.rb

@@ -0,0 +1,53 @@
+module Imparcial
+  module Driver
+    module PostgreSQL
+      module Column
+
+        private
+
+        def sql_for_renaming_column ( options )
+
+          syntax = ''
+        
+          syntax += 'ALTER TABLE ' + quote(options[:table_name]) + ' '
+ 
+          options[:field].each do |old_name, new_name|
+                        
+            syntax += "RENAME COLUMN #{old_name} TO #{new_name}"
+            
+          end
+                      
+          logger.warn(syntax) if @table_logging
+
+          syntax
+        
+        end
+  
+        def sql_for_modifying_columns ( options )
+
+          syntax = ''
+        
+          syntax += 'ALTER TABLE ' + quote(options[:table_name]) + ' '
+ 
+          options[:fields].each do |field|
+            
+            column = field_to_column field
+                        
+            syntax += "ALTER COLUMN #{column[:name]} TYPE #{column[:type]}"
+            syntax += "(#{column[:size]})" if column[:size]
+            syntax += ","
+            
+          end
+          
+          syntax.chop!
+                      
+          logger.warn(syntax) if @table_logging
+
+          syntax
+        
+        end 
+      
+      end
+    end
+  end
+end

data/lib/imparcial/driver/postgre/sql/constraint.rb

@@ -0,0 +1,37 @@
+module Imparcial
+  module Driver
+    module PostgreSQL
+      module Constraint
+
+        private
+
+        def sql_for_adding_auto_increment ( options )
+                   
+          seq_name = 'seq_' + options[:table_name].to_s + '_' + options[:field].to_s 
+          drop_sequence_if_necessary :sequence_name => seq_name
+          create_sequence :sequence_name => seq_name          
+          sql_for_adding_default_value :table_name => options[:table_name], :fields => {options[:field] => "nextval('#{seq_name}')"}
+          
+        end        
+        
+        def sql_for_adding_default_value ( options )
+                   
+          syntax  = 'ALTER TABLE ' + quote(options[:table_name]) + ' '   
+          
+          for field, value in options[:fields]
+          
+            value = quote_value(value) if not value.to_s.match('nextval')
+            syntax += "ALTER COLUMN #{quote(field)} SET DEFAULT #{value},"
+            
+          end
+          
+          syntax.chop!
+                             
+          syntax                  
+          
+        end        
+        
+      end
+    end
+  end
+end

data/lib/imparcial/driver/postgre/sql/index.rb

@@ -0,0 +1,53 @@
+module Imparcial
+  module Driver
+    module PostgreSQL   
+      module Index
+        
+        private
+
+        # Generate SQL statement for verifying if a given index exists.
+
+        def sql_for_index_exists? ( options )
+          
+          %{SELECT 
+              c2.relname as "table",
+              (select attname from pg_attribute where attrelid = i.indexrelid AND attnum > 0) as "column" ,
+              c.relname as "index"             
+            FROM pg_class c
+            JOIN pg_roles r ON r.oid = c.relowner
+            LEFT JOIN pg_namespace n ON n.oid = c.relnamespace
+            LEFT JOIN pg_index i ON i.indexrelid = c.oid
+            LEFT JOIN pg_class c2 ON i.indrelid = c2.oid
+            WHERE c.relkind = 'i'
+            AND n.nspname NOT IN ('pg_catalog', 'pg_toast') AND
+            c2.relname = #{quote_value(options[:table_name])} AND
+            c.relname = #{quote_value(options[:index_name])}
+           }
+          
+        end        
+
+        # Generate SQL statement for retrieving all indexes.
+        
+        def sql_for_getting_indexes ( options )
+          
+          %{SELECT 
+              c2.relname as "table",
+              (select attname from pg_attribute where attrelid = i.indexrelid AND attnum > 0) as "column" ,
+              c.relname as "index"             
+            FROM pg_class c
+            JOIN pg_roles r ON r.oid = c.relowner
+            LEFT JOIN pg_namespace n ON n.oid = c.relnamespace
+            LEFT JOIN pg_index i ON i.indexrelid = c.oid
+            LEFT JOIN pg_class c2 ON i.indrelid = c2.oid
+            WHERE c.relkind = 'i'
+            AND n.nspname NOT IN ('pg_catalog', 'pg_toast') AND
+            c2.relname = #{quote_value(options[:table_name])}
+           }
+          
+        end
+        
+    
+      end
+    end
+  end
+end

data/lib/imparcial/driver/postgre/sql/sequence.rb

@@ -0,0 +1,30 @@
+module Imparcial
+  module Driver
+    module PostgreSQL
+      module Sequence
+      
+        private
+        
+        def sql_for_sequence_exists? ( options )
+      
+          %{
+            SELECT 1 FROM pg_class WHERE relkind = 'S' AND
+            relname = #{quote_value(options[:sequence_name])}
+           } 
+                  
+        end     
+                
+        # Generate SQL statement for retrieving sequences.
+      
+        def sql_for_getting_sequences
+      
+          %{
+            SELECT relname FROM pg_class WHERE relkind = 'S'
+           }
+      
+        end                       
+        
+      end
+    end
+  end
+end

data/lib/imparcial/driver/postgre/sql/table.rb

@@ -0,0 +1,46 @@
+module Imparcial
+  module Driver
+    module PostgreSQL
+      module Table
+        
+        private
+                 
+        # Generate SQL statement for retrieving tables.
+        
+        def sql_for_getting_tables
+        
+          %{
+            SELECT table_name FROM information_schema.tables 
+            WHERE table_schema = 'public'
+           } 
+          
+        end
+                
+        # Generate SQL statement for retrieving columns.
+=begin        
+        def sql_for_retrieving_columns ( options )
+        
+          %{
+            SELECT 
+            A.column_name, A.data_type, A.character_maximum_length,
+            (A.is_nullable = 'YES') as allow_null,
+            (A.column_name = B.column_name) as pk, 
+            A.column_default, 
+            A.column_default LIKE 'nextval%' as auto_inc,
+            A.column_name in (SELECT sa.attname from pg_class sc
+                                JOIN pg_index si ON sc.oid = si.indexrelid
+                                JOIN pg_class sc2 ON sc2.oid = si.indrelid
+                                JOIN pg_attribute sa ON sc.oid = sa.attrelid
+                              WHERE  sc2.relname = #{quote_value(options[:table_name])} AND sa.attname = A.column_name 
+                             ) as column_indexed
+            FROM information_schema.columns A 
+              LEFT JOIN information_schema.key_column_usage B ON  A.table_name = B.table_name 
+            WHERE A.table_name = #{quote_value(options[:table_name])};
+           }
+        
+        end          
+=end        
+      end
+    end
+  end
+end

data/lib/imparcial/driver/postgre/sql.rb

@@ -0,0 +1,15 @@
+FileList[File.dirname(__FILE__)  + '/sql/*.rb'].each do |f|
+  require f
+end
+
+module Imparcial
+  module Driver
+    module PostgreSQL
+      include Table
+      include Sequence
+      include Index
+      include Constraint
+      include Column
+    end
+  end
+end

data/lib/imparcial/driver/postgre/typemap.rb

@@ -0,0 +1,36 @@
+module Imparcial
+  module Driver
+    module PostgreTypemap
+
+      def regular_types
+        
+        super.dup.merge!(:datetime => 'TIMESTAMP')
+        
+      end
+
+      def sql_types
+      
+        super.dup.merge!('CHARACTER VARYING' => :string, 
+        'DOUBLE PRECISION' => :float) 
+        
+      end
+      
+      private
+      
+      # Due to nextval, we have to format the default value
+      
+      def parse_column ( column )
+
+        super
+        
+        if column[:default_value]
+        
+          column[:default_value].gsub!(/::.*/,'')
+          
+        end        
+        
+      end
+                
+    end
+  end
+end

data/lib/imparcial/driver/postgre/util.rb

@@ -0,0 +1,19 @@
+module Imparcial
+  module Driver
+    module PostgreUtil
+    
+      def quote_value ( val )
+      
+        conn.class.quote val
+      
+      end
+        
+      def quote ( val )
+      
+        "\"#{val}\""
+      
+      end      
+    
+    end
+  end
+end

data/lib/imparcial/driver/postgre.rb

@@ -0,0 +1,43 @@
+require 'postgres'
+
+require 'imparcial/driver/postgre/util'
+require 'imparcial/driver/postgre/result'
+require 'imparcial/driver/postgre/typemap'
+require 'imparcial/driver/postgre/sql'
+require 'imparcial/driver/postgre/expression'
+
+module Imparcial
+  module Driver    
+    class PostgreAdapter < AbstractAdapter
+    
+      include PostgreUtil
+      include PostgreTypemap
+      include PostgreSQL
+      include PostgreExpression
+      
+      def adapter_specific_exception
+      
+        PGError
+        
+      end
+    
+      def connect
+
+        @conn = PGconn.connect @host, @port, '', '', @database, @username, @password
+        
+      rescue adapter_specific_exception => ex
+           
+        raise AdapterConnectionError.new(ex.message)
+        
+      end
+          
+      def query ( sql )
+        
+        result  = conn.exec sql
+        @result = PostgreResult.new result
+        
+      end
+        
+    end    
+  end
+end

data/lib/imparcial/driver.rb

@@ -0,0 +1 @@
+require 'imparcial/driver/abstract'