xbar 0.0.1 → 0.4.0

data/spec/config/production_raise_error.json

@@ -0,0 +1,17 @@
+{
+	"__COMMENT": "Production Raise Error",
+	
+	"connections": {
+    "duplicated_shard_name": {"database": "xbar_shard5", "adapter": "mysql2", "host": "localhost"},
+    "duplicated_shard_name": {"database": "xbar_shard4", "adapter": "mysql2", "host": "localhost"}
+	},
+		
+	"environments": {
+    "shards": {
+      "shards": {
+        "history_shard": "duplicated_shard_name",
+        "country_shard": "duplicated_shard_name"
+			}
+    }
+  }
+}

data/spec/config/simple.json

@@ -0,0 +1,22 @@
+{
+    "__COMMENT": "Simple SQLite French Environment",
+
+    "connections": {
+        "paris_m": { "adapter": "sqlite3", "database": "/tmp/paris.sqlite3"},
+	"france_1": { "adapter": "sqlite3", "database": "/tmp/france_1.sqlite3"},
+	"france_2": { "adapter": "sqlite3", "database": "/tmp/france_2.sqlite3"},
+	"france_3": { "adapter": "sqlite3", "database": "/tmp/france_3.sqlite3"}
+    },
+
+    "environments": {
+        "test": {
+            "shards": {
+                "master": "paris_m",
+                "france": ["france_1", "france_2", "france_3"],
+                "france_nord": "france_1",
+		"france_central": "france_2",
+                "france_sud": "france_3"
+            }
+        }
+    }
+}

data/spec/config/single_adapter.json

@@ -0,0 +1,20 @@
+{
+    "__COMMENT": "Acme Corporation Environments with a single adapter used in test environment",
+    "__COMMENT": "Although four connections are defined, only two adapters will be installed",
+
+    "connections": {
+        "mysql_m": "mysql2://root@localhost:3306/master",
+        "inventory_1": "postgres://postgres@127.0.0.1/russia_1",      
+        "inventory_2": "postgres://postgres@127.0.0.1/russia_2",
+        "sales_1": "mysql2://root:graceling@deimos.thirdmode.com:3307/canada"   
+    },
+
+    "environments": {
+        "test": {
+            "shards": {
+                "master": "inventory_1",
+                "sales": "inventory_2"
+            }
+        }
+    }
+}

data/spec/console.rb

@@ -0,0 +1,15 @@
+# From XBar root directory, 
+# irb -I spec
+# require 'console'
+
+require 'rspec'
+require 'spec_helper'
+require 'support/database_models'
+
+module XBar
+  def self.directory
+    File.expand_path(File.dirname(__FILE__))
+  end
+end
+
+# XBar::Mapper.reset

data/spec/migrations/10_create_users_using_replication.rb

@@ -0,0 +1,12 @@
+class CreateUsersUsingReplication < ActiveRecord::Migration
+  using :china
+       
+  def self.up
+    Cat.create!(:name => "Replication")
+  end
+
+  def self.down
+    Cat.delete_all
+  end
+  
+end

data/spec/migrations/11_add_field_in_all_slaves.rb

@@ -0,0 +1,11 @@
+class AddFieldInAllSlaves < ActiveRecord::Migration
+  using(:slave1, :slave2, :slave3, :slave4)
+
+  def self.up
+    Cat.create!(:name => "Slaves")
+  end
+
+  def self.down
+    Cat.delete_all
+  end
+end

data/spec/migrations/12_create_users_using_block.rb

@@ -0,0 +1,23 @@
+class CreateUsersUsingBlock < ActiveRecord::Migration
+  def self.up
+    XBar.using(:brazil) do
+      User.create!(:name => "UsingBlock1")
+      User.create!(:name => "UsingBlock2")
+    end
+
+    XBar.using(:canada) do
+      User.create!(:name => "UsingCanada")
+      User.create!(:name => "UsingCanada2")
+    end
+  end
+
+  def self.down
+    XBar.using(:brazil) do
+      User.delete_all
+    end
+
+    XBar.using(:canada) do
+      User.delete_all
+    end
+  end
+end

