pg_partitioning 0.0.1

data/lib/pg_partitioning/strategies/date.rb

@@ -0,0 +1,73 @@
+require 'pg_partitioning/strategies/base'
+
+module PgPartitioning
+  module Strategies
+    class Date < Base
+      
+      protected
+
+      def valid?
+        if super
+          @error_message = if !(@data_type.include?('timestamp') || @data_type.include?('date'))
+                             I18n.t "pg_partitioning.failure.column_type"
+                           elsif (@cond.blank? || !cond_valid?)
+                             I18n.t "pg_partitioning.failure.blank_cond"
+                           end
+        end
+        @error_message.blank?
+      end
+      
+      def cond_valid?
+        res = []
+        patterns = { year:  /(?<year>Y{1,4})/,
+                     month: /(?<month>M{2})/,
+                     day:   /(?<day>D{2})/,
+                     week:  /(?<week>W{1,2})/,
+                     hour:  /(?<hour>HH24{1})/ }
+        
+        patterns.each do |key, value|
+          match = @cond.match value
+          res << match[key] if match
+        end
+        @cond = res.join '_'
+        !@cond.blank?
+      end
+
+      def create_insert_master_function
+        @sql.execute "CREATE OR REPLACE FUNCTION #{@table_name}_insert_master() RETURNS TRIGGER AS $$
+                       DECLARE
+                         colname text      := '#{@column_name.to_sym}';
+                         colval  timestamp := NEW.#{@column_name.to_sym};
+                         pattern text      := '#{@cond.to_sym}';
+                         sample  text      := TO_CHAR(colval, pattern);
+                         tblname text      := '#{@table_name}_' || sample;
+                       BEGIN
+                         IF NOT EXISTS(SELECT relname FROM pg_class WHERE relname=tblname) THEN
+                         EXECUTE 'CREATE TABLE '
+                                 || tblname 
+                                 || '(check (TO_CHAR(' 
+                                 || quote_ident(colname) 
+                                 || ','
+                                 || quote_literal(pattern)
+                                 || ')=' 
+                                 || quote_literal(sample)
+                                 || ')) INHERITS (' 
+                                 || TG_RELNAME 
+                                 || ');';
+                         EXECUTE 'CREATE INDEX ' 
+                                 || quote_ident('index_' || tblname) 
+                                 || ' ON ' 
+                                 || quote_ident(tblname) 
+                                 || ' (id, '
+                                 || quote_ident(colname)
+                                 || ');';
+                         END IF;
+                         EXECUTE 'INSERT INTO ' || tblname || ' SELECT ($1).*' USING NEW;
+                         RETURN NEW;
+                       END;
+                       $$ LANGUAGE plpgsql;"
+        info I18n.t("pg_partitioning.progress.insert_master", state: "OK")
+      end
+    end
+  end
+end

data/lib/pg_partitioning/strategies/equal.rb

@@ -0,0 +1,43 @@
+require 'pg_partitioning/strategies/base'
+
+module PgPartitioning
+  module Strategies
+    class Equal < Base
+      
+      protected
+      
+      def create_insert_master_function
+        @sql.execute "CREATE OR REPLACE FUNCTION #{@table_name}_insert_master() RETURNS TRIGGER AS $$
+                       DECLARE
+                         colname text := '#{@column_name.to_sym}';
+                         colval  text := NEW.#{@column_name.to_sym};
+                         tblname text := '#{@table_name}_' || colval;
+                       BEGIN
+                         IF NOT EXISTS(SELECT relname FROM pg_class WHERE relname=tblname) THEN
+                         EXECUTE 'CREATE TABLE '
+                                 || tblname 
+                                 || '(check (' 
+                                 || quote_ident(colname) 
+                                 || '=' 
+                                 || quote_literal(colval) 
+                                 || ')) INHERITS (' 
+                                 || TG_RELNAME 
+                                 || ');';
+                         EXECUTE 'CREATE INDEX ' 
+                                 || quote_ident('index_' || tblname) 
+                                 || ' ON ' 
+                                 || quote_ident(tblname) 
+                                 || ' (id, '
+                                 || quote_ident(colname)
+                                 || ');';
+                         END IF;
+                         EXECUTE 'INSERT INTO ' || tblname || ' SELECT ($1).*' USING NEW;
+                         RETURN NEW;
+                       END;
+                       $$ LANGUAGE plpgsql;"
+        info I18n.t("pg_partitioning.progress.insert_master", state: "OK")
+      end
+    end
+  end
+end
+

