xbar 0.0.1 → 0.4.0

data/gemfiles/rails3.gemfile

@@ -0,0 +1,8 @@
+# This file was generated by Appraisal
+
+source "http://rubygems.org"
+
+gem "activerecord", "~> 3.0.11"
+gem "actionpack", "~> 3.0.11"
+
+gemspec :path=>"../"

data/gemfiles/rails3.gemfile.lock

@@ -0,0 +1,74 @@
+PATH
+  remote: /Volumes/Opt/projects/xbar
+  specs:
+    xbar (0.4.0)
+      activerecord
+      mysql2
+
+GEM
+  remote: http://rubygems.org/
+  specs:
+    abstract (1.0.0)
+    actionpack (3.0.11)
+      activemodel (= 3.0.11)
+      activesupport (= 3.0.11)
+      builder (~> 2.1.2)
+      erubis (~> 2.6.6)
+      i18n (~> 0.5.0)
+      rack (~> 1.2.1)
+      rack-mount (~> 0.6.14)
+      rack-test (~> 0.5.7)
+      tzinfo (~> 0.3.23)
+    activemodel (3.0.11)
+      activesupport (= 3.0.11)
+      builder (~> 2.1.2)
+      i18n (~> 0.5.0)
+    activerecord (3.0.11)
+      activemodel (= 3.0.11)
+      activesupport (= 3.0.11)
+      arel (~> 2.0.10)
+      tzinfo (~> 0.3.23)
+    activesupport (3.0.11)
+    appraisal (0.4.0)
+      bundler
+      rake
+    arel (2.0.10)
+    builder (2.1.2)
+    diff-lcs (1.1.3)
+    erubis (2.6.6)
+      abstract (>= 1.0.0)
+    i18n (0.5.0)
+    mysql2 (0.3.11)
+    pg (0.12.2)
+    rack (1.2.5)
+    rack-mount (0.6.14)
+      rack (>= 1.0.0)
+    rack-test (0.5.7)
+      rack (>= 1.0)
+    rake (0.9.2.2)
+    rspec (2.8.0)
+      rspec-core (~> 2.8.0)
+      rspec-expectations (~> 2.8.0)
+      rspec-mocks (~> 2.8.0)
+    rspec-core (2.8.0)
+    rspec-expectations (2.8.0)
+      diff-lcs (~> 1.1.2)
+    rspec-mocks (2.8.0)
+    sqlite3 (1.3.5)
+    syntax (1.0.0)
+    tzinfo (0.3.31)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  actionpack (~> 3.0.11)
+  activerecord (~> 3.0.11)
+  appraisal
+  mysql2
+  pg
+  rake
+  rspec (= 2.8.0)
+  sqlite3
+  syntax
+  xbar!

data/gemfiles/rails31.gemfile

@@ -0,0 +1,8 @@
+# This file was generated by Appraisal
+
+source "http://rubygems.org"
+
+gem "activerecord", "~> 3.1.1"
+gem "actionpack", "~> 3.1.1"
+
+gemspec :path=>"../"

data/gemfiles/rails31.gemfile.lock

