partitioned 0.8.0

data/examples/lib/by_company_id.rb

@@ -0,0 +1,11 @@
+class ByCompanyId < Partitioned::ByForeignKey
+  self.abstract_class = true
+
+  def self.partition_foreign_key
+    return :company_id
+  end
+
+  partitioned do |partition|
+    partition.index :id, :unique => true
+  end
+end

data/examples/lib/command_line_tool_mixin.rb

@@ -0,0 +1,71 @@
+module CommandLineToolMixin
+  def columnized_row(fields, sized)
+    r = []
+    fields.each_with_index do |f, i|
+      r << sprintf("%0-#{sized[i]}s", f.to_s.gsub(/\\n\\r/, '').slice(0, sized[i]))
+    end
+    r.join('   ')
+  end
+
+  def columnized(rows, options = {})
+    sized = {}
+    rows.each do |row|
+      row.each_index do |i|
+        value = row[i]
+        sized[i] = [sized[i].to_i, value.to_s.length].max
+        sized[i] = [options[:max_width], sized[i].to_i].min if options[:max_width]
+      end
+    end
+
+    table = []
+    rows.each { |row| table << "    " + columnized_row(row, sized).rstrip }
+    table.join("\n")
+  end
+
+  def help(command_line_usage, command_line_options)
+    puts(command_line_usage)
+    rows = []
+    command_line_options.keys.sort.each do |long_switch|
+      parameters = command_line_options[long_switch]
+      columns = []
+      switches = "#{long_switch}"
+      if (parameters[:short])
+        switches += " | #{parameters[:short]}"
+      end
+      if (parameters[:argument_note])
+        switches += " #{parameters[:argument_note]}"
+      end
+      columns << switches
+      columns << parameters[:note]
+      columns << (parameters[:environment_variable] or "")
+      rows << columns
+    end
+    puts(columnized(rows))
+  end
+
+  def command_line_options(options, usage = "command [OPTIONS]")
+    options.merge!({
+                       "--?" => {
+                           :short => "-?",
+                           :argument => GetoptLong::NO_ARGUMENT,
+                           :note => "show this help"
+                       },
+                   })
+    get_options = GetOptions.new(options)
+    get_options.each do |option, argument|
+      if option == '--?'
+        help(usage, options)
+        exit 0
+      else
+        if (options[option] && options[option][:constraint])
+          error = options[option][:constraint].call(argument, self)
+          if error
+            puts(error)
+            exit 0
+          end
+        end
+        yield option, argument
+      end
+    end
+  end
+end

data/examples/lib/company.rb

@@ -0,0 +1,29 @@
+class Company < ActiveRecord::Base
+  extend Partitioned::BulkMethodsMixin
+  has_many :employees, :class_name => 'Company', :conditions => "companies.id = employees.companies_id"
+
+  connection.execute <<-SQL
+    create table companies
+    (
+        id               serial not null primary key,
+        created_at       timestamp not null default now(),
+        updated_at       timestamp,
+        name             text null
+    );
+  SQL
+end
+
+COMPANIES = [
+             {
+               :name => 'Fluent Mobile, inc.'
+             },
+             {
+               :name => 'Fiksu, inc.'
+             },
+             {
+               :name => 'AppExchanger, inc.'
+             },
+             {
+               :name => 'FreeMyApps, inc.'
+             },
+]

data/examples/lib/get_options.rb

@@ -0,0 +1,44 @@
+require 'getoptlong'
+
+class GetOptions < GetoptLong
+  ARGUMENT_FLAGS = [
+      NO_ARGUMENT = GetoptLong::NO_ARGUMENT,
+      REQUIRED_ARGUMENT = GetoptLong::REQUIRED_ARGUMENT,
+      OPTIONAL_ARGUMENT = GetoptLong::OPTIONAL_ARGUMENT
+  ]
+
+  def initialize(switches = {})
+    @command_line_switches = switches
+    @environment_variables = {}
+    @getopt_options = []
+
+    @command_line_switches.each do |long_switch, parameters|
+      if parameters[:environment_variable].present?
+        @environment_variables[parameters[:environment_variable]] = long_switch
+      end
+
+      options = []
+      options << long_switch
+      if (parameters[:short])
+        options << parameters[:short]
+      end
+      options << parameters[:argument]
+      @getopt_options << options
+    end
+
+    argv_additions = []
+    for environment_variable_name, value in @environment_variables do
+      if ENV[environment_variable_name]
+        argv_additions << value
+        argv_additions << ENV[environment_variable_name] unless ENV[environment_variable_name].empty?
+      end
+    end
+    for arg in ARGV do
+      argv_additions << arg
+    end
+
+    argv_additions.each_with_index{ |v, i| ARGV[i] = v }
+
+    super(*@getopt_options)
+  end
+end

data/examples/lib/roman.rb

@@ -0,0 +1,41 @@
+Roman_array = [
+               [ 1000000, '_M'  ],
+               [  900000, '_C_M' ],
+               [  500000, '_D'  ],
+               [  400000, '_C_D' ],
+               [  100000, '_C'  ],
+               [   90000, '_X_C' ],
+               [   50000, '_L'  ],
+               [   40000, '_X_L' ],
+               [   10000, '_X'  ],
+               [    9000, '_I_X' ],
+               [    5000, '_V'  ],
+               [    1000, 'M'  ],
+               [     900, 'CM' ],
+               [     500, 'D'  ],
+               [     400, 'CD' ],
+               [     100, 'C'  ],
+               [      90, 'XC' ],
+               [      50, 'L'  ],
+               [      40, 'XL' ],
+               [      10, 'X'  ],
+               [       9, 'IX' ],
+               [       5, 'V'  ],
+               [       4, 'IV' ],
+               [       1, 'I'  ]
+              ]
+
+def to_roman(val)
+  if val < 0 or val >= 5000000
+    raise "out of range '#{val}'"
+  else
+    s = ""
+    Roman_array.each { |pair|
+      while val >= pair[0]
+        s << pair[1]
+        val -= pair[0]
+      end
+    }
+    return s
+  end
+end 

