torquebox-core 2.3.2-java → 3.0.0.beta1-java

data/lib/torquebox/codecs/edn.rb

@@ -0,0 +1,36 @@
+# Copyright 2008-2013 Red Hat, Inc, and individual contributors.
+# 
+# This is free software; you can redistribute it and/or modify it
+# under the terms of the GNU Lesser General Public License as
+# published by the Free Software Foundation; either version 2.1 of
+# the License, or (at your option) any later version.
+# 
+# This software is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+# Lesser General Public License for more details.
+# 
+# You should have received a copy of the GNU Lesser General Public
+# License along with this software; if not, write to the Free
+# Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
+# 02110-1301 USA, or see the FSF site: http://www.fsf.org.
+
+require 'edn'
+
+module TorqueBox
+  module Codecs
+    module EDN
+      class << self
+
+        def encode(data)
+          data.to_edn
+        end
+
+        def decode(data)
+          ::EDN.read(data) unless data.nil?
+        end
+
+      end
+    end
+  end
+end

data/lib/torquebox/codecs/json.rb

@@ -0,0 +1,61 @@
+# Copyright 2008-2013 Red Hat, Inc, and individual contributors.
+# 
+# This is free software; you can redistribute it and/or modify it
+# under the terms of the GNU Lesser General Public License as
+# published by the Free Software Foundation; either version 2.1 of
+# the License, or (at your option) any later version.
+# 
+# This software is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+# Lesser General Public License for more details.
+# 
+# You should have received a copy of the GNU Lesser General Public
+# License along with this software; if not, write to the Free
+# Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
+# 02110-1301 USA, or see the FSF site: http://www.fsf.org.
+
+
+module TorqueBox
+  module Codecs
+    module JSON
+      class << self
+
+        # @api private
+        def require_json
+          # We can't ship our own json, as it may collide with the gem
+          # requirement for the app.
+          if !defined?( ::JSON )
+            begin
+              require 'json'
+            rescue LoadError => ex
+              raise RuntimeError.new( "Unable to load the json gem. Verify that is installed and in your Gemfile (if using Bundler)" )
+            end
+          end
+        end
+
+        def encode(data)
+          require_json
+          begin
+            if ( data.respond_to?( :as_json ) )
+              data = data.as_json
+            end
+            ::JSON.fast_generate( data ) unless data.nil?
+          rescue ::JSON::GeneratorError
+            ::JSON.dump(data)
+          end
+        end
+
+        def decode(data)
+          require_json
+          begin
+            ::JSON.parse( data, :symbolize_names => true ) unless data.nil?
+          rescue ::JSON::ParserError
+            ::JSON.load(data)
+          end
+        end
+
+      end
+    end
+  end
+end

data/lib/torquebox/codecs/marshal.rb

@@ -0,0 +1,35 @@
+# Copyright 2008-2013 Red Hat, Inc, and individual contributors.
+# 
+# This is free software; you can redistribute it and/or modify it
+# under the terms of the GNU Lesser General Public License as
+# published by the Free Software Foundation; either version 2.1 of
+# the License, or (at your option) any later version.
+# 
+# This software is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+# Lesser General Public License for more details.
+# 
+# You should have received a copy of the GNU Lesser General Public
+# License along with this software; if not, write to the Free
+# Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
+# 02110-1301 USA, or see the FSF site: http://www.fsf.org.
+
+
+module TorqueBox
+  module Codecs
+    module Marshal
+      class << self
+
+        def encode(data)
+          ::Marshal.dump(data) unless data.nil?
+        end
+        
+        def decode(data)
+          ::Marshal.restore(data) unless data.nil?
+        end
+
+      end
+    end
+  end
+end

data/lib/torquebox/codecs/marshal_base64.rb