@@ -0,0 +1,83 @@
+PATH
+  remote: /Volumes/Opt/projects/xbar
+  specs:
+    xbar (0.4.0)
+      activerecord
+      mysql2
+
+GEM
+  remote: http://rubygems.org/
+  specs:
+    actionpack (3.1.3)
+      activemodel (= 3.1.3)
+      activesupport (= 3.1.3)
+      builder (~> 3.0.0)
+      erubis (~> 2.7.0)
+      i18n (~> 0.6)
+      rack (~> 1.3.5)
+      rack-cache (~> 1.1)
+      rack-mount (~> 0.8.2)
+      rack-test (~> 0.6.1)
+      sprockets (~> 2.0.3)
+    activemodel (3.1.3)
+      activesupport (= 3.1.3)
+      builder (~> 3.0.0)
+      i18n (~> 0.6)
+    activerecord (3.1.3)
+      activemodel (= 3.1.3)
+      activesupport (= 3.1.3)
+      arel (~> 2.2.1)
+      tzinfo (~> 0.3.29)
+    activesupport (3.1.3)
+      multi_json (~> 1.0)
+    appraisal (0.4.0)
+      bundler
+      rake
+    arel (2.2.1)
+    builder (3.0.0)
+    diff-lcs (1.1.3)
+    erubis (2.7.0)
+    hike (1.2.1)
+    i18n (0.6.0)
+    multi_json (1.0.4)
+    mysql2 (0.3.11)
+    pg (0.12.2)
+    rack (1.3.6)
+    rack-cache (1.1)
+      rack (>= 0.4)
+    rack-mount (0.8.3)
+      rack (>= 1.0.0)
+    rack-test (0.6.1)
+      rack (>= 1.0)
+    rake (0.9.2.2)
+    rspec (2.8.0)
+      rspec-core (~> 2.8.0)
+      rspec-expectations (~> 2.8.0)
+      rspec-mocks (~> 2.8.0)
+    rspec-core (2.8.0)
+    rspec-expectations (2.8.0)
+      diff-lcs (~> 1.1.2)
+    rspec-mocks (2.8.0)
+    sprockets (2.0.3)
+      hike (~> 1.2)
+      rack (~> 1.0)
+      tilt (~> 1.1, != 1.3.0)
+    sqlite3 (1.3.5)
+    syntax (1.0.0)
+    tilt (1.3.3)
+    tzinfo (0.3.31)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  actionpack (~> 3.1.1)
+  activerecord (~> 3.1.1)
+  appraisal
+  mysql2
+  pg
+  rake
+  rspec (= 2.8.0)
+  sqlite3
+  syntax
+  xbar!

data/gemfiles/rails32.gemfile

@@ -0,0 +1,7 @@
+# This file was generated by Appraisal
+
+source "http://rubygems.org"
+
+gem "rails", "3.2.1"
+
+gemspec :path=>"../"

data/gemfiles/rails32.gemfile.lock

@@ -0,0 +1,117 @@
+PATH
+  remote: /opt/projects/xbar
+  specs:
+    xbar (0.4.0)
+      activerecord
+      mysql2
+
+GEM
+  remote: http://rubygems.org/
+  specs:
+    actionmailer (3.2.1)
+      actionpack (= 3.2.1)
+      mail (~> 2.4.0)
+    actionpack (3.2.1)
+      activemodel (= 3.2.1)
+      activesupport (= 3.2.1)
+      builder (~> 3.0.0)
+      erubis (~> 2.7.0)
+      journey (~> 1.0.1)
+      rack (~> 1.4.0)
+      rack-cache (~> 1.1)
+      rack-test (~> 0.6.1)
+      sprockets (~> 2.1.2)
+    activemodel (3.2.1)
+      activesupport (= 3.2.1)
+      builder (~> 3.0.0)
+    activerecord (3.2.1)
+      activemodel (= 3.2.1)
+      activesupport (= 3.2.1)
+      arel (~> 3.0.0)
+      tzinfo (~> 0.3.29)
+    activeresource (3.2.1)
+      activemodel (= 3.2.1)
+      activesupport (= 3.2.1)
+    activesupport (3.2.1)
+      i18n (~> 0.6)
+      multi_json (~> 1.0)
+    appraisal (0.4.0)
+      bundler
+      rake
+    arel (3.0.0)
+    builder (3.0.0)
+    diff-lcs (1.1.3)
+    erubis (2.7.0)
+    hike (1.2.1)
+    i18n (0.6.0)
+    journey (1.0.1)
+    json (1.6.5)
+    mail (2.4.1)
+      i18n (>= 0.4.0)
+      mime-types (~> 1.16)
+      treetop (~> 1.4.8)
+    mime-types (1.17.2)
+    multi_json (1.0.4)
+    mysql2 (0.3.11)
+    pg (0.12.2)
+    polyglot (0.3.3)
+    rack (1.4.1)
+    rack-cache (1.1)
+      rack (>= 0.4)
+    rack-ssl (1.3.2)
+      rack
+    rack-test (0.6.1)
+      rack (>= 1.0)
+    rails (3.2.1)
+      actionmailer (= 3.2.1)
+      actionpack (= 3.2.1)
+      activerecord (= 3.2.1)
+      activeresource (= 3.2.1)
+      activesupport (= 3.2.1)
+      bundler (~> 1.0)
+      railties (= 3.2.1)
+    railties (3.2.1)
+      actionpack (= 3.2.1)
+      activesupport (= 3.2.1)
+      rack-ssl (~> 1.3.2)
+      rake (>= 0.8.7)
+      rdoc (~> 3.4)
+      thor (~> 0.14.6)
+    rake (0.9.2.2)
+    rdoc (3.12)
+      json (~> 1.4)
+    rspec (2.8.0)
+      rspec-core (~> 2.8.0)
+      rspec-expectations (~> 2.8.0)
+      rspec-mocks (~> 2.8.0)
+    rspec-core (2.8.0)
+    rspec-expectations (2.8.0)
+      diff-lcs (~> 1.1.2)
+    rspec-mocks (2.8.0)
+    sprockets (2.1.2)
+      hike (~> 1.2)
+      rack (~> 1.0)
+      tilt (~> 1.1, != 1.3.0)
+    sqlite3 (1.3.5)
+    syntax (1.0.0)
+    thor (0.14.6)
+    tilt (1.3.3)
+    treetop (1.4.10)
+      polyglot
+      polyglot (>= 0.3.1)
+    tzinfo (0.3.31)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  actionpack
+  appraisal
+  mysql2
+  pg
+  rails (= 3.2.1)
+  rake
+  rspec (= 2.8.0)
+  sqlite3
+  syntax
+  xbar!

