Sha256: 147d053c2d7305dd8ffac8fc48f1b3de051c731fa0cfad655f9877cbba50864c
Contents?: true
Size: 1.96 KB
Versions: 5
Compression:
Stored size: 1.96 KB
Contents
module MCollective
module RPC
# Simple class to manage compliant replies to MCollective::RPC
class Reply
attr_accessor :statuscode, :statusmsg, :data
def initialize(action, ddl)
@data = {}
@statuscode = 0
@statusmsg = "OK"
@ddl = ddl
@action = action
begin
initialize_data
rescue Exception => e
Log.warn("Could not pre-populate reply data from the DDL: %s: %s" % [e.class, e.to_s ])
end
end
def initialize_data
unless @ddl.actions.include?(@action)
raise "No action '%s' defined for agent '%s' in the DDL" % [@action, @ddl.pluginname]
end
interface = @ddl.action_interface(@action)
interface[:output].keys.each do |output|
@data[output] = interface[:output][output][:default]
end
end
# Helper to fill in statusmsg and code on failure
def fail(msg, code=1)
@statusmsg = msg
@statuscode = code
end
# Helper that fills in statusmsg and code but also raises an appropriate error
def fail!(msg, code=1)
@statusmsg = msg
@statuscode = code
case code
when 1
raise RPCAborted, msg
when 2
raise UnknownRPCAction, msg
when 3
raise MissingRPCData, msg
when 4
raise InvalidRPCData, msg
else
raise UnknownRPCError, msg
end
end
# Write to the data hash
def []=(key, val)
@data[key] = val
end
# Read from the data hash
def [](key)
@data[key]
end
def fetch(key, default)
@data.fetch(key, default)
end
# Returns a compliant Hash of the reply that should be sent
# over the middleware
def to_hash
return {:statuscode => @statuscode,
:statusmsg => @statusmsg,
:data => @data}
end
end
end
end
Version data entries
5 entries across 5 versions & 1 rubygems