src/examples/openflow_message/hello.rb in trema-0.2.5 vs src/examples/openflow_message/hello.rb in trema-0.2.6

@@ -1,10 +1,8 @@
 #
 # A test example program to send a OFPT_HELLO message.
 #
-# Author: Nick Karanatsios <[e-mail]>
-#
 # Copyright (C) 2008-2012 NEC Corporation
 #
 # This program is free software; you can redistribute it and/or modify
 # it under the terms of the GNU General Public License, version 2, as
 # published by the Free Software Foundation.
@@ -18,38 +16,25 @@
 # with this program; if not, write to the Free Software Foundation, Inc.,
 # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 #
 
 
-require "example"
-
-
 class HelloController < Controller
-  include Example
-
-
-  class << self
-    def run args
-      usage unless Example.options_parse args
+  def start
+    if ARGV.size < 2
+      STDERR.puts "Usage: #{ File.basename ARGV[ 0 ] } COUNT"
+      shutdown!
     end
+    @count = ARGV[ 1 ].to_i
+  end
 
 
-    def usage
-      puts Example.cmd_usage
-      puts "Send count number of hello messages to datapath_id."
-      exit false
+  def switch_ready datapath_id
+    @count.times do
+      send_message datapath_id, Hello.new
     end
   end
-
-
-  def switch_ready msg_datapath_id
-    may_raise_error msg_datapath_id
-    send_nr_msgs Hello
-  end
 end
-
-
-HelloController.run ARGV
 
 
 ### Local variables:
 ### mode: Ruby
 ### coding: utf-8-unix