data/gemfiles/rails4.gemfile

@@ -0,0 +1,9 @@
+# This file was generated by Appraisal
+
+source "http://rubygems.org"
+
+gem "rails", :path=>"/opt/rails/rails-edge"
+gem "journey", :git=>"https://github.com/rails/journey.git"
+gem "arel", :git=>"https://github.com/rails/arel.git"
+
+gemspec :path=>"../"

data/gemfiles/rails4.gemfile.lock

@@ -0,0 +1,134 @@
+GIT
+  remote: https://github.com/rails/arel.git
+  revision: fc787a42c0febcee41e13d16e06b72492ab00468
+  specs:
+    arel (3.0.0.20120112113618)
+
+GIT
+  remote: https://github.com/rails/journey.git
+  revision: 7059e60070861a9c364a50954ee4abb355cc4541
+  specs:
+    journey (1.0.1.20120126102309)
+
+PATH
+  remote: /Volumes/Opt/projects/xbar
+  specs:
+    xbar (0.4.0)
+      activerecord
+      mysql2
+
+PATH
+  remote: /opt/rails/rails-edge
+  specs:
+    actionmailer (4.0.0.beta)
+      actionpack (= 4.0.0.beta)
+      mail (~> 2.4.1)
+    actionpack (4.0.0.beta)
+      activemodel (= 4.0.0.beta)
+      activesupport (= 4.0.0.beta)
+      builder (~> 3.0.0)
+      erubis (~> 2.7.0)
+      journey (~> 1.0.1)
+      rack (~> 1.4.1)
+      rack-cache (~> 1.1)
+      rack-test (~> 0.6.1)
+      sprockets (~> 2.2.0)
+    activemodel (4.0.0.beta)
+      activesupport (= 4.0.0.beta)
+      builder (~> 3.0.0)
+    activerecord (4.0.0.beta)
+      activemodel (= 4.0.0.beta)
+      activesupport (= 4.0.0.beta)
+      arel (~> 3.0.0)
+      tzinfo (~> 0.3.29)
+    activeresource (4.0.0.beta)
+      activemodel (= 4.0.0.beta)
+      activesupport (= 4.0.0.beta)
+    activesupport (4.0.0.beta)
+      i18n (~> 0.6)
+      multi_json (~> 1.0)
+    rails (4.0.0.beta)
+      actionmailer (= 4.0.0.beta)
+      actionpack (= 4.0.0.beta)
+      activerecord (= 4.0.0.beta)
+      activeresource (= 4.0.0.beta)
+      activesupport (= 4.0.0.beta)
+      bundler (~> 1.0)
+      railties (= 4.0.0.beta)
+    railties (4.0.0.beta)
+      actionpack (= 4.0.0.beta)
+      activesupport (= 4.0.0.beta)
+      rack-ssl (~> 1.3.2)
+      rake (>= 0.8.7)
+      rdoc (~> 3.4)
+      thor (~> 0.14.6)
+
+GEM
+  remote: http://rubygems.org/
+  specs:
+    appraisal (0.4.0)
+      bundler
+      rake
+    builder (3.0.0)
+    diff-lcs (1.1.3)
+    erubis (2.7.0)
+    hike (1.2.1)
+    i18n (0.6.0)
+    json (1.6.5)
+    mail (2.4.1)
+      i18n (>= 0.4.0)
+      mime-types (~> 1.16)
+      treetop (~> 1.4.8)
+    mime-types (1.17.2)
+    multi_json (1.0.4)
+    mysql2 (0.3.11)
+    pg (0.12.2)
+    polyglot (0.3.3)
+    rack (1.4.1)
+    rack-cache (1.1)
+      rack (>= 0.4)
+    rack-ssl (1.3.2)
+      rack
+    rack-test (0.6.1)
+      rack (>= 1.0)
+    rake (0.9.2.2)
+    rdoc (3.12)
+      json (~> 1.4)
+    rspec (2.8.0)
+      rspec-core (~> 2.8.0)
+      rspec-expectations (~> 2.8.0)
+      rspec-mocks (~> 2.8.0)
+    rspec-core (2.8.0)
+    rspec-expectations (2.8.0)
+      diff-lcs (~> 1.1.2)
+    rspec-mocks (2.8.0)
+    sprockets (2.2.0)
+      hike (~> 1.2)
+      multi_json (~> 1.0)
+      rack (~> 1.0)
+      tilt (~> 1.1, != 1.3.0)
+    sqlite3 (1.3.5)
+    syntax (1.0.0)
+    thor (0.14.6)
+    tilt (1.3.3)
+    treetop (1.4.10)
+      polyglot
+      polyglot (>= 0.3.1)
+    tzinfo (0.3.31)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  actionpack
+  appraisal
+  arel!
+  journey!
+  mysql2
+  pg
+  rails!
+  rake
+  rspec (= 2.8.0)
+  sqlite3
+  syntax
+  xbar!

