mass_insert 0.1.1 → 0.1.2

data/lib/mass_insert/builder/adapters/helpers/column_value.rb

@@ -0,0 +1,90 @@
+module MassInsert
+  module Builder
+    module Adapters
+      module Helpers
+        class ColumnValue
+
+          attr_accessor :row, :column, :class_name
+
+          def initialize row, column, class_name
+            @row        = row
+            @column     = column
+            @class_name = class_name
+          end
+
+          # Returns a symbol with the column type according to the database.
+          def column_type
+            class_name.columns_hash[@column.to_s].type
+          end
+
+          # Returns the value to this column in the row hash. The value is
+          # finding by symbol or string key to be most flexible.
+          def column_value
+            row.fetch(column){row[@column.to_s]}
+          end
+
+          # Returns the default column value and it's added to the query if
+          # the row hash does not contains column value.
+          def default_value
+            default_db_value ? default_db_value.to_s : "null"
+          end
+
+          # Returns the default database column value using methods that
+          # ActiveRecord provides.
+          def default_db_value
+            class_name.columns_hash[@column.to_s].default
+          end
+
+          # Returns the valid column value to be included in the query.
+          # The value depends of the database engine.
+          def build
+            column_value.nil? ? default_value : send(:"column_value_#{column_type}")
+          end
+
+          # Returns the correct value when column value is string, text,
+          # date, time, datetime, timestamp.
+          def column_value_string
+            "'#{column_value}'"
+          end
+          alias :column_value_text      :column_value_string
+          alias :column_value_date      :column_value_string
+          alias :column_value_time      :column_value_string
+          alias :column_value_datetime  :column_value_string
+          alias :column_value_timestamp :column_value_string
+          alias :column_value_binary    :column_value_string
+
+          # Returns the correct value to integer column.
+          def column_value_integer
+            column_value.to_i.to_s
+          end
+
+          # Returns the correct value to decimal or float column.
+          def column_value_decimal
+            column_value.to_f.to_s
+          end
+          alias :column_value_float :column_value_decimal
+
+          # Returns the correct value to boolean column. This column calls
+          # the correct method according to the database adapter.
+          def column_value_boolean
+            self.send(:"#{Utilities.adapter}_column_value_boolean")
+          end
+
+          # Returns the column value to boolean column to mysql, postgresql
+          # and sqlserver databases.
+          def mysql2_column_value_boolean
+            column_value ? "true" : "false"
+          end
+          alias :postgresql_column_value_boolean :mysql2_column_value_boolean
+
+          # Returns the column value to boolean column to sqlite database.
+          def sqlite3_column_value_boolean
+            column_value ? "1" : "0"
+          end
+          alias :sqlserver_column_value_boolean  :sqlite3_column_value_boolean
+
+        end
+      end
+    end
+  end
+end

data/lib/mass_insert/builder/adapters/mysql2_adapter.rb

@@ -0,0 +1,15 @@
+module MassInsert
+  module Builder
+    module Adapters
+      class Mysql2Adapter < Adapter
+
+        # The method is overwrited because the timestamp format to mysql
+        # adapter does not need accuracy with nanoseconds.
+        def timestamp_format
+          "%Y-%m-%d %H:%M:%S"
+        end
+
+      end
+    end
+  end
+end

data/lib/mass_insert/builder/adapters/postgresql_adapter.rb

@@ -0,0 +1,7 @@
+module MassInsert
+  module Builder
+    module Adapters
+      class PostgreSQLAdapter < Adapter; end
+    end
+  end
+end

data/lib/mass_insert/builder/adapters/sqlite3_adapter.rb

@@ -0,0 +1,27 @@
+module MassInsert
+  module Builder
+    module Adapters
+      class SQLite3Adapter < Adapter
+
+        # Returns the amount of records to each query. Tries to take the
+        # each_slice option value or 500 due it's a standard in SQLite.
+        def values_per_insertion
+          each_slice || 500
+        end
+
+        # The method is overwrited because the queries structure to Sqlite3
+        # adapter are different.
+        def string_values
+          "SELECT #{string_rows_values};"
+        end
+
+        # The method is overwrited because the record separator to sqlite adapter
+        # is different.
+        def string_rows_values
+          values.map{ |row| string_single_row_values(row) }.join(" UNION SELECT ")
+        end
+
+      end
+    end
+  end
+end

data/lib/mass_insert/builder/adapters/sqlserver_adapter.rb