@@ -0,0 +1,36 @@
+# Copyright 2008-2013 Red Hat, Inc, and individual contributors.
+# 
+# This is free software; you can redistribute it and/or modify it
+# under the terms of the GNU Lesser General Public License as
+# published by the Free Software Foundation; either version 2.1 of
+# the License, or (at your option) any later version.
+# 
+# This software is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+# Lesser General Public License for more details.
+# 
+# You should have received a copy of the GNU Lesser General Public
+# License along with this software; if not, write to the Free
+# Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
+# 02110-1301 USA, or see the FSF site: http://www.fsf.org.
+
+require 'base64'
+
+module TorqueBox
+  module Codecs
+    module MarshalBase64
+      class << self
+
+        def encode(data)
+          Base64.encode64(::Marshal.dump(data)) unless data.nil?
+        end
+
+        def decode(data)
+          ::Marshal.restore(Base64.decode64(data)) unless data.nil?
+        end
+        
+      end
+    end
+  end
+end

data/lib/torquebox/codecs/marshal_smart.rb

@@ -0,0 +1,51 @@
+# Copyright 2008-2013 Red Hat, Inc, and individual contributors.
+# 
+# This is free software; you can redistribute it and/or modify it
+# under the terms of the GNU Lesser General Public License as
+# published by the Free Software Foundation; either version 2.1 of
+# the License, or (at your option) any later version.
+# 
+# This software is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+# Lesser General Public License for more details.
+# 
+# You should have received a copy of the GNU Lesser General Public
+# License along with this software; if not, write to the Free
+# Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
+# 02110-1301 USA, or see the FSF site: http://www.fsf.org.
+
+require 'base64'
+
+module TorqueBox
+  module Codecs
+    module MarshalSmart
+      class << self
+
+        # @api private
+        MARSHAL_MARKER = "_|marshalled|_"
+
+        def encode(object)
+          case object
+          when String, Numeric, true, false, nil
+            object
+          else
+            if object.respond_to?(:java_object)
+              object
+            else
+              MARSHAL_MARKER + Base64.encode64(::Marshal.dump(object))
+            end
+          end
+        end
+
+        def decode(object)
+          if object.is_a?(String) && object.start_with?(MARSHAL_MARKER)
+            object = ::Marshal.load(Base64.decode64(object.sub(MARSHAL_MARKER, '')))
+          end
+          object
+        end
+      end
+
+    end
+  end
+end

data/lib/torquebox/codecs.rb

@@ -0,0 +1,65 @@
+# Copyright 2008-2013 Red Hat, Inc, and individual contributors.
+# 
+# This is free software; you can redistribute it and/or modify it
+# under the terms of the GNU Lesser General Public License as
+# published by the Free Software Foundation; either version 2.1 of
+# the License, or (at your option) any later version.
+# 
+# This software is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+# Lesser General Public License for more details.
+# 
+# You should have received a copy of the GNU Lesser General Public
+# License along with this software; if not, write to the Free
+# Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
+# 02110-1301 USA, or see the FSF site: http://www.fsf.org.
+
+# These codecs don't depend on anything outside Ruby stdlib
+require 'torquebox/codecs/marshal'
+require 'torquebox/codecs/marshal_base64'
+require 'torquebox/codecs/marshal_smart'
+
+# These codecs depend on external gems - attempt to load them
+# but ignore any load errors and we'll lazily try again later
+require 'torquebox/codecs/json' rescue nil
+require 'torquebox/codecs/edn' rescue nil
+
+module TorqueBox
+  module Codecs
+    class << self
+
+      def [](key)
+        case key
+        when :edn
+          # This is only so any issues requiring the edn codec bubble
+          # up when it gets used
+          require 'torquebox/codecs/edn' unless defined?(TorqueBox::Codecs::EDN)
+          TorqueBox::Codecs::EDN
+        when :json
+          # This is only so any issues requiring the json codec bubble
+          # up when it gets used
+          require 'torquebox/codecs/json' unless defined?(TorqueBox::Codecs::JSON)
+          TorqueBox::Codecs::JSON
+        when :marshal
+          TorqueBox::Codecs::Marshal
+        when :marshal_base64
+          MarshalBase64
+        when :marshal_smart
+          MarshalSmart
+        else
+          raise "Unsupported codec #{key}"
+        end
+      end
+
+      def encode(data, encoding)
+        self[encoding].encode(data)
+      end
+
+      def decode(data, encoding)
+        self[encoding].decode(data)
+      end
+
+    end
+  end
+end

