arproxy 0.1.0

data/lib/arproxy/base.rb

@@ -0,0 +1,9 @@
+module Arproxy
+  class Base
+    attr_accessor :proxy_chain, :next_proxy
+
+    def execute(sql, name=nil)
+      next_proxy.execute sql, name
+    end
+  end
+end

data/lib/arproxy/chain_tail.rb

@@ -0,0 +1,11 @@
+module Arproxy
+  class ChainTail < Base
+    def initialize(proxy_chain)
+      self.proxy_chain = proxy_chain
+    end
+
+    def execute(sql, name=nil)
+      self.proxy_chain.connection.execute_without_arproxy sql, name
+    end
+  end
+end

data/lib/arproxy/config.rb

@@ -0,0 +1,21 @@
+module Arproxy
+  class Config
+    attr_accessor :adapter, :logger
+    attr_reader :proxies
+
+    def initialize
+      @proxies = []
+    end
+
+    def use(proxy_class, *options)
+      ::Arproxy.logger.debug("Arproxy: Mounting #{proxy_class.inspect} (#{options.inspect})")
+      @proxies << [proxy_class, options]
+    end
+
+    def adapter_class
+      raise Arproxy::Error, "config.adapter must be set" unless @adapter
+      camelized_adapter_name = @adapter.split("_").map(&:capitalize).join
+      eval "::ActiveRecord::ConnectionAdapters::#{camelized_adapter_name}Adapter"
+    end
+  end
+end

data/lib/arproxy/error.rb

@@ -0,0 +1,4 @@
+module Arproxy
+  class Error < Exception
+  end
+end

data/lib/arproxy/proxy_chain.rb

@@ -0,0 +1,26 @@
+module Arproxy
+  autoload :ChainTail, "arproxy/chain_tail"
+
+  class ProxyChain
+    attr_reader :head, :tail
+    attr_accessor :connection
+
+    def initialize(config)
+      @config = config
+      setup_proxy_chain(@config)
+    end
+
+    def setup_proxy_chain(config)
+      @tail = ChainTail.new self
+      @head = config.proxies.reverse.inject(@tail) do |next_proxy, proxy_config|
+        cls, options = proxy_config
+        proxy = cls.new *options
+        proxy.proxy_chain = self
+        proxy.next_proxy = next_proxy
+        proxy
+      end
+    end
+    private :setup_proxy_chain
+
+  end
+end

data/lib/arproxy.rb

@@ -0,0 +1,51 @@
+require "logger"
+
+module Arproxy
+  autoload :Config, "arproxy/config"
+  autoload :ProxyChain, "arproxy/proxy_chain"
+  autoload :Error, "arproxy/error"
+  autoload :Base, "arproxy/base"
+
+  module_function
+  def configure
+    config = Config.new
+    yield config
+    @config = config
+  end
+
+  def enable!
+    @proxy_chain = ProxyChain.new @config
+
+    @config.adapter_class.class_eval do
+      def execute_with_arproxy(sql, name=nil)
+        ::Arproxy.proxy_chain.connection = self
+        ::Arproxy.proxy_chain.head.execute sql, name
+      end
+      alias_method :execute_without_arproxy, :execute
+      alias_method :execute, :execute_with_arproxy
+      ::Arproxy.logger.debug("Arproxy: Enabled")
+    end
+  end
+
+  def disable!
+    @config.adapter_class.class_eval do
+      alias_method :execute, :execute_without_arproxy
+      ::Arproxy.logger.debug("Arproxy: Disabled")
+    end
+    @proxy_chain = nil
+    @config = nil
+  end
+
+  def logger
+    @logger ||= begin
+                  @config && @config.logger ||
+                    defined?(::Rails) && ::Rails.logger ||
+                    ::Logger.new(STDOUT)
+                end
+  end
+
+  def proxy_chain
+    @proxy_chain
+  end
+end
+

metadata

@@ -0,0 +1,70 @@
+--- !ruby/object:Gem::Specification 
+name: arproxy
+version: !ruby/object:Gem::Version 
+  hash: 27
+  prerelease: 
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- Issei Naruta
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2012-04-29 00:00:00 Z
+dependencies: []
+
+description: Arproxy is a proxy between ActiveRecord and database adapter
+email: naruta@cookpad.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/arproxy/base.rb
+- lib/arproxy/chain_tail.rb
+- lib/arproxy/config.rb
+- lib/arproxy/error.rb
+- lib/arproxy/proxy_chain.rb
+- lib/arproxy.rb
+homepage: https://github.com/mirakui/arproxy
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.8.6
+signing_key: 
+specification_version: 3
+summary: Proxy between ActiveRecord and DB adapter
+test_files: []
+