Menu Search

agent_external.rb

require 'cqpid'
require 'qmf2'

class MyAgent < Qmf2::AgentHandler

  def initialize(session, data)
    super(session)
    @data = data
  end

  def authorize_query(query, user_id)
    puts "Authorizing #{user_id}"
    return true
  end

  def get_query(context, query, user_id)
    puts "Get Query"
    context.response(@data)
    context.complete
  end

  def method_call(context, method_name, data_addr, args, user_id)
    puts "Method: #{method_name}"
    context._success
  end

end


class Program

  def initialize(url)
    @url = url
    @sess_options = "{allow-queries:False, external:True}"
  end

  def setup_schema(agent)
    @cls_control = Qmf2::Schema.new(Qmf2::SCHEMA_TYPE_DATA, "org.package", "control")
    @cls_control.add_property(Qmf2::SchemaProperty.new("state", Qmf2::SCHEMA_DATA_STRING))
    agent.register_schema(@cls_control)
  end

  def run
    connection = Cqpid::Connection.new(@url)
    connection.open

    session = Qmf2::AgentSession.new(connection, @sess_options)
    session.set_vendor("package.org")
    session.set_product("external_agent")
    setup_schema(session)
    session.open

    @control = Qmf2::Data.new(@cls_control)
    @control.state = "OPERATIONAL-EXTERNAL"
    @control.set_addr(Qmf2::DataAddr.new("singleton"))

    main = MyAgent.new(session, @control)
    main.run
  end
end

prog = Program.new("localhost")
prog.run

Download this file