data/lib/migrations/1_create_usage_statistics.rb

@@ -0,0 +1,23 @@
+class CreateUsageStatistics < ActiveRecord::Migration
+
+  def up
+    create_table(:usage_statistics) do |t|
+      t.string :shard_name
+      t.string :method
+      t.string :adapter
+      t.string :username
+      t.string :thread_id
+      t.integer :port
+      t.string :host
+      t.string :database_name
+    end
+  end
+
+  def down
+    puts "Dropping table..."
+    drop_table(:usage_statistics)
+    puts "Done dropping table."
+  end
+
+
+end

data/lib/xbar/association.rb

@@ -0,0 +1,49 @@
+require 'active_support/concern'
+
+module XBar::Association
+  
+  extend ActiveSupport::Concern
+  
+  module ClassMethods
+    def has_many(association_id, options = {}, &extension)
+      default_xbar_opts(options)
+      super
+    end
+
+    def has_and_belongs_to_many(association_id, options = {}, &extension)
+      default_xbar_opts(options)
+      super
+    end
+
+    def default_xbar_opts(options)
+      if options[:before_add].is_a?(Array)
+        options[:before_add] << :set_connection_on_association
+      elsif options[:before_add].is_a?(Symbol)
+        options[:before_add] = [:set_connection_on_association, options[:before_add]]
+      else
+        options[:before_add] = :set_connection_on_association
+      end
+
+      if options[:before_remove].is_a?(Array)
+        options[:before_remove] << :set_connection_on_association
+      elsif options[:before_remove].is_a?(Symbol)
+        options[:before_remove] = [:set_connection_on_association, options[:before_remove]]
+      else
+        options[:before_remove] = :set_connection_on_association
+      end
+    end
+  end
+
+  private
+  
+  def set_connection_on_association(record)
+    return if !self.connection.respond_to?(:current_shard) || !self.respond_to?(:current_shard)
+    if !record.current_shard.nil? && !self.current_shard.nil? && record.current_shard != self.current_shard
+      raise "Association Error: Records are from different shards"
+    end
+    record.current_shard = self.connection.current_shard = self.current_shard if should_set_current_shard?
+  end
+ 
+end
+
+ActiveRecord::Base.send(:include, XBar::Association)

data/lib/xbar/association_collection.rb