data/spec/migrations/13_create_users_using_block_and_using.rb

@@ -0,0 +1,15 @@
+class CreateUsersUsingBlockAndUsing < ActiveRecord::Migration
+  using(:brazil)
+
+  def self.up
+    XBar.using(:canada) do
+      User.create!(:name => "Canada")
+    end
+
+    User.create!(:name => "Brazil")
+  end
+
+  def self.down
+    User.delete_all()
+  end
+end

data/spec/migrations/1_create_users_on_master.rb

@@ -0,0 +1,9 @@
+class CreateUsersOnMaster < ActiveRecord::Migration
+  def self.up
+    User.create!(:name => "Master")
+  end
+
+  def self.down
+    User.delete_all
+  end
+end

data/spec/migrations/2_create_users_on_canada.rb

@@ -0,0 +1,11 @@
+class CreateUsersOnCanada < ActiveRecord::Migration
+  using(:canada)
+
+  def self.up
+    User.create!(:name => "Sharding")
+  end
+
+  def self.down
+    User.delete_all
+  end
+end

data/spec/migrations/3_create_users_on_both_shards.rb

@@ -0,0 +1,11 @@
+class CreateUsersOnBothShards < ActiveRecord::Migration
+  using(:canada, :brazil)
+
+  def self.up
+    User.create!(:name => "Both")
+  end
+
+  def self.down
+    User.delete_all
+  end
+end

data/spec/migrations/4_create_users_on_shards_of_a_group.rb

@@ -0,0 +1,11 @@
+class CreateUsersOnShardsOfAGroup < ActiveRecord::Migration
+  history_shard(:country_shard)
+
+  def self.up
+    User.create!(:name => "Group")
+  end
+
+  def self.down
+    User.delete_all()
+  end
+end

data/spec/migrations/5_create_users_on_multiples_groups.rb

@@ -0,0 +1,11 @@
+class CreateUsersOnMultiplesGroups < ActiveRecord::Migration
+  history_shard('country_shard', 'history_shard')
+
+  def self.up
+    User.create!(:name => "MultipleGroup")
+  end
+
+  def self.down
+    User.delete_all()
+  end
+end

data/spec/migrations/6_raise_exception_with_invalid_shard_name.rb

@@ -0,0 +1,11 @@
+class RaiseExceptionWithInvalidShardName < ActiveRecord::Migration
+  using(:amazing_shard)
+
+  def self.up
+    User.create!(:name => "Error")
+  end
+
+  def self.down
+    User.delete_all()
+  end
+end

data/spec/migrations/7_raise_exception_with_invalid_multiple_shard_names.rb

@@ -0,0 +1,11 @@
+class RaiseExceptionWithInvalidMultipleShardNames < ActiveRecord::Migration
+  using(:brazil, :invalid_shard)
+
+  def self.up
+    User.create!(:name => "Error")
+  end
+
+  def self.down
+    User.delete_all()
+  end
+end

data/spec/migrations/8_raise_exception_with_invalid_group_name.rb

@@ -0,0 +1,11 @@
+class RaiseExceptionWithInvalidGroupName < ActiveRecord::Migration
+  history_shard(:invalid_group)
+
+  def self.up
+    User.create!(:name => "Error")
+  end
+
+  def self.down
+    User.delete_all()
+  end
+end

data/spec/migrations/9_raise_exception_with_multiple_invalid_group_names.rb

@@ -0,0 +1,11 @@
+class RaiseExceptionWithMultipleInvalidGroupNames < ActiveRecord::Migration
+  history_shard(:country_shard,:invalid_group)
+
+  def self.up
+    User.create!(:name => "Error")
+  end
+
+  def self.down
+    User.delete_all()
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,25 @@
+require "rubygems"
+require "bundler/setup"
+require "mysql2"
+require "active_record"
+require "action_controller"
+require "xbar"
+require "support/xbar_helper"
+require "support/database_models"
+
+MIGRATIONS_ROOT = File.expand_path(File.join(File.dirname(__FILE__),  'migrations'))
+
+XBar.directory = File.expand_path("../../spec", __FILE__)
+
+RSpec.configure do |config|
+  
+  config.before(:each) do
+    XBar.stub!(:directory).and_return(File.dirname(__FILE__))
+  end
+  
+  config.after(:each) do
+    clean_all_shards
+    XBar::Mapper.reset(xbar_env: 'default', app_env: 'test')
+  end
+  
+end