@@ -0,0 +1,26 @@
+module MassInsert
+  module Builder
+    module Adapters
+      class SQLServerAdapter < Adapter
+
+        # Returns a basic part of the query with the columns definition.
+        def string_columns
+          "([#{columns.join("], [")}]) "
+        end
+
+        # Returns the amount of records to each query. Tries to take the
+        # each_slice option value or 1000 due it's a standard in SQLServer.
+        def values_per_insertion
+          each_slice || 1000
+        end
+
+        # The method is overwrited because the timestamp format to SQLServer
+        # adapter needs accuracy with three nanoseconds.
+        def timestamp_format
+          "%Y-%m-%d %H:%M:%S.%3N"
+        end
+
+      end
+    end
+  end
+end

data/lib/mass_insert/builder/base.rb

@@ -0,0 +1,28 @@
+module MassInsert
+  module Builder
+    class Base
+
+      # This function gets the correct adapter class and returns the
+      # sql string ready to be executed.
+      def build values, options
+        adapter_class.new(values, options).execute
+      end
+
+      # Returns the class of the correct database adapter according to the
+      # database engine that is used in Rails project.
+      def adapter_class
+        case Utilities.adapter
+        when "mysql2"
+          Adapters::Mysql2Adapter
+        when "postgresql"
+          Adapters::PostgreSQLAdapter
+        when "sqlite3"
+          Adapters::SQLite3Adapter
+        when "sqlserver"
+          Adapters::SQLServerAdapter
+        end
+      end
+
+    end
+  end
+end

data/lib/mass_insert/builder/utilities.rb

@@ -0,0 +1,13 @@
+module MassInsert
+  module Builder
+    module Utilities
+
+      # Returns a string that contains the adapter type previosly
+      # configured in Rails project usually in the database.yml file.
+      def self.adapter
+        ActiveRecord::Base.connection.instance_values["config"][:adapter]
+      end
+
+    end
+  end
+end

data/lib/mass_insert/executer.rb

@@ -0,0 +1,13 @@
+module MassInsert
+  # This class is responsible to execute the queries into the database.
+  # Uses the ActiveRecord::Base.connection.execute functionality.
+  class Executer
+
+    def execute queries
+      Array(queries).each do |query|
+        ActiveRecord::Base.connection.execute(query)
+      end
+    end
+
+  end
+end

data/lib/mass_insert/process.rb

@@ -0,0 +1,24 @@
+require 'benchmark'
+
+module MassInsert
+  class Process
+
+    def initialize values, options
+      @values  = values
+      @options = options
+    end
+
+    # Calls the necessary methods to complete the mass insertion process
+    # and the time that each method takes being executed.
+    def start
+      @building_time  = Benchmark.measure do
+        @queries = Builder::Base.new.build(@values, @options)
+      end
+
+      @execution_time = Benchmark.measure do
+        Executer.new.execute(@queries)
+      end
+    end
+
+  end
+end

data/lib/mass_insert/result.rb

@@ -0,0 +1,33 @@
+module MassInsert
+  # This class is responsible to provides the mass insert process
+  # results according to the Process class instance.
+  class Result
+
+    def initialize process
+      @process = process
+    end
+
+    # Returns the time that took to create the query or queries 
+    # string that was persisted.
+    def building_time
+      @process.instance_variable_get(:@building_time).total
+    end
+
+    # Returns the time that took to execute the query or queries
+    # string that was persisted.
+    def execution_time
+      @process.instance_variable_get(:@execution_time).total
+    end
+
+    # Returns the time that took to do all the MassInsert process.
+    def time
+      building_time + execution_time
+    end
+
+    # Returns the amount of records that were persisted.
+    def records
+      @process.instance_variable_get(:@values).count
+    end
+
+  end
+end

data/lib/mass_insert/version.rb

@@ -1,3 +1,3 @@
 module MassInsert
-  VERSION = "0.1.1"
+  VERSION = "0.1.2"
 end

data/mass_insert.gemspec

@@ -8,8 +8,8 @@ Gem::Specification.new do |spec|
   spec.version       = MassInsert::VERSION
   spec.authors       = ["Alejandro Gutiérrez"]
   spec.email         = ["alejandrodevs@gmail.com"]
-  spec.description   = "Mass database insertion in Rails"
-  spec.summary       = "This gem aims to provide an easy and faster way to do single database insertions in Rails."
+  spec.summary       = "Mass database insertion in Rails"
+  spec.description   = "This gem aims to provide an easy and faster way to do single database insertions in Rails."
   spec.homepage      = "https://github.com/alejandrogutierrez/mass_insert"
   spec.license       = "MIT"
 

data/spec/adapters/column_types/binary_spec.rb

@@ -0,0 +1,64 @@
+require 'spec_helper'
+
+describe "Binary" do
+  let!(:values){ [{}] }
+  let!(:options){ Hash.new }
+
+  context "when contains binary value" do
+    context "when is 1" do
+      it "saves the value correctly" do
+        values.first.merge!(checked: 1)
+        User.mass_insert(values, options)
+        expect(User.last.checked).to eq("1")
+      end
+    end
+
+    context "when is 0" do
+      it "saves the value correctly" do
+        values.first.merge!(checked: 0)
+        User.mass_insert(values, options)
+        expect(User.last.checked).to eq("0")
+      end
+    end
+  end
+
+  context "when contains a string" do
+    it "converts string value to binary" do
+      values.first.merge!(checked: "string")
+      User.mass_insert(values, options)
+      expect(User.last.checked).to eq("string")
+    end
+  end
+
+  context "when contains a integer greater than 1" do
+    it "converts integer value to binary" do
+      values.first.merge!(checked: 150)
+      User.mass_insert(values, options)
+      expect(User.last.checked).to eq("150")
+    end
+  end
+
+  context "when contains a decimal" do
+    it "converts decimal value to binary" do
+      values.first.merge!(checked: 25.34)
+      User.mass_insert(values, options)
+      expect(User.last.checked).to eq("25.34")
+    end
+  end
+
+  context "when contains a boolean" do
+    it "converts boolean value to binary" do
+      values.first.merge!(checked: true)
+      User.mass_insert(values, options)
+      expect(User.last.checked).to eq("true")
+    end
+  end
+
+  context "when not exist in values hashes" do
+    it "saves the default value" do
+      values.first.delete(:checked)
+      User.mass_insert(values, options)
+      expect(User.last.checked).to eq(nil)
+    end
+  end
+end

data/spec/adapters/column_types/boolean_spec.rb

@@ -0,0 +1,48 @@
+require 'spec_helper'
+
+describe "Boolean" do
+  let!(:values){ [{}] }
+  let!(:options){ Hash.new }
+
+  context "when contains boolean value" do
+    context "when is true" do
+      it "saves the true value correctly" do
+        values.first.merge!(active: true)
+        User.mass_insert(values, options)
+        expect(User.last.active).to eq(true)
+      end
+    end
+
+    context "when is false" do
+      it "saves the false value correctly" do
+        values.first.merge!(active: false)
+        User.mass_insert(values, options)
+        expect(User.last.active).to eq(false)
+      end
+    end
+  end
+
+  context "when contains a string" do
+    it "converts string value to boolean" do
+      values.first.merge!(active: "string")
+      User.mass_insert(values, options)
+      expect(User.last.active).to eq(true)
+    end
+  end
+
+  context "when contains a decimal" do
+    it "converts decimal value to boolean" do
+      values.first.merge!(active: 25.34)
+      User.mass_insert(values, options)
+      expect(User.last.active).to eq(true)
+    end
+  end
+
+  context "when not exist in values hashes" do
+    it "saves the default value" do
+      values.first.delete(:active)
+      User.mass_insert(values, options)
+      expect(User.last.active).to eq(nil)
+    end
+  end
+end

data/spec/adapters/column_types/decimal_spec.rb

@@ -0,0 +1,59 @@
+require 'spec_helper'
+
+describe "Decimal" do
+  let!(:values){ [{}] }
+  let!(:options){ Hash.new }
+
+  context "when contains an integer" do
+    it "converts integer value to decimal" do
+      values.first.merge!(money: 10)
+      User.mass_insert(values, options)
+      expect(User.last.money).to eq(10.0)
+    end
+  end
+
+  context "when contains a string without digits" do
+    it "converts string value to decimal" do
+      values.first.merge!(money: "string")
+      User.mass_insert(values, options)
+      expect(User.last.money).to eq(0.0)
+    end
+  end
+
+  context "when contains a digits string" do
+    it "converts string value to decimal" do
+      values.first.merge!(money: "100")
+      User.mass_insert(values, options)
+      expect(User.last.money).to eq(100.0)
+    end
+
+    it "converts string value to decimal" do
+      values.first.merge!(money: "200.50")
+      User.mass_insert(values, options)
+      expect(User.last.money).to eq(200.50)
+    end
+  end
+
+  context "when contains a decimal" do
+    it "saves the correct value" do
+      values.first.merge!(money: 20.50)
+      User.mass_insert(values, options)
+      expect(User.last.money).to eq(20.50)
+    end
+  end
+
+  context "when contains a boolean" do
+    it "raises an exception" do
+      values.first.merge!(money: true)
+      expect(lambda{ User.mass_insert(values, options) }).to raise_exception
+    end
+  end
+
+  context "when not exist in values hashes" do
+    it "saves the default value" do
+      values.first.delete(:money)
+      User.mass_insert(values, options)
+      expect(User.last.money).to eq(nil)
+    end
+  end
+end