data/lib/pg_partitioning/strategies/step.rb

@@ -0,0 +1,57 @@
+require 'pg_partitioning/strategies/base'
+
+module PgPartitioning
+  module Strategies
+    class Step < Base
+      
+      def valid?
+        if super
+          @error_message = if !@data_type.include? 'integer'
+                             I18n.t "pg_partitioning.failure.column_type"
+                           elsif @cond.blank?
+                             I18n.t "pg_partitioning.failure.blank_cond"
+                           end
+        end
+        @error_message.blank?
+      end
+        
+      protected
+
+      def create_insert_master_function
+        @sql.execute "CREATE OR REPLACE FUNCTION #{@table_name}_insert_master() RETURNS TRIGGER AS $$
+                       DECLARE
+                         colname   text    := '#{@column_name.to_sym}';
+                         colval    integer := NEW.#{@column_name.to_sym};
+                         threshold integer := '#{@cond}';
+                         step      integer := ROUND(colval / threshold);
+                         tblname   text    := '#{@table_name}_' || step;
+                       BEGIN
+                         IF NOT EXISTS(SELECT relname FROM pg_class WHERE relname=tblname) THEN
+                         EXECUTE 'CREATE TABLE '
+                                 || tblname 
+                                 || '(check (ROUND(' 
+                                 || quote_ident(colname) 
+                                 || '/'
+                                 || threshold 
+                                 || ')=' 
+                                 || step
+                                 || ')) INHERITS (' 
+                                 || TG_RELNAME 
+                                 || ');';
+                         EXECUTE 'CREATE INDEX ' 
+                                 || quote_ident('index_' || tblname) 
+                                 || ' ON ' 
+                                 || quote_ident(tblname) 
+                                 || ' (id, '
+                                 || quote_ident(colname)
+                                 || ');';
+                         END IF;
+                         EXECUTE 'INSERT INTO ' || tblname || ' SELECT ($1).*' USING NEW;
+                         RETURN NEW;
+                       END;
+                       $$ LANGUAGE plpgsql;"
+        info I18n.t("pg_partitioning.progress.insert_master", state: "OK")
+      end
+    end
+  end
+end

data/lib/pg_partitioning/version.rb

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

data/lib/pg_partitioning.rb

@@ -0,0 +1,4 @@
+module PgPartitioning
+  class Engine < Rails::Engine
+  end
+end

data/lib/tasks/pg_partitioning_tasks.rake

@@ -0,0 +1,4 @@
+# desc "Explaining what the task does"
+# task :pg_partitioning do
+#   # Task goes here
+# end

data/spec/db_helpers.rb

@@ -0,0 +1,36 @@
+module DbHelpers
+  def clear_triggers(table)
+    sql = <<-SQL
+            DROP TRIGGER IF EXISTS #{table}_before_insert_trigger ON #{table};
+            DROP TRIGGER IF EXISTS #{table}_after_insert_trigger ON #{table}; 
+          SQL
+    ActiveRecord::Base.connection.execute(sql)
+  end
+  
+  def insert_master_function_exists?(table)
+    sql = <<-SQL
+            SELECT 1 FROM pg_trigger, pg_proc 
+            WHERE pg_proc.oid=pg_trigger.tgfoid 
+            AND pg_trigger.tgname = '#{table}_before_insert_trigger';
+          SQL
+    result = nil
+    ActiveRecord::Base.connection.execute(sql).each do |r|
+      result = r["?column?"]
+    end
+    result.to_i == 1
+  end
+  
+  def after_insert_trigger_exists?(table)
+    sql = <<-SQL
+            SELECT 1 FROM pg_trigger, pg_proc 
+            WHERE pg_proc.oid=pg_trigger.tgfoid 
+            AND pg_trigger.tgname = '#{table}_after_insert_trigger';
+          SQL
+    result = nil
+    ActiveRecord::Base.connection.execute(sql).each do |r|
+      result = r["?column?"]
+    end
+    result.to_i == 1
+  end
+end
+

