vertica_rails_adapter 0.0.1

data/.gitignore

@@ -0,0 +1,18 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp
+*~

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in vertica_rails_adapter.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2013 Sasan
+
+MIT License
+
+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.

data/README.md

@@ -0,0 +1,29 @@
+# VerticaRailsAdapter
+
+TODO: Write a gem description
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'vertica_rails_adapter'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install vertica_rails_adapter
+
+## Usage
+
+TODO: Write usage instructions here
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/lib/active_record/connection_adapters/vertica_adapter.rb

@@ -0,0 +1,174 @@
+require 'active_record/connection_adapters/abstract_adapter'
+require 'active_support/core_ext/kernel/requires'
+
+module ActiveRecord
+  class Base
+
+    ##
+    # Establishes a connection to the database that's used by all Active Record objects
+    ##
+    def self.vertica_connection(config)
+      unless defined? Vertica
+        begin
+          require 'vertica'
+        rescue LoadError
+          raise "Vertica Gem not installed"
+        end
+      end
+
+      config = config.symbolize_keys
+      host = config[:host]
+      port = config[:port] || 5433
+      username = config[:username].to_s if config[:username]
+      password = config[:password].to_s if config[:password]
+      schema = config[:schema].to_s if config[:schema]
+
+      if config.has_key?(:database)
+        database = config[:database]
+      else
+        raise ArgumentError, "No database specified. Missing argument: database."
+      end
+
+      # if config.has_key?(:schema)
+      #   schema = config[:schema]
+      # else
+      #   raise ArgumentError, "Vertica Schema must be specified."
+      # end
+
+      conn = Vertica.connect({ :user => username, 
+                               :password => password, 
+                               :host => host, 
+                               :port => port, 
+                               :database => database, 
+                               :schema => schema })
+
+      options = [host, username, password, database, port,schema]
+
+      ConnectionAdapters::VerticaAdapter.new(conn, options, config)
+    end  
+
+
+    def self.instantiate(record)
+      record.stringify_keys!
+      
+      sti_class = find_sti_class(record[inheritance_column])
+      record_id = sti_class.primary_key && record[sti_class.primary_key]
+      if ActiveRecord::IdentityMap.enabled? && record_id
+        if (column = sti_class.columns_hash[sti_class.primary_key]) && column.number?
+          record_id = record_id.to_i
+        end
+        if instance = IdentityMap.get(sti_class, record_id)
+          instance.reinit_with('attributes' => record)
+        else
+          instance = sti_class.allocate.init_with('attributes' => record)
+          IdentityMap.add(instance)
+        end
+      else
+        instance = sti_class.allocate.init_with('attributes' => record)
+      end
+      
+      instance
+    end
+  end
+  
+  module ConnectionAdapters
+    class VerticaColumn < Column
+      
+    end
+    
+    class VerticaAdapter < AbstractAdapter
+      ADAPTER_NAME = 'Vertica'.freeze
+
+      def adapter_name #:nodoc:
+        ADAPTER_NAME
+      end
+
+      def initialize(connection, connection_options, config)
+        super(connection)
+        @connection_options, @config = connection_options, config
+        @quoted_column_names, @quoted_table_names = {}, {}
+        # connect
+      end
+
+      def active?
+        @connection.opened?
+      end
+      
+      # Disconnects from the database if already connected, and establishes a
+      # new connection with the database.
+      def reconnect!
+        @connection.reset_connection
+      end
+      def reset
+        reconnect!
+      end
+      
+      # Close the connection.
+      def disconnect!
+        @connection.close rescue nil
+      end
+
+      # return raw object
+      def execute(sql, name=nil)
+        log(sql,name) do
+          if block_given?
+            @connection = ::Vertica.connect(@connection.options)
+            @connection.query(sql) {|row| yield row }
+            @connection.close
+          else
+            @connection = ::Vertica.connect(@connection.options)
+            results = @connection.query(sql)
+            @connection.close
+            results
+          end
+        end
+      end
+
+      def schema_name
+        @schema ||= @connection.options[:schema]
+      end
+
+      def tables(name = nil) #:nodoc:
+        sql = "SELECT * FROM tables WHERE table_schema = #{quote_column_name(schema_name)}"
+
+        tables = []
+        execute(sql, name) { |field| tables << field[:table_name] }
+        tables
+      end
+
+      def columns(table_name, name = nil)#:nodoc:
+        sql = "SELECT * FROM columns WHERE table_name = #{quote_column_name(table_name)}"
+
+        columns = []
+        execute(sql, name){ |field| columns << VerticaColumn.new(field[:column_name],field[:column_default],field[:data_type],field[:is_nullable])}
+        columns
+      end
+
+      def select(sql, name = nil, binds = [])
+        rows = []
+        @connection = ::Vertica.connect(@connection.options)
+        @connection.query(sql) {|row| rows << row }
+        @connection.close
+        rows
+      end
+
+      def primary_key(table)
+        ''
+      end
+
+      ## QUOTING
+      def quote_column_name(name) #:nodoc:
+        "'#{name}'"
+      end
+
+      def quote_table_name(name) #:nodoc:
+        if schema_name.blank?
+          name
+        else
+          "#{schema_name}.#{name}"
+        end
+      end
+      
+    end  
+  end
+end