data/lib/torquebox/injectors.rb

@@ -15,26 +15,47 @@
 # Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 # 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 
+require 'torquebox/msc'
 require 'torquebox/registry'
+require 'torquebox/service_registry'
 
 module TorqueBox
-  module Injectors
 
-    def self.analyze_and_inject(&block)
-      fetch( 'runtime-injection-analyzer' ).analyze_and_inject( block )
+  class InjectionError < StandardError
+  end
+
+  def self.fetch(something)
+    unless TorqueBox::Registry.has_key?(something.to_s)
+      handler_registry = TorqueBox::ServiceRegistry['torquebox.core.injection.injectable-handler-registry']
+      # handler_registry should only be nil when running outside of
+      # TorqueBox so we just return the nil value and skip everything
+      # else to facilitate testing outside the container
+      return nil if handler_registry.nil?
+      handler = handler_registry.get_handler(something)
+      raise InjectionError.new("Invalid injection - #{something}") if handler.nil?
+      injectable = handler.handle(something, true)
+      service_name = injectable.get_service_name(TorqueBox::Registry['service-target'],
+                                                 TorqueBox::Registry['deployment-unit'])
+      service = TorqueBox::ServiceRegistry.registry.getService(service_name)
+      raise InjectionError.new("Service not found for injection - #{something}") if service.nil?
+      state = TorqueBox::MSC.wait_for_service_to_start(service, 45)
+      raise InjectionError.new("Injected service failed to start - #{service_name}") if state != 'UP'
+      value = service.value
+      raise InjectionError.new("Injected service had no value - #{service_name}") if value.nil?
+      value = value.convert(JRuby.runtime) if value.respond_to?(:convert)
+      TorqueBox::Registry.merge!(something.to_s => value)
     end
+    TorqueBox::Registry[something.to_s]
+  end
+
+  module Injectors
 
     def fetch(something)
-      TorqueBox::Registry[something.to_s]
+      TorqueBox.fetch(something)
     end
-    alias_method :inject, :fetch
     alias_method :__inject__, :fetch
-    
-    %w{ msc service cdi jndi queue topic }.each do |type|
-      define_method("inject_#{type}".to_sym) do |key|
-        fetch(key)
-      end
 
+    %w{ msc service cdi jndi queue topic }.each do |type|
       define_method("fetch_#{type}".to_sym) do |key|
         fetch(key)
       end
@@ -42,3 +63,13 @@ module TorqueBox
 
   end
 end
+
+# Temporarily workaround TorqueSpec needing an inject(...) method
+# @api private
+module TorqueSpec
+  class Daemon
+    def inject(something)
+      TorqueBox.fetch(something)
+    end
+  end
+end

data/lib/torquebox/kernel.rb

@@ -18,6 +18,7 @@
 require 'torquebox/service_registry'
 
 module TorqueBox
+  # @api private
   class Kernel
 
     def self.kernel=(kernel)

data/lib/torquebox/logger.rb

@@ -19,6 +19,7 @@ require 'logger'
 
 module TorqueBox
 
+  # @api private
   class FallbackLogger < ::Logger
 
     def initialize name = nil
@@ -54,6 +55,8 @@ module TorqueBox
       define_method(method) { true }
     end
 
+    # The minimum log level to actually log, with debug being the lowest
+    # and fatal the highest
     attr_accessor :level
 
     def add(severity, message, progname, &block)