data/examples/start_date.rb

@@ -0,0 +1,621 @@
+#!/usr/bin/env ../spec/dummy/script/rails runner
+# if you use linux, please change previous line to the
+# "#! ../spec/dummy/script/rails runner"
+
+# Before running this example you should execute "bundle install" and "rake db:create".
+# To run this example you should open 'example' directory and execute example with one of the following flags:
+# -C    cleanup data in database and exit;
+# -F    cleanup data in database before creating new data;
+#
+# For example:
+# ./start_date.rb - F
+
+# Initial data:
+#
+#  Companies table is completed by four companies:
+#
+#  create table companies (
+#        id               serial not null primary key,
+#        created_at       timestamp not null default now(),
+#        updated_at       timestamp,
+#        name             text null
+#  );
+#
+#  insert into companies (created_at,id,name) values
+#    ('2012-03-13 13:26:52.184347',1,'Fluent Mobile, inc.'),
+#    ('2012-03-13 13:26:52.184347',2,'Fiksu, inc.'),
+#    ('2012-03-13 13:26:52.184347',3,'AppExchanger, inc.'),
+#    ('2012-03-13 13:26:52.184347',4,'FreeMyApps, inc.');
+#
+#  id |         created_at         | updated_at |        name
+#  ---+----------------------------+------------+---------------------
+#   1 | 2012-03-11 13:26:52.184347 |            | Fluent Mobile, inc.
+#   2 | 2012-03-11 13:26:52.184347 |            | Fiksu, inc.
+#   3 | 2012-03-11 13:26:52.184347 |            | AppExchanger, inc.
+#   4 | 2012-03-11 13:26:52.184347 |            | FreeMyApps, inc.
+#
+#  Employees table is associated with companies table via key - id:
+#
+#    create table employees
+#    (
+#        id               serial not null primary key,
+#        created_at       timestamp not null default now(),
+#        updated_at       timestamp null,
+#        start_date       date not null,
+#        name             text not null,
+#        salary           money not null,
+#        company_id       integer not null
+#    );
+#
+#   id | created_at | updated_at | start_date | name | salary | company_id
+#  ----+------------+------------+------------+------+--------+------------
+#
+# Task:
+#
+#  To increase the speed of requests to the database and to reduce the time
+#  of the request, need to split the Employees table to the partition tables.
+#  Break criterion is a start date(start_date).
+#
+# Implementation:
+#
+#  Class Employee inherits from the abstract class ByStartDate,
+#  which supports partitioning.
+#
+#  class Employee < Partitioned::ByStartDate
+#
+#    Indicates a relationship to the companies table.
+#    belongs_to :company, :class_name => 'Company'
+#
+#    Create a rules for each partition.
+#    Id is a unique index. Foreign key is company_id.
+#    This imposes a restriction on each of partition, that
+#    the column company_id associated with the table of companies
+#    and can not have values ​​that are not in the table companies.
+#    In this example, set up only 4 records in the table companies,
+#    so company_id can not be equal to 5 in any partition
+#    until it is an established company with id = 5.
+#
+#    partitioned do |partition|
+#      partition.index :id, :unique => true
+#      partition.foreign_key :company_id
+#    end
+#  end
+#
+#  Create a schema employees_partitions, within which to store all of our partitions:
+#
+#  Employee.create_infrastructure
+#
+#  Create a partition tables with increments of one month:
+#
+#  dates = Employee.partition_generate_range(START_DATE, END_DATE)
+#  Employee.create_new_partition_tables(dates)
+#
+#  Each of partition has the same structure as that of the employees table:
+#
+#   id | created_at | updated_at | start_date | name | salary | company_id
+#  ----+------------+------------+------------+------+--------+------------
+#
+#  CREATE TABLE "employees_partitions"."p201101" (CHECK (start_date >= '2011-01-01'
+#                                      AND start_date < '2011-02-01')) INHERITS (employees);
+#  CREATE INDEX "p201101_start_date_idx" ON "employees_partitions"."p201101" ("start_date");
+#  CREATE UNIQUE INDEX "p201101_id_udx" ON "employees_partitions"."p201101" ("id");
+#  ALTER TABLE employees_partitions.p201101 add foreign key (company_id) references companies(id);
+#
+#  CREATE TABLE "employees_partitions"."p201102" (CHECK (start_date >= '2011-02-01'
+#                                      AND start_date < '2011-03-01')) INHERITS (employees);
+#  CREATE INDEX "p201102_start_date_idx" ON "employees_partitions"."p201102" ("start_date");
+#  CREATE UNIQUE INDEX "p201102_id_udx" ON "employees_partitions"."p201102" ("id");
+#  ALTER TABLE employees_partitions.p201102 add foreign key (company_id) references companies(id);
+#
+#  CREATE TABLE "employees_partitions"."p201103" (CHECK (start_date >= '2011-03-01'
+#                                      AND start_date < '2011-04-01')) INHERITS (employees);
+#  CREATE INDEX "p201103_start_date_idx" ON "employees_partitions"."p201103" ("start_date");
+#  CREATE UNIQUE INDEX "p201103_id_udx" ON "employees_partitions"."p201103" ("id");
+#  ALTER TABLE employees_partitions.p201103 add foreign key (company_id) references companies(id);
+#
+#  CREATE TABLE "employees_partitions"."p201104" (CHECK (start_date >= '2011-04-01'
+#                                     AND start_date < '2011-05-01')) INHERITS (employees);
+#  CREATE INDEX "p201104_start_date_idx" ON "employees_partitions"."p201104" ("start_date");
+#  CREATE UNIQUE INDEX "p201104_id_udx" ON "employees_partitions"."p201104" ("id");
+#  ALTER TABLE employees_partitions.p201104 add foreign key (company_id) references companies(id);
+#
+#  CREATE TABLE "employees_partitions"."p201105" (CHECK (start_date >= '2011-05-01'
+#                                      AND start_date < '2011-06-01')) INHERITS (employees);
+#  CREATE INDEX "p201105_start_date_idx" ON "employees_partitions"."p201105" ("start_date");
+#  CREATE UNIQUE INDEX "p201105_id_udx" ON "employees_partitions"."p201105" ("id");
+#  ALTER TABLE employees_partitions.p201105 add foreign key (company_id) references companies(id);
+#
+#  CREATE TABLE "employees_partitions"."p201106" (CHECK (start_date >= '2011-06-01'
+#                                      AND start_date < '2011-07-01')) INHERITS (employees);
+#  CREATE INDEX "p201106_start_date_idx" ON "employees_partitions"."p201106" ("start_date");
+#  CREATE UNIQUE INDEX "p201106_id_udx" ON "employees_partitions"."p201106" ("id");
+#  ALTER TABLE employees_partitions.p201106 add foreign key (company_id) references companies(id);
+#
+#  CREATE TABLE "employees_partitions"."p201107" (CHECK (start_date >= '2011-07-01'
+#                                      AND start_date < '2011-08-01')) INHERITS (employees);
+#  CREATE INDEX "p201107_start_date_idx" ON "employees_partitions"."p201107" ("start_date");
+#  CREATE UNIQUE INDEX "p201107_id_udx" ON "employees_partitions"."p201107" ("id");
+#  ALTER TABLE employees_partitions.p201107 add foreign key (company_id) references companies(id);
+#
+#  CREATE TABLE "employees_partitions"."p201108" (CHECK (start_date >= '2011-08-01'
+#                                      AND start_date < '2011-09-01')) INHERITS (employees);
+#  CREATE INDEX "p201108_start_date_idx" ON "employees_partitions"."p201108" ("start_date");
+#  CREATE UNIQUE INDEX "p201108_id_udx" ON "employees_partitions"."p201108" ("id");
+#  ALTER TABLE employees_partitions.p201108 add foreign key (company_id) references companies(id);
+#
+#  CREATE TABLE "employees_partitions"."p201109" (CHECK (start_date >= '2011-09-01'
+#                                      AND start_date < '2011-10-01')) INHERITS (employees);
+#  CREATE INDEX "p201109_start_date_idx" ON "employees_partitions"."p201109" ("start_date");
+#  CREATE UNIQUE INDEX "p201109_id_udx" ON "employees_partitions"."p201109" ("id");
+#  ALTER TABLE employees_partitions.p201109 add foreign key (company_id) references companies(id);
+#
+#  CREATE TABLE "employees_partitions"."p201110" (CHECK (start_date >= '2011-10-01'
+#                                      AND start_date < '2011-11-01')) INHERITS (employees);
+#  CREATE INDEX "p201110_start_date_idx" ON "employees_partitions"."p201110" ("start_date");
+#  CREATE UNIQUE INDEX "p201110_id_udx" ON "employees_partitions"."p201110" ("id");
+#  ALTER TABLE employees_partitions.p201110 add foreign key (company_id) references companies(id);
+#
+#  CREATE TABLE "employees_partitions"."p201111" (CHECK (start_date >= '2011-11-01'
+#                                      AND start_date < '2011-12-01')) INHERITS (employees);
+#  CREATE INDEX "p201111_start_date_idx" ON "employees_partitions"."p201111" ("start_date");
+#  CREATE UNIQUE INDEX "p201111_id_udx" ON "employees_partitions"."p201111" ("id");
+#  ALTER TABLE employees_partitions.p201111 add foreign key (company_id) references companies(id);
+#
+#  CREATE TABLE "employees_partitions"."p201112" (CHECK (start_date >= '2011-12-01'
+#                                            AND start_date < '2012-01-01')) INHERITS (employees);
+#  CREATE INDEX "p201112_start_date_idx" ON "employees_partitions"."p201112" ("start_date");
+#  CREATE UNIQUE INDEX "p201112_id_udx" ON "employees_partitions"."p201112" ("id");
+#  ALTER TABLE employees_partitions.p201112 add foreign key (company_id) references companies(id);
+#
+# You should have the following tables with increments of one month:
+#  employees_partitions.p201101
+#  employees_partitions.p201102
+#  employees_partitions.p201103
+#  employees_partitions.p201104
+#  employees_partitions.p201105
+#  employees_partitions.p201106
+#  employees_partitions.p201107
+#  employees_partitions.p201108
+#  employees_partitions.p201109
+#  employees_partitions.p201110
+#  employees_partitions.p201111
+#  employees_partitions.p201112
+#
+#  Each of partitions inherits from employees table,
+#  thus a new row will automatically be added to the employees table .
+#
+#  To add data, we use the following constructions,
+#  in which employees and employee_data - a random data:
+#
+#  create_many - allows you to add multiple records
+#  Employee.create_many(employees)
+#  create - allows you to add one record
+#  Employee.create(employee_data)
+#  new/save! - allows you to add one record without using "create" method
+#  employee = Employee.new(employee_data)
+#  employee.save!
+#
+#  For update data, we use the following constructions,
+#  in which updates - a random data:
+#
+#  update_many - allows you to update multiple records.
+#  :set_array - additional option, you may read the description
+#  of the method in the file update_many bulk_methods_mixin.rb about this option.
+#  Employee.update_many(updates,  {:set_array => '"salary = #{table_name}.salary +
+#                                                    datatable.salary, updated_at = now()"'})
+#
+#  This construction using for update one record. You also may use update method.
+#  employee = Employee.from_partition(employee_record[:start_date]).find(employee_record[:id])
+#  employee.save
+#
+#  The data get into the employees table ONLY through partition tables.
+#  You can not do an insert row into a table employees directly.
+#  For this purpose special restrictions are imposed on the table employees.
+#
+#  Result:
+#
+#  We have table companies:
+#
+#  id |         created_at         | updated_at |        name
+#  ---+----------------------------+------------+---------------------
+#   1 | 2012-03-11 13:26:52.184347 |            | Fluent Mobile, inc.
+#   2 | 2012-03-11 13:26:52.184347 |            | Fiksu, inc.
+#   3 | 2012-03-11 13:26:52.184347 |            | AppExchanger, inc.
+#   4 | 2012-03-11 13:26:52.184347 |            | FreeMyApps, inc.
+#
+#  Table employees with random data from 1 to 5000:
+#
+#  id  |         created_at         |         updated_at         | start_date |               name                |   salary    | company_id
+#------+----------------------------+----------------------------+------------+-----------------------------------+-------------+------------
+#    1 | 2012-03-27 08:00:17.328727 |                            | 2011-11-12 | Winston J. Sillypants, I          |  $61,403.00 |     4
+#    2 | 2012-03-27 08:00:17.328727 |                            | 2011-11-24 | Winston J. Sillypants, II         |  $94,140.00 |     3
+#    3 | 2012-03-27 08:00:17.328727 | 2012-03-27 08:00:35.840263 | 2011-07-21 | Winston J. Sillypants, III        | $101,512.00 |     3
+#  ...
+# 4998 | 2012-03-27 08:00:29.175598 | 2012-03-27 08:00:29.175598 | 2011-08-31 | Picholine Pimplenad, MMMMCMXCVIII |  $70,642.00 |     1
+# 4999 | 2012-03-27 08:00:29.180265 | 2012-03-27 08:00:29.180265 | 2011-05-19 | Picholine Pimplenad, MMMMCMXCIX   |  $92,530.00 |     1
+# 5000 | 2012-03-27 08:00:29.184878 | 2012-03-27 09:00:43.18046  | 2011-03-16 | Picholine Pimplenad, _V           | $123,110.00 |     1
+#
+#  Partition employees_partitions.p201101 - partition where
+#  start_date >= '2011-01-01' AND start_date < '2011-02-01':
+#
+#  id  |         created_at         |         updated_at         | start_date |               name                |   salary    | company_id
+#------+----------------------------+----------------------------+------------+-----------------------------------+-------------+------------
+#    5 | 2012-03-27 08:00:17.328727 |                            | 2011-01-06 | Winston J. Sillypants, V          | $101,809.00 |     4
+#   16 | 2012-03-27 08:00:17.328727 | 2012-03-27 09:00:43.198146 | 2011-01-11 | Winston J. Sillypants, XVI        | $108,194.00 |     3
+#   18 | 2012-03-27 08:00:17.328727 |                            | 2011-01-29 | Winston J. Sillypants, XVIII      | $131,523.00 |     2
+#  ...
+# 4982 | 2012-03-27 08:00:29.099722 | 2012-03-27 08:00:29.099722 | 2011-01-12 | Picholine Pimplenad, MMMMCMLXXXII | $138,806.00 |     1
+# 4990 | 2012-03-27 08:00:29.138021 | 2012-03-27 08:00:33.090801 | 2011-01-17 | Picholine Pimplenad, MMMMCMXC     |  $73,884.00 |     1
+# 4997 | 2012-03-27 08:00:29.170905 | 2012-03-27 08:00:29.170905 | 2011-01-07 | Picholine Pimplenad, MMMMCMXCVII  |  $71,352.00 |     3
+#
+#  Partition employees_partitions.p201102 - partition where
+#  start_date >= '2011-02-01' AND start_date < '2011-03-01':
+#
+#  id  |         created_at         |         updated_at         | start_date |               name                |   salary    | company_id
+#------+----------------------------+----------------------------+------------+-----------------------------------+-------------+------------
+#   39 | 2012-03-27 08:00:17.328727 | 2012-03-27 09:00:43.124392 | 2011-02-16 | Winston J. Sillypants, XXXIX      |  $61,801.00 |     3
+#  118 | 2012-03-27 08:00:17.328727 |                            | 2011-02-15 | Winston J. Sillypants, CXVIII     |  $77,866.00 |     2
+#  147 | 2012-03-27 08:00:17.328727 | 2012-03-27 09:00:43.124392 | 2011-02-24 | Winston J. Sillypants, CXLVII     | $119,707.00 |     4
+#  ...
+# 4981 | 2012-03-27 08:00:29.094862 | 2012-03-27 08:00:29.094862 | 2011-02-28 | Picholine Pimplenad, MMMMCMLXXXI  | $123,537.00 |     3
+# 4993 | 2012-03-27 08:00:29.152405 | 2012-03-27 09:00:43.124392 | 2011-02-02 | Picholine Pimplenad, MMMMCMXCIII  | $139,969.00 |     3
+# 4995 | 2012-03-27 08:00:29.161747 | 2012-03-27 08:00:29.161747 | 2011-02-24 | Picholine Pimplenad, MMMMCMXCV    | $114,288.00 |     3
+#
+#  Partition employees_partitions.p201103 - partition where
+#  start_date >= '2011-03-01' AND start_date < '2011-04-01':
+#
+#  id  |         created_at         |         updated_at         | start_date |              name                 |   salary    | company_id
+#------+----------------------------+----------------------------+------------+-----------------------------------+-------------+------------
+#   13 | 2012-03-27 08:00:17.328727 | 2012-03-27 08:00:30.051678 | 2011-03-17 | Winston J. Sillypants, XIII       | $127,027.00 |     2
+#   20 | 2012-03-27 08:00:17.328727 |                            | 2011-03-23 | Winston J. Sillypants, XX         |  $67,804.00 |     4
+#   50 | 2012-03-27 08:00:17.328727 |                            | 2011-03-12 | Winston J. Sillypants, L          |  $75,709.00 |     1
+#  ...
+# 4991 | 2012-03-27 08:00:29.142771 | 2012-03-27 08:00:33.477371 | 2011-03-15 | Picholine Pimplenad, MMMMCMXCI    |  $81,774.00 |     4
+# 4996 | 2012-03-27 08:00:29.166363 | 2012-03-27 08:00:34.219636 | 2011-03-17 | Picholine Pimplenad, MMMMCMXCVI   |  $62,954.00 |     2
+# 5000 | 2012-03-27 08:00:29.184878 | 2012-03-27 09:00:43.18046  | 2011-03-16 | Picholine Pimplenad, _V           | $123,110.00 |     1
+#
+#  Partition employees_partitions.p201104 - partition where
+#  start_date >= '2011-04-01' AND start_date < '2011-05-01':
+#
+#  id  |         created_at         |         updated_at         | start_date |                name               |   salary    | company_id
+#------+----------------------------+----------------------------+------------+-----------------------------------+-------------+------------
+#   14 | 2012-03-27 08:00:17.328727 |                            | 2011-04-26 | Winston J. Sillypants, XIV        |  $79,735.00 |     3
+#   30 | 2012-03-27 08:00:17.328727 |                            | 2011-04-05 | Winston J. Sillypants, XXX        | $111,415.00 |     2
+#   44 | 2012-03-27 08:00:17.328727 |                            | 2011-04-26 | Winston J. Sillypants, XLIV       | $119,897.00 |     2
+#  ...
+# 4953 | 2012-03-27 08:00:28.901328 | 2012-03-27 08:00:28.901328 | 2011-04-02 | Picholine Pimplenad, MMMMCMLIII   |  $79,695.00 |     4
+# 4979 | 2012-03-27 08:00:29.085946 | 2012-03-27 08:00:29.085946 | 2011-04-12 | Picholine Pimplenad, MMMMCMLXXIX  | $127,600.00 |     1
+# 4983 | 2012-03-27 08:00:29.105705 | 2012-03-27 08:00:30.791092 | 2011-04-04 | Picholine Pimplenad, MMMMCMLXXXIII| $136,207.00 |     1
+#
+#  Partition employees_partitions.p201105 - partition where
+#  start_date >= '2011-05-01' AND start_date < '2011-06-01':
+#
+#  id  |         created_at         |         updated_at         | start_date |                name               |   salary    | company_id
+#------+----------------------------+----------------------------+------------+-----------------------------------+-------------+------------
+#    7 | 2012-03-27 08:00:17.328727 |                            | 2011-05-06 | Winston J. Sillypants, VII        |  $92,576.00 |     3
+#   41 | 2012-03-27 08:00:17.328727 | 2012-03-27 09:00:43.3198   | 2011-05-21 | Winston J. Sillypants, XLI        | $101,404.00 |     2
+#   76 | 2012-03-27 08:00:17.328727 |                            | 2011-05-13 | Winston J. Sillypants, LXXVI      |  $90,599.00 |     3
+#  ...
+# 4987 | 2012-03-27 08:00:29.123474 | 2012-03-27 08:00:29.123474 | 2011-05-14 | Picholine Pimplenad, MMMMCMLXXXVII| $125,956.00 |     4
+# 4994 | 2012-03-27 08:00:29.157211 | 2012-03-27 08:00:29.157211 | 2011-05-30 | Picholine Pimplenad, MMMMCMXCIV   |  $70,812.00 |     4
+# 4999 | 2012-03-27 08:00:29.180265 | 2012-03-27 08:00:29.180265 | 2011-05-19 | Picholine Pimplenad, MMMMCMXCIX   |  $92,530.00 |     1
+#
+#  Partition employees_partitions.p201106 - partition where
+#  start_date >= '2011-06-01' AND start_date < '2011-07-01':
+#
+#  id  |         created_at         |         updated_at         | start_date |               name                |   salary    | company_id
+#------+----------------------------+----------------------------+------------+-----------------------------------+-------------+------------
+#    9 | 2012-03-27 08:00:17.328727 |                            | 2011-06-11 | Winston J. Sillypants, IX         |  $76,361.00 |     1
+#   23 | 2012-03-27 08:00:17.328727 |                            | 2011-06-07 | Winston J. Sillypants, XXIII      |  $99,101.00 |     2
+#   42 | 2012-03-27 08:00:17.328727 | 2012-03-27 08:00:36.081006 | 2011-06-02 | Winston J. Sillypants, XLII       |  $98,706.00 |     1
+#  ...
+# 4948 | 2012-03-27 08:00:28.878517 | 2012-03-27 09:00:43.3421   | 2011-06-17 | Picholine Pimplenad, MMMMCMXLVIII |  $92,524.00 |     4
+# 4969 | 2012-03-27 08:00:28.973943 | 2012-03-27 08:00:28.973943 | 2011-06-28 | Picholine Pimplenad, MMMMCMLXIX   |  $91,862.00 |     3
+# 4973 | 2012-03-27 08:00:28.991697 | 2012-03-27 08:00:28.991697 | 2011-06-26 | Picholine Pimplenad, MMMMCMLXXIII | $102,840.00 |     2
+#
+#  Partition employees_partitions.p201107 - partition where
+#  start_date >= '2011-07-01' AND start_date < '2011-08-01':
+#
+#  id  |         created_at         |         updated_at         | start_date |               name                |   salary    | company_id
+#------+----------------------------+----------------------------+------------+-----------------------------------+-------------+------------
+#    3 | 2012-03-27 08:00:17.328727 | 2012-03-27 08:00:35.840263 | 2011-07-21 | Winston J. Sillypants, III        | $101,512.00 |     3
+#    4 | 2012-03-27 08:00:17.328727 |                            | 2011-07-26 | Winston J. Sillypants, IV         |  $75,210.00 |     4
+#   10 | 2012-03-27 08:00:17.328727 |                            | 2011-07-16 | Winston J. Sillypants, X          |  $84,192.00 |     3
+#  ...
+# 4951 | 2012-03-27 08:00:28.892169 | 2012-03-27 08:00:28.892169 | 2011-07-13 | Picholine Pimplenad, MMMMCMLI     |  $74,385.00 |     1
+# 4958 | 2012-03-27 08:00:28.924581 | 2012-03-27 08:00:29.740671 | 2011-07-30 | Picholine Pimplenad, MMMMCMLVIII  | $134,710.00 |     3
+# 4964 | 2012-03-27 08:00:28.951394 | 2012-03-27 08:00:35.574577 | 2011-07-02 | Picholine Pimplenad, MMMMCMLXIV   | $111,258.00 |     4
+#
+#  Partition employees_partitions.p201108 - partition where
+#  start_date >= '2011-08-01' AND start_date < '2011-09-01':
+#
+#  id  |         created_at         |         updated_at         | start_date |               name                |   salary    | company_id
+#------+----------------------------+----------------------------+------------+-----------------------------------+-------------+------------
+#   11 | 2012-03-27 08:00:17.328727 |                            | 2011-08-08 | Winston J. Sillypants, XI         | $124,558.00 |     2
+#   28 | 2012-03-27 08:00:17.328727 | 2012-03-27 09:00:43.087144 | 2011-08-12 | Winston J. Sillypants, XXVIII     | $103,416.00 |     2
+#   37 | 2012-03-27 08:00:17.328727 |                            | 2011-08-30 | Winston J. Sillypants, XXXVII     | $119,090.00 |     2
+#  ...
+# 4985 | 2012-03-27 08:00:29.114785 | 2012-03-27 08:00:29.114785 | 2011-08-12 | Picholine Pimplenad, MMMMCMLXXXV  |  $79,474.00 |     3
+# 4989 | 2012-03-27 08:00:29.133224 | 2012-03-27 08:00:35.107938 | 2011-08-29 | Picholine Pimplenad, MMMMCMLXXXIX | $118,545.00 |     2
+# 4998 | 2012-03-27 08:00:29.175598 | 2012-03-27 08:00:29.175598 | 2011-08-31 | Picholine Pimplenad, MMMMCMXCVIII |  $70,642.00 |     1
+#
+#  Partition employees_partitions.p201109 - partition where
+#  start_date >= '2011-09-01' AND start_date < '2011-10-01':
+#
+#  id  |         created_at         |         updated_at         | start_date |                name                |   salary    | company_id
+#------+----------------------------+----------------------------+------------+------------------------------------+-------------+------------
+#   17 | 2012-03-27 08:00:17.328727 |                            | 2011-09-03 | Winston J. Sillypants, XVII        | $137,286.00 |     4
+#   22 | 2012-03-27 08:00:17.328727 |                            | 2011-09-19 | Winston J. Sillypants, XXII        | $108,640.00 |     1
+#   24 | 2012-03-27 08:00:17.328727 |                            | 2011-09-09 | Winston J. Sillypants, XXIV        |  $71,643.00 |     2
+#  ...
+# 4967 | 2012-03-27 08:00:28.965141 | 2012-03-27 08:00:28.965141 | 2011-09-15 | Picholine Pimplenad, MMMMCMLXVII   | $103,039.00 |     1
+# 4978 | 2012-03-27 08:00:29.015989 | 2012-03-27 09:00:43.299117 | 2011-09-26 | Picholine Pimplenad, MMMMCMLXXVIII | $110,453.00 |     3
+# 4988 | 2012-03-27 08:00:29.12854  | 2012-03-27 08:00:34.414851 | 2011-09-15 | Picholine Pimplenad, MMMMCMLXXXVIII| $140,014.00 |     4
+#
+#  Partition employees_partitions.p201110 - partition where
+#  start_date >= '2011-10-01' AND start_date < '2011-11-01':
+#
+#  id  |         created_at         |         updated_at         | start_date |               name                |   salary    | company_id
+#------+----------------------------+----------------------------+------------+-----------------------------------+-------------+------------
+#   15 | 2012-03-27 08:00:17.328727 |                            | 2011-10-29 | Winston J. Sillypants, XV         | $122,755.00 |     4
+#   27 | 2012-03-27 08:00:17.328727 |                            | 2011-10-18 | Winston J. Sillypants, XXVII      |  $62,954.00 |     3
+#   38 | 2012-03-27 08:00:17.328727 |                            | 2011-10-14 | Winston J. Sillypants, XXXVIII    |  $85,840.00 |     1
+#  ...
+# 4976 | 2012-03-27 08:00:29.005547 | 2012-03-27 08:00:29.005547 | 2011-10-10 | Picholine Pimplenad, MMMMCMLXXVI  | $125,095.00 |     3
+# 4984 | 2012-03-27 08:00:29.11038  | 2012-03-27 08:00:33.269888 | 2011-10-11 | Picholine Pimplenad, MMMMCMLXXXIV | $105,346.00 |     3
+# 4992 | 2012-03-27 08:00:29.147872 | 2012-03-27 08:00:29.147872 | 2011-10-08 | Picholine Pimplenad, MMMMCMXCII   |  $65,541.00 |     4
+#
+#  Partition employees_partitions.p201111 - partition where
+#  start_date >= '2011-11-01' AND start_date < '2011-12-01':
+#
+#  id  |         created_at         |         updated_at         | start_date |               name                |   salary    | company_id
+#------+----------------------------+----------------------------+------------+-----------------------------------+-------------+------------
+#    1 | 2012-03-27 08:00:17.328727 |                            | 2011-11-12 | Winston J. Sillypants, I          |  $61,403.00 |     4
+#    2 | 2012-03-27 08:00:17.328727 |                            | 2011-11-24 | Winston J. Sillypants, II         |  $94,140.00 |     3
+#    8 | 2012-03-27 08:00:17.328727 |                            | 2011-11-22 | Winston J. Sillypants, VIII       | $139,234.00 |     1
+#  ...
+# 4950 | 2012-03-27 08:00:28.887451 | 2012-03-27 08:00:28.887451 | 2011-11-18 | Picholine Pimplenad, MMMMCML      |  $78,791.00 |     2
+# 4962 | 2012-03-27 08:00:28.942506 | 2012-03-27 09:00:43.14307  | 2011-11-15 | Picholine Pimplenad, MMMMCMLXII   | $121,325.00 |     3
+# 4974 | 2012-03-27 08:00:28.996189 | 2012-03-27 08:00:28.996189 | 2011-11-21 | Picholine Pimplenad, MMMMCMLXXIV  |  $60,607.00 |     2
+#
+#  Partition employees_partitions.p201112 - partition where
+#  start_date >= '2011-12-01' AND start_date < '2012-01-01':
+#
+#  id  |         created_at         |         updated_at         | start_date |               name                |   salary    | company_id
+#------+----------------------------+----------------------------+------------+-----------------------------------+-------------+------------
+#    6 | 2012-03-27 08:00:17.328727 |                            | 2011-12-22 | Winston J. Sillypants, VI         | $114,347.00 |     2
+#   12 | 2012-03-27 08:00:17.328727 |                            | 2011-12-22 | Winston J. Sillypants, XII        |  $83,008.00 |     4
+#   19 | 2012-03-27 08:00:17.328727 | 2012-03-27 08:00:35.479746 | 2011-12-16 | Winston J. Sillypants, XIX        | $101,796.00 |     4
+#  ...
+# 4939 | 2012-03-27 08:00:28.837839 | 2012-03-27 09:00:43.1619   | 2011-12-16 | Picholine Pimplenad, MMMMCMXXXIX  | $114,202.00 |     1
+# 4968 | 2012-03-27 08:00:28.969687 | 2012-03-27 08:00:33.652371 | 2011-12-13 | Picholine Pimplenad, MMMMCMLXVIII | $116,727.00 |     3
+# 4975 | 2012-03-27 08:00:29.000734 | 2012-03-27 08:00:29.000734 | 2011-12-25 | Picholine Pimplenad, MMMMCMLXXV   | $125,930.00 |     2
+#
+
+require File.expand_path(File.dirname(__FILE__) + "/lib/command_line_tool_mixin")
+require File.expand_path(File.dirname(__FILE__) + "/lib/get_options")
+
+include CommandLineToolMixin
+
+$cleanup = false
+$force = false
+$create_many = 3000
+$create_individual = 1000
+$new_individual = 1000
+$update_many = 1000
+$update_individual = 1000
+
+@options = {
+  "--cleanup" => {
+    :short => "-C",
+    :argument => GetoptLong::NO_ARGUMENT,
+    :note => "cleanup data in database and exit"
+  },
+  "--force" => {
+    :short => "-F",
+    :argument => GetoptLong::NO_ARGUMENT,
+    :note => "cleanup data in database before creating new data"
+  },
+  "--create-many" => {
+    :short => "-m",
+    :argument => GetoptLong::REQUIRED_ARGUMENT,
+    :note => "how many objects to create via create_many",
+    :argument_note => "NUMBER"
+  },
+  "--create-individual" => {
+    :short => "-i",
+    :argument => GetoptLong::REQUIRED_ARGUMENT,
+    :note => "how many objects to create via create",
+    :argument_note => "NUMBER"
+  },
+  "--new-individual" => {
+    :short => "-I",
+    :argument => GetoptLong::REQUIRED_ARGUMENT,
+    :note => "how many objects to create via new/save",
+    :argument_note => "NUMBER"
+  },
+  "--update-individual" => {
+    :short => "-u",
+    :argument => GetoptLong::REQUIRED_ARGUMENT,
+    :note => "how many objects to update indivudually",
+    :argument_note => "NUMBER"
+  },
+  "--update-many" => {
+    :short => "-U",
+    :argument => GetoptLong::REQUIRED_ARGUMENT,
+    :note => "how many objects to update via update_many",
+    :argument_note => "NUMBER"
+  },
+}
+
+command_line_options(@options) do |option,argument|
+  if option == '--cleanup'
+    $cleanup = true
+  elsif option == '--force'
+    $force = true
+  elsif option == '--create-many'
+    $create_many = argument.to_i
+  elsif option == '--create-individual'
+    $create_individual = argument.to_i
+  elsif option == '--new-individual'
+    $new_individual = argument.to_i
+  elsif option == '--update-individual'
+    $update_individual = argument.to_i
+  elsif option == '--update-many'
+    $update_many = argument.to_i
+  end
+end
+
+if $cleanup || $force
+  ActiveRecord::Base.connection.drop_schema("employees_partitions", :cascade => true) rescue nil
+  ActiveRecord::Base.connection.drop_table("employees") rescue nil
+  ActiveRecord::Base.connection.drop_table("companies") rescue nil
+  exit(0) if $cleanup
+end
+
+$total_records = $create_many + $create_individual + $new_individual
+
+puts "total records: #{$total_records}"
+
+START_DATE = Date.parse('2011-01-01')
+END_DATE = Date.parse('2011-12-31')
+
+# the ActiveRecord classes
+
+require File.expand_path(File.dirname(__FILE__) + "/lib/company")
+
+class Partitioned::ByStartDate < Partitioned::ByMonthlyTimeField
+  self.abstract_class = true
+
+  def self.partition_time_field
+    return :start_date
+  end
+end
+
+class Employee < Partitioned::ByStartDate
+  belongs_to :company, :class_name => 'Company'
+  attr_accessible :company_id, :start_date, :salary, :name
+
+  partitioned do |partition|
+    partition.index :id, :unique => true
+    partition.foreign_key :company_id
+  end
+
+  connection.execute <<-SQL
+    create table employees
+    (
+        id               serial not null primary key,
+        created_at       timestamp not null default now(),
+        updated_at       timestamp null,
+        start_date       date not null,
+        name             text not null,
+        salary           money not null,
+        company_id       integer not null
+    );
+  SQL
+end
+
+# You should have the following tables:
+#  public.companies
+#  public.employees
+
+# create the infrastructure for EMPLOYEES table which includes the schema and partition tables
+
+Employee.create_infrastructure
+
+# You should have the following schema:
+#  employees_partitions
+
+dates = Employee.partition_generate_range(START_DATE, END_DATE)
+Employee.create_new_partition_tables(dates)
+
+# You should have the following tables with increments of one month:
+#  employees_partitions.p201101
+#  employees_partitions.p201102
+#  employees_partitions.p201103
+#  employees_partitions.p201104
+#  employees_partitions.p201105
+#  employees_partitions.p201106
+#  employees_partitions.p201107
+#  employees_partitions.p201108
+#  employees_partitions.p201109
+#  employees_partitions.p201110
+#  employees_partitions.p201111
+#  employees_partitions.p201112
+
+# add some companies
+
+Company.create_many(COMPANIES)
+company_ids = Company.all.map(&:id)
+
+# now add some employees across the year.
+
+employees = []
+
+require File.expand_path(File.dirname(__FILE__) + "/lib/roman")
+
+# generates data for employees_partitions and employees tables
+
+base = 0
+(1..$create_many).each do |i|
+  employees << {
+    :name => "Winston J. Sillypants, #{to_roman(base+i)}",
+    :start_date => START_DATE + rand(END_DATE - START_DATE) + rand(1.day.seconds).seconds,
+    :salary => rand(80000) + 60000,
+    :company_id => company_ids[rand company_ids.length]
+  }
+end
+
+puts "creating many #{$create_many}"
+Employee.create_many(employees)
+base += $create_many
+
+puts "creating individual #{$create_individual}"
+(1..$create_individual).each do |i|
+  employee_data = {
+    :name => "Jonathan Crabapple, #{to_roman(base+i)}",
+    :start_date => START_DATE + rand(END_DATE - START_DATE) + rand(1.day.seconds).seconds,
+    :salary => rand(80000) + 60000,
+    :company_id => company_ids[rand company_ids.length]
+  }
+  employees << Employee.create(employee_data)
+end
+base += $create_individual
+
+puts "new individual #{$new_individual}"
+(1..$new_individual).each do |i|
+  employee_data = {
+    :name => "Picholine Pimplenad, #{to_roman(base+i)}",
+    :start_date => START_DATE + rand(END_DATE - START_DATE) + rand(1.day.seconds).seconds,
+    :salary => rand(80000) + 60000,
+    :company_id => company_ids[rand company_ids.length]
+  }
+  employee = Employee.new(employee_data)
+  employee.save
+  employees << employee
+end
+base += $new_individual
+
+updates = {}
+puts "update many #{$update_many}"
+(1..$update_many).each do |i|
+  index = rand(employees.length)
+  employee_record = employees[index]
+  updates[{
+            :id => employee_record[:id],
+            :start_date => employee_record[:start_date]
+          }] = {
+    :salary => 100
+  }
+end
+
+Employee.update_many(updates, {:set_array => '"salary = #{table_name}.salary + datatable.salary, updated_at = now()"'})
+
+puts "update individual #{$update_individual}"
+(1..$update_individual).each do |i|
+  index = rand(employees.length)
+  employee_record = employees[index]
+  employee = Employee.from_partition(employee_record[:start_date]).find(employee_record[:id])
+  employee.salary += 1000
+  employee.save
+end