data/spec/support/database_models.rb

@@ -0,0 +1,78 @@
+# Rails 3.1 needs to do some introspection around the base class, which requires
+# the model be a descendent of ActiveRecord::Base.
+class BlankModel < ActiveRecord::Base; end;
+
+class User < ActiveRecord::Base
+  def awesome_queries
+    XBar.using(:canada) do
+      User.create(:name => "teste")
+    end
+  end
+end
+
+class Client < ActiveRecord::Base
+  unreplicated_model
+  has_many :items
+  has_many :comments, :as => :commentable
+end
+
+class Cat < ActiveRecord::Base
+  using :china
+end
+
+#This class sets its own connection
+class CustomConnection < ActiveRecord::Base
+  xbar_establish_connection(:adapter => "mysql2", :database => "rogue",
+  :username => "root", :password => "")
+end
+
+#This items belongs to a client
+class Item < ActiveRecord::Base
+  belongs_to :client
+  has_many :parts
+end
+
+class Part < ActiveRecord::Base
+  belongs_to :item
+end
+
+class Keyboard < ActiveRecord::Base
+  validates_uniqueness_of :name
+  belongs_to :computer
+end
+
+class Computer < ActiveRecord::Base
+  has_one :keyboard
+end
+
+class Role < ActiveRecord::Base
+  has_and_belongs_to_many :permissions
+end
+
+class Permission < ActiveRecord::Base
+  has_and_belongs_to_many :roles
+end
+
+class Assignment < ActiveRecord::Base
+  belongs_to :programmer
+  belongs_to :project
+end
+
+class Programmer < ActiveRecord::Base
+  has_many :assignments
+  has_many :projects, :through => :assignments
+end
+
+class Project < ActiveRecord::Base
+  has_many :assignments
+  has_many :programmers, :through => :assignments
+end
+
+class Comment < ActiveRecord::Base
+  belongs_to :commentable, :polymorphic => true
+end
+
+
+class Bacon < ActiveRecord::Base
+  xbar_set_table_name "yummy"
+end

data/spec/support/xbar_helper.rb

@@ -0,0 +1,42 @@
+def clean_all_shards
+  if proxy = Thread.current[:connection_proxy]
+    tables = ['schema_migrations', 'users', 'clients', 'cats', 'items', 'keyboards',
+      'computers', 'permissions_roles', 'roles', 'permissions', 'assignments',
+      'projects', 'programmers', "yummy"]
+    proxy.shards.keys.each do |key|
+      tables.each do |t|
+        BlankModel.using(key).connection.execute("DELETE FROM #{t}")
+      end
+    end
+  end
+end
+
+def clean_connection_proxy
+  Thread.current[:connection_proxy] = nil
+end
+
+def migrating_to_version(version, &block)
+  puts "migrating to version called version = #{version}"
+  ActiveRecord::Migrator.run(:up, MIGRATIONS_ROOT, version)
+  yield
+ensure
+  ActiveRecord::Migrator.run(:down, MIGRATIONS_ROOT, version)
+end
+
+def using_environment(environment, &block)
+  prev_env = XBar::Mapper.app_env
+  XBar::Mapper.reset(xbar_env: 'default', app_env: environment.to_s)
+  yield
+ensure
+  XBar::Mapper.reset(xbar_env: 'default', app_env: prev_env)
+end
+
+def set_xbar_env(xbar_env, app_env = nil)
+  opts = {xbar_env: xbar_env.to_s, :clear_cache => true}
+  if app_env && !defined?(Rails)
+    opts[:app_env] = app_env.to_s
+  end
+  XBar::Mapper.reset(opts)
+  # Not needed because reset should clean, I think.  
+  #Thread.current[:connection_proxy].clean_proxy if Thread.current[:connection_proxy]
+end