Added support for Ruby v2 Specialization (#1101)

* Added support for Ruby v2 Specialization
* Added ruby v2 example with builder code

environments/ruby/fission/handler.rb

@@ -7,10 +7,10 @@ module Fission
     def self.call(env)
       context = Context.new(env)
 
-      response = if method(:handler).arity > 0
-        handler(context)
+      response = if $handler.arity > 0
+        $handler.call(context)
       else
-        handler
+        $handler.call
       end
 
       response.is_a?(Array) ? response : Rack::Response.new([response]).finish

environments/ruby/fission/specializer.rb

@@ -1,10 +1,13 @@
 # frozen_string_literal: true
+
 require 'benchmark'
+require 'json'
 
 module Fission
-  CODE_PATH = '/userfunc/user'
-
   module Specializer
+    
+    CODE_PATH = '/userfunc/user'
+
     def self.call(env)
       request = Request.new(env)
 
@@ -12,11 +15,70 @@ module Fission
       time = Benchmark.measure { load CODE_PATH }
       request.logger.info("User code loaded in #{(time.real * 1000).round(3)}ms")
 
-      Rack::Response.new([], 201).finish
+      # set to "handler" for v1 specialization
+      $handler = method(:handler)
 
+      Rack::Response.new([], 201).finish
     rescue => e
       request.logger.error(%(Specialization failed - #{e}\n#{e.backtrace.join("\n")}))
       Rack::Response.new(['500 Internal Server Error'], 500, {}).finish
     end
   end
+
+  module V2
+    module Specializer
+
+      def self.load_vendor(path)
+        gems = Dir[File.join(path, 'vendor/bundle/ruby/*/gems/*/lib')]
+        exts = Dir[File.join(path, 'vendor/bundle/ruby/*/extensions/x86_64-linux/*/*')]
+
+        $LOAD_PATH.unshift(*gems)
+        $LOAD_PATH.unshift(*exts)
+      end
+
+      def self.call(env)
+        request = Request.new(env)
+
+        body = JSON.parse(request.body.read)
+        path = body['filepath']
+        func = body['functionName']
+
+        time = Benchmark.measure do
+          if File.file?(path)
+            # If pointing to just a single file, all we need to do is load it
+            request.logger.debug("Loading file #{path}")
+
+            load path
+          elsif File.directory?(path)
+            # First we want to load all the vendor files
+            load_vendor(path)
+
+            # We then want to get all the .rb files in this src
+            rb_files = File.join(path, "**/*.rb")
+
+            # But we don't want to include the vendor files that we loaded above
+            vendor_dir = File.join(path, 'vendor')
+            src_files = Dir[rb_files].reject {|d| d.start_with?(vendor_dir) }
+
+            request.logger.debug("Loading sources #{src_files}")
+
+            src_files.each do |file|
+              load file
+            end
+          else
+            request.logger.error(%(Specialization failed - could not find src at #{path}}))
+            Rack::Response.new(['500 Internal Server Error'], 500, {}).finish
+
+            return
+          end
+        end
+        request.logger.info("User code loaded in #{(time.real * 1000).round(3)}ms")
+
+        # set global handler for this specialization
+        $handler = method(func)
+
+        Rack::Response.new([], 201).finish
+      end
+    end
+  end
 end