Menu Search

agent_internal.rb

require 'cqpid'
require 'qmf2'

class MyAgent < Qmf2::AgentHandler

  def initialize(session)
    super(session)
  end

  def authorize_query(query, user_id)
    puts "Authorizing #{user_id}"
    return true
  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}"
  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("internal_agent")
    setup_schema(session)
    session.open

    control = Qmf2::Data.new(@cls_control)
    control.state = "OPERATIONAL"
    session.add_data(control)

    main = MyAgent.new(session)
    main.run
  end
end

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

Download this file