Contents

module DbCharmer
  module Sharding
    module ClassMethods
      def self.extended(model)
        model.cattr_accessor(:sharded_connection)
      end

      def shard_for(key, proxy_target = nil, &block)
        raise ArgumentError, "No sharded connection configured!" unless sharded_connection
        conn = sharded_connection.sharder.shard_for_key(key)
        on_db(conn, proxy_target, &block)
      end

      # Run on default shard (if supported by the sharding method)
      def on_default_shard(proxy_target = nil, &block)
        raise ArgumentError, "No sharded connection configured!" unless sharded_connection

        if sharded_connection.support_default_shard?
          shard_for(:default, proxy_target, &block)
        else
          raise ArgumentError, "This model's sharding method does not support default shard"
        end
      end

      # Enumerate shards
      def on_each_shard(proxy_target = nil, &block)
        raise ArgumentError, "No sharded connection configured!" unless sharded_connection

        conns = sharded_connection.shard_connections
        raise ArgumentError, "This model's sharding method does not support shards enumeration" unless conns

        conns.each do |conn|
          on_db(conn, proxy_target, &block)
        end
      end
    end

    #-------------------------------------------------------------
    @@sharded_connections = {}

    def self.register_connection(config)
      name = config[:name] or raise ArgumentError, "No :name in connection!"
      @@sharded_connections[name] = DbCharmer::Sharding::Connection.new(config)
    end

    def self.sharded_connection(name)
      @@sharded_connections[name] or raise ArgumentError, "Invalid sharded connection name!"
    end
  end
end

Version data entries

20 entries across 20 versions & 1 rubygems

Version	Path
db-charmer-1.6.19	lib/db_charmer/sharding.rb
db-charmer-1.6.18	lib/db_charmer/sharding.rb
db-charmer-1.6.17	lib/db_charmer/sharding.rb
db-charmer-1.6.16	lib/db_charmer/sharding.rb
db-charmer-1.6.15	lib/db_charmer/sharding.rb
db-charmer-1.6.14	lib/db_charmer/sharding.rb
db-charmer-1.6.13	lib/db_charmer/sharding.rb
db-charmer-1.6.12	lib/db_charmer/sharding.rb
db-charmer-1.6.11	lib/db_charmer/sharding.rb
db-charmer-1.6.10	lib/db_charmer/sharding.rb
db-charmer-1.6.9	lib/db_charmer/sharding.rb
db-charmer-1.6.8	lib/db_charmer/sharding.rb
db-charmer-1.6.7	lib/db_charmer/sharding.rb
db-charmer-1.6.6	lib/db_charmer/sharding.rb
db-charmer-1.6.5	lib/db_charmer/sharding.rb
db-charmer-1.6.4	lib/db_charmer/sharding.rb
db-charmer-1.6.3	lib/db_charmer/sharding.rb
db-charmer-1.6.2	lib/db_charmer/sharding.rb
db-charmer-1.6.1	lib/db_charmer/sharding.rb
db-charmer-1.6.0	lib/db_charmer/sharding.rb