data/spec/dummy/README.rdoc

@@ -0,0 +1,28 @@
+== README
+
+This README would normally document whatever steps are necessary to get the
+application up and running.
+
+Things you may want to cover:
+
+* Ruby version
+
+* System dependencies
+
+* Configuration
+
+* Database creation
+
+* Database initialization
+
+* How to run the test suite
+
+* Services (job queues, cache servers, search engines, etc.)
+
+* Deployment instructions
+
+* ...
+
+
+Please feel free to use a different markup language if you do not plan to run
+<tt>rake doc:app</tt>.

data/spec/dummy/Rakefile

@@ -0,0 +1,6 @@
+# Add your own tasks in files placed in lib/tasks ending in .rake,
+# for example lib/tasks/capistrano.rake, and they will automatically be available to Rake.
+
+require File.expand_path('../config/application', __FILE__)
+
+Rails.application.load_tasks

data/spec/dummy/app/assets/javascripts/application.js

@@ -0,0 +1,13 @@
+// This is a manifest file that'll be compiled into application.js, which will include all the files
+// listed below.
+//
+// Any JavaScript/Coffee file within this directory, lib/assets/javascripts, vendor/assets/javascripts,
+// or any plugin's vendor/assets/javascripts directory can be referenced here using a relative path.
+//
+// It's not advisable to add code directly here, but if you do, it'll appear at the bottom of the
+// compiled file.
+//
+// Read Sprockets README (https://github.com/rails/sprockets#sprockets-directives) for details
+// about supported directives.
+//
+//= require_tree .

data/spec/dummy/app/assets/stylesheets/application.css

@@ -0,0 +1,15 @@
+/*
+ * This is a manifest file that'll be compiled into application.css, which will include all the files
+ * listed below.
+ *
+ * Any CSS and SCSS file within this directory, lib/assets/stylesheets, vendor/assets/stylesheets,
+ * or any plugin's vendor/assets/stylesheets directory can be referenced here using a relative path.
+ *
+ * You're free to add application-wide styles to this file and they'll appear at the bottom of the
+ * compiled file so the styles you add here take precedence over styles defined in any styles
+ * defined in the other CSS/SCSS files in this directory. It is generally better to create a new
+ * file per style scope.
+ *
+ *= require_tree .
+ *= require_self
+ */

data/spec/dummy/app/controllers/application_controller.rb

@@ -0,0 +1,5 @@
+class ApplicationController < ActionController::Base
+  # Prevent CSRF attacks by raising an exception.
+  # For APIs, you may want to use :null_session instead.
+  protect_from_forgery with: :exception
+end

data/spec/dummy/app/helpers/application_helper.rb

@@ -0,0 +1,2 @@
+module ApplicationHelper
+end

data/spec/dummy/app/models/bandit.rb

@@ -0,0 +1,4 @@
+class Bandit < ActiveRecord::Base
+  belongs_to :gang
+  has_many :crimes, dependent: :destroy
+end

data/spec/dummy/app/models/crime.rb

@@ -0,0 +1,3 @@
+class Crime < ActiveRecord::Base
+  belongs_to :bandit
+end

data/spec/dummy/app/models/gang.rb

@@ -0,0 +1,3 @@
+class Gang < ActiveRecord::Base
+  has_many :bandits, dependent: :destroy
+end

data/spec/dummy/app/views/layouts/application.html.erb

@@ -0,0 +1,14 @@
+<!DOCTYPE html>
+<html>
+<head>
+  <title>Dummy</title>
+  <%= stylesheet_link_tag    'application', media: 'all', 'data-turbolinks-track' => true %>
+  <%= javascript_include_tag 'application', 'data-turbolinks-track' => true %>
+  <%= csrf_meta_tags %>
+</head>
+<body>
+
+<%= yield %>
+
+</body>
+</html>

data/spec/dummy/bin/bundle

