prime_helper.rb in royw-drbman-0.0.3

- old
+ new

@@ -1,6 +1,6 @@
-require 'drb'
+require 'drbman_server'
 
 # A helper object for calculating primes using the Sieve of Eratosthenes
 #
 # == Usage
 # ruby prime_helper.rb foo.example.com 1234
@@ -11,11 +11,11 @@
 #
 # ruby prime_helper.rb
 # will run the service as: druby://localhost:9000
 #
 class PrimeHelper
-  attr_accessor :name
+  include DrbmanServer
   
   # Find the multiples of the give prime number that are less than the 
   # given maximum.
   # @example
   #  multiples_of(5,20) => [10, 15]
@@ -25,20 +25,8 @@
   def multiples_of(prime, maximum)
     a = []
     2.upto((maximum - 1) / prime) { |i| a << (i * prime) }
     a
   end
-  
-  # Stop the DRb service
-  def stop_service
-    DRb.stop_service
-  end
 end
 
-machine = 'localhost'
-machine = ARGV[0] unless ARGV.length < 1
-port = 9000
-port = ARGV[1] unless ARGV.length < 2
-server = PrimeHelper.new
-server.name = "druby://#{machine}:#{port}"
-DRb.start_service(server.name, server)
-DRb.thread.join
+DrbmanServer.start_service(PrimeHelper)