@@ -65,6 +68,16 @@ module TorqueBox
       @logger.send(delegate, *params)
     end
 
+    # @!method debug(message)
+    #   @param [String] message The message to log
+    # @!method info(message)
+    #   @param [String] message The message to log
+    # @!method warn(message)
+    #   @param [String] message The message to log
+    # @!method error(message)
+    #   @param [String] message The message to log
+    # @!method fatal(message)
+    #   @param [String] message The message to log
     def method_missing(method, *args, &block)
       delegate = method
       is_boolean = false

data/lib/torquebox/msc.rb

@@ -15,19 +15,18 @@
 # Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 # 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 
-require 'torquebox/injectors'
+require 'java'
 
 module TorqueBox
   class MSC
     class << self
-      include TorqueBox::Injectors
 
       def service_registry
-        fetch('service-registry')
+        TorqueBox::Registry['service-registry']
       end
 
       def deployment_unit
-        fetch('deployment-unit')
+        TorqueBox::Registry['deployment-unit']
       end
 
       def service_names
@@ -37,6 +36,63 @@ module TorqueBox
       def get_service(service_name)
         service_registry.get_service(service_name)
       end
+
+      # Returns (or yields over the items) of list of
+      # services with (canonical) name matching provided
+      # regular expression.
+      #
+      # @param [Regexp] Regular expression to match the name
+      #
+      # @param Optional block. If block is provided it'll
+      #         iterate over the values. If there is no block
+      #         given this method returns list of services
+      def get_services(regexp, &block)
+        services = []
+
+        TorqueBox::MSC.service_names.each do |name|
+          if name.canonical_name =~ regexp
+            service = TorqueBox::MSC.get_service(name)
+            services << service unless service.nil?
+          end
+        end
+
+        if block
+          services.each do |s|
+            yield s
+          end
+        else
+          services
+        end
+      end
+
+      def java_web_context
+        war_meta_data = deployment_unit.get_attachment(org.jboss.as.web.deployment.WarMetaData::ATTACHMENT_KEY)
+        return nil if war_meta_data.nil? # no web component in this application
+        jboss_web_meta_data = war_meta_data.getMergedJBossWebMetaData
+        virtual_host = jboss_web_meta_data.virtual_hosts.first || 'default-host'
+        context_path = jboss_web_meta_data.context_root
+        context_path = "/#{context_path}" unless context_path.start_with?('/')
+        deployment_service_name = org.jboss.msc.service.ServiceName.parse("jboss.web.deployment")
+        service_name = deployment_service_name.append(virtual_host).append(context_path)
+        get_service(service_name).value
+      end
+
+      # Wait for the given MSC service to start.
+      #
+      # @param [org.jboss.msc.service.Service] MSC service to wait on
+      #
+      # @return String the service state after waiting - one of DOWN,
+      # STARTING, START_FAILED, UP, STOPPING, or REMOVED. This should
+      # be UP unless something went wrong.
+      def wait_for_service_to_start(service, seconds_to_wait)
+        unless service.state.to_s == 'UP'
+          listener = org.torquebox.core.gem.MSCServiceListener.new(service)
+          service.add_listener(listener)
+          service.set_mode(org.jboss.msc.service.ServiceController::Mode::ACTIVE)
+          listener.wait_for_start_or_failure(seconds_to_wait)
+        end
+        service.state.to_s
+      end
     end
   end
 end

data/lib/torquebox/registry.rb

@@ -19,8 +19,9 @@ require 'thread'
 
 module TorqueBox
   class Registry
+    # @api private
     MUTEX = Mutex.new
-    
+
     class << self
       def merge!(hash)
         MUTEX.synchronize do
@@ -33,7 +34,13 @@ module TorqueBox
         MUTEX.synchronize do
           value = registry[key]
         end
-        return value
+        value
+      end
+
+      def has_key?(key)
+        MUTEX.synchronize do
+          registry.has_key?(key)
+        end
       end
 
       def registry