2014-05-22 13:35:45 -04:00
|
|
|
module ActiveJob
|
2014-08-17 16:48:44 -04:00
|
|
|
class DeserializationError < StandardError
|
|
|
|
attr_reader :original_exception
|
|
|
|
|
|
|
|
def initialize(e)
|
|
|
|
super ("Error while trying to deserialize arguments: #{e.message}")
|
|
|
|
@original_exception = e
|
|
|
|
set_backtrace e.backtrace
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2014-08-15 06:04:06 -04:00
|
|
|
module Arguments
|
|
|
|
extend self
|
2014-05-31 10:31:20 -04:00
|
|
|
TYPE_WHITELIST = [ NilClass, Fixnum, Float, String, TrueClass, FalseClass, Bignum ]
|
2014-05-22 13:35:45 -04:00
|
|
|
|
2014-08-15 06:04:06 -04:00
|
|
|
def serialize(arguments)
|
2014-06-05 16:10:41 -04:00
|
|
|
arguments.map { |argument| serialize_argument(argument) }
|
|
|
|
end
|
|
|
|
|
2014-08-15 06:04:06 -04:00
|
|
|
def deserialize(arguments)
|
2014-06-05 16:10:41 -04:00
|
|
|
arguments.map { |argument| deserialize_argument(argument) }
|
|
|
|
end
|
|
|
|
|
|
|
|
private
|
2014-08-15 06:04:06 -04:00
|
|
|
def serialize_argument(argument)
|
2014-05-31 10:31:20 -04:00
|
|
|
case argument
|
2014-08-16 21:06:30 -04:00
|
|
|
when GlobalID::Identification
|
2014-07-02 22:32:40 -04:00
|
|
|
argument.global_id.to_s
|
2014-05-31 10:31:20 -04:00
|
|
|
when *TYPE_WHITELIST
|
2014-05-22 13:35:45 -04:00
|
|
|
argument
|
2014-05-31 10:31:20 -04:00
|
|
|
when Array
|
|
|
|
serialize(argument)
|
2014-06-05 16:10:41 -04:00
|
|
|
when Hash
|
|
|
|
Hash[ argument.map { |key, value| [ serialize_hash_key(key), serialize_argument(value) ] } ]
|
2014-05-22 13:35:45 -04:00
|
|
|
else
|
|
|
|
raise "Unsupported argument type: #{argument.class.name}"
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2014-08-15 06:04:06 -04:00
|
|
|
def deserialize_argument(argument)
|
2014-05-31 10:31:20 -04:00
|
|
|
case argument
|
|
|
|
when Array
|
|
|
|
deserialize(argument)
|
|
|
|
when Hash
|
2014-06-05 16:10:41 -04:00
|
|
|
Hash[ argument.map { |key, value| [ key, deserialize_argument(value) ] } ].with_indifferent_access
|
2014-05-31 10:31:20 -04:00
|
|
|
else
|
2014-08-16 21:06:30 -04:00
|
|
|
GlobalID::Locator.locate(argument) || argument
|
2014-05-31 10:31:20 -04:00
|
|
|
end
|
2014-08-17 16:48:44 -04:00
|
|
|
rescue => e
|
|
|
|
raise DeserializationError.new(e)
|
2014-05-31 10:31:20 -04:00
|
|
|
end
|
|
|
|
|
2014-08-15 06:04:06 -04:00
|
|
|
def serialize_hash_key(key)
|
2014-05-31 10:31:20 -04:00
|
|
|
case key
|
|
|
|
when String, Symbol
|
|
|
|
key.to_s
|
|
|
|
else
|
|
|
|
raise "Unsupported hash key type: #{key.class.name}"
|
|
|
|
end
|
|
|
|
end
|
2014-05-22 13:35:45 -04:00
|
|
|
end
|
|
|
|
end
|