@@ -0,0 +1,3 @@
+#!/usr/bin/env ruby
+ENV['BUNDLE_GEMFILE'] ||= File.expand_path('../../Gemfile', __FILE__)
+load Gem.bin_path('bundler', 'bundle')

data/spec/dummy/bin/rails

@@ -0,0 +1,4 @@
+#!/usr/bin/env ruby
+APP_PATH = File.expand_path('../../config/application', __FILE__)
+require_relative '../config/boot'
+require 'rails/commands'

data/spec/dummy/bin/rake

@@ -0,0 +1,4 @@
+#!/usr/bin/env ruby
+require_relative '../config/boot'
+require 'rake'
+Rake.application.run

data/spec/dummy/bin/setup

@@ -0,0 +1,29 @@
+#!/usr/bin/env ruby
+require 'pathname'
+
+# path to your application root.
+APP_ROOT = Pathname.new File.expand_path('../../',  __FILE__)
+
+Dir.chdir APP_ROOT do
+  # This script is a starting point to setup your application.
+  # Add necessary setup steps to this file:
+
+  puts "== Installing dependencies =="
+  system "gem install bundler --conservative"
+  system "bundle check || bundle install"
+
+  # puts "\n== Copying sample files =="
+  # unless File.exist?("config/database.yml")
+  #   system "cp config/database.yml.sample config/database.yml"
+  # end
+
+  puts "\n== Preparing database =="
+  system "bin/rake db:setup"
+
+  puts "\n== Removing old logs and tempfiles =="
+  system "rm -f log/*"
+  system "rm -rf tmp/cache"
+
+  puts "\n== Restarting application server =="
+  system "touch tmp/restart.txt"
+end

data/spec/dummy/config/application.rb

@@ -0,0 +1,32 @@
+require File.expand_path('../boot', __FILE__)
+
+# Pick the frameworks you want:
+require "active_record/railtie"
+require "action_controller/railtie"
+require "action_mailer/railtie"
+require "action_view/railtie"
+require "sprockets/railtie"
+# require "rails/test_unit/railtie"
+
+Bundler.require(*Rails.groups)
+require "pg_partitioning"
+
+module Dummy
+  class Application < Rails::Application
+    # Settings in config/environments/* take precedence over those specified here.
+    # Application configuration should go into files in config/initializers
+    # -- all .rb files in that directory are automatically loaded.
+
+    # Set Time.zone default to the specified zone and make Active Record auto-convert to this zone.
+    # Run "rake -D time" for a list of tasks for finding time zone names. Default is UTC.
+    # config.time_zone = 'Central Time (US & Canada)'
+
+    # The default locale is :en and all translations from config/locales/*.rb,yml are auto loaded.
+    # config.i18n.load_path += Dir[Rails.root.join('my', 'locales', '*.{rb,yml}').to_s]
+    # config.i18n.default_locale = :de
+
+    # Do not swallow errors in after_commit/after_rollback callbacks.
+    config.active_record.raise_in_transactional_callbacks = true
+  end
+end
+

data/spec/dummy/config/boot.rb

@@ -0,0 +1,5 @@
+# Set up gems listed in the Gemfile.
+ENV['BUNDLE_GEMFILE'] ||= File.expand_path('../../../../Gemfile', __FILE__)
+
+require 'bundler/setup' if File.exist?(ENV['BUNDLE_GEMFILE'])
+$LOAD_PATH.unshift File.expand_path('../../../../lib', __FILE__)

data/spec/dummy/config/database.yml

@@ -0,0 +1,28 @@
+# SQLite version 3.x
+#   gem install sqlite3
+#
+#   Ensure the SQLite 3 gem is defined in your Gemfile
+#   gem 'sqlite3'
+#
+default: &default
+  adapter: postgresql
+  uncoding: unicode
+  pool: 5
+  timeout: 5000
+  username: victor
+  password: mAGArLAm0v 
+
+development:
+  <<: *default
+  database: dev_pg
+
+# Warning: The database defined as "test" will be erased and
+# re-generated from your development database when you run "rake".
+# Do not set this db to the same as development or production.
+test:
+  <<: *default
+  database: test_pg
+
+production:
+  <<: *default
+  database: prod_pg