Brendan StennettandTa-Ching Chen
c717f182b6
* Added support for Ruby v2 Specialization * Added ruby v2 example with builder code
85 lines
2.4 KiB
Ruby
85 lines
2.4 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'benchmark'
|
|
require 'json'
|
|
|
|
module Fission
|
|
module Specializer
|
|
|
|
CODE_PATH = '/userfunc/user'
|
|
|
|
def self.call(env)
|
|
request = Request.new(env)
|
|
|
|
request.logger.info("Codepath defaulting to #{CODE_PATH}")
|
|
time = Benchmark.measure { load CODE_PATH }
|
|
request.logger.info("User code loaded in #{(time.real * 1000).round(3)}ms")
|
|
|
|
# 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
|