teradata-ar 0.0.3 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: e2782e323965d4d88ac3786289701ae7167d31f2
-  data.tar.gz: b7ccdf30cfc3f6dad319747313f09763a40e3074
+  metadata.gz: 498cdd551375bd432cdf0d6f76c5dbdb3fe8ce1b
+  data.tar.gz: 9f092600d613e3d31c3fc9b935e08cde84488d9e
 SHA512:
-  metadata.gz: 7517fb456cb70b3f7c01e5e7cf2b3c9a85bc8c4eb2e256b26133674f591b636b12e47b34492eb4407e3cf77d4a5f260ed7aa0bdeed946cdf6bbeec557eaa63dd
-  data.tar.gz: 534eb4c75cf4ca1f25490683ef7a7cb69f27fd19a65ee0b3fa23eae79ad75a34df62c4d12072a94068b7e8a6296c69ebdb4f2e6b6a14357416f745a9521659ad
+  metadata.gz: 125abcaa7940d177e0c8c8a06d92d0dbdb171df1d67fdd9a4a3a46c7972c4cf439f52d017cb8c018a2891c3a527e0f52f71ee55a07d4979fd3a02b049c79639d
+  data.tar.gz: 43539430c99e2c2270e21a61c2699b10733ac01baaae1b060e3f39cb6a366102a98d5ce46b546e3a9cf84f885c60f7711670827706ade859cf1dbcff9daf2285

data/lib/active_record/connection_adapters/teradata_adapter.rb

@@ -1,8 +1,20 @@
 require 'active_record/connection_adapters/abstract_adapter'
+require 'arel/visitors/bind_visitor'
 
 #gem 'teradata-cli', :path => '~/projects/teradata-cli'
 require 'teradata-cli'
 
+class ActiveRecord::Base
+  def self.table_name_prefix
+    Rails.configuration.database_configuration[Rails.env]['table_name_prefix']
+  end
+end
+
+class ActiveRecord::Migration
+  def self.check_pending!
+  end
+end
+
 module ActiveRecord
   module ConnectionHandling
     def teradata_connection(config)
@@ -20,12 +32,30 @@ module ActiveRecord
   module ConnectionAdapters
     class TeradataAdapter < AbstractAdapter
 
+      class BindSubstitution < Arel::Visitors::ToSql
+        include Arel::Visitors::BindVisitor
+
+        def visit_Arel_Nodes_Limit o
+        end
+        def visit_Arel_Nodes_Offset o
+        end
+
+        def visit_Arel_Nodes_Top o
+          "TOP #{o.expr}"
+        end
+
+      end
+
       def initialize(logger, logon_string, config)
         @logon_string = logon_string
         @charset = config[:charset]
+        @database = config[:database]
+        @table_name_prefix = config[:table_name_prefix]
         @config = config
         connect
         super @connection, logger
+        @visitor = unprepared_visitor
+        configure_connection
       end
 
       def adapter_name
@@ -37,10 +67,160 @@ module ActiveRecord
       end
       private :connect
 
+      def configure_connection
+        execute("DATABASE #{@database}") if @database
+      end
+      private :configure_connection
+
+      # CONNECTION MANAGEMENT ====================================
+
+      def active?
+        @connection.execute_query('SELECT 1')
+        true
+      rescue
+        false
+      end
+
+      def reconnect!
+        super
+        disconnect!
+        connect
+        configure_connection
+      end
+      alias :reset! :reconnect!
+
+      # Disconnects from the database if already connected.
+      # Otherwise, this method does nothing.
+      def disconnect!
+        super
+        unless @connection.nil?
+          @connection.close
+          @connection = nil
+        end
+      end
+
+      # DATABASE STATEMENTS ======================================
+
       def tables(name = nil)