@@ -0,0 +1,69 @@
+module XBar::AssociationCollection
+
+  def self.included(base)
+    if XBar.rails31?
+      base.instance_eval do
+        alias_method_chain :reader, :xbar
+        alias_method_chain :writer, :xbar
+        alias_method_chain :ids_reader, :xbar
+        alias_method_chain :ids_writer, :xbar
+        alias_method_chain :create, :xbar
+        alias_method_chain :create!, :xbar
+        alias_method_chain :build, :xbar
+      end
+    end
+  end
+
+  def build_with_xbar(*args)
+    owner.reload_connection
+    build_without_xbar(*args)
+  end
+
+  def reader_with_xbar(*args)
+    owner.reload_connection
+    reader_without_xbar(*args)
+  end
+
+  def writer_with_xbar(*args)
+    owner.reload_connection
+    writer_without_xbar(*args)
+  end
+
+  def ids_reader_with_xbar(*args)
+    owner.reload_connection
+    ids_reader_without_xbar(*args)
+  end
+
+  def ids_writer_with_xbar(*args)
+    owner.reload_connection
+    ids_writer_without_xbar(*args)
+  end
+
+  def create_with_xbar(*args)
+    owner.reload_connection
+    create_without_xbar(*args)
+  end
+
+  def create_with_xbar!(*args)
+    owner.reload_connection
+    create_without_xbar!(*args)
+  end
+
+  def should_wrap_the_connection?
+    @owner.respond_to?(:current_shard) && @owner.current_shard != nil
+  end
+
+  def count(*args)
+    if should_wrap_the_connection?
+      XBar.using(@owner.current_shard) { super }
+    else
+      super
+    end
+  end
+end
+
+if XBar.rails31?
+  ActiveRecord::Associations::CollectionAssociation.send(:include, XBar::AssociationCollection)
+else
+  ActiveRecord::Associations::AssociationCollection.send(:include, XBar::AssociationCollection)
+end

data/lib/xbar/colors.rb

@@ -0,0 +1,32 @@
+module Colors
+
+  # foreground color
+  BLACK_TEXT =   "\033[30m"
+  RED_TEXT =     "\033[31m"
+  GREEN_TEXT =   "\033[32m"
+  BROWN_TEXT =   "\033[33m"
+  BLUE_TEXT =    "\033[34m"
+  MAGENTA_TEXT = "\033[35m"
+  CYAN_TEXT =    "\033[36m"
+  GRAY_TEXT =    "\033[37m"
+
+  # background color
+  BLACK_BACK =   "\033[40m"
+  RED_BACK =     "\033[41m"
+  GREEN_BACK =   "\033[42m"
+  BROWN_BACK =   "\033[43m"
+  BLUE_BACK =    "\033[44m"
+  MAGENTA_BACK = "\033[45m"
+  CYAN_BACK =    "\033[46m"
+  WHITE_BACK =   "\033[47m"
+
+  # ANSI control chars
+  RESET_COLORS =  "\033[0m"
+  BOLD_ON =      "\033[1m"
+  BLINK_ON =     "\033[5m"
+  REVERSE_ON =   "\033[7m"
+  BOLD_OFF =     "\033[22m"
+  BLINK_OFF =    "\033[25m"
+  REVERSE_OFF =  "\033[27m"
+
+end

data/lib/xbar/has_and_belongs_to_many_association.rb

@@ -0,0 +1,17 @@
+module XBar::HasAndBelongsToManyAssociation
+  def self.included(base)
+    base.instance_eval do
+      alias_method_chain :insert_record, :xbar
+    end
+  end
+
+  def insert_record_with_xbar(record, force = true, validate = true)
+    if should_wrap_the_connection?
+      XBar.using(@owner.current_shard) { insert_record_without_xbar(record, force, validate) }
+    else
+      insert_record_without_xbar(record, force, validate)
+    end
+  end
+end
+
+ActiveRecord::Associations::HasAndBelongsToManyAssociation.send(:include, XBar::HasAndBelongsToManyAssociation)

data/lib/xbar/logger.rb

@@ -0,0 +1,14 @@
+require "logger"
+
+class XBar::Logger < Logger
+  def format_message(severity, timestamp, progname, msg)
+    str = super
+
+    if ActiveRecord::Base.connection.respond_to?(:current_shard)
+      str += "Shard: #{ActiveRecord::Base.connection.current_shard} -"
+    end
+
+    str
+  end
+end
+