data/lib/activerecord-import/active_record/adapters/vertica_adapter.rb

@@ -0,0 +1,6 @@
+require "active_record/connection_adapters/vertica_adapter"
+require "activerecord-import/adapters/vertica_adapter"
+
+class ActiveRecord::ConnectionAdapters::VerticaAdapter
+  include ActiveRecord::Import::VerticaAdapter::InstanceMethods
+end

data/lib/activerecord-import/active_record/adapters/vertica_adapter.rb~

@@ -0,0 +1,6 @@
+require "activerecord/active_record/connection_adapters/vertica_adapter"
+require "activerecord-import/adapters/vertica_adapter"
+
+class ActiveRecord::ConnectionAdapters::VerticaAdapter
+  include ActiveRecord::Import::VerticaAdapter::InstanceMethods
+end

data/lib/activerecord-import/adapters/vertica_adapter.rb

@@ -0,0 +1,4 @@
+module ActiveRecord::Import::VerticaAdapter
+  module InstanceMethods
+  end
+end

data/lib/activerecord-vertica-adapter.rb

@@ -0,0 +1,3 @@
+require 'active_record/connection_adapters/vertica_adapter'
+require 'activerecord-import/active_record/adapters/vertica_adapter'
+require 'activerecord-import/adapters/vertica_adapter'

data/lib/vertica_rails_adapter.rb

@@ -0,0 +1,8 @@
+require "vertica_rails_adapter/version"
+
+require 'active_record/connection_adapters/vertica_adapter'
+require "activerecord-vertica-adapter"
+
+module VerticaRailsAdapter
+  # Your code goes here...
+end

data/lib/vertica_rails_adapter/version.rb

@@ -0,0 +1,3 @@
+module VerticaRailsAdapter
+  VERSION = "0.0.1"
+end

data/vertica_rails_adapter.gemspec

@@ -0,0 +1,21 @@
+# -*- encoding: utf-8 -*-
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'vertica_rails_adapter/version'
+
+Gem::Specification.new do |gem|
+  gem.name          = "vertica_rails_adapter"
+  gem.version       = VerticaRailsAdapter::VERSION
+  gem.authors       = ["Sasan Padidar"]
+  gem.email         = ["sasan@raybeam.com"]
+  gem.description   = %q{Vertica Adapter for ActiveRecord.}
+  gem.summary       = %q{Adds activerecord functionalites to your Vertica connection. The credit goes to https://github.com/zemis for starting up the project.}
+  gem.homepage      = ""
+
+  gem.files         = `git ls-files`.split($/)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.require_paths = ["lib"]
+
+  gem.add_dependency "rails", ">= 3.0.0"
+end

metadata

@@ -0,0 +1,93 @@
+--- !ruby/object:Gem::Specification 
+name: vertica_rails_adapter
+version: !ruby/object:Gem::Version 
+  hash: 29
+  prerelease: 
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Sasan Padidar
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2013-01-25 00:00:00 Z
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  type: :runtime
+  name: rails
+  prerelease: false
+  version_requirements: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 7
+        segments: 
+        - 3
+        - 0
+        - 0
+        version: 3.0.0
+  requirement: *id001
+description: Vertica Adapter for ActiveRecord.
+email: 
+- sasan@raybeam.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- lib/active_record/connection_adapters/vertica_adapter.rb
+- lib/activerecord-import/active_record/adapters/vertica_adapter.rb
+- lib/activerecord-import/active_record/adapters/vertica_adapter.rb~
+- lib/activerecord-import/adapters/vertica_adapter.rb
+- lib/activerecord-vertica-adapter.rb
+- lib/vertica_rails_adapter.rb
+- lib/vertica_rails_adapter/version.rb
+- vertica_rails_adapter.gemspec
+homepage: ""
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.8.22
+signing_key: 
+specification_version: 3
+summary: Adds activerecord functionalites to your Vertica connection. The credit goes to https://github.com/zemis for starting up the project.
+test_files: []
+