-        # FIXME: fetch metadata from database
-        ['shohins']
+        sql = "SELECT TABLENAME FROM DBC.TABLES "
+        clauses = []
+        clauses << "DATABASENAME = '#{@database}'" if @database
+        clauses << "TABLENAME = '#{name}'" if name
+        clauses << "TABLENAME LIKE '#{@table_name_prefix}%'" if @table_name_prefix
+        unless clauses.empty?
+          sql << " WHERE " + clauses.join(' AND ')
+        end
+        rs = execute(sql)
+        rs.entries.collect {|record| record['TableName'].strip}
       end
+
+      # Executes the SQL statement in the context of this connection.
+      def execute(sql, name = nil)
+        if name == :skip_logging
+          @connection.query(sql)
+        else
+          log(sql, name) { @connection.query(sql) }
+        end
+      end
+
+      # Executes +sql+ statement in the context of this connection using
+      # +binds+ as the bind substitutes. +name+ is logged along with
+      # the executed +sql+ statement.
+      def exec_query(sql, name = 'SQL', binds = [])
+        result = execute(sql, name)
+        if result && result.count > 0
+          ActiveRecord::Result.new(result.entries[0].keys, result.entries.collect{|r|r.collect{|v|v}})
+        else
+          ActiveRecord::Result.new([],[])
+        end
+      end
+
+      # Returns an ActiveRecord::Result instance.
+      def select(sql, name = nil, binds = [])
+        exec_query(sql, name)
+      end
+
+      # Returns an array of arrays containing the field values.
+      # Order is the same as that returned by +columns+.
+      def select_rows(sql, name = nil)
+        result = execute(sql, name)
+        if result && result.count > 0
+          result.entries.collect{|r|r.collect{|v|v}}
+        else
+          []
+        end
+      end
+
+      def execute_update(sql, name = nil)
+        log(sql, name) { @connection.execute_update(sql) }
+      end
+
+      def begin_db_transaction
+        execute_update "BEGIN TRANSACTION"
+      end
+
+      def commit_db_transaction
+        execute_update "END TRANSACTION"
+      end
+
+      def rollback_db_transaction
+        execute_update "ROLLBACK"
+      end
+
+      # Returns an array of +Column+ objects for the table specified by +table_name+.
+      def columns(table_name)#:nodoc:
+        sql = "SELECT * FROM DBC.COLUMNS WHERE TABLENAME='#{table_name}'"
+        sql << " AND DATABASENAME='#{@database}'" if @database
+        rs = execute(sql)
+        rs.entries.collect do |record|
+          new_column(
+            record['ColumnName'].strip,
+            record['DefaultValue'],
+            build_sql_type(record),
+            record['Nullable'].strip == "Y"
+          )
+        end
+      end
+
+      def primary_key(table_name)
+        columns(table_name).select{|col| col.primary == true}[0].name rescue 'id'
+      end
+
+      # Overridden by the adapters to instantiate their specific Column type.
+      def new_column(field, default, type, null) # :nodoc:
+        Column.new(field, default, type, null)
+      end
+
+      # construct the sql type from the teradata column info
+      def build_sql_type(column)
+        field_type = get_field_type(column['ColumnType'])
+        case field_type
+        when "varchar"
+          "#{field_type}(#{column['ColumnLength']})"
+        when "decimal"
+          "#{field_type}(#{column['DecimalTotalDigits']},#{column['DecimalFractionalDigits']})"
+        else
+          field_type
+        end
+      end
+
+      def get_field_type(type)
+        case type.strip
+        when "I", "I1"
+          "int"
+        when "CV"
+          "varchar"
+        when "DA"
+          "date"
+        when "D"
+          "decimal"
+        when "TS"
+          "datetime"
+        else
+          raise "Column type #{type} not supported"
+        end
+      end
+
     end
   end
 end

data/lib/teradata/ar/version.rb

@@ -1,5 +1,5 @@
 module Teradata
   module Ar
-    VERSION = "0.0.3"
+    VERSION = "0.0.4"
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: teradata-ar
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
 platform: ruby
 authors:
 - Giuseppe Privitera
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-10-04 00:00:00.000000000 Z
+date: 2013-10-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler